Merge pull request #130 from Scale3-Labs/dylan/add-table-check-hot-fix

adding table exist checks
Scale3-Labs · May 31, 2024 · 0a0cb33 · 0a0cb33
2 parents aa8e62a + 44c9e9d
commit 0a0cb33
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/lib/services/trace_service.ts b/lib/services/trace_service.ts
@@ -115,6 +115,10 @@ export class TraceService implements ITraceService {
 
   async GetSpanById(span_id: string, project_id: string): Promise<Span> {
     try {
+      const tableExists = await this.client.checkTableExists(project_id);
+      if (!tableExists) {
+        return {} as Span;
+      }
       const query = sql.select().from(project_id).where({ span_id });
       const span: Span[] = await this.client.find<Span[]>(query);
       return span[0];
@@ -127,6 +131,10 @@ export class TraceService implements ITraceService {
 
   async GetTraceById(trace_id: string, project_id: string): Promise<Span[]> {
     try {
+      const tableExists = await this.client.checkTableExists(project_id);
+      if (!tableExists) {
+        return [];
+      }
       const query = sql.select().from(project_id).where({ trace_id });
       const span: Span[] = await this.client.find<Span[]>(query);
       return span;
@@ -305,6 +313,10 @@ export class TraceService implements ITraceService {
 
   async GetFailedSpans(project_id: string): Promise<Span[]> {
     try {
+      const tableExists = await this.client.checkTableExists(project_id);
+      if (!tableExists) {
+        return [];
+      }
       const query = sql
         .select()
         .from(project_id)
@@ -403,6 +415,10 @@ export class TraceService implements ITraceService {
     lastNHours = 168
   ): Promise<Span[]> {
     try {
+      const tableExists = await this.client.checkTableExists(project_id);
+      if (!tableExists) {
+        return [];
+      }
       const query = sql.select().from(project_id);
       if (!lastNHours) {
         return await this.client.find<Span[]>(query);