knex 3.0.1 → 3.2.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (199) hide show
  1. package/CHANGELOG.md +2431 -2358
  2. package/CONTRIBUTING.md +190 -194
  3. package/LICENSE +22 -22
  4. package/README.md +156 -149
  5. package/UPGRADING.md +245 -245
  6. package/bin/cli.js +516 -475
  7. package/bin/knexfile-runtime-error.js +27 -0
  8. package/bin/utils/cli-config-utils.js +217 -212
  9. package/bin/utils/constants.js +7 -7
  10. package/bin/utils/migrationsLister.js +37 -37
  11. package/knex.js +23 -23
  12. package/knex.mjs +11 -11
  13. package/lib/builder-interface-augmenter.js +120 -120
  14. package/lib/client.js +585 -495
  15. package/lib/constants.js +61 -61
  16. package/lib/dialects/better-sqlite3/index.js +101 -77
  17. package/lib/dialects/cockroachdb/crdb-columncompiler.js +14 -14
  18. package/lib/dialects/cockroachdb/crdb-querybuilder.js +11 -11
  19. package/lib/dialects/cockroachdb/crdb-querycompiler.js +122 -122
  20. package/lib/dialects/cockroachdb/crdb-tablecompiler.js +46 -37
  21. package/lib/dialects/cockroachdb/crdb-viewcompiler.js +15 -15
  22. package/lib/dialects/cockroachdb/index.js +86 -86
  23. package/lib/dialects/index.js +34 -34
  24. package/lib/dialects/mssql/index.js +498 -500
  25. package/lib/dialects/mssql/mssql-formatter.js +34 -34
  26. package/lib/dialects/mssql/query/mssql-querycompiler.js +601 -601
  27. package/lib/dialects/mssql/schema/mssql-columncompiler.js +185 -185
  28. package/lib/dialects/mssql/schema/mssql-compiler.js +91 -91
  29. package/lib/dialects/mssql/schema/mssql-tablecompiler.js +393 -378
  30. package/lib/dialects/mssql/schema/mssql-viewcompiler.js +55 -55
  31. package/lib/dialects/mssql/transaction.js +176 -176
  32. package/lib/dialects/mysql/index.js +317 -201
  33. package/lib/dialects/mysql/query/mysql-querybuilder.js +14 -0
  34. package/lib/dialects/mysql/query/mysql-querycompiler.js +292 -274
  35. package/lib/dialects/mysql/schema/mysql-columncompiler.js +193 -193
  36. package/lib/dialects/mysql/schema/mysql-compiler.js +60 -60
  37. package/lib/dialects/mysql/schema/mysql-tablecompiler.js +426 -381
  38. package/lib/dialects/mysql/schema/mysql-viewbuilder.js +21 -21
  39. package/lib/dialects/mysql/schema/mysql-viewcompiler.js +15 -15
  40. package/lib/dialects/mysql/transaction.js +46 -46
  41. package/lib/dialects/mysql2/index.js +53 -53
  42. package/lib/dialects/mysql2/transaction.js +44 -44
  43. package/lib/dialects/oracle/DEAD_CODE.md +5 -5
  44. package/lib/dialects/oracle/index.js +92 -92
  45. package/lib/dialects/oracle/query/oracle-querycompiler.js +343 -343
  46. package/lib/dialects/oracle/schema/internal/incrementUtils.js +22 -20
  47. package/lib/dialects/oracle/schema/internal/trigger.js +155 -135
  48. package/lib/dialects/oracle/schema/oracle-columnbuilder.js +17 -17
  49. package/lib/dialects/oracle/schema/oracle-columncompiler.js +126 -126
  50. package/lib/dialects/oracle/schema/oracle-compiler.js +124 -122
  51. package/lib/dialects/oracle/schema/oracle-tablecompiler.js +210 -190
  52. package/lib/dialects/oracle/utils.js +107 -87
  53. package/lib/dialects/oracledb/index.js +381 -327
  54. package/lib/dialects/oracledb/query/oracledb-querycompiler.js +481 -481
  55. package/lib/dialects/oracledb/schema/oracledb-columncompiler.js +61 -61
  56. package/lib/dialects/oracledb/schema/oracledb-tablecompiler.js +19 -19
  57. package/lib/dialects/oracledb/schema/oracledb-viewbuilder.js +13 -13
  58. package/lib/dialects/oracledb/schema/oracledb-viewcompiler.js +19 -19
  59. package/lib/dialects/oracledb/transaction.js +98 -98
  60. package/lib/dialects/oracledb/utils.js +208 -208
  61. package/lib/dialects/pgnative/index.js +60 -60
  62. package/lib/dialects/postgres/execution/pg-transaction.js +19 -19
  63. package/lib/dialects/postgres/index.js +373 -358
  64. package/lib/dialects/postgres/query/pg-querybuilder.js +43 -43
  65. package/lib/dialects/postgres/query/pg-querycompiler.js +400 -400
  66. package/lib/dialects/postgres/schema/pg-columncompiler.js +162 -156
  67. package/lib/dialects/postgres/schema/pg-compiler.js +138 -138
  68. package/lib/dialects/postgres/schema/pg-tablecompiler.js +331 -304
  69. package/lib/dialects/postgres/schema/pg-viewbuilder.js +21 -21
  70. package/lib/dialects/postgres/schema/pg-viewcompiler.js +35 -35
  71. package/lib/dialects/redshift/index.js +86 -86
  72. package/lib/dialects/redshift/query/redshift-querycompiler.js +163 -163
  73. package/lib/dialects/redshift/schema/redshift-columnbuilder.js +22 -22
  74. package/lib/dialects/redshift/schema/redshift-columncompiler.js +67 -67
  75. package/lib/dialects/redshift/schema/redshift-compiler.js +14 -14
  76. package/lib/dialects/redshift/schema/redshift-tablecompiler.js +134 -122
  77. package/lib/dialects/redshift/schema/redshift-viewcompiler.js +11 -11
  78. package/lib/dialects/redshift/transaction.js +32 -32
  79. package/lib/dialects/sqlite3/execution/sqlite-transaction.js +172 -25
  80. package/lib/dialects/sqlite3/index.js +263 -250
  81. package/lib/dialects/sqlite3/query/sqlite-querybuilder.js +33 -33
  82. package/lib/dialects/sqlite3/query/sqlite-querycompiler.js +341 -334
  83. package/lib/dialects/sqlite3/schema/ddl.js +380 -400
  84. package/lib/dialects/sqlite3/schema/internal/compiler.js +327 -327
  85. package/lib/dialects/sqlite3/schema/internal/parser-combinator.js +161 -161
  86. package/lib/dialects/sqlite3/schema/internal/parser.js +638 -638
  87. package/lib/dialects/sqlite3/schema/internal/sqlite-ddl-operations.js +41 -41
  88. package/lib/dialects/sqlite3/schema/internal/tokenizer.js +38 -38
  89. package/lib/dialects/sqlite3/schema/internal/utils.js +12 -12
  90. package/lib/dialects/sqlite3/schema/sqlite-columncompiler.js +50 -50
  91. package/lib/dialects/sqlite3/schema/sqlite-compiler.js +80 -80
  92. package/lib/dialects/sqlite3/schema/sqlite-tablecompiler.js +364 -347
  93. package/lib/dialects/sqlite3/schema/sqlite-viewcompiler.js +40 -40
  94. package/lib/execution/batch-insert.js +51 -51
  95. package/lib/execution/internal/delay.js +6 -6
  96. package/lib/execution/internal/ensure-connection-callback.js +41 -41
  97. package/lib/execution/internal/query-executioner.js +62 -62
  98. package/lib/execution/runner.js +325 -325
  99. package/lib/execution/transaction.js +417 -409
  100. package/lib/formatter/formatterUtils.js +42 -42
  101. package/lib/formatter/rawFormatter.js +84 -84
  102. package/lib/formatter/wrappingFormatter.js +253 -250
  103. package/lib/formatter.js +25 -25
  104. package/lib/index.js +3 -3
  105. package/lib/knex-builder/FunctionHelper.js +80 -80
  106. package/lib/knex-builder/Knex.js +59 -59
  107. package/lib/knex-builder/internal/config-resolver.js +57 -57
  108. package/lib/knex-builder/internal/parse-connection.js +87 -87
  109. package/lib/knex-builder/make-knex.js +345 -345
  110. package/lib/logger.js +76 -76
  111. package/lib/migrations/common/MigrationsLoader.js +36 -36
  112. package/lib/migrations/migrate/MigrationGenerator.js +84 -84
  113. package/lib/migrations/migrate/Migrator.js +632 -599
  114. package/lib/migrations/migrate/migrate-stub.js +17 -17
  115. package/lib/migrations/migrate/migration-list-resolver.js +33 -33
  116. package/lib/migrations/migrate/migrator-configuration-merger.js +58 -58
  117. package/lib/migrations/migrate/sources/fs-migrations.js +74 -74
  118. package/lib/migrations/migrate/stub/cjs.stub +15 -15
  119. package/lib/migrations/migrate/stub/coffee.stub +13 -13
  120. package/lib/migrations/migrate/stub/eg.stub +14 -14
  121. package/lib/migrations/migrate/stub/js-schema.stub +22 -22
  122. package/lib/migrations/migrate/stub/js.stub +22 -22
  123. package/lib/migrations/migrate/stub/knexfile-coffee.stub +34 -34
  124. package/lib/migrations/migrate/stub/knexfile-eg.stub +43 -43
  125. package/lib/migrations/migrate/stub/knexfile-js.stub +47 -47
  126. package/lib/migrations/migrate/stub/knexfile-ls.stub +35 -35
  127. package/lib/migrations/migrate/stub/knexfile-ts.stub +47 -47
  128. package/lib/migrations/migrate/stub/ls.stub +14 -14
  129. package/lib/migrations/migrate/stub/mjs.stub +23 -23
  130. package/lib/migrations/migrate/stub/ts-schema.stub +21 -21
  131. package/lib/migrations/migrate/stub/ts.stub +21 -21
  132. package/lib/migrations/migrate/table-creator.js +77 -77
  133. package/lib/migrations/migrate/table-resolver.js +27 -27
  134. package/lib/migrations/seed/Seeder.js +137 -137
  135. package/lib/migrations/seed/seed-stub.js +13 -13
  136. package/lib/migrations/seed/seeder-configuration-merger.js +60 -60
  137. package/lib/migrations/seed/sources/fs-seeds.js +65 -65
  138. package/lib/migrations/seed/stub/coffee.stub +9 -9
  139. package/lib/migrations/seed/stub/eg.stub +11 -11
  140. package/lib/migrations/seed/stub/js.stub +13 -13
  141. package/lib/migrations/seed/stub/ls.stub +11 -11
  142. package/lib/migrations/seed/stub/mjs.stub +12 -12
  143. package/lib/migrations/seed/stub/ts.stub +13 -13
  144. package/lib/migrations/util/fs.js +86 -86
  145. package/lib/migrations/util/import-file.js +12 -12
  146. package/lib/migrations/util/is-module-type.js +9 -9
  147. package/lib/migrations/util/template.js +52 -52
  148. package/lib/migrations/util/timestamp.js +14 -14
  149. package/lib/query/analytic.js +52 -52
  150. package/lib/query/constants.js +15 -15
  151. package/lib/query/joinclause.js +270 -270
  152. package/lib/query/method-constants.js +136 -136
  153. package/lib/query/querybuilder.js +1793 -1793
  154. package/lib/query/querycompiler.js +1634 -1591
  155. package/lib/raw.js +139 -139
  156. package/lib/ref.js +39 -39
  157. package/lib/schema/builder.js +115 -115
  158. package/lib/schema/columnbuilder.js +146 -146
  159. package/lib/schema/columncompiler.js +307 -307
  160. package/lib/schema/compiler.js +187 -187
  161. package/lib/schema/internal/helpers.js +55 -55
  162. package/lib/schema/tablebuilder.js +379 -376
  163. package/lib/schema/tablecompiler.js +450 -433
  164. package/lib/schema/viewbuilder.js +92 -92
  165. package/lib/schema/viewcompiler.js +138 -138
  166. package/lib/util/finally-mixin.js +13 -13
  167. package/lib/util/helpers.js +95 -95
  168. package/lib/util/is.js +32 -32
  169. package/lib/util/nanoid.js +40 -40
  170. package/lib/util/noop.js +1 -1
  171. package/lib/util/save-async-stack.js +14 -14
  172. package/lib/util/security.js +32 -26
  173. package/lib/util/string.js +190 -190
  174. package/lib/util/timeout.js +29 -29
  175. package/package.json +285 -266
  176. package/scripts/act-testing/act.sh +19 -0
  177. package/scripts/act-testing/merged-no-label.json +11 -0
  178. package/scripts/act-testing/merged-patch-labeled.json +12 -0
  179. package/scripts/act-testing/merged-skip-labeled.json +12 -0
  180. package/scripts/act-testing/not-merged-patch-labeled.json +12 -0
  181. package/scripts/build-for-release.sh +122 -0
  182. package/scripts/build.js +125 -125
  183. package/scripts/clean.js +31 -31
  184. package/scripts/docker-compose.yml +150 -152
  185. package/scripts/format-changelog.js +55 -0
  186. package/scripts/next-release-howto.md +24 -24
  187. package/scripts/oracledb-install-driver-libs.sh +82 -82
  188. package/scripts/release.sh +36 -36
  189. package/scripts/runkit-example.js +35 -35
  190. package/scripts/stress-test/README.txt +18 -18
  191. package/scripts/stress-test/docker-compose.yml +55 -57
  192. package/scripts/stress-test/knex-stress-test.js +212 -212
  193. package/scripts/stress-test/mysql2-random-hanging-every-now-and-then.js +149 -149
  194. package/scripts/stress-test/mysql2-sudden-exit-without-error.js +101 -101
  195. package/scripts/stress-test/reconnect-test-mysql-based-drivers.js +188 -188
  196. package/types/index.d.ts +3321 -3272
  197. package/types/result.d.ts +27 -27
  198. package/types/tables.d.ts +4 -4
  199. package/scripts/update_gitignore_for_tsc_output.js +0 -90
@@ -1,500 +1,498 @@
1
- // MSSQL Client
2
- // -------
3
- const map = require('lodash/map');
4
- const isNil = require('lodash/isNil');
5
-
6
- const Client = require('../../client');
7
- const MSSQL_Formatter = require('./mssql-formatter');
8
- const Transaction = require('./transaction');
9
- const QueryCompiler = require('./query/mssql-querycompiler');
10
- const SchemaCompiler = require('./schema/mssql-compiler');
11
- const TableCompiler = require('./schema/mssql-tablecompiler');
12
- const ViewCompiler = require('./schema/mssql-viewcompiler');
13
- const ColumnCompiler = require('./schema/mssql-columncompiler');
14
- const QueryBuilder = require('../../query/querybuilder');
15
- const { setHiddenProperty } = require('../../util/security');
16
-
17
- const debug = require('debug')('knex:mssql');
18
-
19
- const SQL_INT4 = { MIN: -2147483648, MAX: 2147483647 };
20
- const SQL_BIGINT_SAFE = { MIN: -9007199254740991, MAX: 9007199254740991 };
21
-
22
- // Always initialize with the "QueryBuilder" and "QueryCompiler" objects, which
23
- // extend the base 'lib/query/builder' and 'lib/query/compiler', respectively.
24
- class Client_MSSQL extends Client {
25
- constructor(config = {}) {
26
- super(config);
27
- }
28
-
29
- /**
30
- * @param {import('knex').Config} options
31
- */
32
- _generateConnection() {
33
- const settings = this.connectionSettings;
34
- settings.options = settings.options || {};
35
-
36
- /** @type {import('tedious').ConnectionConfig} */
37
- const cfg = {
38
- authentication: {
39
- type: settings.type || 'default',
40
- options: {
41
- userName: settings.userName || settings.user,
42
- password: settings.password,
43
- domain: settings.domain,
44
- token: settings.token,
45
- clientId: settings.clientId,
46
- clientSecret: settings.clientSecret,
47
- tenantId: settings.tenantId,
48
- msiEndpoint: settings.msiEndpoint,
49
- },
50
- },
51
- server: settings.server || settings.host,
52
- options: {
53
- database: settings.database,
54
- encrypt: settings.encrypt || false,
55
- port: settings.port || 1433,
56
- connectTimeout: settings.connectionTimeout || settings.timeout || 15000,
57
- requestTimeout: !isNil(settings.requestTimeout)
58
- ? settings.requestTimeout
59
- : 15000,
60
- rowCollectionOnDone: false,
61
- rowCollectionOnRequestCompletion: false,
62
- useColumnNames: false,
63
- tdsVersion: settings.options.tdsVersion || '7_4',
64
- appName: settings.options.appName || 'knex',
65
- trustServerCertificate: false,
66
- ...settings.options,
67
- },
68
- };
69
-
70
- if (cfg.authentication.options.password) {
71
- setHiddenProperty(cfg.authentication.options);
72
- }
73
-
74
- // tedious always connect via tcp when port is specified
75
- if (cfg.options.instanceName) delete cfg.options.port;
76
-
77
- if (isNaN(cfg.options.requestTimeout)) cfg.options.requestTimeout = 15000;
78
- if (cfg.options.requestTimeout === Infinity) cfg.options.requestTimeout = 0;
79
- if (cfg.options.requestTimeout < 0) cfg.options.requestTimeout = 0;
80
-
81
- if (settings.debug) {
82
- cfg.options.debug = {
83
- packet: true,
84
- token: true,
85
- data: true,
86
- payload: true,
87
- };
88
- }
89
-
90
- return cfg;
91
- }
92
-
93
- _driver() {
94
- const tds = require('tedious');
95
-
96
- return tds;
97
- }
98
-
99
- formatter() {
100
- return new MSSQL_Formatter(this, ...arguments);
101
- }
102
-
103
- transaction() {
104
- return new Transaction(this, ...arguments);
105
- }
106
-
107
- queryCompiler() {
108
- return new QueryCompiler(this, ...arguments);
109
- }
110
-
111
- schemaCompiler() {
112
- return new SchemaCompiler(this, ...arguments);
113
- }
114
-
115
- tableCompiler() {
116
- return new TableCompiler(this, ...arguments);
117
- }
118
-
119
- viewCompiler() {
120
- return new ViewCompiler(this, ...arguments);
121
- }
122
- queryBuilder() {
123
- const b = new QueryBuilder(this);
124
- return b;
125
- }
126
-
127
- columnCompiler() {
128
- return new ColumnCompiler(this, ...arguments);
129
- }
130
-
131
- wrapIdentifierImpl(value) {
132
- if (value === '*') {
133
- return '*';
134
- }
135
-
136
- return `[${value.replace(/[[\]]+/g, '')}]`;
137
- }
138
-
139
- // Get a raw connection, called by the `pool` whenever a new
140
- // connection needs to be added to the pool.
141
- acquireRawConnection() {
142
- return new Promise((resolver, rejecter) => {
143
- debug('connection::connection new connection requested');
144
- const Driver = this._driver();
145
- const settings = Object.assign({}, this._generateConnection());
146
-
147
- const connection = new Driver.Connection(settings);
148
-
149
- connection.connect((err) => {
150
- if (err) {
151
- debug('connection::connect error: %s', err.message);
152
- return rejecter(err);
153
- }
154
-
155
- debug('connection::connect connected to server');
156
-
157
- connection.connected = true;
158
- connection.on('error', (e) => {
159
- debug('connection::error message=%s', e.message);
160
- connection.__knex__disposed = e;
161
- connection.connected = false;
162
- });
163
-
164
- connection.once('end', () => {
165
- connection.connected = false;
166
- connection.__knex__disposed = 'Connection to server was terminated.';
167
- debug('connection::end connection ended.');
168
- });
169
-
170
- return resolver(connection);
171
- });
172
- });
173
- }
174
-
175
- validateConnection(connection) {
176
- return connection && connection.connected;
177
- }
178
-
179
- // Used to explicitly close a connection, called internally by the pool
180
- // when a connection times out or the pool is shutdown.
181
- destroyRawConnection(connection) {
182
- debug('connection::destroy');
183
-
184
- return new Promise((resolve) => {
185
- connection.once('end', () => {
186
- resolve();
187
- });
188
-
189
- connection.close();
190
- });
191
- }
192
-
193
- // Position the bindings for the query.
194
- positionBindings(sql) {
195
- let questionCount = -1;
196
- return sql.replace(/\\?\?/g, (match) => {
197
- if (match === '\\?') {
198
- return '?';
199
- }
200
-
201
- questionCount += 1;
202
- return `@p${questionCount}`;
203
- });
204
- }
205
-
206
- _chomp(connection) {
207
- if (connection.state.name === 'LoggedIn') {
208
- const nextRequest = this.requestQueue.pop();
209
- if (nextRequest) {
210
- debug(
211
- 'connection::query executing query, %d more in queue',
212
- this.requestQueue.length
213
- );
214
-
215
- connection.execSql(nextRequest);
216
- }
217
- }
218
- }
219
-
220
- _enqueueRequest(request, connection) {
221
- this.requestQueue.push(request);
222
- this._chomp(connection);
223
- }
224
-
225
- _makeRequest(query, callback) {
226
- const Driver = this._driver();
227
- const sql = typeof query === 'string' ? query : query.sql;
228
- let rowCount = 0;
229
-
230
- if (!sql) throw new Error('The query is empty');
231
-
232
- debug('request::request sql=%s', sql);
233
-
234
- const request = new Driver.Request(sql, (err, remoteRowCount) => {
235
- if (err) {
236
- debug('request::error message=%s', err.message);
237
- return callback(err);
238
- }
239
-
240
- rowCount = remoteRowCount;
241
- debug('request::callback rowCount=%d', rowCount);
242
- });
243
-
244
- request.on('prepared', () => {
245
- debug('request %s::request prepared', this.id);
246
- });
247
-
248
- request.on('done', (rowCount, more) => {
249
- debug('request::done rowCount=%d more=%s', rowCount, more);
250
- });
251
-
252
- request.on('doneProc', (rowCount, more) => {
253
- debug(
254
- 'request::doneProc id=%s rowCount=%d more=%s',
255
- request.id,
256
- rowCount,
257
- more
258
- );
259
- });
260
-
261
- request.on('doneInProc', (rowCount, more) => {
262
- debug(
263
- 'request::doneInProc id=%s rowCount=%d more=%s',
264
- request.id,
265
- rowCount,
266
- more
267
- );
268
- });
269
-
270
- request.once('requestCompleted', () => {
271
- debug('request::completed id=%s', request.id);
272
- return callback(null, rowCount);
273
- });
274
-
275
- request.on('error', (err) => {
276
- debug('request::error id=%s message=%s', request.id, err.message);
277
- return callback(err);
278
- });
279
-
280
- return request;
281
- }
282
-
283
- // Grab a connection, run the query via the MSSQL streaming interface,
284
- // and pass that through to the stream we've sent back to the client.
285
- _stream(connection, query, /** @type {NodeJS.ReadWriteStream} */ stream) {
286
- return new Promise((resolve, reject) => {
287
- const request = this._makeRequest(query, (err) => {
288
- if (err) {
289
- stream.emit('error', err);
290
- return reject(err);
291
- }
292
-
293
- resolve();
294
- });
295
-
296
- request.on('row', (row) => {
297
- stream.write(
298
- row.reduce(
299
- (prev, curr) => ({
300
- ...prev,
301
- [curr.metadata.colName]: curr.value,
302
- }),
303
- {}
304
- )
305
- );
306
- });
307
- request.on('error', (err) => {
308
- stream.emit('error', err);
309
- reject(err);
310
- });
311
- request.once('requestCompleted', () => {
312
- stream.end();
313
- resolve();
314
- });
315
-
316
- this._assignBindings(request, query.bindings);
317
- this._enqueueRequest(request, connection);
318
- });
319
- }
320
-
321
- _assignBindings(request, bindings) {
322
- if (Array.isArray(bindings)) {
323
- for (let i = 0; i < bindings.length; i++) {
324
- const binding = bindings[i];
325
- this._setReqInput(request, i, binding);
326
- }
327
- }
328
- }
329
-
330
- _scaleForBinding(binding) {
331
- if (binding % 1 === 0) {
332
- throw new Error(`The binding value ${binding} must be a decimal number.`);
333
- }
334
-
335
- return { scale: 10 };
336
- }
337
-
338
- _typeForBinding(binding) {
339
- const Driver = this._driver();
340
-
341
- if (
342
- this.connectionSettings.options &&
343
- this.connectionSettings.options.mapBinding
344
- ) {
345
- const result = this.connectionSettings.options.mapBinding(binding);
346
- if (result) {
347
- return [result.value, result.type];
348
- }
349
- }
350
-
351
- switch (typeof binding) {
352
- case 'string':
353
- return [binding, Driver.TYPES.NVarChar];
354
- case 'boolean':
355
- return [binding, Driver.TYPES.Bit];
356
- case 'number': {
357
- if (binding % 1 !== 0) {
358
- return [binding, Driver.TYPES.Float];
359
- }
360
-
361
- if (binding < SQL_INT4.MIN || binding > SQL_INT4.MAX) {
362
- if (binding < SQL_BIGINT_SAFE.MIN || binding > SQL_BIGINT_SAFE.MAX) {
363
- throw new Error(
364
- `Bigint must be safe integer or must be passed as string, saw ${binding}`
365
- );
366
- }
367
-
368
- return [binding, Driver.TYPES.BigInt];
369
- }
370
-
371
- return [binding, Driver.TYPES.Int];
372
- }
373
- default: {
374
- if (binding instanceof Date) {
375
- return [binding, Driver.TYPES.DateTime];
376
- }
377
-
378
- if (binding instanceof Buffer) {
379
- return [binding, Driver.TYPES.VarBinary];
380
- }
381
-
382
- return [binding, Driver.TYPES.NVarChar];
383
- }
384
- }
385
- }
386
-
387
- // Runs the query on the specified connection, providing the bindings
388
- // and any other necessary prep work.
389
- _query(connection, query) {
390
- return new Promise((resolve, reject) => {
391
- const rows = [];
392
- const request = this._makeRequest(query, (err, count) => {
393
- if (err) {
394
- return reject(err);
395
- }
396
-
397
- query.response = rows;
398
-
399
- process.nextTick(() => this._chomp(connection));
400
-
401
- resolve(query);
402
- });
403
-
404
- request.on('row', (row) => {
405
- debug('request::row');
406
- rows.push(row);
407
- });
408
-
409
- this._assignBindings(request, query.bindings);
410
- this._enqueueRequest(request, connection);
411
- });
412
- }
413
-
414
- // sets a request input parameter. Detects bigints and decimals and sets type appropriately.
415
- _setReqInput(req, i, inputBinding) {
416
- const [binding, tediousType] = this._typeForBinding(inputBinding);
417
- const bindingName = 'p'.concat(i);
418
- let options;
419
-
420
- if (typeof binding === 'number' && binding % 1 !== 0) {
421
- options = this._scaleForBinding(binding);
422
- }
423
-
424
- debug(
425
- 'request::binding pos=%d type=%s value=%s',
426
- i,
427
- tediousType.name,
428
- binding
429
- );
430
-
431
- if (Buffer.isBuffer(binding)) {
432
- options = {
433
- length: 'max',
434
- };
435
- }
436
-
437
- req.addParameter(bindingName, tediousType, binding, options);
438
- }
439
-
440
- // Process the response as returned from the query.
441
- processResponse(query, runner) {
442
- if (query == null) return;
443
- let { response } = query;
444
- const { method } = query;
445
-
446
- if (query.output) {
447
- return query.output.call(runner, response);
448
- }
449
-
450
- response = response.map((row) =>
451
- row.reduce((columns, r) => {
452
- const colName = r.metadata.colName;
453
-
454
- if (columns[colName]) {
455
- if (!Array.isArray(columns[colName])) {
456
- columns[colName] = [columns[colName]];
457
- }
458
-
459
- columns[colName].push(r.value);
460
- } else {
461
- columns[colName] = r.value;
462
- }
463
-
464
- return columns;
465
- }, {})
466
- );
467
-
468
- if (query.output) return query.output.call(runner, response);
469
- switch (method) {
470
- case 'select':
471
- return response;
472
- case 'first':
473
- return response[0];
474
- case 'pluck':
475
- return map(response, query.pluck);
476
- case 'insert':
477
- case 'del':
478
- case 'update':
479
- case 'counter':
480
- if (query.returning) {
481
- if (query.returning === '@@rowcount') {
482
- return response[0][''];
483
- }
484
- }
485
- return response;
486
- default:
487
- return response;
488
- }
489
- }
490
- }
491
-
492
- Object.assign(Client_MSSQL.prototype, {
493
- requestQueue: [],
494
-
495
- dialect: 'mssql',
496
-
497
- driverName: 'mssql',
498
- });
499
-
500
- module.exports = Client_MSSQL;
1
+ // MSSQL Client
2
+ // -------
3
+ const map = require('lodash/map');
4
+ const isNil = require('lodash/isNil');
5
+
6
+ const Client = require('../../client');
7
+ const MSSQL_Formatter = require('./mssql-formatter');
8
+ const Transaction = require('./transaction');
9
+ const QueryCompiler = require('./query/mssql-querycompiler');
10
+ const SchemaCompiler = require('./schema/mssql-compiler');
11
+ const TableCompiler = require('./schema/mssql-tablecompiler');
12
+ const ViewCompiler = require('./schema/mssql-viewcompiler');
13
+ const ColumnCompiler = require('./schema/mssql-columncompiler');
14
+ const QueryBuilder = require('../../query/querybuilder');
15
+ const { setHiddenProperty } = require('../../util/security');
16
+
17
+ const debug = require('debug')('knex:mssql');
18
+
19
+ const SQL_INT4 = { MIN: -2147483648, MAX: 2147483647 };
20
+ const SQL_BIGINT_SAFE = { MIN: -9007199254740991, MAX: 9007199254740991 };
21
+
22
+ // Always initialize with the "QueryBuilder" and "QueryCompiler" objects, which
23
+ // extend the base 'lib/query/builder' and 'lib/query/compiler', respectively.
24
+ class Client_MSSQL extends Client {
25
+ constructor(config = {}) {
26
+ super(config);
27
+ }
28
+
29
+ /**
30
+ * @param {import('knex').Config} options
31
+ */
32
+ _generateConnection() {
33
+ const settings = this.connectionSettings;
34
+ settings.options = settings.options || {};
35
+
36
+ /** @type {import('tedious').ConnectionConfig} */
37
+ const cfg = {
38
+ authentication: {
39
+ type: settings.type || 'default',
40
+ options: {
41
+ userName: settings.userName || settings.user,
42
+ password: settings.password,
43
+ domain: settings.domain,
44
+ token: settings.token,
45
+ clientId: settings.clientId,
46
+ clientSecret: settings.clientSecret,
47
+ tenantId: settings.tenantId,
48
+ msiEndpoint: settings.msiEndpoint,
49
+ },
50
+ },
51
+ server: settings.server || settings.host,
52
+ options: {
53
+ database: settings.database,
54
+ encrypt: settings.encrypt || false,
55
+ port: settings.port || 1433,
56
+ connectTimeout: settings.connectionTimeout || settings.timeout || 15000,
57
+ requestTimeout: !isNil(settings.requestTimeout)
58
+ ? settings.requestTimeout
59
+ : 15000,
60
+ rowCollectionOnDone: false,
61
+ rowCollectionOnRequestCompletion: false,
62
+ useColumnNames: false,
63
+ tdsVersion: settings.options.tdsVersion || '7_4',
64
+ appName: settings.options.appName || 'knex',
65
+ trustServerCertificate: false,
66
+ ...settings.options,
67
+ },
68
+ };
69
+
70
+ if (cfg.authentication.options.password) {
71
+ setHiddenProperty(cfg.authentication.options);
72
+ }
73
+
74
+ // tedious always connect via tcp when port is specified
75
+ if (cfg.options.instanceName) delete cfg.options.port;
76
+
77
+ if (isNaN(cfg.options.requestTimeout)) cfg.options.requestTimeout = 15000;
78
+ if (cfg.options.requestTimeout === Infinity) cfg.options.requestTimeout = 0;
79
+ if (cfg.options.requestTimeout < 0) cfg.options.requestTimeout = 0;
80
+
81
+ if (settings.debug) {
82
+ cfg.options.debug = {
83
+ packet: true,
84
+ token: true,
85
+ data: true,
86
+ payload: true,
87
+ };
88
+ }
89
+
90
+ return cfg;
91
+ }
92
+
93
+ _driver() {
94
+ const tds = require('tedious');
95
+
96
+ return tds;
97
+ }
98
+
99
+ formatter() {
100
+ return new MSSQL_Formatter(this, ...arguments);
101
+ }
102
+
103
+ transaction() {
104
+ return new Transaction(this, ...arguments);
105
+ }
106
+
107
+ queryCompiler() {
108
+ return new QueryCompiler(this, ...arguments);
109
+ }
110
+
111
+ schemaCompiler() {
112
+ return new SchemaCompiler(this, ...arguments);
113
+ }
114
+
115
+ tableCompiler() {
116
+ return new TableCompiler(this, ...arguments);
117
+ }
118
+
119
+ viewCompiler() {
120
+ return new ViewCompiler(this, ...arguments);
121
+ }
122
+ queryBuilder() {
123
+ const b = new QueryBuilder(this);
124
+ return b;
125
+ }
126
+
127
+ columnCompiler() {
128
+ return new ColumnCompiler(this, ...arguments);
129
+ }
130
+
131
+ wrapIdentifierImpl(value) {
132
+ if (value === '*') {
133
+ return '*';
134
+ }
135
+
136
+ return `[${value.replace(/[[\]]+/g, '')}]`;
137
+ }
138
+
139
+ // Get a raw connection, called by the `pool` whenever a new
140
+ // connection needs to be added to the pool.
141
+ acquireRawConnection() {
142
+ return new Promise((resolver, rejecter) => {
143
+ debug('connection::connection new connection requested');
144
+ const Driver = this._driver();
145
+ const settings = Object.assign({}, this._generateConnection());
146
+
147
+ const connection = new Driver.Connection(settings);
148
+
149
+ connection.connect((err) => {
150
+ if (err) {
151
+ debug('connection::connect error: %s', err.message);
152
+ return rejecter(err);
153
+ }
154
+
155
+ debug('connection::connect connected to server');
156
+
157
+ connection.connected = true;
158
+ connection.on('error', (e) => {
159
+ debug('connection::error message=%s', e.message);
160
+ connection.__knex__disposed = e;
161
+ connection.connected = false;
162
+ });
163
+
164
+ connection.once('end', () => {
165
+ connection.connected = false;
166
+ connection.__knex__disposed = 'Connection to server was terminated.';
167
+ debug('connection::end connection ended.');
168
+ });
169
+
170
+ return resolver(connection);
171
+ });
172
+ });
173
+ }
174
+
175
+ validateConnection(connection) {
176
+ return connection && connection.connected;
177
+ }
178
+
179
+ // Used to explicitly close a connection, called internally by the pool
180
+ // when a connection times out or the pool is shutdown.
181
+ destroyRawConnection(connection) {
182
+ debug('connection::destroy');
183
+
184
+ return new Promise((resolve) => {
185
+ connection.once('end', () => {
186
+ resolve();
187
+ });
188
+
189
+ connection.close();
190
+ });
191
+ }
192
+
193
+ // Position the bindings for the query.
194
+ positionBindings(sql) {
195
+ let questionCount = -1;
196
+ return sql.replace(/\\?\?/g, (match) => {
197
+ if (match === '\\?') {
198
+ return '?';
199
+ }
200
+
201
+ questionCount += 1;
202
+ return `@p${questionCount}`;
203
+ });
204
+ }
205
+
206
+ _chomp(connection) {
207
+ if (connection.state.name === 'LoggedIn') {
208
+ const nextRequest = this.requestQueue.pop();
209
+ if (nextRequest) {
210
+ debug(
211
+ 'connection::query executing query, %d more in queue',
212
+ this.requestQueue.length
213
+ );
214
+
215
+ connection.execSql(nextRequest);
216
+ }
217
+ }
218
+ }
219
+
220
+ _enqueueRequest(request, connection) {
221
+ this.requestQueue.push(request);
222
+ this._chomp(connection);
223
+ }
224
+
225
+ _makeRequest(query, callback) {
226
+ const Driver = this._driver();
227
+ const sql = typeof query === 'string' ? query : query.sql;
228
+ let rowCount = 0;
229
+
230
+ if (!sql) throw new Error('The query is empty');
231
+
232
+ debug('request::request sql=%s', sql);
233
+
234
+ const request = new Driver.Request(sql, (err, remoteRowCount) => {
235
+ if (err) {
236
+ debug('request::error message=%s', err.message);
237
+ return callback(err);
238
+ }
239
+
240
+ rowCount = remoteRowCount;
241
+ debug('request::callback rowCount=%d', rowCount);
242
+ });
243
+
244
+ request.on('prepared', () => {
245
+ debug('request %s::request prepared', this.id);
246
+ });
247
+
248
+ request.on('done', (rowCount, more) => {
249
+ debug('request::done rowCount=%d more=%s', rowCount, more);
250
+ });
251
+
252
+ request.on('doneProc', (rowCount, more) => {
253
+ debug(
254
+ 'request::doneProc id=%s rowCount=%d more=%s',
255
+ request.id,
256
+ rowCount,
257
+ more
258
+ );
259
+ });
260
+
261
+ request.on('doneInProc', (rowCount, more) => {
262
+ debug(
263
+ 'request::doneInProc id=%s rowCount=%d more=%s',
264
+ request.id,
265
+ rowCount,
266
+ more
267
+ );
268
+ });
269
+
270
+ request.once('requestCompleted', () => {
271
+ debug('request::completed id=%s', request.id);
272
+ return callback(null, rowCount);
273
+ });
274
+
275
+ request.on('error', (err) => {
276
+ debug('request::error id=%s message=%s', request.id, err.message);
277
+ return callback(err);
278
+ });
279
+
280
+ return request;
281
+ }
282
+
283
+ // Grab a connection, run the query via the MSSQL streaming interface,
284
+ // and pass that through to the stream we've sent back to the client.
285
+ _stream(connection, query, /** @type {NodeJS.ReadWriteStream} */ stream) {
286
+ return new Promise((resolve, reject) => {
287
+ const request = this._makeRequest(query, (err) => {
288
+ if (err) {
289
+ stream.emit('error', err);
290
+ return reject(err);
291
+ }
292
+
293
+ resolve();
294
+ });
295
+
296
+ request.on('row', (row) => {
297
+ stream.write(
298
+ row.reduce((prev, curr) => {
299
+ prev[curr.metadata.colName] = curr.value;
300
+
301
+ return prev;
302
+ }, {})
303
+ );
304
+ });
305
+ request.on('error', (err) => {
306
+ stream.emit('error', err);
307
+ reject(err);
308
+ });
309
+ request.once('requestCompleted', () => {
310
+ stream.end();
311
+ resolve();
312
+ });
313
+
314
+ this._assignBindings(request, query.bindings);
315
+ this._enqueueRequest(request, connection);
316
+ });
317
+ }
318
+
319
+ _assignBindings(request, bindings) {
320
+ if (Array.isArray(bindings)) {
321
+ for (let i = 0; i < bindings.length; i++) {
322
+ const binding = bindings[i];
323
+ this._setReqInput(request, i, binding);
324
+ }
325
+ }
326
+ }
327
+
328
+ _scaleForBinding(binding) {
329
+ if (binding % 1 === 0) {
330
+ throw new Error(`The binding value ${binding} must be a decimal number.`);
331
+ }
332
+
333
+ return { scale: 10 };
334
+ }
335
+
336
+ _typeForBinding(binding) {
337
+ const Driver = this._driver();
338
+
339
+ if (
340
+ this.connectionSettings.options &&
341
+ this.connectionSettings.options.mapBinding
342
+ ) {
343
+ const result = this.connectionSettings.options.mapBinding(binding);
344
+ if (result) {
345
+ return [result.value, result.type];
346
+ }
347
+ }
348
+
349
+ switch (typeof binding) {
350
+ case 'string':
351
+ return [binding, Driver.TYPES.NVarChar];
352
+ case 'boolean':
353
+ return [binding, Driver.TYPES.Bit];
354
+ case 'number': {
355
+ if (binding % 1 !== 0) {
356
+ return [binding, Driver.TYPES.Float];
357
+ }
358
+
359
+ if (binding < SQL_INT4.MIN || binding > SQL_INT4.MAX) {
360
+ if (binding < SQL_BIGINT_SAFE.MIN || binding > SQL_BIGINT_SAFE.MAX) {
361
+ throw new Error(
362
+ `Bigint must be safe integer or must be passed as string, saw ${binding}`
363
+ );
364
+ }
365
+
366
+ return [binding, Driver.TYPES.BigInt];
367
+ }
368
+
369
+ return [binding, Driver.TYPES.Int];
370
+ }
371
+ default: {
372
+ if (binding instanceof Date) {
373
+ return [binding, Driver.TYPES.DateTime];
374
+ }
375
+
376
+ if (binding instanceof Buffer) {
377
+ return [binding, Driver.TYPES.VarBinary];
378
+ }
379
+
380
+ return [binding, Driver.TYPES.NVarChar];
381
+ }
382
+ }
383
+ }
384
+
385
+ // Runs the query on the specified connection, providing the bindings
386
+ // and any other necessary prep work.
387
+ _query(connection, query) {
388
+ return new Promise((resolve, reject) => {
389
+ const rows = [];
390
+ const request = this._makeRequest(query, (err, count) => {
391
+ if (err) {
392
+ return reject(err);
393
+ }
394
+
395
+ query.response = rows;
396
+
397
+ process.nextTick(() => this._chomp(connection));
398
+
399
+ resolve(query);
400
+ });
401
+
402
+ request.on('row', (row) => {
403
+ debug('request::row');
404
+ rows.push(row);
405
+ });
406
+
407
+ this._assignBindings(request, query.bindings);
408
+ this._enqueueRequest(request, connection);
409
+ });
410
+ }
411
+
412
+ // sets a request input parameter. Detects bigints and decimals and sets type appropriately.
413
+ _setReqInput(req, i, inputBinding) {
414
+ const [binding, tediousType] = this._typeForBinding(inputBinding);
415
+ const bindingName = 'p'.concat(i);
416
+ let options;
417
+
418
+ if (typeof binding === 'number' && binding % 1 !== 0) {
419
+ options = this._scaleForBinding(binding);
420
+ }
421
+
422
+ debug(
423
+ 'request::binding pos=%d type=%s value=%s',
424
+ i,
425
+ tediousType.name,
426
+ binding
427
+ );
428
+
429
+ if (Buffer.isBuffer(binding)) {
430
+ options = {
431
+ length: 'max',
432
+ };
433
+ }
434
+
435
+ req.addParameter(bindingName, tediousType, binding, options);
436
+ }
437
+
438
+ // Process the response as returned from the query.
439
+ processResponse(query, runner) {
440
+ if (query == null) return;
441
+ let { response } = query;
442
+ const { method } = query;
443
+
444
+ if (query.output) {
445
+ return query.output.call(runner, response);
446
+ }
447
+
448
+ response = response.map((row) =>
449
+ row.reduce((columns, r) => {
450
+ const colName = r.metadata.colName;
451
+
452
+ if (columns[colName]) {
453
+ if (!Array.isArray(columns[colName])) {
454
+ columns[colName] = [columns[colName]];
455
+ }
456
+
457
+ columns[colName].push(r.value);
458
+ } else {
459
+ columns[colName] = r.value;
460
+ }
461
+
462
+ return columns;
463
+ }, {})
464
+ );
465
+
466
+ if (query.output) return query.output.call(runner, response);
467
+ switch (method) {
468
+ case 'select':
469
+ return response;
470
+ case 'first':
471
+ return response[0];
472
+ case 'pluck':
473
+ return map(response, query.pluck);
474
+ case 'insert':
475
+ case 'del':
476
+ case 'update':
477
+ case 'counter':
478
+ if (query.returning) {
479
+ if (query.returning === '@@rowcount') {
480
+ return response[0][''];
481
+ }
482
+ }
483
+ return response;
484
+ default:
485
+ return response;
486
+ }
487
+ }
488
+ }
489
+
490
+ Object.assign(Client_MSSQL.prototype, {
491
+ requestQueue: [],
492
+
493
+ dialect: 'mssql',
494
+
495
+ driverName: 'mssql',
496
+ });
497
+
498
+ module.exports = Client_MSSQL;