Use Execer if available

Signed-off-by: Andrew Thornton <[email protected]>

Author: zeripath

models/sql_postgres_with_schema.go

@@ -36,6 +36,21 @@ func (d *postgresSchemaDriver) Open(name string) (driver.Conn, error) {
 	if err != nil {
 		return conn, err
 	}
+	schemaValue, _ := driver.String.ConvertValue(setting.Database.Schema)
+
+	// golangci lint is incorrect here - there is no benefit to using driver.ExecerContext here
+	// and in any case pq does not implement it
+	if execer, ok := conn.(driver.Execer); ok { //nolint
+		_, err := execer.Exec(`SELECT set_config(
+			'search_path',
+			$1 || ',' || current_setting('search_path'),
+			false)`, []driver.Value{schemaValue}) //nolint
+		if err != nil {
+			_ = conn.Close()
+			return nil, err
+		}
+		return conn, nil
+	}
 
 	stmt, err := conn.Prepare(`SELECT set_config(
 		'search_path',
@@ -48,7 +63,6 @@ func (d *postgresSchemaDriver) Open(name string) (driver.Conn, error) {
 	defer stmt.Close()
 
 	// driver.String.ConvertValue will never return err for string
-	schemaValue, _ := driver.String.ConvertValue(setting.Database.Schema)
 
 	// golangci lint is incorrect here - there is no benefit to using stmt.ExecWithContext here
 	_, err = stmt.Exec([]driver.Value{schemaValue}) //nolint
@@ -57,5 +71,5 @@ func (d *postgresSchemaDriver) Open(name string) (driver.Conn, error) {
 		return nil, err
 	}
 
-	return conn, err
+	return conn, nil
 }