rubydb 0.1.4 → 0.1.6

data/lib/rubydb/server/server.rb

@@ -1,364 +1,363 @@
-# frozen_string_literal: true
-
-require "socket"
-require "thread"
-require "time"
-require "json"
-require "fileutils"
-require "monitor"
-require "openssl"
-
-require_relative "listener"
-require_relative "connection"
-require_relative "session"
-require_relative "connection_pool"
-require_relative "worker_pool"
-require_relative "worker"
-require_relative "request_handler"
-require_relative "lifecycle"
-require_relative "../protocol/protocol"
-require_relative "../protocol/handshake"
-require_relative "../storage/engine"
-require_relative "../transactions/transaction_manager"
-
-module RubyDB
-  module Server
-    # Server - Main database server
-    class Server
-      attr_reader :config, :engine, :listener, :connection_pool
-      attr_reader :worker_pool, :request_handler, :lifecycle, :stats
-
-      def initialize(config = {})
-        @config = {
-          host: config[:host] || "localhost",
-          port: config[:port] || 7432,
-          max_connections: config[:max_connections] || 100,
-          min_workers: config[:min_workers] || 2,
-          max_workers: config[:max_workers] || 20,
-          worker_queue_size: config[:worker_queue_size] || 1000,
-          read_timeout: config[:read_timeout] || 30,
-          write_timeout: config[:write_timeout] || 30,
-          idle_timeout: config[:idle_timeout] || 300,
-          max_request_size: config[:max_request_size] || 10 * 1024 * 1024, # 10MB
-          data_dir: config[:data_dir] || "data",
-          log_dir: config[:log_dir] || "log",
-          pid_file: config[:pid_file] || "rubydb.pid",
-          daemonize: config[:daemonize] || false,
-          authentication: config[:authentication] || { method: "none" },
-          ssl: config[:ssl] || { enabled: false }
-        }.merge(config)
-
-        @stats = {
-          connections_total: 0,
-          connections_active: 0,
-          connections_rejected: 0,
-          requests_processed: 0,
-          requests_failed: 0,
-          bytes_received: 0,
-          bytes_sent: 0,
-          started_at: nil,
-          uptime_seconds: 0,
-          last_activity: nil
-        }
-
-        @lock = Monitor.new
-        @running = false
-        @shutdown = false
-        @stopping = false
-
-        validate_authentication_config!
-        validate_ssl_config!
-        validate_resource_limits!
-
-        # Setup directories
-        setup_directories
-
-        # Initialize components
-        initialize_components
-      end
-
-      def start
-        @lock.synchronize do
-          return if @running
-
-          @running = true
-          @shutdown = false
-          @stats[:started_at] = Time.now
-
-          # Daemonize if configured
-          daemonize if @config[:daemonize]
-
-          # Write PID file
-          write_pid_file
-
-          # Start components
-          @listener.start
-          @connection_pool.start
-          @worker_pool.start
-          @lifecycle.start
-
-          puts "RubyDB Server v#{RubyDB::VERSION} started on #{@config[:host]}:#{@config[:port]}"
-          puts "Data directory: #{@config[:data_dir]}"
-          puts "Max connections: #{@config[:max_connections]}"
-
-          # Update stats thread
-          start_stats_thread
-
-          true
-        end
-      end
-
-      def stop
-        @lock.synchronize do
-          return unless @running
-          return true if @stopping
-
-          @stopping = true
-
-          @shutdown = true
-          puts "Shutting down RubyDB Server..."
-
-          # Stop accepting new connections
-          @listener.stop
-
-          # Close all connections
-          @connection_pool.close_all
-
-          # Stop workers
-          @worker_pool.stop
-
-          # Stop lifecycle
-          @lifecycle.stop
-
-          # Release the database file handle before the data directory is
-          # removed by callers (notably on Windows).
-          @engine&.close
-
-          # Remove PID file
-          remove_pid_file
-
-          @running = false
-          @stopping = false
-
-          puts "RubyDB Server stopped"
-          true
-        end
-      end
-
-      def restart
-        stop
-        sleep(1)
-        start
-      end
-
-      def running?
-        @running
-      end
-
-      def stats
-        @lock.synchronize do
-          elapsed = @stats[:started_at] ? (Time.now - @stats[:started_at]).to_i : 0
-          @stats[:uptime_seconds] = elapsed
-          pool_stats = @connection_pool&.stats || {}
-          @stats[:connections_active] = pool_stats[:active_connections] || 0
-          @stats[:connections_total] = pool_stats[:total_connections] || 0
-          @stats[:requests_processed] = pool_stats[:requests_processed] || 0
-          @stats[:requests_failed] = pool_stats[:requests_failed] || 0
-
-          @stats.merge({
-            engine_stats: @engine&.stats || {},
-            connection_pool_stats: @connection_pool&.stats || {},
-            worker_pool_stats: @worker_pool&.stats || {},
-            memory_usage: get_memory_usage,
-            pid: Process.pid
-          })
-        end
-      end
-
-      private
-
-      def initialize_components
-        # Initialize storage engine
-        db_path = File.join(@config[:data_dir], "rubydb.rdb")
-        @engine = Storage::Engine.new(db_path, @config)
-
-        # Initialize transaction manager
-        @transaction_manager = Transactions::TransactionManager.new(@engine, @config)
-
-        # Initialize protocol
-        @protocol = Protocol::Protocol.new(
-          format: :json,
-          compression: @config[:compression] || false,
-          server_info: {
-            version: RubyDB::VERSION,
-            pid: Process.pid,
-            started_at: Time.now.iso8601,
-            default_auth: authentication_method,
-            authentication_credentials: authentication_credentials
-          }
-        )
-
-        # Initialize connection pool
-        @connection_pool = ConnectionPool.new(@config, @protocol, @engine)
-
-        # Initialize worker pool
-        @worker_pool = WorkerPool.new(@config, @engine, @transaction_manager)
-
-        # Initialize request handler
-        @request_handler = RequestHandler.new(@engine, @transaction_manager, @config)
-
-        # Initialize listener
-        @listener = Listener.new(@config, @connection_pool, @worker_pool)
-
-        # Initialize lifecycle
-        @lifecycle = Lifecycle.new(self, @config)
-      end
-
-      def setup_directories
-        [:data_dir, :log_dir].each do |dir|
-          path = @config[dir]
-          FileUtils.mkdir_p(path) unless Dir.exist?(path)
-        end
-      end
-
-      def authentication_method
-        auth = @config[:authentication] || {}
-        (auth[:method] || auth["method"] || "none").to_s.tr("_", "-")
-      end
-
-      def authentication_credentials
-        auth = @config[:authentication] || {}
-        auth[:credentials] || auth["credentials"] || auth
-      end
-
-      def validate_authentication_config!
-        method = authentication_method
-        supported = %w[none password md5 scram-sha-256]
-        raise ServerError, "Unsupported authentication method: #{method}" unless supported.include?(method)
-
-        return if method == "none"
-
-        credentials = authentication_credentials
-        users = credentials[:users] || credentials["users"]
-        complete = users && !users.empty? ||
-          ((credentials[:username] || credentials["username"]) &&
-           (credentials[:password] || credentials["password"]))
-        raise ServerError, "Authentication credentials are required for #{method}" unless complete
-      end
-
-      def validate_ssl_config!
-        ssl = @config[:ssl] || {}
-        enabled = ssl == true || ssl[:enabled] || ssl["enabled"]
-        return unless enabled
-
-        raise ServerError, "SSL configuration must include certificate and key paths" if ssl == true
-
-        cert = ssl[:cert_file] || ssl["cert_file"]
-        key = ssl[:key_file] || ssl["key_file"]
-        raise ServerError, "ssl.cert_file and ssl.key_file are required when SSL is enabled" unless cert && key
-        raise ServerError, "SSL certificate file does not exist" unless File.file?(cert)
-        raise ServerError, "SSL key file does not exist" unless File.file?(key)
-
-        begin
-          OpenSSL::X509::Certificate.new(File.binread(cert))
-          OpenSSL::PKey.read(File.binread(key))
-        rescue OpenSSL::SSL::SSLError, OpenSSL::X509::CertificateError, OpenSSL::PKey::PKeyError, ArgumentError => e
-          raise ServerError, "SSL certificate or key is invalid: #{e.message}"
-        end
-
-        ca = ssl[:ca_file] || ssl["ca_file"]
-        if (ssl[:verify_peer] || ssl["verify_peer"]) && !ca
-          raise ServerError, "ssl.ca_file is required when peer verification is enabled"
-        end
-        raise ServerError, "SSL CA file does not exist" if ca && !File.file?(ca)
-
-        min_version = ssl[:min_version] || ssl["min_version"] || :TLS1_2
-        unless %i[TLS1_2 TLS1_3].include?(min_version.to_s.to_sym)
-          raise ServerError, "SSL min_version must be TLS1_2 or TLS1_3"
-        end
-      end
-
-      def validate_resource_limits!
-        integer_limits = {
-          max_connections: 1..10_000,
-          min_workers: 1..1_024,
-          max_workers: 1..1_024,
-          worker_queue_size: 1..100_000,
-          max_request_size: 1_024..(64 * 1024 * 1024)
-        }
-        integer_limits.each do |name, allowed|
-          value = Integer(@config[name])
-          raise ServerError, "#{name} must be between #{allowed.begin} and #{allowed.end}" unless allowed.cover?(value)
-          @config[name] = value
-        rescue ArgumentError, TypeError
-          raise ServerError, "#{name} must be an integer"
-        end
-        raise ServerError, "min_workers cannot exceed max_workers" if @config[:min_workers] > @config[:max_workers]
-
-        %i[read_timeout write_timeout idle_timeout].each do |name|
-          value = Float(@config[name])
-          raise ServerError, "#{name} must be greater than zero" unless value.positive?
-          @config[name] = value
-        rescue ArgumentError, TypeError
-          raise ServerError, "#{name} must be a number"
-        end
-      end
-
-      def daemonize
-        return unless @config[:daemonize]
-
-        Process.daemon(true, true)
-        $stdout.reopen(File.join(@config[:log_dir], "rubydb.out"), "a")
-        $stderr.reopen(File.join(@config[:log_dir], "rubydb.err"), "a")
-      end
-
-      def write_pid_file
-        File.write(@config[:pid_file], Process.pid.to_s)
-      end
-
-      def remove_pid_file
-        File.delete(@config[:pid_file]) if File.exist?(@config[:pid_file])
-      end
-
-      def start_stats_thread
-        Thread.new do
-          while @running
-            sleep(10)
-            update_stats
-          end
-        end
-      end
-
-      def update_stats
-        @lock.synchronize do
-          @stats[:connections_active] = @connection_pool&.active_count || 0
-          @stats[:connections_total] = @connection_pool&.total_connections || 0
-          @stats[:requests_processed] = @request_handler&.stats&.dig(:requests_processed) || 0
-        end
-      end
-
-      def get_memory_usage
-        # Get memory usage in MB
-        if File.exist?("/proc/self/status")
-          File.read("/proc/self/status").each_line do |line|
-            if line.start_with?("VmRSS:")
-              return line.split[1].to_i / 1024
-            end
-          end
-        end
-
-        # Fallback - run ps command
-        begin
-          output = `ps -o rss= -p #{Process.pid}`
-          return output.strip.to_i / 1024 if output && !output.empty?
-        rescue
-          # Ignore
-        end
-
-        0
-      rescue
-        0
-      end
-    end
-  end
-end
+# frozen_string_literal: true
+
+require "socket"
+require "time"
+require "json"
+require "fileutils"
+require "monitor"
+require "openssl"
+
+require_relative "listener"
+require_relative "connection"
+require_relative "session"
+require_relative "connection_pool"
+require_relative "worker_pool"
+require_relative "worker"
+require_relative "request_handler"
+require_relative "lifecycle"
+require_relative "../protocol/protocol"
+require_relative "../protocol/handshake"
+require_relative "../storage/engine"
+require_relative "../transactions/transaction_manager"
+
+module RubyDB
+  module Server
+    # Server - Main database server
+    class Server
+      attr_reader :config, :engine, :listener, :connection_pool
+      attr_reader :worker_pool, :request_handler, :lifecycle, :stats
+
+      def initialize(config = {})
+        @config = {
+          host: config[:host] || "localhost",
+          port: config[:port] || 7432,
+          max_connections: config[:max_connections] || 100,
+          min_workers: config[:min_workers] || 2,
+          max_workers: config[:max_workers] || 20,
+          worker_queue_size: config[:worker_queue_size] || 1000,
+          read_timeout: config[:read_timeout] || 30,
+          write_timeout: config[:write_timeout] || 30,
+          idle_timeout: config[:idle_timeout] || 300,
+          max_request_size: config[:max_request_size] || 10 * 1024 * 1024, # 10MB
+          data_dir: config[:data_dir] || "data",
+          log_dir: config[:log_dir] || "log",
+          pid_file: config[:pid_file] || "rubydb.pid",
+          daemonize: config[:daemonize] || false,
+          authentication: config[:authentication] || {method: "none"},
+          ssl: config[:ssl] || {enabled: false}
+        }.merge(config)
+
+        @stats = {
+          connections_total: 0,
+          connections_active: 0,
+          connections_rejected: 0,
+          requests_processed: 0,
+          requests_failed: 0,
+          bytes_received: 0,
+          bytes_sent: 0,
+          started_at: nil,
+          uptime_seconds: 0,
+          last_activity: nil
+        }
+
+        @lock = Monitor.new
+        @running = false
+        @shutdown = false
+        @stopping = false
+
+        validate_authentication_config!
+        validate_ssl_config!
+        validate_resource_limits!
+
+        # Setup directories
+        setup_directories
+
+        # Initialize components
+        initialize_components
+      end
+
+      def start
+        @lock.synchronize do
+          return if @running
+
+          @running = true
+          @shutdown = false
+          @stats[:started_at] = Time.now
+
+          # Daemonize if configured
+          daemonize if @config[:daemonize]
+
+          # Write PID file
+          write_pid_file
+
+          # Start components
+          @listener.start
+          @connection_pool.start
+          @worker_pool.start
+          @lifecycle.start
+
+          puts "RubyDB Server v#{RubyDB::VERSION} started on #{@config[:host]}:#{@config[:port]}"
+          puts "Data directory: #{@config[:data_dir]}"
+          puts "Max connections: #{@config[:max_connections]}"
+
+          # Update stats thread
+          start_stats_thread
+
+          true
+        end
+      end
+
+      def stop
+        @lock.synchronize do
+          return unless @running
+          return true if @stopping
+
+          @stopping = true
+
+          @shutdown = true
+          puts "Shutting down RubyDB Server..."
+
+          # Stop accepting new connections
+          @listener.stop
+
+          # Close all connections
+          @connection_pool.close_all
+
+          # Stop workers
+          @worker_pool.stop
+
+          # Stop lifecycle
+          @lifecycle.stop
+
+          # Release the database file handle before the data directory is
+          # removed by callers (notably on Windows).
+          @engine&.close
+
+          # Remove PID file
+          remove_pid_file
+
+          @running = false
+          @stopping = false
+
+          puts "RubyDB Server stopped"
+          true
+        end
+      end
+
+      def restart
+        stop
+        sleep(1)
+        start
+      end
+
+      def running?
+        @running
+      end
+
+      def stats
+        @lock.synchronize do
+          elapsed = @stats[:started_at] ? (Time.now - @stats[:started_at]).to_i : 0
+          @stats[:uptime_seconds] = elapsed
+          pool_stats = @connection_pool&.stats || {}
+          @stats[:connections_active] = pool_stats[:active_connections] || 0
+          @stats[:connections_total] = pool_stats[:total_connections] || 0
+          @stats[:requests_processed] = pool_stats[:requests_processed] || 0
+          @stats[:requests_failed] = pool_stats[:requests_failed] || 0
+
+          @stats.merge({
+            engine_stats: @engine&.stats || {},
+            connection_pool_stats: @connection_pool&.stats || {},
+            worker_pool_stats: @worker_pool&.stats || {},
+            memory_usage: get_memory_usage,
+            pid: Process.pid
+          })
+        end
+      end
+
+      private
+
+      def initialize_components
+        # Initialize storage engine
+        db_path = File.join(@config[:data_dir], "rubydb.rdb")
+        @engine = Storage::Engine.new(db_path, @config)
+
+        # Initialize transaction manager
+        @transaction_manager = Transactions::TransactionManager.new(@engine, @config)
+
+        # Initialize protocol
+        @protocol = Protocol::Protocol.new(
+          format: :json,
+          compression: @config[:compression] || false,
+          server_info: {
+            version: RubyDB::VERSION,
+            pid: Process.pid,
+            started_at: Time.now.iso8601,
+            default_auth: authentication_method,
+            authentication_credentials: authentication_credentials
+          }
+        )
+
+        # Initialize connection pool
+        @connection_pool = ConnectionPool.new(@config, @protocol, @engine)
+
+        # Initialize worker pool
+        @worker_pool = WorkerPool.new(@config, @engine, @transaction_manager)
+
+        # Initialize request handler
+        @request_handler = RequestHandler.new(@engine, @transaction_manager, @config)
+
+        # Initialize listener
+        @listener = Listener.new(@config, @connection_pool, @worker_pool)
+
+        # Initialize lifecycle
+        @lifecycle = Lifecycle.new(self, @config)
+      end
+
+      def setup_directories
+        [:data_dir, :log_dir].each do |dir|
+          path = @config[dir]
+          FileUtils.mkdir_p(path) unless Dir.exist?(path)
+        end
+      end
+
+      def authentication_method
+        auth = @config[:authentication] || {}
+        (auth[:method] || auth["method"] || "none").to_s.tr("_", "-")
+      end
+
+      def authentication_credentials
+        auth = @config[:authentication] || {}
+        auth[:credentials] || auth["credentials"] || auth
+      end
+
+      def validate_authentication_config!
+        method = authentication_method
+        supported = %w[none password md5 scram-sha-256]
+        raise ServerError, "Unsupported authentication method: #{method}" unless supported.include?(method)
+
+        return if method == "none"
+
+        credentials = authentication_credentials
+        users = credentials[:users] || credentials["users"]
+        complete = users && !users.empty? ||
+          ((credentials[:username] || credentials["username"]) &&
+           (credentials[:password] || credentials["password"]))
+        raise ServerError, "Authentication credentials are required for #{method}" unless complete
+      end
+
+      def validate_ssl_config!
+        ssl = @config[:ssl] || {}
+        enabled = ssl == true || ssl[:enabled] || ssl["enabled"]
+        return unless enabled
+
+        raise ServerError, "SSL configuration must include certificate and key paths" if ssl == true
+
+        cert = ssl[:cert_file] || ssl["cert_file"]
+        key = ssl[:key_file] || ssl["key_file"]
+        raise ServerError, "ssl.cert_file and ssl.key_file are required when SSL is enabled" unless cert && key
+        raise ServerError, "SSL certificate file does not exist" unless File.file?(cert)
+        raise ServerError, "SSL key file does not exist" unless File.file?(key)
+
+        begin
+          OpenSSL::X509::Certificate.new(File.binread(cert))
+          OpenSSL::PKey.read(File.binread(key))
+        rescue OpenSSL::SSL::SSLError, OpenSSL::X509::CertificateError, OpenSSL::PKey::PKeyError, ArgumentError => e
+          raise ServerError, "SSL certificate or key is invalid: #{e.message}"
+        end
+
+        ca = ssl[:ca_file] || ssl["ca_file"]
+        if (ssl[:verify_peer] || ssl["verify_peer"]) && !ca
+          raise ServerError, "ssl.ca_file is required when peer verification is enabled"
+        end
+        raise ServerError, "SSL CA file does not exist" if ca && !File.file?(ca)
+
+        min_version = ssl[:min_version] || ssl["min_version"] || :TLS1_2
+        unless %i[TLS1_2 TLS1_3].include?(min_version.to_s.to_sym)
+          raise ServerError, "SSL min_version must be TLS1_2 or TLS1_3"
+        end
+      end
+
+      def validate_resource_limits!
+        integer_limits = {
+          max_connections: 1..10_000,
+          min_workers: 1..1_024,
+          max_workers: 1..1_024,
+          worker_queue_size: 1..100_000,
+          max_request_size: 1_024..(64 * 1024 * 1024)
+        }
+        integer_limits.each do |name, allowed|
+          value = Integer(@config[name])
+          raise ServerError, "#{name} must be between #{allowed.begin} and #{allowed.end}" unless allowed.cover?(value)
+          @config[name] = value
+        rescue ArgumentError, TypeError
+          raise ServerError, "#{name} must be an integer"
+        end
+        raise ServerError, "min_workers cannot exceed max_workers" if @config[:min_workers] > @config[:max_workers]
+
+        %i[read_timeout write_timeout idle_timeout].each do |name|
+          value = Float(@config[name])
+          raise ServerError, "#{name} must be greater than zero" unless value.positive?
+          @config[name] = value
+        rescue ArgumentError, TypeError
+          raise ServerError, "#{name} must be a number"
+        end
+      end
+
+      def daemonize
+        return unless @config[:daemonize]
+
+        Process.daemon(true, true)
+        $stdout.reopen(File.join(@config[:log_dir], "rubydb.out"), "a")
+        $stderr.reopen(File.join(@config[:log_dir], "rubydb.err"), "a")
+      end
+
+      def write_pid_file
+        File.write(@config[:pid_file], Process.pid.to_s)
+      end
+
+      def remove_pid_file
+        File.delete(@config[:pid_file]) if File.exist?(@config[:pid_file])
+      end
+
+      def start_stats_thread
+        Thread.new do
+          while @running
+            sleep(10)
+            update_stats
+          end
+        end
+      end
+
+      def update_stats
+        @lock.synchronize do
+          @stats[:connections_active] = @connection_pool&.active_count || 0
+          @stats[:connections_total] = @connection_pool&.total_connections || 0
+          @stats[:requests_processed] = @request_handler&.stats&.dig(:requests_processed) || 0
+        end
+      end
+
+      def get_memory_usage
+        # Get memory usage in MB
+        if File.exist?("/proc/self/status")
+          File.read("/proc/self/status").each_line do |line|
+            if line.start_with?("VmRSS:")
+              return line.split[1].to_i / 1024
+            end
+          end
+        end
+
+        # Fallback - run ps command
+        begin
+          output = `ps -o rss= -p #{Process.pid}`
+          return output.strip.to_i / 1024 if output && !output.empty?
+        rescue
+          # Ignore
+        end
+
+        0
+      rescue
+        0
+      end
+    end
+  end
+end