feat: add NewDataSourceFromDb function

dictyBase · Sep 16, 2021 · 0499d23 · 0499d23
1 parent b6678c6
commit 0499d23
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/storage/arangodb/datasource.go b/storage/arangodb/datasource.go
@@ -50,6 +50,23 @@ func validateAll(ai ...interface{}) error {
 	return nil
 }
 
+func NewDataSourceFromDb(db *manager.Database, collP *CollectionParams) (storage.DataSource, error) {
+	if err := validateAll(collP); err != nil {
+		return &arangoSource{}, err
+	}
+	oc, err := CreateCollection(db, collP)
+	if err != nil {
+		return &arangoSource{}, err
+	}
+	return &arangoSource{
+		database: db,
+		termc:    oc.Term,
+		relc:     oc.Rel,
+		graphc:   oc.Cv,
+		obog:     oc.Obog,
+	}, nil
+}
+
 func NewDataSource(connP *ConnectParams, collP *CollectionParams) (storage.DataSource, error) {
 	var ds *arangoSource
 	if err := validateAll(connP, collP); err != nil {