knex 2.4.2 → 2.5.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 (190) hide show
  1. package/CHANGELOG.md +54 -18
  2. package/CONTRIBUTING.md +194 -194
  3. package/LICENSE +22 -22
  4. package/README.md +147 -148
  5. package/UPGRADING.md +233 -233
  6. package/bin/cli.js +473 -473
  7. package/bin/utils/cli-config-utils.js +210 -210
  8. package/bin/utils/constants.js +7 -7
  9. package/bin/utils/migrationsLister.js +37 -37
  10. package/knex.js +23 -23
  11. package/knex.mjs +11 -0
  12. package/lib/builder-interface-augmenter.js +120 -120
  13. package/lib/client.js +495 -475
  14. package/lib/constants.js +61 -61
  15. package/lib/dialects/better-sqlite3/index.js +77 -72
  16. package/lib/dialects/cockroachdb/crdb-columncompiler.js +14 -14
  17. package/lib/dialects/cockroachdb/crdb-querybuilder.js +11 -11
  18. package/lib/dialects/cockroachdb/crdb-querycompiler.js +122 -122
  19. package/lib/dialects/cockroachdb/crdb-tablecompiler.js +37 -37
  20. package/lib/dialects/cockroachdb/crdb-viewcompiler.js +15 -15
  21. package/lib/dialects/cockroachdb/index.js +86 -86
  22. package/lib/dialects/index.js +33 -33
  23. package/lib/dialects/mssql/index.js +500 -495
  24. package/lib/dialects/mssql/mssql-formatter.js +34 -34
  25. package/lib/dialects/mssql/query/mssql-querycompiler.js +601 -600
  26. package/lib/dialects/mssql/schema/mssql-columncompiler.js +185 -185
  27. package/lib/dialects/mssql/schema/mssql-compiler.js +91 -91
  28. package/lib/dialects/mssql/schema/mssql-tablecompiler.js +378 -378
  29. package/lib/dialects/mssql/schema/mssql-viewcompiler.js +55 -55
  30. package/lib/dialects/mssql/transaction.js +176 -176
  31. package/lib/dialects/mysql/index.js +201 -201
  32. package/lib/dialects/mysql/query/mysql-querycompiler.js +274 -274
  33. package/lib/dialects/mysql/schema/mysql-columncompiler.js +193 -193
  34. package/lib/dialects/mysql/schema/mysql-compiler.js +60 -60
  35. package/lib/dialects/mysql/schema/mysql-tablecompiler.js +381 -381
  36. package/lib/dialects/mysql/schema/mysql-viewbuilder.js +21 -21
  37. package/lib/dialects/mysql/schema/mysql-viewcompiler.js +15 -15
  38. package/lib/dialects/mysql/transaction.js +46 -46
  39. package/lib/dialects/mysql2/index.js +53 -33
  40. package/lib/dialects/mysql2/transaction.js +44 -44
  41. package/lib/dialects/oracle/DEAD_CODE.md +5 -5
  42. package/lib/dialects/oracle/index.js +92 -92
  43. package/lib/dialects/oracle/query/oracle-querycompiler.js +343 -342
  44. package/lib/dialects/oracle/schema/internal/incrementUtils.js +20 -20
  45. package/lib/dialects/oracle/schema/internal/trigger.js +135 -135
  46. package/lib/dialects/oracle/schema/oracle-columnbuilder.js +17 -17
  47. package/lib/dialects/oracle/schema/oracle-columncompiler.js +126 -126
  48. package/lib/dialects/oracle/schema/oracle-compiler.js +122 -122
  49. package/lib/dialects/oracle/schema/oracle-tablecompiler.js +190 -190
  50. package/lib/dialects/oracle/utils.js +87 -87
  51. package/lib/dialects/oracledb/index.js +327 -327
  52. package/lib/dialects/oracledb/query/oracledb-querycompiler.js +481 -481
  53. package/lib/dialects/oracledb/schema/oracledb-columncompiler.js +61 -55
  54. package/lib/dialects/oracledb/schema/oracledb-tablecompiler.js +19 -19
  55. package/lib/dialects/oracledb/schema/oracledb-viewbuilder.js +13 -13
  56. package/lib/dialects/oracledb/schema/oracledb-viewcompiler.js +19 -19
  57. package/lib/dialects/oracledb/transaction.js +98 -98
  58. package/lib/dialects/oracledb/utils.js +208 -208
  59. package/lib/dialects/pgnative/index.js +60 -60
  60. package/lib/dialects/postgres/execution/pg-transaction.js +19 -12
  61. package/lib/dialects/postgres/index.js +358 -358
  62. package/lib/dialects/postgres/query/pg-querybuilder.js +43 -38
  63. package/lib/dialects/postgres/query/pg-querycompiler.js +400 -395
  64. package/lib/dialects/postgres/schema/pg-columncompiler.js +156 -156
  65. package/lib/dialects/postgres/schema/pg-compiler.js +138 -138
  66. package/lib/dialects/postgres/schema/pg-tablecompiler.js +304 -299
  67. package/lib/dialects/postgres/schema/pg-viewbuilder.js +21 -21
  68. package/lib/dialects/postgres/schema/pg-viewcompiler.js +35 -35
  69. package/lib/dialects/redshift/index.js +86 -86
  70. package/lib/dialects/redshift/query/redshift-querycompiler.js +163 -163
  71. package/lib/dialects/redshift/schema/redshift-columnbuilder.js +22 -22
  72. package/lib/dialects/redshift/schema/redshift-columncompiler.js +67 -67
  73. package/lib/dialects/redshift/schema/redshift-compiler.js +14 -14
  74. package/lib/dialects/redshift/schema/redshift-tablecompiler.js +122 -122
  75. package/lib/dialects/redshift/schema/redshift-viewcompiler.js +11 -11
  76. package/lib/dialects/redshift/transaction.js +32 -25
  77. package/lib/dialects/sqlite3/execution/sqlite-transaction.js +25 -18
  78. package/lib/dialects/sqlite3/index.js +250 -250
  79. package/lib/dialects/sqlite3/query/sqlite-querybuilder.js +33 -33
  80. package/lib/dialects/sqlite3/query/sqlite-querycompiler.js +334 -334
  81. package/lib/dialects/sqlite3/schema/ddl.js +400 -400
  82. package/lib/dialects/sqlite3/schema/internal/compiler.js +327 -327
  83. package/lib/dialects/sqlite3/schema/internal/parser-combinator.js +161 -161
  84. package/lib/dialects/sqlite3/schema/internal/parser.js +638 -638
  85. package/lib/dialects/sqlite3/schema/internal/sqlite-ddl-operations.js +41 -41
  86. package/lib/dialects/sqlite3/schema/internal/tokenizer.js +38 -38
  87. package/lib/dialects/sqlite3/schema/internal/utils.js +12 -12
  88. package/lib/dialects/sqlite3/schema/sqlite-columncompiler.js +50 -50
  89. package/lib/dialects/sqlite3/schema/sqlite-compiler.js +80 -80
  90. package/lib/dialects/sqlite3/schema/sqlite-tablecompiler.js +347 -347
  91. package/lib/dialects/sqlite3/schema/sqlite-viewcompiler.js +40 -40
  92. package/lib/execution/batch-insert.js +51 -51
  93. package/lib/execution/internal/delay.js +6 -6
  94. package/lib/execution/internal/ensure-connection-callback.js +41 -41
  95. package/lib/execution/internal/query-executioner.js +62 -62
  96. package/lib/execution/runner.js +325 -307
  97. package/lib/execution/transaction.js +409 -401
  98. package/lib/formatter/formatterUtils.js +42 -42
  99. package/lib/formatter/rawFormatter.js +84 -84
  100. package/lib/formatter/wrappingFormatter.js +250 -250
  101. package/lib/formatter.js +25 -25
  102. package/lib/index.js +3 -3
  103. package/lib/knex-builder/FunctionHelper.js +80 -54
  104. package/lib/knex-builder/Knex.js +59 -59
  105. package/lib/knex-builder/internal/config-resolver.js +57 -57
  106. package/lib/knex-builder/internal/parse-connection.js +87 -87
  107. package/lib/knex-builder/make-knex.js +345 -340
  108. package/lib/logger.js +76 -76
  109. package/lib/migrations/common/MigrationsLoader.js +36 -36
  110. package/lib/migrations/migrate/MigrationGenerator.js +84 -84
  111. package/lib/migrations/migrate/Migrator.js +598 -598
  112. package/lib/migrations/migrate/migrate-stub.js +17 -17
  113. package/lib/migrations/migrate/migration-list-resolver.js +33 -33
  114. package/lib/migrations/migrate/migrator-configuration-merger.js +58 -58
  115. package/lib/migrations/migrate/sources/fs-migrations.js +74 -74
  116. package/lib/migrations/migrate/stub/cjs.stub +15 -15
  117. package/lib/migrations/migrate/stub/coffee.stub +13 -13
  118. package/lib/migrations/migrate/stub/eg.stub +14 -14
  119. package/lib/migrations/migrate/stub/js-schema.stub +22 -22
  120. package/lib/migrations/migrate/stub/js.stub +22 -22
  121. package/lib/migrations/migrate/stub/knexfile-coffee.stub +34 -34
  122. package/lib/migrations/migrate/stub/knexfile-eg.stub +43 -43
  123. package/lib/migrations/migrate/stub/knexfile-js.stub +47 -47
  124. package/lib/migrations/migrate/stub/knexfile-ls.stub +35 -35
  125. package/lib/migrations/migrate/stub/knexfile-ts.stub +47 -47
  126. package/lib/migrations/migrate/stub/ls.stub +14 -14
  127. package/lib/migrations/migrate/stub/mjs.stub +23 -23
  128. package/lib/migrations/migrate/stub/ts-schema.stub +21 -21
  129. package/lib/migrations/migrate/stub/ts.stub +21 -21
  130. package/lib/migrations/migrate/table-creator.js +77 -77
  131. package/lib/migrations/migrate/table-resolver.js +27 -27
  132. package/lib/migrations/seed/Seeder.js +137 -137
  133. package/lib/migrations/seed/seed-stub.js +13 -13
  134. package/lib/migrations/seed/seeder-configuration-merger.js +60 -60
  135. package/lib/migrations/seed/sources/fs-seeds.js +65 -65
  136. package/lib/migrations/seed/stub/coffee.stub +9 -9
  137. package/lib/migrations/seed/stub/eg.stub +11 -11
  138. package/lib/migrations/seed/stub/js.stub +13 -13
  139. package/lib/migrations/seed/stub/ls.stub +11 -11
  140. package/lib/migrations/seed/stub/mjs.stub +12 -12
  141. package/lib/migrations/seed/stub/ts.stub +13 -13
  142. package/lib/migrations/util/fs.js +86 -86
  143. package/lib/migrations/util/import-file.js +12 -12
  144. package/lib/migrations/util/is-module-type.js +9 -9
  145. package/lib/migrations/util/template.js +52 -52
  146. package/lib/migrations/util/timestamp.js +14 -14
  147. package/lib/query/analytic.js +52 -52
  148. package/lib/query/constants.js +15 -15
  149. package/lib/query/joinclause.js +270 -270
  150. package/lib/query/method-constants.js +136 -135
  151. package/lib/query/querybuilder.js +1793 -1794
  152. package/lib/query/querycompiler.js +1591 -1580
  153. package/lib/raw.js +139 -139
  154. package/lib/ref.js +39 -39
  155. package/lib/schema/builder.js +115 -115
  156. package/lib/schema/columnbuilder.js +146 -146
  157. package/lib/schema/columncompiler.js +307 -307
  158. package/lib/schema/compiler.js +187 -187
  159. package/lib/schema/internal/helpers.js +55 -55
  160. package/lib/schema/tablebuilder.js +376 -376
  161. package/lib/schema/tablecompiler.js +433 -433
  162. package/lib/schema/viewbuilder.js +92 -92
  163. package/lib/schema/viewcompiler.js +138 -138
  164. package/lib/util/finally-mixin.js +13 -13
  165. package/lib/util/helpers.js +95 -95
  166. package/lib/util/is.js +32 -32
  167. package/lib/util/nanoid.js +40 -40
  168. package/lib/util/noop.js +1 -1
  169. package/lib/util/save-async-stack.js +14 -14
  170. package/lib/util/security.js +26 -0
  171. package/lib/util/string.js +190 -190
  172. package/lib/util/timeout.js +29 -29
  173. package/package.json +12 -10
  174. package/scripts/build.js +125 -125
  175. package/scripts/clean.js +31 -31
  176. package/scripts/docker-compose.yml +152 -152
  177. package/scripts/next-release-howto.md +24 -24
  178. package/scripts/oracledb-install-driver-libs.sh +82 -82
  179. package/scripts/release.sh +34 -34
  180. package/scripts/runkit-example.js +34 -34
  181. package/scripts/stress-test/README.txt +18 -18
  182. package/scripts/stress-test/docker-compose.yml +57 -57
  183. package/scripts/stress-test/knex-stress-test.js +208 -208
  184. package/scripts/stress-test/mysql2-random-hanging-every-now-and-then.js +145 -145
  185. package/scripts/stress-test/mysql2-sudden-exit-without-error.js +100 -100
  186. package/scripts/stress-test/reconnect-test-mysql-based-drivers.js +184 -184
  187. package/scripts/update_gitignore_for_tsc_output.js +90 -90
  188. package/types/index.d.ts +3273 -3233
  189. package/types/result.d.ts +27 -27
  190. package/types/tables.d.ts +4 -4
@@ -1,18 +1,18 @@
1
- # Test scripts to evaluate stability of drivers / pool etc.
2
-
3
- # To run this test you need to be in this directory + have node >= 8
4
- # and startup docker containers with proxy and sql servers
5
-
6
- docker-compose up --no-start
7
- docker-compose start
8
-
9
- # Select different test script to run:
10
-
11
- node mysql2-random-hanging-every-now-and-then.js 2> /dev/null | grep -B500 -A2 -- "- STATS"
12
- node mysql2-sudden-exit-without-error
13
- node knex-stress-test.js | grep -A 3 -- "- STATS "
14
- node reconnect-test-mysql-based-drivers.js 2> /dev/null | grep -A 3 -- "- STATS "
15
-
16
- # Shut down docker instances when done:
17
-
18
- docker-compose down
1
+ # Test scripts to evaluate stability of drivers / pool etc.
2
+
3
+ # To run this test you need to be in this directory + have node >= 8
4
+ # and startup docker containers with proxy and sql servers
5
+
6
+ docker-compose up --no-start
7
+ docker-compose start
8
+
9
+ # Select different test script to run:
10
+
11
+ node mysql2-random-hanging-every-now-and-then.js 2> /dev/null | grep -B500 -A2 -- "- STATS"
12
+ node mysql2-sudden-exit-without-error
13
+ node knex-stress-test.js | grep -A 3 -- "- STATS "
14
+ node reconnect-test-mysql-based-drivers.js 2> /dev/null | grep -A 3 -- "- STATS "
15
+
16
+ # Shut down docker instances when done:
17
+
18
+ docker-compose down
@@ -1,57 +1,57 @@
1
- version: '3'
2
-
3
- services:
4
- toxiproxy:
5
- image: shopify/toxiproxy
6
- ports:
7
- - "8474:8474"
8
- - "23306:23306"
9
- - "25432:25432"
10
- - "25433:25433"
11
- - "21521:21521"
12
- - "21433:21433"
13
- links:
14
- - "mysql"
15
- - "postgresql"
16
- - "pgnative"
17
- - "oracledb"
18
- - "mssql"
19
-
20
- mysql:
21
- image: mysql:5.7
22
- ports:
23
- - "33306:3306"
24
- environment:
25
- - TZ=UTC
26
- - MYSQL_ROOT_PASSWORD=mysqlrootpassword
27
-
28
- postgresql:
29
- image: mdillon/postgis
30
- ports:
31
- - "35432:5432"
32
- environment:
33
- - POSTGRES_PASSWORD=postgresrootpassword
34
- - POSTGRES_USER=postgres
35
-
36
- pgnative:
37
- image: mdillon/postgis
38
- ports:
39
- - "35433:5432"
40
- environment:
41
- - POSTGRES_PASSWORD=postgresrootpassword
42
- - POSTGRES_USER=postgres
43
-
44
- oracledb:
45
- image: quillbuilduser/oracle-18-xe
46
- ports:
47
- - "31521:1521"
48
- environment:
49
- - ORACLE_ALLOW_REMOTE=true
50
-
51
- mssql:
52
- image: microsoft/mssql-server-linux:2017-latest
53
- ports:
54
- - "31433:1433"
55
- environment:
56
- - ACCEPT_EULA=Y
57
- - SA_PASSWORD=S0meVeryHardPassword
1
+ version: '3'
2
+
3
+ services:
4
+ toxiproxy:
5
+ image: shopify/toxiproxy
6
+ ports:
7
+ - "8474:8474"
8
+ - "23306:23306"
9
+ - "25432:25432"
10
+ - "25433:25433"
11
+ - "21521:21521"
12
+ - "21433:21433"
13
+ links:
14
+ - "mysql"
15
+ - "postgresql"
16
+ - "pgnative"
17
+ - "oracledb"
18
+ - "mssql"
19
+
20
+ mysql:
21
+ image: mysql:5.7
22
+ ports:
23
+ - "33306:3306"
24
+ environment:
25
+ - TZ=UTC
26
+ - MYSQL_ROOT_PASSWORD=mysqlrootpassword
27
+
28
+ postgresql:
29
+ image: mdillon/postgis
30
+ ports:
31
+ - "35432:5432"
32
+ environment:
33
+ - POSTGRES_PASSWORD=postgresrootpassword
34
+ - POSTGRES_USER=postgres
35
+
36
+ pgnative:
37
+ image: mdillon/postgis
38
+ ports:
39
+ - "35433:5432"
40
+ environment:
41
+ - POSTGRES_PASSWORD=postgresrootpassword
42
+ - POSTGRES_USER=postgres
43
+
44
+ oracledb:
45
+ image: quillbuilduser/oracle-18-xe
46
+ ports:
47
+ - "31521:1521"
48
+ environment:
49
+ - ORACLE_ALLOW_REMOTE=true
50
+
51
+ mssql:
52
+ image: microsoft/mssql-server-linux:2017-latest
53
+ ports:
54
+ - "31433:1433"
55
+ environment:
56
+ - ACCEPT_EULA=Y
57
+ - SA_PASSWORD=S0meVeryHardPassword
@@ -1,208 +1,208 @@
1
- const Knex = require('../../lib');
2
-
3
- const toxiproxy = require('toxiproxy-node-client');
4
- const toxicli = new toxiproxy.Toxiproxy('http://localhost:8474');
5
- const rp = require('request-promise-native');
6
- const delay = require('../../lib/execution/internal/delay');
7
-
8
- // init instances
9
- const pg = Knex({
10
- client: 'pg',
11
- connection:
12
- 'postgres://postgres:postgresrootpassword@localhost:25432/postgres',
13
- pool: { max: 50 },
14
- });
15
-
16
- const pgnative = Knex({
17
- client: 'pgnative',
18
- connection:
19
- 'postgres://postgres:postgresrootpassword@localhost:25433/postgres',
20
- pool: { max: 50 },
21
- });
22
-
23
- const mysql2 = Knex({
24
- client: 'mysql2',
25
- connection:
26
- 'mysql://root:mysqlrootpassword@localhost:23306/?charset=utf8&connectTimeout=500',
27
- pool: { max: 50 },
28
- });
29
-
30
- const mysql = Knex({
31
- client: 'mysql',
32
- connection:
33
- 'mysql://root:mysqlrootpassword@localhost:23306/?charset=utf8&connectTimeout=500',
34
- pool: { max: 50 },
35
- });
36
-
37
- const mssql = Knex({
38
- client: 'mssql',
39
- connection: {
40
- port: 21433,
41
- user: 'sa',
42
- password: 'S0meVeryHardPassword',
43
- server: 'localhost',
44
- requestTimeout: 500,
45
- },
46
- pool: { max: 50 },
47
- });
48
-
49
- /* TODO: figure out how to nicely install oracledb node driver on osx
50
- const oracledb = Knex({
51
- client: 'oracledb',
52
- connection: {
53
- user : "travis",
54
- password : "travis",
55
- connectString : "localhost/XE",
56
- // https://github.com/oracle/node-oracledb/issues/525
57
- stmtCacheSize : 0
58
- },
59
- pool: { max: 50 }
60
- });
61
- */
62
-
63
- const counters = {};
64
-
65
- function setQueryCounters(instance, name) {
66
- const counts = (counters[name] = { queries: 0, results: 0, errors: 0 });
67
- instance.on('query', () => (counts.queries += 1));
68
- instance.on('query-response', () => (counts.results += 1));
69
- instance.on('query-error', () => (counts.errors += 1));
70
- }
71
-
72
- setQueryCounters(pg, 'pg');
73
- setQueryCounters(mysql, 'mysql');
74
- setQueryCounters(mysql2, 'mysql2');
75
- setQueryCounters(mssql, 'mssql');
76
-
77
- const _ = require('lodash');
78
-
79
- // start printing out counters
80
- let lastCounters = _.cloneDeep(counters);
81
-
82
- setInterval(() => {
83
- const reqsPerSec = {};
84
- for (let key of Object.keys(counters)) {
85
- reqsPerSec[key] = {
86
- queries: (counters[key].queries - lastCounters[key].queries) / 2,
87
- results: (counters[key].results - lastCounters[key].results) / 2,
88
- errors: (counters[key].errors - lastCounters[key].errors) / 2,
89
- };
90
- }
91
- console.log(
92
- '------------------------ STATS PER SECOND ------------------------'
93
- );
94
- console.dir(reqsPerSec, { colors: true });
95
- console.log(
96
- '------------------------------- EOS ------------------------------'
97
- );
98
- lastCounters = _.cloneDeep(counters);
99
- }, 2000);
100
-
101
- async function killConnectionsPg(client) {
102
- return client.raw(`SELECT pg_terminate_backend(pg_stat_activity.pid)
103
- FROM pg_stat_activity
104
- WHERE pg_stat_activity.datname = 'postgres'
105
- AND pid <> pg_backend_pid()`);
106
- }
107
-
108
- async function killConnectionsMyslq(client) {
109
- const [rows, colDefs] = await client.raw(`SHOW FULL PROCESSLIST`);
110
- await Promise.all(rows.map((row) => client.raw(`KILL ${row.Id}`)));
111
- }
112
-
113
- async function killConnectionsMssql() {
114
- const rows = await mssql('sys.dm_exec_sessions').select('session_id');
115
- await Promise.all(rows.map((row) => mssql.raw(`KILL ${row.session_id}`)));
116
- }
117
-
118
- async function main() {
119
- async function loopQueries(prefix, query) {
120
- const queries = () => [...Array(50).fill(query)];
121
-
122
- while (true) {
123
- try {
124
- await Promise.all(queries());
125
- } catch (err) {
126
- console.log(prefix, err);
127
- }
128
- }
129
- }
130
-
131
- async function recreateProxy(serviceName, listenPort, proxyToPort) {
132
- try {
133
- await rp.delete({
134
- url: `${toxicli.host}/proxies/${serviceName}`,
135
- });
136
- } catch (err) {}
137
-
138
- const proxy = await toxicli.createProxy({
139
- name: serviceName,
140
- listen: `0.0.0.0:${listenPort}`,
141
- upstream: `${serviceName}:${proxyToPort}`,
142
- });
143
-
144
- // add some latency
145
- await proxy.addToxic(
146
- new toxiproxy.Toxic(proxy, {
147
- type: 'latency',
148
- attributes: { latency: 1, jitter: 1 },
149
- })
150
- );
151
-
152
- // cause connections to be closed every 500 bytes
153
- await proxy.addToxic(
154
- new toxiproxy.Toxic(proxy, {
155
- type: 'limit_data',
156
- attributes: { bytes: 5000 },
157
- })
158
- );
159
- }
160
-
161
- // create TCP proxies for simulating bad connections etc.
162
- async function recreateProxies() {
163
- await recreateProxy('postgresql', 25432, 5432);
164
- await recreateProxy('postgresql', 25433, 5433);
165
- await recreateProxy('mysql', 23306, 3306);
166
- await recreateProxy('oracledb', 21521, 1521);
167
- await recreateProxy('mssql', 21433, 1433);
168
- }
169
-
170
- await recreateProxies();
171
-
172
- loopQueries('PSQL:', pg.raw('select 1'));
173
- loopQueries('PSQL TO:', pg.raw('select 1').timeout(20));
174
-
175
- loopQueries('PGNATIVE:', pgnative.raw('select 1'));
176
- loopQueries('PGNATIVE TO:', pgnative.raw('select 1').timeout(20));
177
-
178
- loopQueries('MYSQL:', mysql.raw('select 1'));
179
- loopQueries('MYSQL TO:', mysql.raw('select 1').timeout(20));
180
-
181
- // mysql2 still crashes app (without connection killer nor timeouts)
182
- // https://github.com/sidorares/node-mysql2/issues/731
183
- // loopQueries('MYSQL2:', mysql2.raw('select 1'));
184
- // loopQueries('MYSQL2 TO:', mysql2.raw('select 1').timeout(20));
185
-
186
- loopQueries('MSSQL:', mssql.raw('select 1'));
187
- loopQueries('MSSQL TO:', mssql.raw('select 1').timeout(20));
188
-
189
- setInterval(recreateProxies, 2000);
190
-
191
- while (true) {
192
- await delay(20); // kill everything every quite often from server side
193
- try {
194
- await Promise.all([
195
- killConnectionsPg(pg),
196
- killConnectionsPg(pgnative),
197
- killConnectionsMyslq(mysql),
198
- // killConnectionsMyslq(mysql2),
199
- killConnectionsMssql(),
200
- ]);
201
- } catch (err) {
202
- console.log('KILLER ERROR:', err);
203
- }
204
- }
205
- }
206
-
207
- process.on('exit', () => console.log('- STATS PRINT NEAR END LOGS ')); // marker for grep...
208
- main();
1
+ const Knex = require('../../lib');
2
+
3
+ const toxiproxy = require('toxiproxy-node-client');
4
+ const toxicli = new toxiproxy.Toxiproxy('http://localhost:8474');
5
+ const rp = require('request-promise-native');
6
+ const delay = require('../../lib/execution/internal/delay');
7
+
8
+ // init instances
9
+ const pg = Knex({
10
+ client: 'pg',
11
+ connection:
12
+ 'postgres://postgres:postgresrootpassword@localhost:25432/postgres',
13
+ pool: { max: 50 },
14
+ });
15
+
16
+ const pgnative = Knex({
17
+ client: 'pgnative',
18
+ connection:
19
+ 'postgres://postgres:postgresrootpassword@localhost:25433/postgres',
20
+ pool: { max: 50 },
21
+ });
22
+
23
+ const mysql2 = Knex({
24
+ client: 'mysql2',
25
+ connection:
26
+ 'mysql://root:mysqlrootpassword@localhost:23306/?charset=utf8&connectTimeout=500',
27
+ pool: { max: 50 },
28
+ });
29
+
30
+ const mysql = Knex({
31
+ client: 'mysql',
32
+ connection:
33
+ 'mysql://root:mysqlrootpassword@localhost:23306/?charset=utf8&connectTimeout=500',
34
+ pool: { max: 50 },
35
+ });
36
+
37
+ const mssql = Knex({
38
+ client: 'mssql',
39
+ connection: {
40
+ port: 21433,
41
+ user: 'sa',
42
+ password: 'S0meVeryHardPassword',
43
+ server: 'localhost',
44
+ requestTimeout: 500,
45
+ },
46
+ pool: { max: 50 },
47
+ });
48
+
49
+ /* TODO: figure out how to nicely install oracledb node driver on osx
50
+ const oracledb = Knex({
51
+ client: 'oracledb',
52
+ connection: {
53
+ user : "travis",
54
+ password : "travis",
55
+ connectString : "localhost/XE",
56
+ // https://github.com/oracle/node-oracledb/issues/525
57
+ stmtCacheSize : 0
58
+ },
59
+ pool: { max: 50 }
60
+ });
61
+ */
62
+
63
+ const counters = {};
64
+
65
+ function setQueryCounters(instance, name) {
66
+ const counts = (counters[name] = { queries: 0, results: 0, errors: 0 });
67
+ instance.on('query', () => (counts.queries += 1));
68
+ instance.on('query-response', () => (counts.results += 1));
69
+ instance.on('query-error', () => (counts.errors += 1));
70
+ }
71
+
72
+ setQueryCounters(pg, 'pg');
73
+ setQueryCounters(mysql, 'mysql');
74
+ setQueryCounters(mysql2, 'mysql2');
75
+ setQueryCounters(mssql, 'mssql');
76
+
77
+ const _ = require('lodash');
78
+
79
+ // start printing out counters
80
+ let lastCounters = _.cloneDeep(counters);
81
+
82
+ setInterval(() => {
83
+ const reqsPerSec = {};
84
+ for (let key of Object.keys(counters)) {
85
+ reqsPerSec[key] = {
86
+ queries: (counters[key].queries - lastCounters[key].queries) / 2,
87
+ results: (counters[key].results - lastCounters[key].results) / 2,
88
+ errors: (counters[key].errors - lastCounters[key].errors) / 2,
89
+ };
90
+ }
91
+ console.log(
92
+ '------------------------ STATS PER SECOND ------------------------'
93
+ );
94
+ console.dir(reqsPerSec, { colors: true });
95
+ console.log(
96
+ '------------------------------- EOS ------------------------------'
97
+ );
98
+ lastCounters = _.cloneDeep(counters);
99
+ }, 2000);
100
+
101
+ async function killConnectionsPg(client) {
102
+ return client.raw(`SELECT pg_terminate_backend(pg_stat_activity.pid)
103
+ FROM pg_stat_activity
104
+ WHERE pg_stat_activity.datname = 'postgres'
105
+ AND pid <> pg_backend_pid()`);
106
+ }
107
+
108
+ async function killConnectionsMyslq(client) {
109
+ const [rows, colDefs] = await client.raw(`SHOW FULL PROCESSLIST`);
110
+ await Promise.all(rows.map((row) => client.raw(`KILL ${row.Id}`)));
111
+ }
112
+
113
+ async function killConnectionsMssql() {
114
+ const rows = await mssql('sys.dm_exec_sessions').select('session_id');
115
+ await Promise.all(rows.map((row) => mssql.raw(`KILL ${row.session_id}`)));
116
+ }
117
+
118
+ async function main() {
119
+ async function loopQueries(prefix, query) {
120
+ const queries = () => [...Array(50).fill(query)];
121
+
122
+ while (true) {
123
+ try {
124
+ await Promise.all(queries());
125
+ } catch (err) {
126
+ console.log(prefix, err);
127
+ }
128
+ }
129
+ }
130
+
131
+ async function recreateProxy(serviceName, listenPort, proxyToPort) {
132
+ try {
133
+ await rp.delete({
134
+ url: `${toxicli.host}/proxies/${serviceName}`,
135
+ });
136
+ } catch (err) {}
137
+
138
+ const proxy = await toxicli.createProxy({
139
+ name: serviceName,
140
+ listen: `0.0.0.0:${listenPort}`,
141
+ upstream: `${serviceName}:${proxyToPort}`,
142
+ });
143
+
144
+ // add some latency
145
+ await proxy.addToxic(
146
+ new toxiproxy.Toxic(proxy, {
147
+ type: 'latency',
148
+ attributes: { latency: 1, jitter: 1 },
149
+ })
150
+ );
151
+
152
+ // cause connections to be closed every 500 bytes
153
+ await proxy.addToxic(
154
+ new toxiproxy.Toxic(proxy, {
155
+ type: 'limit_data',
156
+ attributes: { bytes: 5000 },
157
+ })
158
+ );
159
+ }
160
+
161
+ // create TCP proxies for simulating bad connections etc.
162
+ async function recreateProxies() {
163
+ await recreateProxy('postgresql', 25432, 5432);
164
+ await recreateProxy('postgresql', 25433, 5433);
165
+ await recreateProxy('mysql', 23306, 3306);
166
+ await recreateProxy('oracledb', 21521, 1521);
167
+ await recreateProxy('mssql', 21433, 1433);
168
+ }
169
+
170
+ await recreateProxies();
171
+
172
+ loopQueries('PSQL:', pg.raw('select 1'));
173
+ loopQueries('PSQL TO:', pg.raw('select 1').timeout(20));
174
+
175
+ loopQueries('PGNATIVE:', pgnative.raw('select 1'));
176
+ loopQueries('PGNATIVE TO:', pgnative.raw('select 1').timeout(20));
177
+
178
+ loopQueries('MYSQL:', mysql.raw('select 1'));
179
+ loopQueries('MYSQL TO:', mysql.raw('select 1').timeout(20));
180
+
181
+ // mysql2 still crashes app (without connection killer nor timeouts)
182
+ // https://github.com/sidorares/node-mysql2/issues/731
183
+ // loopQueries('MYSQL2:', mysql2.raw('select 1'));
184
+ // loopQueries('MYSQL2 TO:', mysql2.raw('select 1').timeout(20));
185
+
186
+ loopQueries('MSSQL:', mssql.raw('select 1'));
187
+ loopQueries('MSSQL TO:', mssql.raw('select 1').timeout(20));
188
+
189
+ setInterval(recreateProxies, 2000);
190
+
191
+ while (true) {
192
+ await delay(20); // kill everything every quite often from server side
193
+ try {
194
+ await Promise.all([
195
+ killConnectionsPg(pg),
196
+ killConnectionsPg(pgnative),
197
+ killConnectionsMyslq(mysql),
198
+ // killConnectionsMyslq(mysql2),
199
+ killConnectionsMssql(),
200
+ ]);
201
+ } catch (err) {
202
+ console.log('KILLER ERROR:', err);
203
+ }
204
+ }
205
+ }
206
+
207
+ process.on('exit', () => console.log('- STATS PRINT NEAR END LOGS ')); // marker for grep...
208
+ main();