Module: ActiveRecord::ConnectionAdapters::SQLServer::DatabaseStatements

Defined in:
lib/composite_primary_keys/connection_adapters/sqlserver/database_statements.rb

Instance Method Summary collapse

Instance Method Details

#sql_for_insert(sql, pk, binds) ⇒ Object 



5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
 
# File 'lib/composite_primary_keys/connection_adapters/sqlserver/database_statements.rb', line 5

def sql_for_insert(sql, pk, binds)
  if pk.nil?
    table_name = query_requires_identity_insert?(sql)
    pk = primary_key(table_name)
  end

  sql = if pk && use_output_inserted? && !database_prefix_remote_server?
          # CPK
          #quoted_pk = SQLServer::Utils.extract_identifiers(pk).quoted
          quoted_pk = Array(pk).map {|subkey| SQLServer::Utils.extract_identifiers(subkey).quoted}

          table_name ||= get_table_name(sql)
          exclude_output_inserted = exclude_output_inserted_table_name?(table_name, sql)
          if exclude_output_inserted
            id_sql_type = exclude_output_inserted.is_a?(TrueClass) ? "bigint" : exclude_output_inserted
            # CPK
            # <<~SQL.squish
            #   DECLARE @ssaIdInsertTable table (#{quoted_pk} #{id_sql_type});
            #   #{sql.dup.insert sql.index(/ (DEFAULT )?VALUES/), " OUTPUT INSERTED.#{quoted_pk} INTO @ssaIdInsertTable"}
            #   SELECT CAST(#{quoted_pk.join(',')} AS #{id_sql_type}) FROM @ssaIdInsertTable
            # SQL
            "              DECLARE @ssaIdInsertTable table (\#{quoted_pk.map {|subkey| \"\#{subkey} \#{id_sql_type}\"}.join(\", \")});\n              \#{sql.dup.insert sql.index(/ (DEFAULT )?VALUES/), \" OUTPUT INSERTED.\#{quoted_pk.join(', INSERTED.')} INTO @ssaIdInsertTable\"}\n              SELECT \#{quoted_pk.map {|subkey| \"CAST(\#{subkey} AS \#{id_sql_type}) \#{subkey}\"}.join(\", \")} FROM @ssaIdInsertTable\n            SQL\n          else\n            # CPK\n            # sql.dup.insert sql.index(/ (DEFAULT )?VALUES/), \" OUTPUT INSERTED.\#{quoted_pk}\"\n            sql.dup.insert sql.index(/ (DEFAULT )?VALUES/), \" OUTPUT INSERTED.\#{quoted_pk.join(', INSERTED.')}\"\n          end\n        else\n          \"\#{sql}; SELECT CAST(SCOPE_IDENTITY() AS bigint) AS Ident\"\n        end\n  super\nend\n".squish