sequel 4.36.0 → 5.61.0

data/lib/sequel/adapters/mock.rb

@@ -1,8 +1,9 @@
 # frozen-string-literal: true
 
+require_relative 'utils/unmodified_identifiers'
+
 module Sequel
   module Mock
-    # Connection class for Sequel's mock adapter.
     class Connection
       # Sequel::Mock::Database object that created this connection
       attr_reader :db
@@ -28,66 +29,9 @@ module Sequel
       end
     end
 
-    # Database class for Sequel's mock adapter.
     class Database < Sequel::Database
       set_adapter_scheme :mock
 
-      # Map of database type names to module names, used for handling
-      # mock adapters for specific database types.
-      SHARED_ADAPTERS = {
-        'access'=>'Access',
-        'cubrid'=>'Cubrid',
-        'db2'=>'DB2',
-        'firebird'=>'Firebird',
-        'informix'=>'Informix',
-        'mssql'=>'MSSQL',
-        'mysql'=>'MySQL',
-        'oracle'=>'Oracle',
-        'postgres'=>'Postgres',
-        'sqlanywhere'=>'SqlAnywhere',
-        'sqlite'=>'SQLite'
-      }
-
-      # Procs to run for specific database types to get the mock adapter
-      # to work with the shared adapter
-      SHARED_ADAPTER_SETUP = {
-        'postgres' => lambda do |db|
-          db.instance_eval do
-            @server_version = 90400
-            initialize_postgres_adapter
-          end
-          db.extend(Module.new do
-            def bound_variable_arg(arg, conn)
-              arg
-            end
-
-            def primary_key(table)
-              :id
-            end
-          end)
-        end,
-        'oracle' => lambda do |db|
-          db.instance_eval do
-            @primary_key_sequences = {}
-          end
-        end,
-        'mysql' => lambda do |db|
-          db.instance_eval do
-            @server_version = 50617
-          end
-        end,
-        'mssql' => lambda do |db|
-          db.instance_eval do
-            @server_version = 11000000
-          end
-        end,
-        'sqlite' => lambda do |db|
-          db.instance_eval do
-            @sqlite_version = 30804
-          end
-        end
-      }
-
       # Set the autogenerated primary key integer
       # to be returned when running an insert query.
       # Argument types supported:
@@ -102,14 +46,22 @@ module Sequel
       #         the value returned
       # Class :: Should be an Exception subclass, will create a new
       #          instance an raise it wrapped in a DatabaseError.
-      attr_writer :autoid
+      def autoid=(v)
+        @autoid = case v
+        when Integer
+          i = v - 1
+          proc{@mutex.synchronize{i+=1}}
+        else
+          v
+        end
+      end
 
       # Set the columns to set in the dataset when the dataset fetches
       # rows.  Argument types supported:
       # nil :: Set no columns
-      # Array of Symbols: Used for all datasets
-      # Array (otherwise): First retrieval gets the first value in the
-      #                    array, second gets the second value, etc.
+      # Array of Symbols :: Used for all datasets
+      # Array (otherwise) :: First retrieval gets the first value in the
+      #                      array, second gets the second value, etc.
       # Proc :: Called with the select SQL query, uses the value
       #         returned, which should be an array of symbols
       attr_writer :columns
@@ -174,9 +126,11 @@ module Sequel
       # Return all stored SQL queries, and clear the cache
       # of SQL queries.
       def sqls
-        s = @sqls.dup
-        @sqls.clear
-        s
+        @mutex.synchronize do
+          s = @sqls.dup
+          @sqls.clear
+          s
+        end
       end
 
       # Enable use of savepoints.
@@ -186,26 +140,12 @@ module Sequel
 
       private
 
-      def _autoid(sql, v, ds=nil)
-        case v
-        when Integer
-          if ds
-            ds.autoid += 1 if ds.autoid.is_a?(Integer)
-          else
-            @autoid += 1
-          end
-          v
-        else
-          _nextres(v, sql, nil)
-        end
-      end
-
       def _execute(c, sql, opts=OPTS, &block)
         sql += " -- args: #{opts[:arguments].inspect}" if opts[:arguments]
         sql += " -- #{@opts[:append]}" if @opts[:append]
         sql += " -- #{c.server.is_a?(Symbol) ? c.server : c.server.inspect}" if c.server != :default
         log_connection_yield(sql, c){} unless opts[:log] == false
-        @sqls << sql 
+        @mutex.synchronize{@sqls << sql}
 
         ds = opts[:dataset]
         begin
@@ -216,8 +156,15 @@ module Sequel
             if meth == :numrows
               _numrows(sql, (ds.numrows if ds) || @numrows)
             else
-              v = ds.autoid if ds
-              _autoid(sql, v || @autoid, (ds if v))
+              if ds
+                @mutex.synchronize do
+                  v = ds.autoid
+                  if v.is_a?(Integer)
+                    ds.send(:cache_set, :_autoid, v + 1)
+                  end
+                  v
+                end
+              end || _nextres(@autoid, sql, nil)
             end
           end
         rescue => e
@@ -233,7 +180,7 @@ module Sequel
           if f.all?{|h| h.is_a?(Hash)}
             f.each{|h| yield h.dup}
           else
-            _fetch(sql, f.shift, &block)
+            _fetch(sql, @mutex.synchronize{f.shift}, &block)
           end
         when Proc
           h = f.call(sql)
@@ -246,7 +193,7 @@ module Sequel
           if f < Exception
             raise f
           else
-            raise Error, "Invalid @autoid/@numrows attribute: #{v.inspect}"
+            raise Error, "Invalid @fetch attribute: #{v.inspect}"
           end
         when nil
           # nothing
@@ -260,7 +207,7 @@ module Sequel
         when Integer
           v
         when Array
-          v.empty? ? default : _nextres(v.shift, sql, default)
+          v.empty? ? default : _nextres(@mutex.synchronize{v.shift}, sql, default)
         when Proc
           v.call(sql)
         when Class
@@ -290,18 +237,37 @@ module Sequel
       # :sqls :: The array to store the SQL queries in.
       def adapter_initialize
         opts = @opts
+        @mutex = Mutex.new
         @sqls = opts[:sqls] || []
-        if mod_name = SHARED_ADAPTERS[opts[:host]]
-          @shared_adapter = true
-          require "sequel/adapters/shared/#{opts[:host]}"
-          extend Sequel.const_get(mod_name)::DatabaseMethods
-          extend_datasets Sequel.const_get(mod_name)::DatasetMethods
-          if pr = SHARED_ADAPTER_SETUP[opts[:host]]
-            pr.call(self)
+        @shared_adapter = false
+
+        case db_type = opts[:host] 
+        when String, Symbol
+          db_type = db_type.to_sym
+          unless mod = Sequel.synchronize{SHARED_ADAPTER_MAP[db_type]}
+            begin
+              require "sequel/adapters/shared/#{db_type}"
+            rescue LoadError
+            else
+              mod = Sequel.synchronize{SHARED_ADAPTER_MAP[db_type]}
+            end
           end
-        else
-          @shared_adapter = false
+
+          if mod
+            @shared_adapter = true
+            extend(mod::DatabaseMethods)
+            extend_datasets(mod::DatasetMethods)
+            if mod.respond_to?(:mock_adapter_setup)
+              mod.mock_adapter_setup(self)
+            end
+          end
+        end
+
+        unless @shared_adapter
+          extend UnmodifiedIdentifiers::DatabaseMethods
+          extend_datasets UnmodifiedIdentifiers::DatasetMethods
         end
+
         self.autoid = opts[:autoid]
         self.columns = opts[:columns]
         self.fetch = opts[:fetch]
@@ -317,7 +283,7 @@ module Sequel
             if cs.all?{|c| c.is_a?(Symbol)}
               ds.columns(*cs)
             else
-              columns(ds, sql, cs.shift)
+              columns(ds, sql, @mutex.synchronize{cs.shift})
             end
           end
         when Proc
@@ -329,16 +295,12 @@ module Sequel
         end
       end
 
-      def quote_identifiers_default
-        shared_adapter? ? super : false
+      def dataset_class_default
+        Dataset
       end
 
-      def identifier_input_method_default
-        shared_adapter? ? super : nil
-      end
-
-      def identifier_output_method_default
-        shared_adapter? ? super : nil
+      def quote_identifiers_default
+        shared_adapter? ? super : false
       end
 
       def shared_adapter?
@@ -347,16 +309,20 @@ module Sequel
     end
 
     class Dataset < Sequel::Dataset
-      Database::DatasetClass = self
-
-      # Override the databases's autoid setting for this dataset
-      attr_accessor :autoid
+      # The autoid setting for this dataset, if it has been overridden
+      def autoid
+        cache_get(:_autoid) || @opts[:autoid]
+      end
 
-      # Override the databases's fetch setting for this dataset
-      attr_accessor :_fetch
+      # The fetch setting for this dataset, if it has been overridden
+      def _fetch
+        cache_get(:_fetch) || @opts[:fetch]
+      end
 
-      # Override the databases's numrows setting for this dataset
-      attr_accessor :numrows
+      # The numrows setting for this dataset, if it has been overridden
+      def numrows
+        cache_get(:_numrows) || @opts[:numrows]
+      end
 
       # If arguments are provided, use them to set the columns
       # for this dataset and return self.  Otherwise, use the
@@ -374,6 +340,25 @@ module Sequel
         execute(sql, &block)
       end
 
+      def quote_identifiers?
+        @opts.fetch(:quote_identifiers, db.send(:quote_identifiers_default))
+      end
+
+      # Return cloned dataset with the autoid setting modified
+      def with_autoid(autoid)
+        clone(:autoid=>autoid)
+      end
+
+      # Return cloned dataset with the fetch setting modified
+      def with_fetch(fetch)
+        clone(:fetch=>fetch)
+      end
+
+      # Return cloned dataset with the numrows setting modified
+      def with_numrows(numrows)
+        clone(:numrows=>numrows)
+      end
+
       private
 
       def execute(sql, opts=OPTS, &block)
@@ -387,6 +372,10 @@ module Sequel
       def execute_insert(sql, opts=OPTS, &block)
         super(sql, opts.merge(:dataset=>self), &block)
       end
+
+      def non_sql_option?(key)
+        super || key == :fetch || key == :numrows || key == :autoid
+      end
     end
   end
 end