feat: database/datasource: add DataSourceSet{}

grokify · Oct 28, 2023 · 112083a · 112083a
1 parent 944ea45
commit 112083a
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/database/datasource/datasource_set.go b/database/datasource/datasource_set.go
@@ -0,0 +1,22 @@
+package datasource
+
+import (
+	"fmt"
+
+	"github.com/grokify/mogo/type/maputil"
+)
+
+type DataSourceSet struct {
+	Data map[string]DataSource
+}
+
+func (dss DataSourceSet) GetDataSource(key string) (DataSource, error) {
+	if ds, ok := dss.Data[key]; ok {
+		return ds, nil
+	}
+	return DataSource{}, fmt.Errorf("key not found (%s)", key)
+}
+
+func (dss DataSourceSet) Keys() []string {
+	return maputil.Keys(dss.Data)
+}