bulker: snowflake: don't load primary keys for empty tables

jitsucom · Dec 4, 2024 · 8b03619 · 8b03619
1 parent eae4800
commit 8b03619
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 4 deletions.
diff --git a/bulkerlib/implementations/sql/snowflake.go b/bulkerlib/implementations/sql/snowflake.go
@@ -291,6 +291,10 @@ func (s *Snowflake) GetTableSchema(ctx context.Context, namespace string, tableN
 				Statement: query,
 			})
 	}
+
+	if table.ColumnsCount() == 0 {
+		return table, nil
+	}
 
 	primaryKeyName, pkFields, err := s.getPrimaryKey(ctx, namespace, tableName)
 	if err != nil {

diff --git a/config-keeper/router.go b/config-keeper/router.go
@@ -57,10 +57,17 @@ func NewRouter(appContext *Context) *Router {
 			}
 			repStatuses[name] = status
 		}
-		c.JSON(http.StatusOK, gin.H{
-			"status":       utils.Ternary(healthy, "pass", "fail"),
-			"repositories": repStatuses,
-		})
+		if healthy {
+			c.JSON(http.StatusOK, gin.H{
+				"status":       "pass",
+				"repositories": repStatuses,
+			})
+		} else {
+			c.JSON(http.StatusServiceUnavailable, gin.H{
+				"status":       "fail",
+				"repositories": repStatuses,
+			})
+		}
 	})
 
 	return router