Module: MKIt

Includes:
ActiveRecord::Tasks
Defined in:
lib/mkit/ctypes.rb,
lib/mkit.rb,
lib/mkit/utils.rb,
lib/mkit/status.rb,
lib/mkit/haproxy.rb,
lib/mkit/version.rb,
lib/mkit/mkit_dns.rb,
lib/mkit/cmd_runner.rb,
lib/mkit/exceptions.rb,
lib/mkit/job_manager.rb,
lib/mkit/sagas/asaga.rb,
lib/mkit/config/config.rb,
lib/mkit/mkit_interface.rb,
lib/mkit/app/mkit_server.rb,
lib/mkit/docker_listener.rb,
lib/mkit/workers/aworker.rb,
lib/mkit/sagas/saga_manager.rb,
lib/mkit/workers/pod_worker.rb,
lib/mkit/app/helpers/haproxy.rb,
lib/mkit/sagas/create_pod_saga.rb,
lib/mkit/app/helpers/erb_helper.rb,
lib/mkit/workers/haproxy_worker.rb,
lib/mkit/workers/service_worker.rb,
lib/mkit/workers/worker_manager.rb,
lib/mkit/app/helpers/docker_helper.rb,
lib/mkit/app/helpers/services_helper.rb,
lib/mkit/config/load_default_configs.rb,
lib/mkit/app/helpers/interface_helper.rb

Overview

requires Hash.to_o

Defined Under Namespace

Modules: CType, Config, DockerHelper, ERBHelper, HAProxy, Initializers, InterfaceHelper, PoolStatus, ServicesHelper, Status, Templates, Utils Classes: ASaga, AWorker, AppAlreadyDeployedException, BaseException, CmdRunner, CmdRunnerException, CreatePodSaga, DNS, DockerListener, Error, HAProxyWorker, Interface, InvalidPortMappingTypeException, JobManager, MKItCType, MKItStatus, PodNotFoundException, PodWorker, PoolExaustedException, SagaManager, Server, ServiceAlreadyExists, ServiceNameMismatch, ServiceNotFoundException, ServiceWorker, WorkerManager

Constant Summary collapse

System =
Dry::Container.new
VERSION =
"0.3.0"

Class Method Summary collapse

Class Method Details

.configure(options:) ⇒ Object



38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
# File 'lib/mkit.rb', line 38

def self.configure(options:)
  @root = MKIt::Utils.root
  MKItLogger.debug!
  #
  # config dir
  if ENV["RACK_ENV"] != "development"
    @config_dir = options[:config_dir].nil? ? '/etc/mkit' : options[:config_dir]
  else
    @config_dir = options[:config_dir].nil? ? "#{@root}/config" : options[:config_dir]
  end
  MKIt::Utils.set_config_dir(@config_dir)
  # create dirs
  if ENV["RACK_ENV"] != "development" || !options[:config_dir].nil?
    check_config_files = false
    if ! File.exists?(@config_dir)
      FileUtils.mkdir_p(@config_dir)
      check_config_files = true
    end
    FileUtils.mkdir_p('/var/lib/mkitd') unless File.exists?('/var/lib/mkitd')
    FileUtils.cp( "#{@root}/config/mkit_config.yml", @config_dir) unless File.exists?("#{@config_dir}/mkit_config.yml")
    FileUtils.cp( "#{@root}/config/database.yml", @config_dir) unless File.exists?("#{@config_dir}/database.yml")
    FileUtils.cp( "#{@root}/config/mkitd_config.sh", @config_dir) unless File.exists?("#{@config_dir}/mkitd_config.sh")
    if check_config_files
      MKItLogger.info "Configuration files copied to #{@config_dir}. Please check it and restart."
      exit
    end
  end
  #
  # load configuration
  MKIt::Initializers.load_my_configuration
  #
  # run config based tasks
  FileUtils.mkdir_p(MKIt::Config.mkit.haproxy.config_dir)
  # ...haproxy defaults file
  FileUtils.cp(
    "#{MKIt::Utils.root}/lib/mkit/app/templates/haproxy/0000_defaults.cfg",
    MKIt::Config.mkit.haproxy.config_dir
  ) unless File.exists?("#{MKIt::Config.mkit.haproxy.config_dir}/0000_defaults.cfg")
  #
  #  conn = { adapter: "sqlite3", database: ":memory:" }
  #  ActiveRecord::Base.establish_connection(conn)
  #  include ActiveRecord::Tasks
  #  DatabaseTasks.database_configuration = YAML.load_file('my_database_config.yml')
  #  DatabaseTasks.db_dir = 'db'
  #
  DatabaseTasks.database_configuration = MKIt::Utils.load_db_config
  DatabaseTasks.env=MKIt::Config.mkit.database.env
  DatabaseTasks.migrations_paths=[ "#{@root}/db/migrate" ]
  DatabaseTasks.db_dir="db"
  DatabaseTasks.root=@root
end

.establish_db_connectionObject



90
91
92
93
94
95
96
97
98
99
100
# File 'lib/mkit.rb', line 90

def self.establish_db_connection
  #
  ActiveRecord::Base.establish_connection(DatabaseTasks.database_configuration[DatabaseTasks.env])
  ActiveRecord::Base.connection.migration_context.migrations_paths.clear
  ActiveRecord::Base.connection.migration_context.migrations_paths << "#{@root}/db/migrate"
  #
  MKItLogger.debug "database_tasks migration paths     #{DatabaseTasks.migrations_paths}"
  MKItLogger.debug "active_record_base migration_paths #{ActiveRecord::Base.connection.migration_context.migrations_paths.inspect}"
  MKItLogger.debug "active_record_base configurations  #{ActiveRecord::Base.configurations.inspect}"
  MKItLogger.debug "active_record_base conn_db_config  #{ActiveRecord::Base.connection_db_config.inspect}"
end

.migrateObject



102
103
104
# File 'lib/mkit.rb', line 102

def self.migrate
  ActiveRecord::Base.connection.migration_context.migrate if ActiveRecord::Base.connection.migration_context.needs_migration?
end

.restore_operationObject



106
107
108
109
110
111
112
113
# File 'lib/mkit.rb', line 106

def self.restore_operation
  MKItLogger.info "restoring operations..."
  # create interfaces of deployed apps  otherwise haproxy won't start
  Service.all.each { | srv |
    srv.deploy_network
    srv.update_status!
  }
end

.startup(options: {}) ⇒ Object



115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
# File 'lib/mkit.rb', line 115

def self.startup(options: {})
  self.configure(options: options)
  self.establish_db_connection
  self.migrate

  MKIt::Initializers.load_default_configs
  MKIt::Interface.up

  System.register(:job_manager, memoize: true) {
    MKIt::JobManager.new
  }
  System.register(:mkit_dns, memoize: true) {
    Thread.new {
      dns = MKIt::DNS.new
      dns.run
    }
  }
  System.register(:docker_listener, memoize: true) {
    MKIt::DockerListener.new
  }
  # watchdog feature is to be re-evaluated
  # System.register(:watchdog, memoize: true) {
  #  MKIt::WatchdogManager.new
  # }

  # register workers
  WorkerManager.register_workers    
  SagaManager.register_workers
  #
  System[:job_manager].start
  System[:docker_listener].start
  # watchdog feature is to be re-evaluated
  # System[:watchdog].start
  System[:mkit_dns].run
  #
  self.restore_operation
  #
  MKItLogger.debug "restarting proxy..."
  MKIt::HAProxy.restart
  MKItLogger.info "MKIt is up and running!"
end