feat(pkg/oomstore): add Revision to API ImportBatchFeatures

oom-ai · Nov 8, 2021 · 76e7e3c · 76e7e3c
1 parent 1a44029
commit 76e7e3c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/pkg/oomstore/import.go b/pkg/oomstore/import.go
@@ -79,6 +79,7 @@ func (s *OomStore) ImportBatchFeatures(ctx context.Context, opt types.ImportBatc
 		Entity:    entity,
 		Features:  features,
 		Header:    header,
+		Revision:  opt.Revision,
 		CsvReader: csvReader,
 	})
 	if err != nil {
@@ -90,5 +91,6 @@ func (s *OomStore) ImportBatchFeatures(ctx context.Context, opt types.ImportBatc
 		GroupName:   opt.GroupName,
 		DataTable:   dataTable,
 		Description: opt.Description,
+		Anchored:    opt.Revision != nil,
 	})
 }
diff --git a/pkg/oomstore/types/options.go b/pkg/oomstore/types/options.go
@@ -43,6 +43,7 @@ type ImportBatchFeaturesOpt struct {
 	GroupName   string
 	Description string
 	DataSource  CsvDataSource
+	Revision    *int64
 }
 
 type CsvDataSource struct {