rubydb 0.1.4 → 0.1.6

data/lib/rubydb/client/result.rb

@@ -1,123 +1,127 @@
-# frozen_string_literal: true
-
-module RubyDB
-  module Client
-    # Result - Query result set
-    class Result
-      attr_reader :columns, :rows, :row_count, :affected_rows, :sql
-      attr_reader :command_tag, :statement_id, :transaction_id
-      attr_reader :error
-
-      def initialize(data = {})
-        @columns = data[:columns] || []
-        @rows = data[:rows] || []
-        @row_count = data[:row_count] || @rows.size
-        @affected_rows = data[:affected_rows] || 0
-        @sql = data[:sql]
-        @command_tag = data[:command_tag] || "SELECT"
-        @statement_id = data[:statement_id]
-        @transaction_id = data[:transaction_id]
-        @success = data[:success] != false
-        @error = data[:error]
-        @warnings = data[:warnings] || []
-        @notice = data[:notice]
-        @execution_time_ms = data[:execution_time_ms] || 0
-        @created_at = Time.now
-        @position = 0
-      end
-
-      def success?
-        @success
-      end
-
-      def error?
-        !@success && !@error.nil?
-      end
-
-      def empty?
-        @rows.empty?
-      end
-
-      def first
-        @rows.first
-      end
-
-      def last
-        @rows.last
-      end
-
-      def [](index)
-        @rows[index]
-      end
-
-      def each
-        @rows.each { |row| yield row }
-      end
-
-      def each_with_index
-        @rows.each_with_index { |row, idx| yield row, idx }
-      end
-
-      def each_row
-        @rows.each { |row| yield row }
-      end
-
-      def each_column
-        @columns.each { |col| yield col }
-      end
-
-      def column_names
-        @columns.map { |c| c[:name] || c }
-      end
-
-      def column_types
-        @columns.map { |c| c[:type] || "text" }
-      end
-
-      def to_a
-        @rows
-      end
-
-      def to_hash
-        {
-          columns: @columns,
-          rows: @rows,
-          row_count: @row_count,
-          affected_rows: @affected_rows,
-          command_tag: @command_tag,
-          success: @success,
-          error: @error,
-          execution_time_ms: @execution_time_ms
-        }
-      end
-
-      def to_json
-        JSON.generate(to_hash)
-      end
-
-      def inspect
-        rows_preview = @rows.first(3).map(&:inspect).join(", ")
-        more = @rows.size > 3 ? "..." : ""
-        "#<Result rows=#{@row_count} columns=#{@columns.size} data=[#{rows_preview}#{more}]>"
-      end
-
-      def to_s
-        if @rows.empty?
-          "Result: #{@command_tag} (0 rows)"
-        else
-          "Result: #{@command_tag} (#{@row_count} rows)"
-        end
-      end
-
-      # Enumerable support
-      include Enumerable
-
-      def size
-        @row_count
-      end
-
-      alias length size
-      alias count size
-    end
-  end
-end
+# frozen_string_literal: true
+
+module RubyDB
+  module Client
+    # Result - Query result set
+    class Result
+      attr_reader :columns, :rows, :row_count, :affected_rows, :sql
+      attr_reader :command_tag, :statement_id, :transaction_id, :row_id, :inserted_id
+      attr_reader :error
+
+      def initialize(data = {})
+        @columns = data[:columns] || []
+        @rows = data[:rows] || []
+        @row_count = data[:row_count] || @rows.size
+        @affected_rows = data[:affected_rows] || 0
+        @sql = data[:sql]
+        @command_tag = data[:command_tag] || "SELECT"
+        @statement_id = data[:statement_id]
+        @transaction_id = data[:transaction_id]
+        @row_id = data[:row_id]
+        @inserted_id = data[:inserted_id]
+        @success = data[:success] != false
+        @error = data[:error]
+        @warnings = data[:warnings] || []
+        @notice = data[:notice]
+        @execution_time_ms = data[:execution_time_ms] || 0
+        @created_at = Time.now
+        @position = 0
+      end
+
+      def success?
+        @success
+      end
+
+      def error?
+        !@success && !@error.nil?
+      end
+
+      def empty?
+        @rows.empty?
+      end
+
+      def first
+        @rows.first
+      end
+
+      def last
+        @rows.last
+      end
+
+      def [](index)
+        @rows[index]
+      end
+
+      def each
+        @rows.each { |row| yield row }
+      end
+
+      def each_with_index
+        @rows.each_with_index { |row, idx| yield row, idx }
+      end
+
+      def each_row
+        @rows.each { |row| yield row }
+      end
+
+      def each_column
+        @columns.each { |col| yield col }
+      end
+
+      def column_names
+        @columns.map { |c| c[:name] || c }
+      end
+
+      def column_types
+        @columns.map { |c| c[:type] || "text" }
+      end
+
+      def to_a
+        @rows
+      end
+
+      def to_hash
+        {
+          columns: @columns,
+          rows: @rows,
+          row_count: @row_count,
+          affected_rows: @affected_rows,
+          command_tag: @command_tag,
+          row_id: @row_id,
+          inserted_id: @inserted_id,
+          success: @success,
+          error: @error,
+          execution_time_ms: @execution_time_ms
+        }
+      end
+
+      def to_json
+        JSON.generate(to_hash)
+      end
+
+      def inspect
+        rows_preview = @rows.first(3).map(&:inspect).join(", ")
+        more = (@rows.size > 3) ? "..." : ""
+        "#<Result rows=#{@row_count} columns=#{@columns.size} data=[#{rows_preview}#{more}]>"
+      end
+
+      def to_s
+        if @rows.empty?
+          "Result: #{@command_tag} (0 rows)"
+        else
+          "Result: #{@command_tag} (#{@row_count} rows)"
+        end
+      end
+
+      # Enumerable support
+      include Enumerable
+
+      def size
+        @row_count
+      end
+
+      alias_method :length, :size
+      alias_method :count, :size
+    end
+  end
+end

data/lib/rubydb/client/statement.rb

@@ -1,52 +1,52 @@
-# frozen_string_literal: true
-
-module RubyDB
-  module Client
-    # Statement - SQL statement
-    class Statement
-      attr_reader :sql, :params, :client, :result
-
-      def initialize(sql, client)
-        @sql = sql
-        @client = client
-        @result = nil
-        @executed = false
-        @params = []
-        @param_count = sql.scan(/\$/).size
-        @created_at = Time.now
-        @lock = Mutex.new
-      end
-
-      def execute(params = [])
-        @lock.synchronize do
-          @params = params
-          @result = @client.query(@sql, params)
-          @executed = true
-          @result
-        end
-      end
-
-      def execute_batch(params_list)
-        @lock.synchronize do
-          results = []
-          params_list.each do |params|
-            results << execute(params)
-          end
-          results
-        end
-      end
-
-      def executed?
-        @executed
-      end
-
-      def to_s
-        @sql
-      end
-
-      def inspect
-        "#<Statement sql=\"#{@sql[0..50]}#{'...' if @sql.length > 50}\">"
-      end
-    end
-  end
-end
+# frozen_string_literal: true
+
+module RubyDB
+  module Client
+    # Statement - SQL statement
+    class Statement
+      attr_reader :sql, :params, :client, :result
+
+      def initialize(sql, client)
+        @sql = sql
+        @client = client
+        @result = nil
+        @executed = false
+        @params = []
+        @param_count = sql.scan("$").size
+        @created_at = Time.now
+        @lock = Mutex.new
+      end
+
+      def execute(params = [])
+        @lock.synchronize do
+          @params = params
+          @result = @client.query(@sql, params)
+          @executed = true
+          @result
+        end
+      end
+
+      def execute_batch(params_list)
+        @lock.synchronize do
+          results = []
+          params_list.each do |params|
+            results << execute(params)
+          end
+          results
+        end
+      end
+
+      def executed?
+        @executed
+      end
+
+      def to_s
+        @sql
+      end
+
+      def inspect
+        "#<Statement sql=\"#{@sql[0..50]}#{"..." if @sql.length > 50}\">"
+      end
+    end
+  end
+end

data/lib/rubydb/client/transaction.rb

@@ -1,130 +1,130 @@
-# frozen_string_literal: true
-
-module RubyDB
-  module Client
-    # Transaction - Database transaction
-    class Transaction
-      attr_reader :id, :client, :started_at, :options
-
-      def initialize(client, options = {})
-        @client = client
-        @options = options
-        @id = "txn_#{Time.now.to_i}_#{rand(1000)}"
-        @started_at = Time.now
-        @active = true
-        @committed = false
-        @rolled_back = false
-        @savepoints = []
-        @lock = Mutex.new
-      end
-
-      def query(sql, params = [])
-        @lock.synchronize do
-          raise ClientError, "Transaction is not active" unless @active
-          @client.query(sql, params)
-        end
-      end
-
-      def prepare(sql)
-        @lock.synchronize do
-          raise ClientError, "Transaction is not active" unless @active
-          @client.prepare(sql)
-        end
-      end
-
-      def execute(statement_id, params = [])
-        @lock.synchronize do
-          raise ClientError, "Transaction is not active" unless @active
-          @client.execute(statement_id, params)
-        end
-      end
-
-      def savepoint(name)
-        @lock.synchronize do
-          raise ClientError, "Transaction is not active" unless @active
-          @savepoints << name
-          @client.query("SAVEPOINT #{name}")
-        end
-      end
-
-      def rollback_to_savepoint(name)
-        @lock.synchronize do
-          raise ClientError, "Transaction is not active" unless @active
-          unless @savepoints.include?(name)
-            raise ClientError, "Savepoint '#{name}' not found"
-          end
-          @client.query("ROLLBACK TO SAVEPOINT #{name}")
-          @savepoints.delete(name)
-        end
-      end
-
-      def release_savepoint(name)
-        @lock.synchronize do
-          raise ClientError, "Transaction is not active" unless @active
-          unless @savepoints.include?(name)
-            raise ClientError, "Savepoint '#{name}' not found"
-          end
-          @client.query("RELEASE SAVEPOINT #{name}")
-          @savepoints.delete(name)
-        end
-      end
-
-      def commit
-        @lock.synchronize do
-          raise ClientError, "Transaction is not active" unless @active
-        end
-        @client.commit
-      end
-
-      def rollback
-        @lock.synchronize do
-          raise ClientError, "Transaction is not active" unless @active
-        end
-        @client.rollback
-      end
-
-      def active?
-        @active
-      end
-
-      def committed?
-        @committed
-      end
-
-      def rolled_back?
-        @rolled_back
-      end
-
-      # Used by Client#commit/#rollback after the wire request has completed.
-      def mark_committed
-        @lock.synchronize do
-          @active = false
-          @committed = true
-        end
-      end
-
-      def mark_rolled_back
-        @lock.synchronize do
-          @active = false
-          @rolled_back = true
-        end
-      end
-
-      def to_hash
-        {
-          id: @id,
-          started_at: @started_at.iso8601,
-          active: @active,
-          committed: @committed,
-          rolled_back: @rolled_back,
-          savepoints: @savepoints,
-          options: @options
-        }
-      end
-
-      def inspect
-        "#<Transaction id=#{@id} active=#{@active}>"
-      end
-    end
-  end
-end
+# frozen_string_literal: true
+
+module RubyDB
+  module Client
+    # Transaction - Database transaction
+    class Transaction
+      attr_reader :id, :client, :started_at, :options
+
+      def initialize(client, options = {})
+        @client = client
+        @options = options
+        @id = "txn_#{Time.now.to_i}_#{rand(1000)}"
+        @started_at = Time.now
+        @active = true
+        @committed = false
+        @rolled_back = false
+        @savepoints = []
+        @lock = Mutex.new
+      end
+
+      def query(sql, params = [])
+        @lock.synchronize do
+          raise ClientError, "Transaction is not active" unless @active
+          @client.query(sql, params)
+        end
+      end
+
+      def prepare(sql)
+        @lock.synchronize do
+          raise ClientError, "Transaction is not active" unless @active
+          @client.prepare(sql)
+        end
+      end
+
+      def execute(statement_id, params = [])
+        @lock.synchronize do
+          raise ClientError, "Transaction is not active" unless @active
+          @client.execute(statement_id, params)
+        end
+      end
+
+      def savepoint(name)
+        @lock.synchronize do
+          raise ClientError, "Transaction is not active" unless @active
+          @savepoints << name
+          @client.query("SAVEPOINT #{name}")
+        end
+      end
+
+      def rollback_to_savepoint(name)
+        @lock.synchronize do
+          raise ClientError, "Transaction is not active" unless @active
+          unless @savepoints.include?(name)
+            raise ClientError, "Savepoint '#{name}' not found"
+          end
+          @client.query("ROLLBACK TO SAVEPOINT #{name}")
+          @savepoints.delete(name)
+        end
+      end
+
+      def release_savepoint(name)
+        @lock.synchronize do
+          raise ClientError, "Transaction is not active" unless @active
+          unless @savepoints.include?(name)
+            raise ClientError, "Savepoint '#{name}' not found"
+          end
+          @client.query("RELEASE SAVEPOINT #{name}")
+          @savepoints.delete(name)
+        end
+      end
+
+      def commit
+        @lock.synchronize do
+          raise ClientError, "Transaction is not active" unless @active
+        end
+        @client.commit
+      end
+
+      def rollback
+        @lock.synchronize do
+          raise ClientError, "Transaction is not active" unless @active
+        end
+        @client.rollback
+      end
+
+      def active?
+        @active
+      end
+
+      def committed?
+        @committed
+      end
+
+      def rolled_back?
+        @rolled_back
+      end
+
+      # Used by Client#commit/#rollback after the wire request has completed.
+      def mark_committed
+        @lock.synchronize do
+          @active = false
+          @committed = true
+        end
+      end
+
+      def mark_rolled_back
+        @lock.synchronize do
+          @active = false
+          @rolled_back = true
+        end
+      end
+
+      def to_hash
+        {
+          id: @id,
+          started_at: @started_at.iso8601,
+          active: @active,
+          committed: @committed,
+          rolled_back: @rolled_back,
+          savepoints: @savepoints,
+          options: @options
+        }
+      end
+
+      def inspect
+        "#<Transaction id=#{@id} active=#{@active}>"
+      end
+    end
+  end
+end

data/lib/rubydb/concurrency/concurrency.rb

@@ -1,19 +1,19 @@
-# lib/rubydb/concurrency.rb
-# frozen_string_literal: true
-
-require "etc"
-
-# Main entry point for Concurrency module
-require_relative "concurrency/scheduler"
-require_relative "concurrency/latch"
-require_relative "concurrency/mutex"
-require_relative "concurrency/rw_lock"
-require_relative "concurrency/deadlock_detector"
-require_relative "concurrency/lock_graph"
-require_relative "concurrency/worker_pool"
-
-module RubyDB
-  module Concurrency
-    # All classes are now loaded
-  end
-end
+# lib/rubydb/concurrency.rb
+# frozen_string_literal: true
+
+require "etc"
+
+# Main entry point for Concurrency module
+require_relative "concurrency/scheduler"
+require_relative "concurrency/latch"
+require_relative "concurrency/mutex"
+require_relative "concurrency/rw_lock"
+require_relative "concurrency/deadlock_detector"
+require_relative "concurrency/lock_graph"
+require_relative "concurrency/worker_pool"
+
+module RubyDB
+  module Concurrency
+    # All classes are now loaded
+  end
+end