feat(internal/database): modify db method CreateRevision

oom-ai · Nov 8, 2021 · 833dff3 · 833dff3
1 parent cbaf4a1
commit 833dff3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/internal/database/metadata/postgres/revision.go b/internal/database/metadata/postgres/revision.go
@@ -14,7 +14,7 @@ import (
 )
 
 func (db *DB) ListRevision(ctx context.Context, opt metadata.ListRevisionOpt) ([]*types.Revision, error) {
-	revisions := []*types.Revision{}
+	var revisions []*types.Revision
 	query := "SELECT * FROM feature_group_revision"
 	cond, args, err := buildListRevisionCond(opt)
 	if err != nil {
@@ -73,8 +73,8 @@ func (db *DB) GetRevision(ctx context.Context, opt metadata.GetRevisionOpt) (*ty
 }
 
 func (db *DB) CreateRevision(ctx context.Context, opt metadata.CreateRevisionOpt) error {
-	query := "INSERT INTO feature_group_revision(group_name, revision, data_table, description) VALUES ($1, $2, $3, $4)"
-	_, err := db.ExecContext(ctx, query, opt.GroupName, opt.Revision, opt.DataTable, opt.Description)
+	query := "INSERT INTO feature_group_revision(group_name, revision, data_table, anchored, description) VALUES ($1, $2, $3, $4, $5)"
+	_, err := db.ExecContext(ctx, query, opt.GroupName, opt.Revision, opt.DataTable, opt.Anchored, opt.Description)
 	if err != nil {
 		if e2, ok := err.(*pq.Error); ok {
 			if e2.Code == pgerrcode.UniqueViolation {

diff --git a/internal/database/metadata/types.go b/internal/database/metadata/types.go
@@ -17,6 +17,7 @@ type CreateRevisionOpt struct {
 	GroupName   string
 	DataTable   string
 	Description string
+	Anchored    bool
 }
 
 type GetRevisionOpt struct {