autotel 2.1.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 (272) hide show
  1. package/LICENSE +21 -0
  2. package/README.md +1946 -0
  3. package/dist/chunk-2LNRY4QK.js +273 -0
  4. package/dist/chunk-2LNRY4QK.js.map +1 -0
  5. package/dist/chunk-3HENGDW2.js +587 -0
  6. package/dist/chunk-3HENGDW2.js.map +1 -0
  7. package/dist/chunk-4OAT42CA.cjs +73 -0
  8. package/dist/chunk-4OAT42CA.cjs.map +1 -0
  9. package/dist/chunk-5GWX5LFW.js +70 -0
  10. package/dist/chunk-5GWX5LFW.js.map +1 -0
  11. package/dist/chunk-5R2M36QB.js +195 -0
  12. package/dist/chunk-5R2M36QB.js.map +1 -0
  13. package/dist/chunk-5ZN622AO.js +73 -0
  14. package/dist/chunk-5ZN622AO.js.map +1 -0
  15. package/dist/chunk-77MSMAUQ.cjs +498 -0
  16. package/dist/chunk-77MSMAUQ.cjs.map +1 -0
  17. package/dist/chunk-ABPEQ6RK.cjs +596 -0
  18. package/dist/chunk-ABPEQ6RK.cjs.map +1 -0
  19. package/dist/chunk-BWYGJKRB.js +95 -0
  20. package/dist/chunk-BWYGJKRB.js.map +1 -0
  21. package/dist/chunk-BZHG5IZ4.js +73 -0
  22. package/dist/chunk-BZHG5IZ4.js.map +1 -0
  23. package/dist/chunk-G7VZBCD6.cjs +35 -0
  24. package/dist/chunk-G7VZBCD6.cjs.map +1 -0
  25. package/dist/chunk-GVLK7YUU.cjs +30 -0
  26. package/dist/chunk-GVLK7YUU.cjs.map +1 -0
  27. package/dist/chunk-HCCXC7XG.js +205 -0
  28. package/dist/chunk-HCCXC7XG.js.map +1 -0
  29. package/dist/chunk-HE6T6FIX.cjs +203 -0
  30. package/dist/chunk-HE6T6FIX.cjs.map +1 -0
  31. package/dist/chunk-KIXWPOCO.cjs +100 -0
  32. package/dist/chunk-KIXWPOCO.cjs.map +1 -0
  33. package/dist/chunk-KVGNW3FC.js +87 -0
  34. package/dist/chunk-KVGNW3FC.js.map +1 -0
  35. package/dist/chunk-LITNXTTT.js +3 -0
  36. package/dist/chunk-LITNXTTT.js.map +1 -0
  37. package/dist/chunk-M4ANN7RL.js +114 -0
  38. package/dist/chunk-M4ANN7RL.js.map +1 -0
  39. package/dist/chunk-NC52UBR2.cjs +32 -0
  40. package/dist/chunk-NC52UBR2.cjs.map +1 -0
  41. package/dist/chunk-NHCNRQD3.cjs +212 -0
  42. package/dist/chunk-NHCNRQD3.cjs.map +1 -0
  43. package/dist/chunk-NZ72VDNY.cjs +4 -0
  44. package/dist/chunk-NZ72VDNY.cjs.map +1 -0
  45. package/dist/chunk-P6JUDYNO.js +57 -0
  46. package/dist/chunk-P6JUDYNO.js.map +1 -0
  47. package/dist/chunk-RJYY7BWX.js +1349 -0
  48. package/dist/chunk-RJYY7BWX.js.map +1 -0
  49. package/dist/chunk-TRI4V5BF.cjs +126 -0
  50. package/dist/chunk-TRI4V5BF.cjs.map +1 -0
  51. package/dist/chunk-UL33I6IS.js +139 -0
  52. package/dist/chunk-UL33I6IS.js.map +1 -0
  53. package/dist/chunk-URRW6M2C.cjs +61 -0
  54. package/dist/chunk-URRW6M2C.cjs.map +1 -0
  55. package/dist/chunk-UY3UYPBZ.cjs +77 -0
  56. package/dist/chunk-UY3UYPBZ.cjs.map +1 -0
  57. package/dist/chunk-W3253FGB.cjs +277 -0
  58. package/dist/chunk-W3253FGB.cjs.map +1 -0
  59. package/dist/chunk-W7LHZVQF.js +26 -0
  60. package/dist/chunk-W7LHZVQF.js.map +1 -0
  61. package/dist/chunk-WBWNM6LB.cjs +1360 -0
  62. package/dist/chunk-WBWNM6LB.cjs.map +1 -0
  63. package/dist/chunk-WFJ7L2RV.js +494 -0
  64. package/dist/chunk-WFJ7L2RV.js.map +1 -0
  65. package/dist/chunk-X4RMFFMR.js +28 -0
  66. package/dist/chunk-X4RMFFMR.js.map +1 -0
  67. package/dist/chunk-Y4Y2S7BM.cjs +92 -0
  68. package/dist/chunk-Y4Y2S7BM.cjs.map +1 -0
  69. package/dist/chunk-YLPNXZFI.cjs +143 -0
  70. package/dist/chunk-YLPNXZFI.cjs.map +1 -0
  71. package/dist/chunk-YTXEZ4SD.cjs +77 -0
  72. package/dist/chunk-YTXEZ4SD.cjs.map +1 -0
  73. package/dist/chunk-Z6ZWNWWR.js +30 -0
  74. package/dist/chunk-Z6ZWNWWR.js.map +1 -0
  75. package/dist/config.cjs +26 -0
  76. package/dist/config.cjs.map +1 -0
  77. package/dist/config.d.cts +75 -0
  78. package/dist/config.d.ts +75 -0
  79. package/dist/config.js +5 -0
  80. package/dist/config.js.map +1 -0
  81. package/dist/db.cjs +233 -0
  82. package/dist/db.cjs.map +1 -0
  83. package/dist/db.d.cts +123 -0
  84. package/dist/db.d.ts +123 -0
  85. package/dist/db.js +228 -0
  86. package/dist/db.js.map +1 -0
  87. package/dist/decorators.cjs +67 -0
  88. package/dist/decorators.cjs.map +1 -0
  89. package/dist/decorators.d.cts +91 -0
  90. package/dist/decorators.d.ts +91 -0
  91. package/dist/decorators.js +65 -0
  92. package/dist/decorators.js.map +1 -0
  93. package/dist/event-subscriber.cjs +6 -0
  94. package/dist/event-subscriber.cjs.map +1 -0
  95. package/dist/event-subscriber.d.cts +116 -0
  96. package/dist/event-subscriber.d.ts +116 -0
  97. package/dist/event-subscriber.js +3 -0
  98. package/dist/event-subscriber.js.map +1 -0
  99. package/dist/event-testing.cjs +21 -0
  100. package/dist/event-testing.cjs.map +1 -0
  101. package/dist/event-testing.d.cts +110 -0
  102. package/dist/event-testing.d.ts +110 -0
  103. package/dist/event-testing.js +4 -0
  104. package/dist/event-testing.js.map +1 -0
  105. package/dist/event.cjs +30 -0
  106. package/dist/event.cjs.map +1 -0
  107. package/dist/event.d.cts +282 -0
  108. package/dist/event.d.ts +282 -0
  109. package/dist/event.js +13 -0
  110. package/dist/event.js.map +1 -0
  111. package/dist/exporters.cjs +17 -0
  112. package/dist/exporters.cjs.map +1 -0
  113. package/dist/exporters.d.cts +1 -0
  114. package/dist/exporters.d.ts +1 -0
  115. package/dist/exporters.js +4 -0
  116. package/dist/exporters.js.map +1 -0
  117. package/dist/functional.cjs +46 -0
  118. package/dist/functional.cjs.map +1 -0
  119. package/dist/functional.d.cts +478 -0
  120. package/dist/functional.d.ts +478 -0
  121. package/dist/functional.js +13 -0
  122. package/dist/functional.js.map +1 -0
  123. package/dist/http.cjs +189 -0
  124. package/dist/http.cjs.map +1 -0
  125. package/dist/http.d.cts +169 -0
  126. package/dist/http.d.ts +169 -0
  127. package/dist/http.js +184 -0
  128. package/dist/http.js.map +1 -0
  129. package/dist/index.cjs +333 -0
  130. package/dist/index.cjs.map +1 -0
  131. package/dist/index.d.cts +758 -0
  132. package/dist/index.d.ts +758 -0
  133. package/dist/index.js +143 -0
  134. package/dist/index.js.map +1 -0
  135. package/dist/instrumentation.cjs +182 -0
  136. package/dist/instrumentation.cjs.map +1 -0
  137. package/dist/instrumentation.d.cts +49 -0
  138. package/dist/instrumentation.d.ts +49 -0
  139. package/dist/instrumentation.js +179 -0
  140. package/dist/instrumentation.js.map +1 -0
  141. package/dist/logger.cjs +19 -0
  142. package/dist/logger.cjs.map +1 -0
  143. package/dist/logger.d.cts +146 -0
  144. package/dist/logger.d.ts +146 -0
  145. package/dist/logger.js +6 -0
  146. package/dist/logger.js.map +1 -0
  147. package/dist/metric-helpers.cjs +31 -0
  148. package/dist/metric-helpers.cjs.map +1 -0
  149. package/dist/metric-helpers.d.cts +13 -0
  150. package/dist/metric-helpers.d.ts +13 -0
  151. package/dist/metric-helpers.js +6 -0
  152. package/dist/metric-helpers.js.map +1 -0
  153. package/dist/metric-testing.cjs +21 -0
  154. package/dist/metric-testing.cjs.map +1 -0
  155. package/dist/metric-testing.d.cts +110 -0
  156. package/dist/metric-testing.d.ts +110 -0
  157. package/dist/metric-testing.js +4 -0
  158. package/dist/metric-testing.js.map +1 -0
  159. package/dist/metric.cjs +26 -0
  160. package/dist/metric.cjs.map +1 -0
  161. package/dist/metric.d.cts +240 -0
  162. package/dist/metric.d.ts +240 -0
  163. package/dist/metric.js +9 -0
  164. package/dist/metric.js.map +1 -0
  165. package/dist/processors.cjs +17 -0
  166. package/dist/processors.cjs.map +1 -0
  167. package/dist/processors.d.cts +1 -0
  168. package/dist/processors.d.ts +1 -0
  169. package/dist/processors.js +4 -0
  170. package/dist/processors.js.map +1 -0
  171. package/dist/sampling.cjs +40 -0
  172. package/dist/sampling.cjs.map +1 -0
  173. package/dist/sampling.d.cts +260 -0
  174. package/dist/sampling.d.ts +260 -0
  175. package/dist/sampling.js +7 -0
  176. package/dist/sampling.js.map +1 -0
  177. package/dist/semantic-helpers.cjs +35 -0
  178. package/dist/semantic-helpers.cjs.map +1 -0
  179. package/dist/semantic-helpers.d.cts +442 -0
  180. package/dist/semantic-helpers.d.ts +442 -0
  181. package/dist/semantic-helpers.js +14 -0
  182. package/dist/semantic-helpers.js.map +1 -0
  183. package/dist/tail-sampling-processor.cjs +13 -0
  184. package/dist/tail-sampling-processor.cjs.map +1 -0
  185. package/dist/tail-sampling-processor.d.cts +27 -0
  186. package/dist/tail-sampling-processor.d.ts +27 -0
  187. package/dist/tail-sampling-processor.js +4 -0
  188. package/dist/tail-sampling-processor.js.map +1 -0
  189. package/dist/testing.cjs +286 -0
  190. package/dist/testing.cjs.map +1 -0
  191. package/dist/testing.d.cts +291 -0
  192. package/dist/testing.d.ts +291 -0
  193. package/dist/testing.js +263 -0
  194. package/dist/testing.js.map +1 -0
  195. package/dist/trace-context-DRZdUvVY.d.cts +181 -0
  196. package/dist/trace-context-DRZdUvVY.d.ts +181 -0
  197. package/dist/trace-helpers.cjs +54 -0
  198. package/dist/trace-helpers.cjs.map +1 -0
  199. package/dist/trace-helpers.d.cts +524 -0
  200. package/dist/trace-helpers.d.ts +524 -0
  201. package/dist/trace-helpers.js +5 -0
  202. package/dist/trace-helpers.js.map +1 -0
  203. package/dist/tracer-provider.cjs +21 -0
  204. package/dist/tracer-provider.cjs.map +1 -0
  205. package/dist/tracer-provider.d.cts +169 -0
  206. package/dist/tracer-provider.d.ts +169 -0
  207. package/dist/tracer-provider.js +4 -0
  208. package/dist/tracer-provider.js.map +1 -0
  209. package/package.json +280 -0
  210. package/src/baggage-span-processor.test.ts +202 -0
  211. package/src/baggage-span-processor.ts +98 -0
  212. package/src/circuit-breaker.test.ts +341 -0
  213. package/src/circuit-breaker.ts +184 -0
  214. package/src/config.test.ts +94 -0
  215. package/src/config.ts +169 -0
  216. package/src/db.test.ts +252 -0
  217. package/src/db.ts +447 -0
  218. package/src/decorators.test.ts +203 -0
  219. package/src/decorators.ts +188 -0
  220. package/src/env-config.test.ts +246 -0
  221. package/src/env-config.ts +158 -0
  222. package/src/event-queue.test.ts +222 -0
  223. package/src/event-queue.ts +203 -0
  224. package/src/event-subscriber.ts +136 -0
  225. package/src/event-testing.ts +197 -0
  226. package/src/event.test.ts +718 -0
  227. package/src/event.ts +556 -0
  228. package/src/exporters.ts +96 -0
  229. package/src/functional.test.ts +1059 -0
  230. package/src/functional.ts +2295 -0
  231. package/src/http.test.ts +487 -0
  232. package/src/http.ts +424 -0
  233. package/src/index.ts +158 -0
  234. package/src/init.customization.test.ts +210 -0
  235. package/src/init.integrations.test.ts +366 -0
  236. package/src/init.openllmetry.test.ts +282 -0
  237. package/src/init.protocol.test.ts +215 -0
  238. package/src/init.ts +1426 -0
  239. package/src/instrumentation.test.ts +108 -0
  240. package/src/instrumentation.ts +308 -0
  241. package/src/logger.test.ts +117 -0
  242. package/src/logger.ts +246 -0
  243. package/src/metric-helpers.ts +47 -0
  244. package/src/metric-testing.ts +197 -0
  245. package/src/metric.ts +434 -0
  246. package/src/metrics.test.ts +205 -0
  247. package/src/operation-context.ts +93 -0
  248. package/src/processors.ts +106 -0
  249. package/src/rate-limiter.test.ts +199 -0
  250. package/src/rate-limiter.ts +98 -0
  251. package/src/sampling.test.ts +513 -0
  252. package/src/sampling.ts +428 -0
  253. package/src/semantic-helpers.test.ts +311 -0
  254. package/src/semantic-helpers.ts +584 -0
  255. package/src/shutdown.test.ts +311 -0
  256. package/src/shutdown.ts +222 -0
  257. package/src/stub.integration.test.ts +361 -0
  258. package/src/tail-sampling-processor.test.ts +226 -0
  259. package/src/tail-sampling-processor.ts +51 -0
  260. package/src/testing.ts +670 -0
  261. package/src/trace-context.ts +470 -0
  262. package/src/trace-helpers.new.test.ts +278 -0
  263. package/src/trace-helpers.test.ts +242 -0
  264. package/src/trace-helpers.ts +690 -0
  265. package/src/tracer-provider.test.ts +183 -0
  266. package/src/tracer-provider.ts +266 -0
  267. package/src/track.test.ts +153 -0
  268. package/src/track.ts +120 -0
  269. package/src/validation.test.ts +306 -0
  270. package/src/validation.ts +239 -0
  271. package/src/variable-name-inference.test.ts +178 -0
  272. package/src/variable-name-inference.ts +242 -0
package/dist/db.cjs ADDED
@@ -0,0 +1,233 @@
1
+ 'use strict';
2
+
3
+ var chunkY4Y2S7BM_cjs = require('./chunk-Y4Y2S7BM.cjs');
4
+ require('./chunk-URRW6M2C.cjs');
5
+ require('./chunk-G7VZBCD6.cjs');
6
+ var api = require('@opentelemetry/api');
7
+
8
+ async function tracebQuery(dbSystem, operation, fn, attributes) {
9
+ const config = chunkY4Y2S7BM_cjs.getConfig();
10
+ const tracer = config.tracer;
11
+ const spanName = `${dbSystem}.${operation}`;
12
+ return tracer.startActiveSpan(spanName, async (span) => {
13
+ const startTime = performance.now();
14
+ try {
15
+ span.setAttributes({
16
+ "db.system": dbSystem,
17
+ "db.operation": operation,
18
+ ...attributes
19
+ });
20
+ const result = await fn();
21
+ const duration = performance.now() - startTime;
22
+ span.setStatus({ code: api.SpanStatusCode.OK });
23
+ span.setAttribute("db.duration_ms", duration);
24
+ if (Array.isArray(result)) {
25
+ span.setAttribute("db.result_count", result.length);
26
+ }
27
+ return result;
28
+ } catch (error) {
29
+ const duration = performance.now() - startTime;
30
+ span.setStatus({
31
+ code: api.SpanStatusCode.ERROR,
32
+ message: error instanceof Error ? error.message : "Unknown error"
33
+ });
34
+ span.setAttributes({
35
+ "db.duration_ms": duration,
36
+ "error.type": error instanceof Error ? error.constructor.name : "Unknown",
37
+ "error.message": error instanceof Error ? error.message : "Unknown error"
38
+ });
39
+ throw error;
40
+ } finally {
41
+ span.end();
42
+ }
43
+ });
44
+ }
45
+ function inferDbOperation(methodName) {
46
+ const lower = methodName.toLowerCase();
47
+ if (lower.includes("find") || lower.includes("get") || lower.includes("list"))
48
+ return "SELECT";
49
+ if (lower.includes("create") || lower.includes("insert")) return "INSERT";
50
+ if (lower.includes("update") || lower.includes("modify")) return "UPDATE";
51
+ if (lower.includes("delete") || lower.includes("remove")) return "DELETE";
52
+ if (lower.includes("count")) return "COUNT";
53
+ return "QUERY";
54
+ }
55
+ function inferTableName(methodName) {
56
+ const patterns = [
57
+ /find([A-Z][a-zA-Z]+)/,
58
+ /get([A-Z][a-zA-Z]+)/,
59
+ /list([A-Z][a-zA-Z]+)/,
60
+ /create([A-Z][a-zA-Z]+)/,
61
+ /update([A-Z][a-zA-Z]+)/,
62
+ /delete([A-Z][a-zA-Z]+)/,
63
+ /remove([A-Z][a-zA-Z]+)/
64
+ ];
65
+ for (const pattern of patterns) {
66
+ const match = methodName.match(pattern);
67
+ if (match && match[1]) {
68
+ return match[1].toLowerCase();
69
+ }
70
+ }
71
+ return void 0;
72
+ }
73
+ function sanitizeSqlQuery(query) {
74
+ return query.replaceAll(/'[^']*'/g, "'?'").replaceAll(/"[^"]*"/g, '"?"').replaceAll(/\b\d+\b/g, "?").trim();
75
+ }
76
+ var DB_OPERATIONS = {
77
+ SELECT: "SELECT",
78
+ INSERT: "INSERT",
79
+ UPDATE: "UPDATE",
80
+ DELETE: "DELETE",
81
+ COUNT: "COUNT",
82
+ AGGREGATE: "AGGREGATE"
83
+ };
84
+ var DB_SYSTEMS = {
85
+ POSTGRESQL: "postgresql",
86
+ MYSQL: "mysql",
87
+ MONGODB: "mongodb",
88
+ REDIS: "redis",
89
+ SQLITE: "sqlite",
90
+ MSSQL: "mssql"
91
+ };
92
+ var INSTRUMENTED_SYMBOL = Symbol.for("autotel.db.instrumented");
93
+ function instrumentDatabase(client, options) {
94
+ if (client[INSTRUMENTED_SYMBOL]) {
95
+ return client;
96
+ }
97
+ const {
98
+ dbSystem,
99
+ dbName,
100
+ methods,
101
+ skipMethods = [],
102
+ sanitizeQuery = true,
103
+ slowQueryThresholdMs = 1e3
104
+ } = options;
105
+ const config = chunkY4Y2S7BM_cjs.getConfig();
106
+ const tracer = config.tracer;
107
+ const methodsToInstrument = methods || extractDatabaseMethods(client);
108
+ const skipSet = new Set(skipMethods);
109
+ for (const methodName of methodsToInstrument) {
110
+ if (skipSet.has(methodName)) continue;
111
+ if (methodName.startsWith("_")) continue;
112
+ const method = client[methodName];
113
+ if (typeof method !== "function") continue;
114
+ const originalMethod = method;
115
+ client[methodName] = async function(...args) {
116
+ const operation = inferDbOperation(methodName);
117
+ const table = inferTableName(methodName);
118
+ const spanName = table ? `${dbSystem}.${operation} ${table}` : `${dbSystem}.${operation}`;
119
+ return tracer.startActiveSpan(spanName, async (span) => {
120
+ const startTime = performance.now();
121
+ try {
122
+ span.setAttributes({
123
+ "db.system": dbSystem,
124
+ "db.operation": operation
125
+ });
126
+ if (dbName) {
127
+ span.setAttribute("db.name", dbName);
128
+ }
129
+ if (table) {
130
+ span.setAttribute("db.sql.table", table);
131
+ }
132
+ const query = extractQueryFromArgs(args);
133
+ if (query) {
134
+ span.setAttribute(
135
+ "db.statement",
136
+ sanitizeQuery ? sanitizeSqlQuery(query) : query
137
+ );
138
+ }
139
+ const result = await originalMethod.apply(this, args);
140
+ const duration = performance.now() - startTime;
141
+ span.setStatus({ code: api.SpanStatusCode.OK });
142
+ span.setAttributes({
143
+ "db.duration_ms": duration
144
+ });
145
+ if (duration > slowQueryThresholdMs) {
146
+ span.setAttribute("db.slow_query", true);
147
+ span.setAttribute(
148
+ "db.slow_query_threshold_ms",
149
+ slowQueryThresholdMs
150
+ );
151
+ }
152
+ if (Array.isArray(result)) {
153
+ span.setAttribute("db.result_count", result.length);
154
+ }
155
+ return result;
156
+ } catch (error) {
157
+ const duration = performance.now() - startTime;
158
+ span.setStatus({
159
+ code: api.SpanStatusCode.ERROR,
160
+ message: error instanceof Error ? error.message : "Unknown error"
161
+ });
162
+ span.setAttributes({
163
+ "db.duration_ms": duration,
164
+ "error.type": error instanceof Error ? error.constructor.name : "Unknown",
165
+ "error.message": error instanceof Error ? error.message : "Unknown error"
166
+ });
167
+ span.recordException(
168
+ error instanceof Error ? error : new Error(String(error))
169
+ );
170
+ throw error;
171
+ } finally {
172
+ span.end();
173
+ }
174
+ });
175
+ };
176
+ Object.defineProperty(client[methodName], "name", {
177
+ value: methodName,
178
+ configurable: true
179
+ });
180
+ }
181
+ client[INSTRUMENTED_SYMBOL] = true;
182
+ return client;
183
+ }
184
+ function extractDatabaseMethods(client) {
185
+ const methods = [];
186
+ const proto = Object.getPrototypeOf(client);
187
+ for (const key of Object.getOwnPropertyNames(client)) {
188
+ if (typeof client[key] === "function" && !key.startsWith("_")) {
189
+ methods.push(key);
190
+ }
191
+ }
192
+ if (proto) {
193
+ for (const key of Object.getOwnPropertyNames(proto)) {
194
+ if (typeof proto[key] === "function" && !key.startsWith("_") && key !== "constructor") {
195
+ methods.push(key);
196
+ }
197
+ }
198
+ }
199
+ return [...new Set(methods)];
200
+ }
201
+ function extractQueryFromArgs(args) {
202
+ if (args.length === 0) return void 0;
203
+ const firstArg = args[0];
204
+ if (typeof firstArg === "string") {
205
+ return firstArg;
206
+ }
207
+ if (firstArg && typeof firstArg === "object") {
208
+ if ("sql" in firstArg && typeof firstArg.sql === "string") {
209
+ return firstArg.sql;
210
+ }
211
+ if ("text" in firstArg && typeof firstArg.text === "string") {
212
+ return firstArg.text;
213
+ }
214
+ if ("toQuery" in firstArg && typeof firstArg.toQuery === "function") {
215
+ try {
216
+ const queryResult = firstArg.toQuery();
217
+ if (typeof queryResult === "string") return queryResult;
218
+ if (queryResult && typeof queryResult === "object" && "sql" in queryResult) {
219
+ return queryResult.sql;
220
+ }
221
+ } catch {
222
+ }
223
+ }
224
+ }
225
+ return void 0;
226
+ }
227
+
228
+ exports.DB_OPERATIONS = DB_OPERATIONS;
229
+ exports.DB_SYSTEMS = DB_SYSTEMS;
230
+ exports.instrumentDatabase = instrumentDatabase;
231
+ exports.tracebQuery = tracebQuery;
232
+ //# sourceMappingURL=db.cjs.map
233
+ //# sourceMappingURL=db.cjs.map
@@ -0,0 +1 @@
1
+ {"version":3,"sources":["../src/db.ts"],"names":["getConfig","SpanStatusCode"],"mappings":";;;;;;;AAsCA,eAAsB,WAAA,CACpB,QAAA,EACA,SAAA,EACA,EAAA,EACA,UAAA,EACY;AACZ,EAAA,MAAM,SAASA,2BAAA,EAAU;AACzB,EAAA,MAAM,SAAS,MAAA,CAAO,MAAA;AAEtB,EAAA,MAAM,QAAA,GAAW,CAAA,EAAG,QAAQ,CAAA,CAAA,EAAI,SAAS,CAAA,CAAA;AAEzC,EAAA,OAAO,MAAA,CAAO,eAAA,CAAgB,QAAA,EAAU,OAAO,IAAA,KAAS;AACtD,IAAA,MAAM,SAAA,GAAY,YAAY,GAAA,EAAI;AAElC,IAAA,IAAI;AACF,MAAA,IAAA,CAAK,aAAA,CAAc;AAAA,QACjB,WAAA,EAAa,QAAA;AAAA,QACb,cAAA,EAAgB,SAAA;AAAA,QAChB,GAAG;AAAA,OACJ,CAAA;AAED,MAAA,MAAM,MAAA,GAAS,MAAM,EAAA,EAAG;AAExB,MAAA,MAAM,QAAA,GAAW,WAAA,CAAY,GAAA,EAAI,GAAI,SAAA;AACrC,MAAA,IAAA,CAAK,SAAA,CAAU,EAAE,IAAA,EAAMC,kBAAA,CAAe,IAAI,CAAA;AAC1C,MAAA,IAAA,CAAK,YAAA,CAAa,kBAAkB,QAAQ,CAAA;AAE5C,MAAA,IAAI,KAAA,CAAM,OAAA,CAAQ,MAAM,CAAA,EAAG;AACzB,QAAA,IAAA,CAAK,YAAA,CAAa,iBAAA,EAAmB,MAAA,CAAO,MAAM,CAAA;AAAA,MACpD;AAEA,MAAA,OAAO,MAAA;AAAA,IACT,SAAS,KAAA,EAAO;AACd,MAAA,MAAM,QAAA,GAAW,WAAA,CAAY,GAAA,EAAI,GAAI,SAAA;AAErC,MAAA,IAAA,CAAK,SAAA,CAAU;AAAA,QACb,MAAMA,kBAAA,CAAe,KAAA;AAAA,QACrB,OAAA,EAAS,KAAA,YAAiB,KAAA,GAAQ,KAAA,CAAM,OAAA,GAAU;AAAA,OACnD,CAAA;AAED,MAAA,IAAA,CAAK,aAAA,CAAc;AAAA,QACjB,gBAAA,EAAkB,QAAA;AAAA,QAClB,YAAA,EACE,KAAA,YAAiB,KAAA,GAAQ,KAAA,CAAM,YAAY,IAAA,GAAO,SAAA;AAAA,QACpD,eAAA,EACE,KAAA,YAAiB,KAAA,GAAQ,KAAA,CAAM,OAAA,GAAU;AAAA,OAC5C,CAAA;AAED,MAAA,MAAM,KAAA;AAAA,IACR,CAAA,SAAE;AACA,MAAA,IAAA,CAAK,GAAA,EAAI;AAAA,IACX;AAAA,EACF,CAAC,CAAA;AACH;AAIA,SAAS,iBAAiB,UAAA,EAA4B;AACpD,EAAA,MAAM,KAAA,GAAQ,WAAW,WAAA,EAAY;AACrC,EAAA,IAAI,KAAA,CAAM,QAAA,CAAS,MAAM,CAAA,IAAK,KAAA,CAAM,SAAS,KAAK,CAAA,IAAK,KAAA,CAAM,QAAA,CAAS,MAAM,CAAA;AAC1E,IAAA,OAAO,QAAA;AACT,EAAA,IAAI,KAAA,CAAM,SAAS,QAAQ,CAAA,IAAK,MAAM,QAAA,CAAS,QAAQ,GAAG,OAAO,QAAA;AACjE,EAAA,IAAI,KAAA,CAAM,SAAS,QAAQ,CAAA,IAAK,MAAM,QAAA,CAAS,QAAQ,GAAG,OAAO,QAAA;AACjE,EAAA,IAAI,KAAA,CAAM,SAAS,QAAQ,CAAA,IAAK,MAAM,QAAA,CAAS,QAAQ,GAAG,OAAO,QAAA;AACjE,EAAA,IAAI,KAAA,CAAM,QAAA,CAAS,OAAO,CAAA,EAAG,OAAO,OAAA;AACpC,EAAA,OAAO,OAAA;AACT;AAEA,SAAS,eAAe,UAAA,EAAwC;AAM9D,EAAA,MAAM,QAAA,GAAW;AAAA,IACf,sBAAA;AAAA,IACA,qBAAA;AAAA,IACA,sBAAA;AAAA,IACA,wBAAA;AAAA,IACA,wBAAA;AAAA,IACA,wBAAA;AAAA,IACA;AAAA,GACF;AAEA,EAAA,KAAA,MAAW,WAAW,QAAA,EAAU;AAC9B,IAAA,MAAM,KAAA,GAAQ,UAAA,CAAW,KAAA,CAAM,OAAO,CAAA;AACtC,IAAA,IAAI,KAAA,IAAS,KAAA,CAAM,CAAC,CAAA,EAAG;AACrB,MAAA,OAAO,KAAA,CAAM,CAAC,CAAA,CAAE,WAAA,EAAY;AAAA,IAC9B;AAAA,EACF;AAEA,EAAA,OAAO,MAAA;AACT;AAEA,SAAS,iBAAiB,KAAA,EAAuB;AAG/C,EAAA,OAAO,KAAA,CACJ,UAAA,CAAW,UAAA,EAAY,KAAK,CAAA,CAC5B,UAAA,CAAW,UAAA,EAAY,KAAK,CAAA,CAC5B,UAAA,CAAW,UAAA,EAAY,GAAG,EAC1B,IAAA,EAAK;AACV;AAKO,IAAM,aAAA,GAAgB;AAAA,EAC3B,MAAA,EAAQ,QAAA;AAAA,EACR,MAAA,EAAQ,QAAA;AAAA,EACR,MAAA,EAAQ,QAAA;AAAA,EACR,MAAA,EAAQ,QAAA;AAAA,EACR,KAAA,EAAO,OAAA;AAAA,EACP,SAAA,EAAW;AACb;AAKO,IAAM,UAAA,GAAa;AAAA,EACxB,UAAA,EAAY,YAAA;AAAA,EACZ,KAAA,EAAO,OAAA;AAAA,EACP,OAAA,EAAS,SAAA;AAAA,EACT,KAAA,EAAO,OAAA;AAAA,EACP,MAAA,EAAQ,QAAA;AAAA,EACR,KAAA,EAAO;AACT;AAGA,IAAM,mBAAA,GAAsB,MAAA,CAAO,GAAA,CAAI,yBAAyB,CAAA;AAkEzD,SAAS,kBAAA,CACd,QACA,OAAA,EACG;AAGH,EAAA,IAAK,MAAA,CAAe,mBAAmB,CAAA,EAAG;AACxC,IAAA,OAAO,MAAA;AAAA,EACT;AAEA,EAAA,MAAM;AAAA,IACJ,QAAA;AAAA,IACA,MAAA;AAAA,IACA,OAAA;AAAA,IACA,cAAc,EAAC;AAAA,IACf,aAAA,GAAgB,IAAA;AAAA,IAChB,oBAAA,GAAuB;AAAA,GACzB,GAAI,OAAA;AAEJ,EAAA,MAAM,SAASD,2BAAA,EAAU;AACzB,EAAA,MAAM,SAAS,MAAA,CAAO,MAAA;AAGtB,EAAA,MAAM,mBAAA,GAAsB,OAAA,IAAW,sBAAA,CAAuB,MAAM,CAAA;AACpE,EAAA,MAAM,OAAA,GAAU,IAAI,GAAA,CAAI,WAAW,CAAA;AAEnC,EAAA,KAAA,MAAW,cAAc,mBAAA,EAAqB;AAC5C,IAAA,IAAI,OAAA,CAAQ,GAAA,CAAI,UAAU,CAAA,EAAG;AAC7B,IAAA,IAAI,UAAA,CAAW,UAAA,CAAW,GAAG,CAAA,EAAG;AAGhC,IAAA,MAAM,MAAA,GAAU,OAAe,UAAU,CAAA;AACzC,IAAA,IAAI,OAAO,WAAW,UAAA,EAAY;AAGlC,IAAA,MAAM,cAAA,GAAiB,MAAA;AAIvB,IAAC,MAAA,CAAe,UAAU,CAAA,GAAI,eAAA,GAA4B,IAAA,EAAa;AACrE,MAAA,MAAM,SAAA,GAAY,iBAAiB,UAAU,CAAA;AAC7C,MAAA,MAAM,KAAA,GAAQ,eAAe,UAAU,CAAA;AAEvC,MAAA,MAAM,QAAA,GAAW,KAAA,GACb,CAAA,EAAG,QAAQ,CAAA,CAAA,EAAI,SAAS,CAAA,CAAA,EAAI,KAAK,CAAA,CAAA,GACjC,CAAA,EAAG,QAAQ,CAAA,CAAA,EAAI,SAAS,CAAA,CAAA;AAE5B,MAAA,OAAO,MAAA,CAAO,eAAA,CAAgB,QAAA,EAAU,OAAO,IAAA,KAAS;AACtD,QAAA,MAAM,SAAA,GAAY,YAAY,GAAA,EAAI;AAElC,QAAA,IAAI;AACF,UAAA,IAAA,CAAK,aAAA,CAAc;AAAA,YACjB,WAAA,EAAa,QAAA;AAAA,YACb,cAAA,EAAgB;AAAA,WACjB,CAAA;AAED,UAAA,IAAI,MAAA,EAAQ;AACV,YAAA,IAAA,CAAK,YAAA,CAAa,WAAW,MAAM,CAAA;AAAA,UACrC;AAEA,UAAA,IAAI,KAAA,EAAO;AACT,YAAA,IAAA,CAAK,YAAA,CAAa,gBAAgB,KAAK,CAAA;AAAA,UACzC;AAGA,UAAA,MAAM,KAAA,GAAQ,qBAAqB,IAAI,CAAA;AACvC,UAAA,IAAI,KAAA,EAAO;AACT,YAAA,IAAA,CAAK,YAAA;AAAA,cACH,cAAA;AAAA,cACA,aAAA,GAAgB,gBAAA,CAAiB,KAAK,CAAA,GAAI;AAAA,aAC5C;AAAA,UACF;AAGA,UAAA,MAAM,MAAA,GAAS,MAAM,cAAA,CAAe,KAAA,CAAM,MAAM,IAAI,CAAA;AAEpD,UAAA,MAAM,QAAA,GAAW,WAAA,CAAY,GAAA,EAAI,GAAI,SAAA;AAErC,UAAA,IAAA,CAAK,SAAA,CAAU,EAAE,IAAA,EAAMC,kBAAA,CAAe,IAAI,CAAA;AAC1C,UAAA,IAAA,CAAK,aAAA,CAAc;AAAA,YACjB,gBAAA,EAAkB;AAAA,WACnB,CAAA;AAGD,UAAA,IAAI,WAAW,oBAAA,EAAsB;AACnC,YAAA,IAAA,CAAK,YAAA,CAAa,iBAAiB,IAAI,CAAA;AACvC,YAAA,IAAA,CAAK,YAAA;AAAA,cACH,4BAAA;AAAA,cACA;AAAA,aACF;AAAA,UACF;AAGA,UAAA,IAAI,KAAA,CAAM,OAAA,CAAQ,MAAM,CAAA,EAAG;AACzB,YAAA,IAAA,CAAK,YAAA,CAAa,iBAAA,EAAmB,MAAA,CAAO,MAAM,CAAA;AAAA,UACpD;AAEA,UAAA,OAAO,MAAA;AAAA,QACT,SAAS,KAAA,EAAO;AACd,UAAA,MAAM,QAAA,GAAW,WAAA,CAAY,GAAA,EAAI,GAAI,SAAA;AAErC,UAAA,IAAA,CAAK,SAAA,CAAU;AAAA,YACb,MAAMA,kBAAA,CAAe,KAAA;AAAA,YACrB,OAAA,EAAS,KAAA,YAAiB,KAAA,GAAQ,KAAA,CAAM,OAAA,GAAU;AAAA,WACnD,CAAA;AAED,UAAA,IAAA,CAAK,aAAA,CAAc;AAAA,YACjB,gBAAA,EAAkB,QAAA;AAAA,YAClB,YAAA,EACE,KAAA,YAAiB,KAAA,GAAQ,KAAA,CAAM,YAAY,IAAA,GAAO,SAAA;AAAA,YACpD,eAAA,EACE,KAAA,YAAiB,KAAA,GAAQ,KAAA,CAAM,OAAA,GAAU;AAAA,WAC5C,CAAA;AAED,UAAA,IAAA,CAAK,eAAA;AAAA,YACH,iBAAiB,KAAA,GAAQ,KAAA,GAAQ,IAAI,KAAA,CAAM,MAAA,CAAO,KAAK,CAAC;AAAA,WAC1D;AAEA,UAAA,MAAM,KAAA;AAAA,QACR,CAAA,SAAE;AACA,UAAA,IAAA,CAAK,GAAA,EAAI;AAAA,QACX;AAAA,MACF,CAAC,CAAA;AAAA,IACH,CAAA;AAIA,IAAA,MAAA,CAAO,cAAA,CAAgB,MAAA,CAAe,UAAU,CAAA,EAAG,MAAA,EAAQ;AAAA,MACzD,KAAA,EAAO,UAAA;AAAA,MACP,YAAA,EAAc;AAAA,KACf,CAAA;AAAA,EACH;AAIA,EAAC,MAAA,CAAe,mBAAmB,CAAA,GAAI,IAAA;AAEvC,EAAA,OAAO,MAAA;AACT;AAKA,SAAS,uBAAuB,MAAA,EAA0B;AACxD,EAAA,MAAM,UAAoB,EAAC;AAC3B,EAAA,MAAM,KAAA,GAAQ,MAAA,CAAO,cAAA,CAAe,MAAM,CAAA;AAG1C,EAAA,KAAA,MAAW,GAAA,IAAO,MAAA,CAAO,mBAAA,CAAoB,MAAM,CAAA,EAAG;AAEpD,IAAA,IAAI,OAAQ,OAAe,GAAG,CAAA,KAAM,cAAc,CAAC,GAAA,CAAI,UAAA,CAAW,GAAG,CAAA,EAAG;AACtE,MAAA,OAAA,CAAQ,KAAK,GAAG,CAAA;AAAA,IAClB;AAAA,EACF;AAGA,EAAA,IAAI,KAAA,EAAO;AACT,IAAA,KAAA,MAAW,GAAA,IAAO,MAAA,CAAO,mBAAA,CAAoB,KAAK,CAAA,EAAG;AACnD,MAAA,IACE,OAAO,KAAA,CAAM,GAAG,CAAA,KAAM,UAAA,IACtB,CAAC,GAAA,CAAI,UAAA,CAAW,GAAG,CAAA,IACnB,GAAA,KAAQ,aAAA,EACR;AACA,QAAA,OAAA,CAAQ,KAAK,GAAG,CAAA;AAAA,MAClB;AAAA,IACF;AAAA,EACF;AAEA,EAAA,OAAO,CAAC,GAAG,IAAI,GAAA,CAAI,OAAO,CAAC,CAAA;AAC7B;AAMA,SAAS,qBAAqB,IAAA,EAAiC;AAC7D,EAAA,IAAI,IAAA,CAAK,MAAA,KAAW,CAAA,EAAG,OAAO,MAAA;AAE9B,EAAA,MAAM,QAAA,GAAW,KAAK,CAAC,CAAA;AAGvB,EAAA,IAAI,OAAO,aAAa,QAAA,EAAU;AAChC,IAAA,OAAO,QAAA;AAAA,EACT;AAGA,EAAA,IAAI,QAAA,IAAY,OAAO,QAAA,KAAa,QAAA,EAAU;AAC5C,IAAA,IAAI,KAAA,IAAS,QAAA,IAAY,OAAO,QAAA,CAAS,QAAQ,QAAA,EAAU;AACzD,MAAA,OAAO,QAAA,CAAS,GAAA;AAAA,IAClB;AAEA,IAAA,IAAI,MAAA,IAAU,QAAA,IAAY,OAAO,QAAA,CAAS,SAAS,QAAA,EAAU;AAC3D,MAAA,OAAO,QAAA,CAAS,IAAA;AAAA,IAClB;AAEA,IAAA,IAAI,SAAA,IAAa,QAAA,IAAY,OAAO,QAAA,CAAS,YAAY,UAAA,EAAY;AACnE,MAAA,IAAI;AACF,QAAA,MAAM,WAAA,GAAc,SAAS,OAAA,EAAQ;AACrC,QAAA,IAAI,OAAO,WAAA,KAAgB,QAAA,EAAU,OAAO,WAAA;AAC5C,QAAA,IACE,WAAA,IACA,OAAO,WAAA,KAAgB,QAAA,IACvB,SAAS,WAAA,EACT;AACA,UAAA,OAAO,WAAA,CAAY,GAAA;AAAA,QACrB;AAAA,MACF,CAAA,CAAA,MAAQ;AAAA,MAER;AAAA,IACF;AAAA,EACF;AAEA,EAAA,OAAO,MAAA;AACT","file":"db.cjs","sourcesContent":["/**\n * Database Instrumentation Helpers\n *\n * Optional import: Not included in main bundle\n * Import from: 'autotel/db'\n *\n * Provides functional utilities for database query instrumentation.\n * Works with Prisma, Drizzle, TypeORM, raw SQL, and more.\n *\n * @example\n * ```typescript\n * import { instrumentDatabase } from 'autotel/db'\n *\n * const db = drizzle(pool)\n * instrumentDatabase(db, { dbSystem: 'postgresql', dbName: 'myapp' })\n *\n * // Now all queries are automatically trace\n * await db.select().from(users)\n * ```\n */\n\nimport { SpanStatusCode } from '@opentelemetry/api';\nimport { getConfig } from './config';\n\n/**\n * Helper: Trace a single database query\n *\n * @example\n * ```typescript\n * import { tracebQuery } from 'autotel/db'\n *\n * const users = await tracebQuery(\n * 'postgresql',\n * 'SELECT',\n * () => db.query('SELECT * FROM users WHERE active = true')\n * )\n * ```\n */\nexport async function tracebQuery<T>(\n dbSystem: string,\n operation: string,\n fn: () => Promise<T>,\n attributes?: Record<string, string | number>,\n): Promise<T> {\n const config = getConfig();\n const tracer = config.tracer;\n\n const spanName = `${dbSystem}.${operation}`;\n\n return tracer.startActiveSpan(spanName, async (span) => {\n const startTime = performance.now();\n\n try {\n span.setAttributes({\n 'db.system': dbSystem,\n 'db.operation': operation,\n ...attributes,\n });\n\n const result = await fn();\n\n const duration = performance.now() - startTime;\n span.setStatus({ code: SpanStatusCode.OK });\n span.setAttribute('db.duration_ms', duration);\n\n if (Array.isArray(result)) {\n span.setAttribute('db.result_count', result.length);\n }\n\n return result;\n } catch (error) {\n const duration = performance.now() - startTime;\n\n span.setStatus({\n code: SpanStatusCode.ERROR,\n message: error instanceof Error ? error.message : 'Unknown error',\n });\n\n span.setAttributes({\n 'db.duration_ms': duration,\n 'error.type':\n error instanceof Error ? error.constructor.name : 'Unknown',\n 'error.message':\n error instanceof Error ? error.message : 'Unknown error',\n });\n\n throw error;\n } finally {\n span.end();\n }\n });\n}\n\n// Helper functions\n\nfunction inferDbOperation(methodName: string): string {\n const lower = methodName.toLowerCase();\n if (lower.includes('find') || lower.includes('get') || lower.includes('list'))\n return 'SELECT';\n if (lower.includes('create') || lower.includes('insert')) return 'INSERT';\n if (lower.includes('update') || lower.includes('modify')) return 'UPDATE';\n if (lower.includes('delete') || lower.includes('remove')) return 'DELETE';\n if (lower.includes('count')) return 'COUNT';\n return 'QUERY';\n}\n\nfunction inferTableName(methodName: string): string | undefined {\n // Extract table name from method patterns like:\n // findUser -> user\n // listUsers -> users\n // createOrder -> order\n\n const patterns = [\n /find([A-Z][a-zA-Z]+)/,\n /get([A-Z][a-zA-Z]+)/,\n /list([A-Z][a-zA-Z]+)/,\n /create([A-Z][a-zA-Z]+)/,\n /update([A-Z][a-zA-Z]+)/,\n /delete([A-Z][a-zA-Z]+)/,\n /remove([A-Z][a-zA-Z]+)/,\n ];\n\n for (const pattern of patterns) {\n const match = methodName.match(pattern);\n if (match && match[1]) {\n return match[1].toLowerCase();\n }\n }\n\n return undefined;\n}\n\nfunction sanitizeSqlQuery(query: string): string {\n // Remove string literals and sensitive values (PII, credentials, etc.)\n // Preserves query structure for debugging while protecting data\n return query\n .replaceAll(/'[^']*'/g, \"'?'\")\n .replaceAll(/\"[^\"]*\"/g, '\"?\"')\n .replaceAll(/\\b\\d+\\b/g, '?') // Replace literal numbers\n .trim();\n}\n\n/**\n * Common database operation metrics\n */\nexport const DB_OPERATIONS = {\n SELECT: 'SELECT',\n INSERT: 'INSERT',\n UPDATE: 'UPDATE',\n DELETE: 'DELETE',\n COUNT: 'COUNT',\n AGGREGATE: 'AGGREGATE',\n} as const;\n\n/**\n * Common database systems\n */\nexport const DB_SYSTEMS = {\n POSTGRESQL: 'postgresql',\n MYSQL: 'mysql',\n MONGODB: 'mongodb',\n REDIS: 'redis',\n SQLITE: 'sqlite',\n MSSQL: 'mssql',\n} as const;\n\n// Symbol for idempotency - prevents double-instrumentation\nconst INSTRUMENTED_SYMBOL = Symbol.for('autotel.db.instrumented');\n\n/**\n * Options for instrumentDatabase\n */\nexport interface InstrumentDatabaseOptions {\n /** Database system (e.g., 'postgresql', 'mysql') */\n dbSystem: string;\n /** Database name (optional) */\n dbName?: string;\n /** Method names to instrument (if not provided, instruments common patterns) */\n methods?: string[];\n /** Method names to skip */\n skipMethods?: string[];\n /** Sanitize queries (remove sensitive data) - default: true */\n sanitizeQuery?: boolean;\n /** Slow query threshold in milliseconds - default: 1000ms */\n slowQueryThresholdMs?: number;\n}\n\n/**\n * Instrument a database client instance with OpenTelemetry tracing\n *\n * This is a function-based alternative to @DbInstrumented decorator.\n * Modifies the client in-place and returns it (idempotent - safe to call multiple times).\n *\n * Inspired by otel-drizzle and other otel instrumentation packages.\n *\n * @example Drizzle ORM\n * ```typescript\n * import { drizzle } from 'drizzle-orm/node-postgres'\n * import { instrumentDatabase } from 'autotel/db'\n *\n * const db = drizzle(pool)\n * instrumentDatabase(db, { dbSystem: 'postgresql', dbName: 'myapp' })\n *\n * // Now all db queries are automatically trace\n * await db.select().from(users)\n * ```\n *\n * @example Prisma\n * ```typescript\n * import { PrismaClient } from '@prisma/client'\n * import { instrumentDatabase } from 'autotel/db'\n *\n * const prisma = new PrismaClient()\n * instrumentDatabase(prisma, {\n * dbSystem: 'postgresql',\n * methods: ['findMany', 'findUnique', 'create', 'update', 'delete']\n * })\n *\n * // All specified methods are trace\n * await prisma.user.findMany()\n * ```\n *\n * @example Generic database client\n * ```typescript\n * import { instrumentDatabase } from 'autotel/db'\n *\n * const db = createDatabaseClient()\n * instrumentDatabase(db, {\n * dbSystem: 'mongodb',\n * methods: ['find', 'findOne', 'insertOne', 'updateOne', 'deleteOne']\n * })\n * ```\n */\nexport function instrumentDatabase<T extends object>(\n client: T,\n options: InstrumentDatabaseOptions,\n): T {\n // Idempotency check - if already instrumented, return as-is\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n if ((client as any)[INSTRUMENTED_SYMBOL]) {\n return client;\n }\n\n const {\n dbSystem,\n dbName,\n methods,\n skipMethods = [],\n sanitizeQuery = true,\n slowQueryThresholdMs = 1000,\n } = options;\n\n const config = getConfig();\n const tracer = config.tracer;\n\n // Determine which methods to instrument\n const methodsToInstrument = methods || extractDatabaseMethods(client);\n const skipSet = new Set(skipMethods);\n\n for (const methodName of methodsToInstrument) {\n if (skipSet.has(methodName)) continue;\n if (methodName.startsWith('_')) continue; // Skip private methods\n\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n const method = (client as any)[methodName];\n if (typeof method !== 'function') continue;\n\n // Preserve the original method\n const originalMethod = method;\n\n // Wrap the method\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n (client as any)[methodName] = async function (this: T, ...args: any[]) {\n const operation = inferDbOperation(methodName);\n const table = inferTableName(methodName);\n\n const spanName = table\n ? `${dbSystem}.${operation} ${table}`\n : `${dbSystem}.${operation}`;\n\n return tracer.startActiveSpan(spanName, async (span) => {\n const startTime = performance.now();\n\n try {\n span.setAttributes({\n 'db.system': dbSystem,\n 'db.operation': operation,\n });\n\n if (dbName) {\n span.setAttribute('db.name', dbName);\n }\n\n if (table) {\n span.setAttribute('db.sql.table', table);\n }\n\n // Try to extract query from arguments (common patterns)\n const query = extractQueryFromArgs(args);\n if (query) {\n span.setAttribute(\n 'db.statement',\n sanitizeQuery ? sanitizeSqlQuery(query) : query,\n );\n }\n\n // Execute original method\n const result = await originalMethod.apply(this, args);\n\n const duration = performance.now() - startTime;\n\n span.setStatus({ code: SpanStatusCode.OK });\n span.setAttributes({\n 'db.duration_ms': duration,\n });\n\n // Mark slow queries\n if (duration > slowQueryThresholdMs) {\n span.setAttribute('db.slow_query', true);\n span.setAttribute(\n 'db.slow_query_threshold_ms',\n slowQueryThresholdMs,\n );\n }\n\n // Track result count if it's an array\n if (Array.isArray(result)) {\n span.setAttribute('db.result_count', result.length);\n }\n\n return result;\n } catch (error) {\n const duration = performance.now() - startTime;\n\n span.setStatus({\n code: SpanStatusCode.ERROR,\n message: error instanceof Error ? error.message : 'Unknown error',\n });\n\n span.setAttributes({\n 'db.duration_ms': duration,\n 'error.type':\n error instanceof Error ? error.constructor.name : 'Unknown',\n 'error.message':\n error instanceof Error ? error.message : 'Unknown error',\n });\n\n span.recordException(\n error instanceof Error ? error : new Error(String(error)),\n );\n\n throw error;\n } finally {\n span.end();\n }\n });\n };\n\n // Preserve function name\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n Object.defineProperty((client as any)[methodName], 'name', {\n value: methodName,\n configurable: true,\n });\n }\n\n // Mark as instrumented\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n (client as any)[INSTRUMENTED_SYMBOL] = true;\n\n return client;\n}\n\n/**\n * Extract method names from a database client that should be instrumented\n */\nfunction extractDatabaseMethods(client: object): string[] {\n const methods: string[] = [];\n const proto = Object.getPrototypeOf(client);\n\n // Get own methods\n for (const key of Object.getOwnPropertyNames(client)) {\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n if (typeof (client as any)[key] === 'function' && !key.startsWith('_')) {\n methods.push(key);\n }\n }\n\n // Get prototype methods\n if (proto) {\n for (const key of Object.getOwnPropertyNames(proto)) {\n if (\n typeof proto[key] === 'function' &&\n !key.startsWith('_') &&\n key !== 'constructor'\n ) {\n methods.push(key);\n }\n }\n }\n\n return [...new Set(methods)]; // Deduplicate\n}\n\n/**\n * Try to extract SQL query from common argument patterns\n */\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nfunction extractQueryFromArgs(args: any[]): string | undefined {\n if (args.length === 0) return undefined;\n\n const firstArg = args[0];\n\n // String query (raw SQL)\n if (typeof firstArg === 'string') {\n return firstArg;\n }\n\n // Object with sql property\n if (firstArg && typeof firstArg === 'object') {\n if ('sql' in firstArg && typeof firstArg.sql === 'string') {\n return firstArg.sql;\n }\n // PostgreSQL-style query object\n if ('text' in firstArg && typeof firstArg.text === 'string') {\n return firstArg.text;\n }\n // Query builder pattern\n if ('toQuery' in firstArg && typeof firstArg.toQuery === 'function') {\n try {\n const queryResult = firstArg.toQuery();\n if (typeof queryResult === 'string') return queryResult;\n if (\n queryResult &&\n typeof queryResult === 'object' &&\n 'sql' in queryResult\n ) {\n return queryResult.sql as string;\n }\n } catch {\n // Ignore errors from toQuery()\n }\n }\n }\n\n return undefined;\n}\n"]}
package/dist/db.d.cts ADDED
@@ -0,0 +1,123 @@
1
+ /**
2
+ * Database Instrumentation Helpers
3
+ *
4
+ * Optional import: Not included in main bundle
5
+ * Import from: 'autotel/db'
6
+ *
7
+ * Provides functional utilities for database query instrumentation.
8
+ * Works with Prisma, Drizzle, TypeORM, raw SQL, and more.
9
+ *
10
+ * @example
11
+ * ```typescript
12
+ * import { instrumentDatabase } from 'autotel/db'
13
+ *
14
+ * const db = drizzle(pool)
15
+ * instrumentDatabase(db, { dbSystem: 'postgresql', dbName: 'myapp' })
16
+ *
17
+ * // Now all queries are automatically trace
18
+ * await db.select().from(users)
19
+ * ```
20
+ */
21
+ /**
22
+ * Helper: Trace a single database query
23
+ *
24
+ * @example
25
+ * ```typescript
26
+ * import { tracebQuery } from 'autotel/db'
27
+ *
28
+ * const users = await tracebQuery(
29
+ * 'postgresql',
30
+ * 'SELECT',
31
+ * () => db.query('SELECT * FROM users WHERE active = true')
32
+ * )
33
+ * ```
34
+ */
35
+ declare function tracebQuery<T>(dbSystem: string, operation: string, fn: () => Promise<T>, attributes?: Record<string, string | number>): Promise<T>;
36
+ /**
37
+ * Common database operation metrics
38
+ */
39
+ declare const DB_OPERATIONS: {
40
+ readonly SELECT: "SELECT";
41
+ readonly INSERT: "INSERT";
42
+ readonly UPDATE: "UPDATE";
43
+ readonly DELETE: "DELETE";
44
+ readonly COUNT: "COUNT";
45
+ readonly AGGREGATE: "AGGREGATE";
46
+ };
47
+ /**
48
+ * Common database systems
49
+ */
50
+ declare const DB_SYSTEMS: {
51
+ readonly POSTGRESQL: "postgresql";
52
+ readonly MYSQL: "mysql";
53
+ readonly MONGODB: "mongodb";
54
+ readonly REDIS: "redis";
55
+ readonly SQLITE: "sqlite";
56
+ readonly MSSQL: "mssql";
57
+ };
58
+ /**
59
+ * Options for instrumentDatabase
60
+ */
61
+ interface InstrumentDatabaseOptions {
62
+ /** Database system (e.g., 'postgresql', 'mysql') */
63
+ dbSystem: string;
64
+ /** Database name (optional) */
65
+ dbName?: string;
66
+ /** Method names to instrument (if not provided, instruments common patterns) */
67
+ methods?: string[];
68
+ /** Method names to skip */
69
+ skipMethods?: string[];
70
+ /** Sanitize queries (remove sensitive data) - default: true */
71
+ sanitizeQuery?: boolean;
72
+ /** Slow query threshold in milliseconds - default: 1000ms */
73
+ slowQueryThresholdMs?: number;
74
+ }
75
+ /**
76
+ * Instrument a database client instance with OpenTelemetry tracing
77
+ *
78
+ * This is a function-based alternative to @DbInstrumented decorator.
79
+ * Modifies the client in-place and returns it (idempotent - safe to call multiple times).
80
+ *
81
+ * Inspired by otel-drizzle and other otel instrumentation packages.
82
+ *
83
+ * @example Drizzle ORM
84
+ * ```typescript
85
+ * import { drizzle } from 'drizzle-orm/node-postgres'
86
+ * import { instrumentDatabase } from 'autotel/db'
87
+ *
88
+ * const db = drizzle(pool)
89
+ * instrumentDatabase(db, { dbSystem: 'postgresql', dbName: 'myapp' })
90
+ *
91
+ * // Now all db queries are automatically trace
92
+ * await db.select().from(users)
93
+ * ```
94
+ *
95
+ * @example Prisma
96
+ * ```typescript
97
+ * import { PrismaClient } from '@prisma/client'
98
+ * import { instrumentDatabase } from 'autotel/db'
99
+ *
100
+ * const prisma = new PrismaClient()
101
+ * instrumentDatabase(prisma, {
102
+ * dbSystem: 'postgresql',
103
+ * methods: ['findMany', 'findUnique', 'create', 'update', 'delete']
104
+ * })
105
+ *
106
+ * // All specified methods are trace
107
+ * await prisma.user.findMany()
108
+ * ```
109
+ *
110
+ * @example Generic database client
111
+ * ```typescript
112
+ * import { instrumentDatabase } from 'autotel/db'
113
+ *
114
+ * const db = createDatabaseClient()
115
+ * instrumentDatabase(db, {
116
+ * dbSystem: 'mongodb',
117
+ * methods: ['find', 'findOne', 'insertOne', 'updateOne', 'deleteOne']
118
+ * })
119
+ * ```
120
+ */
121
+ declare function instrumentDatabase<T extends object>(client: T, options: InstrumentDatabaseOptions): T;
122
+
123
+ export { DB_OPERATIONS, DB_SYSTEMS, type InstrumentDatabaseOptions, instrumentDatabase, tracebQuery };
package/dist/db.d.ts ADDED
@@ -0,0 +1,123 @@
1
+ /**
2
+ * Database Instrumentation Helpers
3
+ *
4
+ * Optional import: Not included in main bundle
5
+ * Import from: 'autotel/db'
6
+ *
7
+ * Provides functional utilities for database query instrumentation.
8
+ * Works with Prisma, Drizzle, TypeORM, raw SQL, and more.
9
+ *
10
+ * @example
11
+ * ```typescript
12
+ * import { instrumentDatabase } from 'autotel/db'
13
+ *
14
+ * const db = drizzle(pool)
15
+ * instrumentDatabase(db, { dbSystem: 'postgresql', dbName: 'myapp' })
16
+ *
17
+ * // Now all queries are automatically trace
18
+ * await db.select().from(users)
19
+ * ```
20
+ */
21
+ /**
22
+ * Helper: Trace a single database query
23
+ *
24
+ * @example
25
+ * ```typescript
26
+ * import { tracebQuery } from 'autotel/db'
27
+ *
28
+ * const users = await tracebQuery(
29
+ * 'postgresql',
30
+ * 'SELECT',
31
+ * () => db.query('SELECT * FROM users WHERE active = true')
32
+ * )
33
+ * ```
34
+ */
35
+ declare function tracebQuery<T>(dbSystem: string, operation: string, fn: () => Promise<T>, attributes?: Record<string, string | number>): Promise<T>;
36
+ /**
37
+ * Common database operation metrics
38
+ */
39
+ declare const DB_OPERATIONS: {
40
+ readonly SELECT: "SELECT";
41
+ readonly INSERT: "INSERT";
42
+ readonly UPDATE: "UPDATE";
43
+ readonly DELETE: "DELETE";
44
+ readonly COUNT: "COUNT";
45
+ readonly AGGREGATE: "AGGREGATE";
46
+ };
47
+ /**
48
+ * Common database systems
49
+ */
50
+ declare const DB_SYSTEMS: {
51
+ readonly POSTGRESQL: "postgresql";
52
+ readonly MYSQL: "mysql";
53
+ readonly MONGODB: "mongodb";
54
+ readonly REDIS: "redis";
55
+ readonly SQLITE: "sqlite";
56
+ readonly MSSQL: "mssql";
57
+ };
58
+ /**
59
+ * Options for instrumentDatabase
60
+ */
61
+ interface InstrumentDatabaseOptions {
62
+ /** Database system (e.g., 'postgresql', 'mysql') */
63
+ dbSystem: string;
64
+ /** Database name (optional) */
65
+ dbName?: string;
66
+ /** Method names to instrument (if not provided, instruments common patterns) */
67
+ methods?: string[];
68
+ /** Method names to skip */
69
+ skipMethods?: string[];
70
+ /** Sanitize queries (remove sensitive data) - default: true */
71
+ sanitizeQuery?: boolean;
72
+ /** Slow query threshold in milliseconds - default: 1000ms */
73
+ slowQueryThresholdMs?: number;
74
+ }
75
+ /**
76
+ * Instrument a database client instance with OpenTelemetry tracing
77
+ *
78
+ * This is a function-based alternative to @DbInstrumented decorator.
79
+ * Modifies the client in-place and returns it (idempotent - safe to call multiple times).
80
+ *
81
+ * Inspired by otel-drizzle and other otel instrumentation packages.
82
+ *
83
+ * @example Drizzle ORM
84
+ * ```typescript
85
+ * import { drizzle } from 'drizzle-orm/node-postgres'
86
+ * import { instrumentDatabase } from 'autotel/db'
87
+ *
88
+ * const db = drizzle(pool)
89
+ * instrumentDatabase(db, { dbSystem: 'postgresql', dbName: 'myapp' })
90
+ *
91
+ * // Now all db queries are automatically trace
92
+ * await db.select().from(users)
93
+ * ```
94
+ *
95
+ * @example Prisma
96
+ * ```typescript
97
+ * import { PrismaClient } from '@prisma/client'
98
+ * import { instrumentDatabase } from 'autotel/db'
99
+ *
100
+ * const prisma = new PrismaClient()
101
+ * instrumentDatabase(prisma, {
102
+ * dbSystem: 'postgresql',
103
+ * methods: ['findMany', 'findUnique', 'create', 'update', 'delete']
104
+ * })
105
+ *
106
+ * // All specified methods are trace
107
+ * await prisma.user.findMany()
108
+ * ```
109
+ *
110
+ * @example Generic database client
111
+ * ```typescript
112
+ * import { instrumentDatabase } from 'autotel/db'
113
+ *
114
+ * const db = createDatabaseClient()
115
+ * instrumentDatabase(db, {
116
+ * dbSystem: 'mongodb',
117
+ * methods: ['find', 'findOne', 'insertOne', 'updateOne', 'deleteOne']
118
+ * })
119
+ * ```
120
+ */
121
+ declare function instrumentDatabase<T extends object>(client: T, options: InstrumentDatabaseOptions): T;
122
+
123
+ export { DB_OPERATIONS, DB_SYSTEMS, type InstrumentDatabaseOptions, instrumentDatabase, tracebQuery };