[-] fix update metrics in PostgreSQL, fixes #532

cybertec-postgresql · Aug 29, 2024 · 10f998d · 10f998d
1 parent af4696b
commit 10f998d
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/internal/db/conn.go b/internal/db/conn.go
@@ -42,7 +42,7 @@ type PgxPoolIface interface {
 	Stat() *pgxpool.Stat
 }
 
-func MarshallParam(v any) any {
+func MarshallParamToJSONB(v any) any {
 	if v == nil {
 		return nil
 	}

diff --git a/internal/metrics/postgres.go b/internal/metrics/postgres.go
@@ -132,8 +132,8 @@ func (dmrw *dbMetricReaderWriter) UpdateMetric(metricName string, metric Metric)
 	_, err := dmrw.configDb.Exec(dmrw.ctx, `UPDATE pgwatch.metric SET 
 	sqls = $2, init_sql = $3, description = $4, node_status = $5, gauges = $6, is_instance_level = $7, storage_name = $8
 	WHERE name = $1`,
-		metricName, db.MarshallParam(metric.SQLs), metric.InitSQL, metric.Description,
-		metric.NodeStatus, db.MarshallParam(metric.Gauges), metric.IsInstanceLevel, metric.StorageName)
+		metricName, db.MarshallParamToJSONB(metric.SQLs), metric.InitSQL, metric.Description,
+		metric.NodeStatus, metric.Gauges, metric.IsInstanceLevel, metric.StorageName)
 	return err
 }
 
@@ -145,6 +145,6 @@ func (dmrw *dbMetricReaderWriter) DeletePreset(presetName string) error {
 func (dmrw *dbMetricReaderWriter) UpdatePreset(presetName string, preset Preset) error {
 	sql := `INSERT INTO pgwatch.preset(name, description, metrics) VALUES ($1, $2, $3)
 	ON CONFLICT (name) DO UPDATE SET description = $2, metrics = $3`
-	_, err := dmrw.configDb.Exec(dmrw.ctx, sql, presetName, preset.Description, db.MarshallParam(preset.Metrics))
+	_, err := dmrw.configDb.Exec(dmrw.ctx, sql, presetName, preset.Description, db.MarshallParamToJSONB(preset.Metrics))
 	return err
 }
diff --git a/internal/sources/postgres.go b/internal/sources/postgres.go
@@ -49,7 +49,7 @@ func (r *dbSourcesReaderWriter) WriteSources(dbs Sources) error {
 }
 
 func (r *dbSourcesReaderWriter) updateDatabase(conn db.PgxIface, md Source) (err error) {
-	m := db.MarshallParam
+	m := db.MarshallParamToJSONB
 	sql := `insert into pgwatch.source(
 	name, 
 	"group",