Fix CI regression, #1658 merged too soon (#1659)

Also need to use `activity.IsActivity` outside activity specific code
PeerDB-io · May 1, 2024 · f7c7ac2 · f7c7ac2
1 parent 67d7718
commit f7c7ac2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/flow/connectors/postgres/postgres.go b/flow/connectors/postgres/postgres.go
@@ -724,7 +724,9 @@ func (c *PostgresConnector) GetTableSchema(
 ) (*protos.GetTableSchemaBatchOutput, error) {
 	res := make(map[string]*protos.TableSchema)
 	for _, tableName := range req.TableIdentifiers {
-		activity.RecordHeartbeat(ctx, fmt.Sprintf("fetching schema for table %s", tableName))
+		if activity.IsActivity(ctx) {
+			activity.RecordHeartbeat(ctx, "fetching schema for table "+tableName)
+		}
 		tableSchema, err := c.getTableSchemaForTable(ctx, tableName, req.System)
 		if err != nil {
 			return nil, err