wip

kubeshop · Sep 13, 2023 · 8ef6441 · 8ef6441
1 parent 80164f7
commit 8ef6441
Show file tree

Hide file tree

Showing 4 changed files with 20 additions and 9 deletions.
diff --git a/server/datastore/datastore_repository.go b/server/datastore/datastore_repository.go
@@ -91,7 +91,7 @@ func (r *Repository) Update(ctx context.Context, dataStore DataStore) (DataStore
 	}
 	defer tx.Rollback()
 
-	query, params := sqlutil.Tenant(ctx, deleteQuery, DataStoreSingleID)
+	query, params := sqlutil.TenantWithReplacedID(ctx, deleteQuery, DataStoreSingleID)
 	_, err = tx.ExecContext(ctx, query, params...)
 	if err != nil {
 		return DataStore{}, fmt.Errorf("datastore repository sql exec delete: %w", err)
@@ -103,6 +103,9 @@ func (r *Repository) Update(ctx context.Context, dataStore DataStore) (DataStore
 	}
 
 	tenantID := sqlutil.TenantID(ctx)
+	if tenantID != nil {
+		dataStore.ID = id.ID(*tenantID)
+	}
 
 	_, err = tx.ExecContext(ctx, insertQuery,
 		dataStore.ID,
@@ -132,7 +135,7 @@ func (r *Repository) Delete(ctx context.Context, id id.ID) error {
 	}
 	defer tx.Rollback()
 
-	query, params := sqlutil.Tenant(ctx, deleteQuery, id)
+	query, params := sqlutil.TenantWithReplacedID(ctx, deleteQuery, id)
 	_, err = tx.ExecContext(ctx, query, params...)
 	if err != nil {
 		return fmt.Errorf("datastore repository sql exec delete: %w", err)
@@ -167,7 +170,7 @@ func (r *Repository) Current(ctx context.Context) (DataStore, error) {
 }
 
 func (r *Repository) Get(ctx context.Context, id id.ID) (DataStore, error) {
-	query, params := sqlutil.Tenant(ctx, getQuery, id)
+	query, params := sqlutil.TenantWithReplacedID(ctx, getQuery, id)
 	row := r.db.QueryRowContext(ctx, query, params...)
 
 	dataStore, err := r.readRow(row)

diff --git a/server/migrations/33_add_composite_pkey.down.sql b/server/migrations/33_add_composite_pkey.down.sql
diff --git a/server/migrations/33_add_composite_pkey.up.sql b/server/migrations/33_add_composite_pkey.up.sql
diff --git a/server/pkg/sqlutil/tenant.go b/server/pkg/sqlutil/tenant.go
@@ -35,6 +35,20 @@ func TenantWithPrefix(ctx context.Context, query string, prefix string, params .
 	return query + condition, append(params, *tenantID)
 }
 
+func TenantWithReplacedID(ctx context.Context, query string, params ...any) (string, []any) {
+	tenantID := TenantID(ctx)
+	if tenantID == nil {
+		return query, params
+	}
+
+	prefix := getQueryPrefix(query)
+	paramNumber := len(params) + 1
+	condition := fmt.Sprintf(" %s tenant_id = $%d", prefix, paramNumber)
+
+	var newParams []any
+	return query + condition, append(newParams, *tenantID, *tenantID)
+}
+
 func TenantID(ctx context.Context) *string {
 	tenantID := ctx.Value(middleware.TenantIDKey)