fix: fix pgsql lookup sql (#2476)

Signed-off-by: yisaer <[email protected]>
lf-edge · Dec 12, 2023 · ee9a1dc · ee9a1dc
1 parent 518299d
commit ee9a1dc
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/extensions/sources/sql/ext/sqlLookup.go b/extensions/sources/sql/ext/sqlLookup.go
@@ -144,9 +144,9 @@ func (g defaultSQLGen) buildQuery(fields []string, keys []string, values []inter
 	return query
 }
 
-type sqlServerSQLGen struct{ table string }
+type noQuoteSQLGen struct{ table string }
 
-func (g sqlServerSQLGen) buildQuery(fields []string, keys []string, values []interface{}) string {
+func (g noQuoteSQLGen) buildQuery(fields []string, keys []string, values []interface{}) string {
 	query := "SELECT "
 	if len(fields) == 0 {
 		query += "*"
@@ -175,8 +175,8 @@ func (g sqlServerSQLGen) buildQuery(fields []string, keys []string, values []int
 
 func (s *sqlLookupSource) buildQuery(fields []string, keys []string, values []interface{}) string {
 	switch strings.ToLower(s.driver) {
-	case "sqlserver", "mssql":
-		return sqlServerSQLGen{table: s.table}.buildQuery(fields, keys, values)
+	case "sqlserver", "mssql", "postgres":
+		return noQuoteSQLGen{table: s.table}.buildQuery(fields, keys, values)
 	default:
 		return defaultSQLGen{table: s.table}.buildQuery(fields, keys, values)
 	}