WIP: Knex tracing

thomasboyt · Jul 17, 2019 · 3bb9f36 · 3bb9f36
1 parent 018529a
commit 3bb9f36
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/api/src/models/feed.ts b/api/src/models/feed.ts
@@ -6,6 +6,7 @@ import { ENTRY_PAGE_LIMIT } from '../constants';
 import { FeedSongItem, FeedItem, FeedMixtapeItem } from '../resources';
 import { serializeSong, selectSongsQuery } from './song';
 import validateOrThrow from '../util/validateOrThrow';
+import traceQuery from '../util/traceQuery';
 
 function serializeFeedSongItem(row: any): FeedSongItem {
   return {
@@ -101,7 +102,7 @@ export async function getFeedByUserId(
 
   query = query.limit(ENTRY_PAGE_LIMIT);
 
-  const rows = await query;
+  const rows = await traceQuery(query);
 
   return rows.map(
     (row: any): FeedItem => {

diff --git a/api/src/util/traceQuery.ts b/api/src/util/traceQuery.ts
@@ -0,0 +1,11 @@
+import * as Knex from 'knex';
+import { get as getAgent } from '@google-cloud/trace-agent';
+
+export default function traceQuery(query: Knex.QueryBuilder): PromiseLike<any> {
+  const tracer = getAgent();
+  const span = tracer.createChildSpan({ name: 'knex' });
+
+  return query.finally(() => {
+    span.endSpan();
+  });
+}