LogAn/lib/logan/inc/main.rb

require 'sbdb'
require 'safebox'
require 'robustserver'
require 'socket'
require 'logan/inc'
require 'logan/loglines'
require 'logan/cache'

module LogAn::Logging
	class  << self
		def log lvl, *txt
			$stderr.puts( ([Time.now, lvl]+txt).inspect)
		end
		alias method_missing log
	end
	def method_missing *paras
		self.class.log *paras
	end
end

module LogAn::Inc
	class Main < RobustServer
		def cache ret, type, &e
			type ||= 1+4
			ret = LogAn::AutoValueConvertHash.new ret, &e  if type&4 > 0 or e
			ret = LogAn::Cache.new ret, type&3  if type&3 > 0
			ret
		end

		# Open Store.
		def store env, db, type = nil, flags = nil, &e
			LogAn::Logging.info :store, :open, "sids.cnf", db, type
			cache env[ 'sids.cnf', db, :flags => flags || SBDB::CREATE | SBDB::AUTO_COMMIT], type, &e
		end

		# Open Config.
		def config env, db, type = nil, flags = nil, &e
			LogAn::Logging.info :config, :open, "sids.cnf", db, type
			cache env[ 'sids.cnf', db, :flags => flags || SBDB::RDONLY], type, &e
		end

		# Prepare Server.
		#
		# * conf:
		#    logs
		#    : Where to store log-databases? default: ./logs
		#    etc
		#    : Where to find config-databases? default: ./etc
		#    server
		#    : Server-Configuration. default { port: 1087 }
		def initialize conf
			super
			@conf = {}

			# Copy config - changes possible
			conf.each {|key, val| @conf[key]= val }

			# Default directories
			%w[logs etc].each {|key| @conf[key.to_sym] = key }

			# Open Loglines-databases
			@logs = LogAn::Loglines.new @conf[:logs]
			LogAn::Inc::FileParser::Base.logdb = @logs

			# Open config-databases
			Dir.mkdir @conf[:etc]  rescue Errno::EEXIST
			@etc = SBDB::Env.new( @conf[:etc],
					log_config: SBDB::Env::LOG_IN_MEMORY | SBDB::Env::LOG_AUTO_REMOVE,
					flags: SBDB::CREATE | SBDB::Env::INIT_TXN | Bdb::DB_INIT_MPOOL)

			# Open configs
			begin
				configs = @conf[:configs] = {}
				%w[hosts files].each {|key| configs[key.to_sym] = config( @etc, key) {|l| l } }
				configs[:fileparser] = config @etc, 'fileparser', &Safebox.method( :eval)
				LogAn::Inc::SID0.config = configs
			end

			# Open seeks-database
			begin
				stores = @conf[:stores] = {}
				db = @etc[ 'sids.store', 'seeks', SBDB::Recno, SBDB::CREATE | SBDB::AUTO_COMMIT]
				db = LogAn::AutoValueConvertHash.new( db, lambda {|val| val.pack( 'NN') }) {|val| (val||0.chr*8).unpack( 'NN') }
				stores[:seeks] = LogAn::Cache.new db
				LogAn::Inc::FileParser::Base.store = LogAn::Inc::SID0.store = stores
			end

			# Select-framework
			@select = LogAn::Inc::Select.new
			status = lambda do
				@select.at Time.now+5, &status
				LogAn::Logging.info @select
				@conf[:stores].each{|key, db| db.flush!}
			end
			status.call

			# Prepare Inc-server - create server
			@serv = LogAn::Inc::Server.new :sock => TCPServer.new( *@conf[:server]), :config => @conf[:configs], :select => @select
			LogAn::Logging.debug @serv

			# Shutdown on signals
			@sigs[:INT] = @sigs[:TERM] = method( :shutdown)

		rescue Object
			# It's better to close everything, because BDB doesn't like unexpected exits
			self.at_exit
			raise $!
		end

		# Will be called at exit.  Will close all opened BDB::Env
		def at_exit
			LogAn::Logging.info :at_exit
			@logs and @logs.close
			@etc and @etc.close
		end

		# Shutdown Server cleanly. First shutdown TCPServer.
		def shutdown signal = nil
			LogAn::Logging.info :signal, signal, Signal[signal]  if signal
			@serv.close
			exit 0
		end

		# Runs server.  Don't use it!  Use #main.
		def run
			@serv.run
		end
	end
end