Improve TakeBackupOperation handler

cmd/ydbcp/main.go

@@ -49,7 +49,7 @@ func (s *server) GetBackup(ctx context.Context, request *pb.GetBackupRequest) (*
 		return nil, fmt.Errorf("failed to parse uuid %s: %w", request.GetId(), err)
 	}
 	backups, err := s.driver.SelectBackups(
-		ctx, queries.MakeReadTableQuery(
+		ctx, queries.NewReadTableQuery(
 			queries.WithTableName("Backups"),
 			queries.WithSelectFields(queries.AllBackupFields...),
 			queries.WithQueryFilters(
@@ -118,7 +118,7 @@ func (s *server) MakeBackup(ctx context.Context, req *pb.MakeBackupRequest) (*pb
 func (s *server) ListBackups(ctx context.Context, request *pb.ListBackupsRequest) (*pb.ListBackupsResponse, error) {
 	xlog.Debug(ctx, "ListBackups", zap.String("request", request.String()))
 	backups, err := s.driver.SelectBackups(
-		ctx, queries.MakeReadTableQuery(
+		ctx, queries.NewReadTableQuery(
 			queries.WithTableName("Backups"),
 			queries.WithSelectFields(queries.AllBackupFields...),
 			queries.WithQueryFilters(
@@ -156,7 +156,7 @@ func (s *server) ListOperations(ctx context.Context, request *pb.ListOperationsR
 ) {
 	xlog.Debug(ctx, "ListOperations", zap.String("request", request.String()))
 	operations, err := s.driver.SelectOperations(
-		ctx, queries.MakeReadTableQuery(
+		ctx, queries.NewReadTableQuery(
 			queries.WithTableName("Operations"),
 			queries.WithSelectFields(queries.AllOperationFields...),
 			queries.WithQueryFilters(
@@ -270,7 +270,10 @@ func main() {
 
 	handlersRegistry := processor.NewOperationHandlerRegistry()
 	err = handlersRegistry.Add(
-		types.OperationTypeTB, handlers.MakeTBOperationHandler(dbConnector, client.NewClientYdbConnector()),
+		types.OperationTypeTB,
+		handlers.NewTBOperationHandler(
+			dbConnector, client.NewClientYdbConnector(), configInstance, queries.NewWriteTableQuery,
+		),
 	)
 	if err != nil {
 		xlog.Error(ctx, "failed to register TB handler", zap.Error(err))
@@ -279,7 +282,7 @@ func main() {
 
 	err = handlersRegistry.Add(
 		types.OperationTypeRB,
-		handlers.MakeRBOperationHandler(dbConnector, client.NewClientYdbConnector(), configInstance),
+		handlers.NewRBOperationHandler(dbConnector, client.NewClientYdbConnector(), configInstance),
 	)
 
 	if err != nil {

go.mod

@@ -22,9 +22,9 @@ require (
 	github.com/golang-jwt/jwt/v4 v4.4.1 // indirect
 	github.com/pmezard/go-difflib v1.0.0 // indirect
 	go.uber.org/multierr v1.10.0 // indirect
-	golang.org/x/net v0.25.0 // indirect
-	golang.org/x/sync v0.6.0 // indirect
-	golang.org/x/sys v0.20.0 // indirect
-	golang.org/x/text v0.15.0 // indirect
+	golang.org/x/net v0.26.0 // indirect
+	golang.org/x/sync v0.7.0 // indirect
+	golang.org/x/sys v0.21.0 // indirect
+	golang.org/x/text v0.16.0 // indirect
 	google.golang.org/genproto/googleapis/rpc v0.0.0-20240528184218-531527333157 // indirect
 )

go.sum

@@ -103,28 +103,28 @@ golang.org/x/net v0.0.0-20190311183353-d8887717615a/go.mod h1:t9HGtf8HONx5eT2rtn
 golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
 golang.org/x/net v0.0.0-20200822124328-c89045814202/go.mod h1:/O7V0waA8r7cgGh81Ro3o1hOxt32SMVPicZroKQ2sZA=
 golang.org/x/net v0.0.0-20201021035429-f5854403a974/go.mod h1:sp8m0HH+o8qH0wwXwYZr8TS3Oi6o0r6Gce1SSxlDquU=
-golang.org/x/net v0.25.0 h1:d/OCCoBEUq33pjydKrGQhw7IlUPI2Oylr+8qLx49kac=
-golang.org/x/net v0.25.0/go.mod h1:JkAGAh7GEvH74S6FOH42FLoXpXbE/aqXSrIQjXgsiwM=
+golang.org/x/net v0.26.0 h1:soB7SVo0PWrY4vPW/+ay0jKDNScG2X9wFeYlXIvJsOQ=
+golang.org/x/net v0.26.0/go.mod h1:5YKkiSynbBIh3p6iOc/vibscux0x38BZDkn8sCUPxHE=
 golang.org/x/oauth2 v0.0.0-20180821212333-d2e6202438be/go.mod h1:N/0e6XlmueqKjAGxoOufVs8QHGRruUQn6yWY3a++T0U=
 golang.org/x/oauth2 v0.0.0-20200107190931-bf48bf16ab8d/go.mod h1:gOpvHmFTYa4IltrdGE7lF6nIHvwfUNPOp7c8zoXwtLw=
 golang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
 golang.org/x/sync v0.0.0-20181108010431-42b317875d0f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
 golang.org/x/sync v0.0.0-20181221193216-37e7f081c4d4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
 golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
-golang.org/x/sync v0.6.0 h1:5BMeUDZ7vkXGfEr1x9B4bRcTH4lpkTkpdh0T/J+qjbQ=
-golang.org/x/sync v0.6.0/go.mod h1:Czt+wKu1gCyEFDUtn0jG5QVvpJ6rzVqr5aXyt9drQfk=
+golang.org/x/sync v0.7.0 h1:YsImfSBoP9QPYL0xyKJPq0gcaJdG3rInoqxTWbfQu9M=
+golang.org/x/sync v0.7.0/go.mod h1:Czt+wKu1gCyEFDUtn0jG5QVvpJ6rzVqr5aXyt9drQfk=
 golang.org/x/sys v0.0.0-20180830151530-49385e6e1522/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
 golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
 golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
 golang.org/x/sys v0.0.0-20200323222414-85ca7c5b95cd/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
 golang.org/x/sys v0.0.0-20200930185726-fdedc70b468f/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
 golang.org/x/sys v0.0.0-20210119212857-b64e53b001e4/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
-golang.org/x/sys v0.20.0 h1:Od9JTbYCk261bKm4M/mw7AklTlFYIa0bIp9BgSm1S8Y=
-golang.org/x/sys v0.20.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
+golang.org/x/sys v0.21.0 h1:rF+pYz3DAGSQAxAu1CbC7catZg4ebC4UIeIhKxBZvws=
+golang.org/x/sys v0.21.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
 golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
 golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
-golang.org/x/text v0.15.0 h1:h1V/4gjBv8v9cjcR6+AR5+/cIYK5N/WAgiv4xlsEtAk=
-golang.org/x/text v0.15.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU=
+golang.org/x/text v0.16.0 h1:a94ExnEXNtEwYLGJSIUxnWoxoRz/ZcCsV63ROupILh4=
+golang.org/x/text v0.16.0/go.mod h1:GhwF1Be+LQoKShO3cGOHzqOgRrGaYc9AvblQOmPVHnI=
 golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
 golang.org/x/tools v0.0.0-20190114222345-bf090417da8b/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
 golang.org/x/tools v0.0.0-20190226205152-f727befe758c/go.mod h1:9Yl7xja0Znq3iFh3HoIrodX9oNMXvdceNzlUR8zjMvY=
@@ -147,8 +147,8 @@ google.golang.org/grpc v1.27.0/go.mod h1:qbnxyOmOxrQa7FizSgH+ReBfzJrCY1pSN7KXBS8
 google.golang.org/grpc v1.33.1/go.mod h1:fr5YgcSWrqhRRxogOsw7RzIpsmvOZ6IcH4kBYTpR3n0=
 google.golang.org/grpc v1.36.0/go.mod h1:qjiiYl8FncCW8feJPdyg3v6XW24KsRHe+dy9BAGRRjU=
 google.golang.org/grpc v1.47.0/go.mod h1:vN9eftEi1UMyUsIF80+uQXhHjbXYbm0uXoFCACuMGWk=
-google.golang.org/grpc v1.64.0 h1:KH3VH9y/MgNQg1dE7b3XfVK0GsPSIzJwdF617gUSbvY=
-google.golang.org/grpc v1.64.0/go.mod h1:oxjF8E3FBnjp+/gVFYdWacaLDx9na1aqy9oovLpxQYg=
+google.golang.org/grpc v1.64.1 h1:LKtvyfbX3UGVPFcGqJ9ItpVWW6oN/2XqTxfAnwRRXiA=
+google.golang.org/grpc v1.64.1/go.mod h1:hiQF4LFZelK2WKaP6W0L92zGHtiQdZxk8CrSdvyjeP0=
 google.golang.org/protobuf v0.0.0-20200109180630-ec00e32a8dfd/go.mod h1:DFci5gLYBciE7Vtevhsrf46CRTquxDuWsQurQQe4oz8=
 google.golang.org/protobuf v0.0.0-20200221191635-4d8936d0db64/go.mod h1:kwYJMbMJ01Woi6D6+Kah6886xMZcty6N08ah7+eCXa0=
 google.golang.org/protobuf v0.0.0-20200228230310-ab0ca4ff8a60/go.mod h1:cfTl7dwQJ+fmap5saPgwCLgHXTUD7jkjRqWcaiX5VyM=

internal/connectors/db/connector.go

@@ -47,6 +47,7 @@ type DBConnector interface {
 	CreateOperation(context.Context, types.Operation) (types.ObjectID, error)
 	CreateBackup(context.Context, types.Backup) (types.ObjectID, error)
 	UpdateBackup(context context.Context, id types.ObjectID, backupState string) error
+	ExecuteUpsert(ctx context.Context, queryBuilder queries.WriteTableQuery) error
 	Close()
 }
 
@@ -238,7 +239,7 @@ func (d *YdbConnector) SelectBackupsByStatus(
 	return DoStructSelect[types.Backup](
 		ctx,
 		d,
-		queries.MakeReadTableQuery(
+		queries.NewReadTableQuery(
 			queries.WithTableName("Backups"),
 			queries.WithSelectFields("id"),
 			queries.WithQueryFilters(
@@ -280,7 +281,7 @@ func (d *YdbConnector) ActiveOperations(ctx context.Context) (
 	return DoInterfaceSelect[types.Operation](
 		ctx,
 		d,
-		queries.MakeReadTableQuery(
+		queries.NewReadTableQuery(
 			queries.WithTableName("Operations"),
 			queries.WithSelectFields(queries.AllOperationFields...),
 			queries.WithQueryFilters(
@@ -300,14 +301,14 @@ func (d *YdbConnector) ActiveOperations(ctx context.Context) (
 func (d *YdbConnector) UpdateOperation(
 	ctx context.Context, operation types.Operation,
 ) error {
-	return d.ExecuteUpsert(ctx, queries.MakeWriteTableQuery(queries.WithUpdateOperation(operation)))
+	return d.ExecuteUpsert(ctx, queries.NewWriteTableQuery().WithUpdateOperation(operation))
 }
 
 func (d *YdbConnector) CreateOperation(
 	ctx context.Context, operation types.Operation,
 ) (types.ObjectID, error) {
 	operation.SetId(types.GenerateObjectID())
-	err := d.ExecuteUpsert(ctx, queries.MakeWriteTableQuery(queries.WithCreateOperation(operation)))
+	err := d.ExecuteUpsert(ctx, queries.NewWriteTableQuery().WithCreateOperation(operation))
 	if err != nil {
 		return types.ObjectID{}, err
 	}
@@ -319,7 +320,7 @@ func (d *YdbConnector) CreateBackup(
 ) (types.ObjectID, error) {
 	id := types.GenerateObjectID()
 	backup.ID = id
-	err := d.ExecuteUpsert(ctx, queries.MakeWriteTableQuery(queries.WithCreateBackup(backup)))
+	err := d.ExecuteUpsert(ctx, queries.NewWriteTableQuery().WithCreateBackup(backup))
 	if err != nil {
 		return types.ObjectID{}, err
 	}
@@ -333,5 +334,5 @@ func (d *YdbConnector) UpdateBackup(
 		ID:     id,
 		Status: backupStatus,
 	}
-	return d.ExecuteUpsert(context, queries.MakeWriteTableQuery(queries.WithCreateBackup(backup)))
+	return d.ExecuteUpsert(context, queries.NewWriteTableQuery().WithUpdateBackup(backup))
 }

internal/connectors/db/mock.go

@@ -140,8 +140,26 @@ func (c *MockDBConnector) GetOperation(
 	)
 }
 
-func (d *MockDBConnector) SelectOperations(
-	ctx context.Context, queryBuilder queries.ReadTableQuery,
+func (c *MockDBConnector) GetBackup(
+	_ context.Context, backupID types.ObjectID,
+) (types.Backup, error) {
+	if backup, exist := c.backups[backupID]; exist {
+		return backup, nil
+	}
+	return types.Backup{}, fmt.Errorf(
+		"backup not found, id %s", backupID.String(),
+	)
+}
+
+func (c *MockDBConnector) SelectOperations(
+	_ context.Context, _ queries.ReadTableQuery,
 ) ([]types.Operation, error) {
 	return nil, errors.New("Do not call this method")
 }
+
+func (c *MockDBConnector) ExecuteUpsert(_ context.Context, queryBuilder queries.WriteTableQuery) error {
+	queryBuilderMock := queryBuilder.(*queries.WriteTableQueryMock)
+	c.operations[queryBuilderMock.Operation.GetId()] = queryBuilderMock.Operation
+	c.backups[queryBuilderMock.Backup.ID] = queryBuilderMock.Backup
+	return nil
+}

internal/connectors/db/process_result_set.go

@@ -5,6 +5,7 @@ import (
 	"github.com/google/uuid"
 	"github.com/ydb-platform/ydb-go-sdk/v3/table/result"
 	"github.com/ydb-platform/ydb-go-sdk/v3/table/result/named"
+	"time"
 	"ydbcp/internal/types"
 )
 
@@ -70,21 +71,23 @@ func ReadOperationFromResultSet(res result.Result) (types.Operation, error) {
 		operationId   types.ObjectID
 		containerId   string
 		operationType string
+		createdAt     time.Time
+		database      string
 
-		operationStateBuf *string
 		backupId          *types.ObjectID
 		ydbOperationId    *string
-		database          *string
+		operationStateBuf *string
 	)
 	err := res.ScanNamed(
 		named.Required("id", &operationId),
 		named.Required("container_id", &containerId),
 		named.Required("type", &operationType),
+		named.Required("created_at", &createdAt),
+		named.Required("database", &database),
 
-		named.Optional("status", &operationStateBuf),
 		named.Optional("backup_id", &backupId),
 		named.Optional("operation_id", &ydbOperationId),
-		named.Optional("database", &database),
+		named.Optional("status", &operationStateBuf),
 	)
 	if err != nil {
 		return nil, err
@@ -93,31 +96,37 @@ func ReadOperationFromResultSet(res result.Result) (types.Operation, error) {
 	if operationStateBuf != nil {
 		operationState = types.OperationState(*operationStateBuf)
 	}
+	ydbOpId := ""
+	if ydbOperationId != nil {
+		ydbOpId = *ydbOperationId
+	}
 	if operationType == string(types.OperationTypeTB) {
-		if backupId == nil || database == nil || ydbOperationId == nil {
-			return nil, fmt.Errorf("failed to read required fields of operation %s", operationId.String())
+		if backupId == nil {
+			return nil, fmt.Errorf("failed to read backup_id for TB operation: %s", operationId.String())
 		}
 		return &types.TakeBackupOperation{
 			Id:                  operationId,
-			BackupId:            types.ObjectID(*backupId),
+			BackupId:            *backupId,
 			ContainerID:         containerId,
 			State:               operationState,
 			Message:             "",
-			YdbConnectionParams: types.GetYdbConnectionParams(*database),
-			YdbOperationId:      *ydbOperationId,
+			YdbConnectionParams: types.GetYdbConnectionParams(database),
+			YdbOperationId:      ydbOpId,
+			CreatedAt:           createdAt,
 		}, nil
 	} else if operationType == string(types.OperationTypeRB) {
-		if backupId == nil || database == nil || ydbOperationId == nil {
-			return nil, fmt.Errorf("failed to read required fields of operation %s", operationId.String())
+		if backupId == nil {
+			return nil, fmt.Errorf("failed to read backup_id for TB operation: %s", operationId.String())
 		}
 		return &types.RestoreBackupOperation{
 			Id:                  operationId,
-			BackupId:            types.ObjectID(*backupId),
+			BackupId:            *backupId,
 			ContainerID:         containerId,
 			State:               operationState,
 			Message:             "",
-			YdbConnectionParams: types.GetYdbConnectionParams(*database),
-			YdbOperationId:      *ydbOperationId,
+			YdbConnectionParams: types.GetYdbConnectionParams(database),
+			YdbOperationId:      ydbOpId,
+			CreatedAt:           createdAt,
 		}, nil
 	}
 

internal/connectors/db/yql/queries/read.go

@@ -52,7 +52,7 @@ type ReadTableQueryImpl struct {
 
 type ReadTableQueryOption func(*ReadTableQueryImpl)
 
-func MakeReadTableQuery(options ...ReadTableQueryOption) *ReadTableQueryImpl {
+func NewReadTableQuery(options ...ReadTableQueryOption) *ReadTableQueryImpl {
 	d := &ReadTableQueryImpl{}
 	d.filters = make([][]table_types.Value, 0)
 	d.filterFields = make([]string, 0)

internal/connectors/db/yql/queries/read_test.go

@@ -24,7 +24,7 @@ SELECT column1, column2, column3 FROM table1 WHERE (column1 = $param0 OR column1
 			table.ValueParam("$param3", table_types.StringValueFromString("yyy")),
 		)
 	)
-	builder := MakeReadTableQuery(
+	builder := NewReadTableQuery(
 		WithTableName("table1"),
 		WithSelectFields("column1", "column2", "column3"),
 		WithQueryFilters(

internal/connectors/db/yql/queries/write.go

@@ -14,6 +14,10 @@ import (
 
 type WriteTableQuery interface {
 	FormatQuery(ctx context.Context) (*FormatQueryResult, error)
+	WithCreateBackup(backup types.Backup) WriteTableQuery
+	WithCreateOperation(operation types.Operation) WriteTableQuery
+	WithUpdateBackup(backup types.Backup) WriteTableQuery
+	WithUpdateOperation(operation types.Operation) WriteTableQuery
 }
 
 type WriteTableQueryImpl struct {
@@ -123,42 +127,36 @@ func BuildCreateBackupQuery(b types.Backup, index int) WriteSingleTableQueryImpl
 	return d
 }
 
-type WriteTableQueryOption func(*WriteTableQueryImpl)
+type WriteTableQueryImplOption func(*WriteTableQueryImpl)
 
-func MakeWriteTableQuery(options ...WriteTableQueryOption) *WriteTableQueryImpl {
-	d := &WriteTableQueryImpl{}
-	for _, opt := range options {
-		opt(d)
-	}
-	return d
+type WriteTableQueryMockOption func(*WriteTableQueryMock)
+
+func NewWriteTableQuery() WriteTableQuery {
+	return &WriteTableQueryImpl{}
 }
 
-func WithCreateBackup(backup types.Backup) WriteTableQueryOption {
-	return func(d *WriteTableQueryImpl) {
-		index := len(d.tableQueries)
-		d.tableQueries = append(d.tableQueries, BuildCreateBackupQuery(backup, index))
-	}
+func (d *WriteTableQueryImpl) WithCreateBackup(backup types.Backup) WriteTableQuery {
+	index := len(d.tableQueries)
+	d.tableQueries = append(d.tableQueries, BuildCreateBackupQuery(backup, index))
+	return d
 }
 
-func WithUpdateBackup(backup types.Backup) WriteTableQueryOption {
-	return func(d *WriteTableQueryImpl) {
-		index := len(d.tableQueries)
-		d.tableQueries = append(d.tableQueries, BuildUpdateBackupQuery(backup, index))
-	}
+func (d *WriteTableQueryImpl) WithUpdateBackup(backup types.Backup) WriteTableQuery {
+	index := len(d.tableQueries)
+	d.tableQueries = append(d.tableQueries, BuildUpdateBackupQuery(backup, index))
+	return d
 }
 
-func WithUpdateOperation(operation types.Operation) WriteTableQueryOption {
-	return func(d *WriteTableQueryImpl) {
-		index := len(d.tableQueries)
-		d.tableQueries = append(d.tableQueries, BuildUpdateOperationQuery(operation, index))
-	}
+func (d *WriteTableQueryImpl) WithUpdateOperation(operation types.Operation) WriteTableQuery {
+	index := len(d.tableQueries)
+	d.tableQueries = append(d.tableQueries, BuildUpdateOperationQuery(operation, index))
+	return d
 }
 
-func WithCreateOperation(operation types.Operation) WriteTableQueryOption {
-	return func(d *WriteTableQueryImpl) {
-		index := len(d.tableQueries)
-		d.tableQueries = append(d.tableQueries, BuildCreateOperationQuery(operation, index))
-	}
+func (d *WriteTableQueryImpl) WithCreateOperation(operation types.Operation) WriteTableQuery {
+	index := len(d.tableQueries)
+	d.tableQueries = append(d.tableQueries, BuildCreateOperationQuery(operation, index))
+	return d
 }
 
 func (d *WriteSingleTableQueryImpl) DeclareParameters() string {
@@ -180,9 +178,14 @@ func (d *WriteTableQueryImpl) FormatQuery(ctx context.Context) (*FormatQueryResu
 			return nil, errors.New("No table")
 		}
 		declares := t.DeclareParameters()
+		paramNames := t.GetParamNames()
+		keyParam := fmt.Sprintf("%s = %s", t.upsertFields[0], paramNames[0])
+		updates := make([]string, 0)
+		for j := 1; j < len(t.upsertFields); j++ {
+			updates = append(updates, fmt.Sprintf("%s = %s", t.upsertFields[j], paramNames[j]))
+		}
 		queryStrings[i] = fmt.Sprintf(
-			"%s;\nUPSERT INTO %s (%s) VALUES (%s)", declares, t.tableName, strings.Join(t.upsertFields, ", "),
-			strings.Join(t.GetParamNames(), ", "),
+			"%s;\nUPDATE %s SET %s WHERE %s", declares, t.tableName, strings.Join(updates, ", "), keyParam,
 		)
 		for _, p := range t.tableQueryParams {
 			allParams = append(allParams, p)