cfgdatabase 1.2.4 → 1.2.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2944c3a5397dbac2ff52e00e1c7f1ddd97f020d39163c6b8782102f704ee59c0
-  data.tar.gz: 4a290291434d2b6e2af52d94f342a28f56cd67447aac83f4c3905cf081dc2b8c
+  metadata.gz: 606c2258083d60aa81e7d114894d9b22fff92c33036981ce1053371a2897607b
+  data.tar.gz: 3dd8a996e9a7a233eb698bc2ae93b25aafabbf0bac0f156650daac3e911dd3da
 SHA512:
-  metadata.gz: 64de309e76937c5bb812c32c97dccc239c7c19bc64b5bd2643d62da7345b5aafe18466de8b0138258bc058975ec17ed54830a2a911ea223ab973990c00fb81d5
-  data.tar.gz: 97d304a78602270719b7e7240d41fc5d2d652bfa2d1f1184ff73957e6fc4edba27a329b4bb7fe3d0f585352b5139fcdd126e933223fbe35a956643f8908fb5df
+  metadata.gz: 7f271b2a238c361ee4e497b0bcde8dc6650ea2aa370c266f906dd225cf67b8b79f5df168e294baae2dec22ddb5dfb27d52b04232bc3cd24b2afc79f86e0e7c38
+  data.tar.gz: b444cc1b1bb12fbb7ee69f8d18ab820e95fe5dda08ea8a1e6316e46b09688ab776526fc9c8ca6333e55de125b7ac33b5d292f8431ad33873a454c9cdbe848dca

data/utils/migrate.rb

@@ -0,0 +1,186 @@
+#!/usr/bin/env ruby
+# encoding: utf-8
+
+require 'bundler/setup'
+require 'fileutils'
+require 'pathname'
+require 'bunny'
+require 'logger'
+require 'yaml'
+require 'optparse'
+require 'thor'
+require 'monkey-hash'
+require 'app-config'
+require 'app-logger'
+require 'sequel'
+require 'pg'
+require 'app-database'
+
+App::Config.init( approot: Pathname.new( Dir.pwd ).expand_path )
+App::Logger.new
+App::Database.init if Cfg.db?
+Sequel.extension :migration
+
+# тупой способ определения запуска в корне проекта
+launchdir = Dir.pwd
+if launchdir != Cfg.root || ! File.exist?( "#{ Cfg.root }/Gemfile" )
+  puts "\n\tВнимание! Следует запускать migrate.rb в корне проекта.\n\n"
+  raise "\n\tВнимание! Следует запускать migrate.rb в корне проекта.\n\n"
+end
+
+class Dbtask < Thor
+  package_name 'db'
+  desc 'init', 'Создать папки и записать нулевую миграцию'
+  def init
+    mydir       = Pathname( __dir__ ).expand_path.to_s
+    migrations  = Pathname( "#{ Cfg.root }/db/migrations" ).expand_path.to_s
+    Log.info{"Создаю папку #{ migrations }"}
+    FileUtils.mkdir_p migrations
+    list = Dir[ "#{ mydir }/migrations/*rb" ]
+    Log.info{"Копирую файл#{ list.count == 1 ? '' : 'ы' } с миграциями:\n#{ list.join("\n") }"}
+    FileUtils.cp list, "#{ migrations }/"
+  end
+
+  desc "g class_name", "Создать файл миграции в db/migrations"
+  def g(class_name)
+    tstamp = Time.now.strftime "%Y%m%d%H%M%S"
+    fname = ''
+    if Dir[ "#{ Cfg.root }/db/migrations/#{ tstamp }_*.rb" ].any?
+      counter = 0
+      while Dir[ "#{ Cfg.root }/db/migrations/#{ tstamp }#{ '%02d' % counter }_*.rb" ].any? do
+        counter += 1
+      end
+      fname = "#{ Cfg.root }/db/migrations/#{ tstamp }_#{ '%02d' % counter }_#{ class_name }.rb"
+    else
+      fname = "#{ Cfg.root }/db/migrations/#{ tstamp }_#{ class_name }.rb"
+    end
+    Log.info{ "Создаю файлик миграции #{ fname }" }
+    File.open( fname, 'w' ) do |f|
+      f.write <<~EFILE
+      Sequel.migration do
+        up do
+          create_table :#{ class_name } do
+            primary_key :id, type: :Bignum
+
+            column :created_at, DateTime, null: false, index: true, default: Sequel.lit("now()")
+            column :updated_at, DateTime, null: false, index: true, default: Sequel.lit("now()")
+          end
+          run <<~EUP
+            DO $$
+            BEGIN
+              --triggers
+              IF NOT EXISTS (SELECT 1 FROM pg_trigger WHERE tgname = '#{ class_name }_update_timestamp') THEN
+                CREATE TRIGGER #{ class_name }_update_timestamp
+                  BEFORE INSERT OR UPDATE ON #{ class_name }
+                  FOR EACH ROW EXECUTE PROCEDURE update_timestamp();
+              END IF;
+            END $$;
+          EUP
+        end
+        down { run 'DROP TABLE #{ class_name } CASCADE' }
+      end
+      EFILE
+    end
+  end
+
+  desc "m", "Мигрировать миграции, можно добавить имя файла"
+  def m(point = nil)
+    point = point_from_filename(point) || nil
+    Log.warn{ "Мигрирую базу #{ point }" }
+    if point
+      Sequel::Migrator.run(Db, "db/migrations", target: point )
+    else
+      Sequel::Migrator.run(Db, "db/migrations" )
+    end
+    v
+  end
+
+  desc "r", "Откатить базу в ноль, либо к указанному файлу"
+  def r(point = '0')
+    point = point_from_filename(point) || 0
+    Log.warn{ "Откатываю базу #{ point }" }
+    if point
+      Sequel::Migrator.run(Db, "db/migrations", :target => point )
+    else
+      Sequel::Migrator.run(Db, "db/migrations" )
+    end
+    v
+  end
+
+  desc "v" , "Напечатать текущую версию в базе"
+  def v
+    version =
+    if Db.tables.include?(:schema_migrations)
+      (f = Db[:schema_migrations].all).any? ? f.last[:filename] : 'пусто'
+    else
+      'пусто'
+    end
+    puts "Последняя миграция: #{ version }"
+  end
+
+  desc "create", "Создать базу"
+  def create
+    rootdb = superdb
+    Log.warn{ "Создаю пользователя: #{ Cfg.db.user } и базу: #{ Cfg.db.database }." }
+    begin
+      rootdb["CREATE USER #{ Cfg.db.user  } WITH LOGIN PASSWORD '#{ Cfg.db.password }'"].all
+    rescue Exception => e
+      Log.info{ e.message }
+    end
+    begin
+      rootdb["CREATE DATABASE #{ Cfg.db.database } OWNER #{ Cfg.db.user  }"].all
+    rescue Exception => e
+      Log.info{ e.message }
+    end
+  end
+
+  desc "scratch", 'Удалить базу, загрузить все миграции заново'
+  def scratch(db = nil)
+    db = superdb
+    Log.debug{"Отключение от текущей базы"}
+    Db.disconnect
+    unless db.test_connection
+      Log.warn{"Не смог подключиться к базе для махинаций. #{ superuser.inspect }"}
+      exit 255
+    end
+    Log.warn{ "Удаляю базу #{ Cfg.db.database }" }
+    begin
+      db << "DROP DATABASE #{ Cfg.db.database }"
+    rescue Exception => e
+      Log.error e.message
+    end
+    create
+    m
+  end
+
+  no_commands do
+    # подключается к базе с правами админа
+    # жёстко закодировано, что админ базы 'postgres' без пароля, и схема 'public'
+    def superdb
+      if ! @rootdb || ! @rootdb.test_connection
+        superuser = Marshal.load(Marshal.dump( Cfg.db ))
+        superuser[:adapter]  = 'postgres'
+        superuser[:user]     = 'postgres'
+        superuser[:database] = 'postgres'
+        superuser.delete :password
+        Log.debug{"Попытка административного подключения #{ superuser.inspect }"}
+        @rootdb = Sequel.connect( superuser )
+      end
+      @rootdb
+    end
+
+    def point_from_filename(n)
+      Log.debug{"Поиск миграции по куску имени: '#{ n }'."}
+      return nil unless n
+      unless n =~ /^\d+/
+        Pathname.new( Dir["#{ Cfg.root }/db/migrations/*#{ n }*.rb"].sort.last ).basename.to_s[/(\d+)/, 1].to_i
+      else
+        n[/^(\d+)/, 1].to_i
+      end
+    end
+
+  end
+
+end
+
+Dbtask.start

data/utils/migrations/00000000000000_created_at_trigger.rb

@@ -0,0 +1,26 @@
+Sequel.migration do
+  up do
+    run <<~ESTOREDPROC
+    CREATE OR REPLACE FUNCTION insert_timestamp() RETURNS trigger
+      LANGUAGE plpgsql
+      AS $$
+    BEGIN
+      IF NEW.created_at IS NULL THEN
+        NEW.created_at := now();
+      END IF;
+      RETURN NEW;
+    END $$;
+
+    CREATE OR REPLACE FUNCTION update_timestamp() RETURNS trigger
+      LANGUAGE plpgsql
+      AS $$
+    BEGIN
+      IF NEW.updated_at IS NULL THEN
+        NEW.updated_at := now();
+      END IF;
+      RETURN NEW;
+    END $$;
+    ESTOREDPROC
+  end
+  down { run 'DROP FUNCTION IF EXISTS insert_timestamp(); DROP FUNCTION IF EXISTS update_timestamp();' }
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: cfgdatabase
 version: !ruby/object:Gem::Version
-  version: 1.2.4
+  version: 1.2.5
 platform: ruby
 authors:
 - deemytch
 autorequire:
-bindir: bin
+bindir: utils
 cert_chain: []
-date: 2021-02-10 00:00:00.000000000 Z
+date: 2021-03-26 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: sequel
@@ -68,11 +68,14 @@ dependencies:
         version: '0'
 description:
 email: aspamkiller@yandex.ru
-executables: []
+executables:
+- migrate.rb
 extensions: []
 extra_rdoc_files: []
 files:
 - lib/app-database.rb
+- utils/migrate.rb
+- utils/migrations/00000000000000_created_at_trigger.rb
 homepage: https://github.com/deemytch/cfgdatabase
 licenses:
 - GPL-2.0
@@ -92,8 +95,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.1.4
+rubygems_version: 3.2.13
 signing_key:
 specification_version: 4
-summary: Удобная загрузка настроек Sequel.
+summary: Удобная загрузка настроек Sequel и миграции для PostgreSQL.
 test_files: []