Fix query for add new DC
This commit is contained in:
parent
da155464fa
commit
e33d85412c
|
@ -129,7 +129,14 @@ func (r *Repository) RemoveDatacenter(ctx context.Context, fileID int64, dc stri
|
||||||
|
|
||||||
// AddNewDC adds the dc name to the list of datacenters, if it doesn't exist already, for a given file, model and user. It also updates the size of the embedding
|
// AddNewDC adds the dc name to the list of datacenters, if it doesn't exist already, for a given file, model and user. It also updates the size of the embedding
|
||||||
func (r *Repository) AddNewDC(ctx context.Context, fileID int64, model ente.Model, userID int64, size int, dc string) error {
|
func (r *Repository) AddNewDC(ctx context.Context, fileID int64, model ente.Model, userID int64, size int, dc string) error {
|
||||||
res, err := r.DB.ExecContext(ctx, `UPDATE embeddings SET size = $1, datacenters = array_append(COALESCE(datacenters, ARRAY[]::s3region[]), $2::s3region) WHERE file_id = $3 AND model = $4 AND owner_id = $5`, size, dc, fileID, model, userID)
|
res, err := r.DB.ExecContext(ctx, `
|
||||||
|
UPDATE embeddings
|
||||||
|
SET size = $1,
|
||||||
|
datacenters = CASE
|
||||||
|
WHEN $2::s3region = ANY(datacenters) THEN datacenters
|
||||||
|
ELSE array_append(datacenters, $2::s3region)
|
||||||
|
END
|
||||||
|
WHERE file_id = $3 AND model = $4 AND owner_id = $5`, size, dc, fileID, model, userID)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return stacktrace.Propagate(err, "")
|
return stacktrace.Propagate(err, "")
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue