create postgres extension (#18)

this extension is required for `datasets` because of UUID.
eapolinario · Nov 19, 2019 · 90ac9a8 · 90ac9a8
1 parent 63e5803
commit 90ac9a8
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/datacalog/pkg/repositories/handle.go b/datacalog/pkg/repositories/handle.go
@@ -66,6 +66,10 @@ func (h *DBHandle) CreateDB(dbName string) error {
 }
 
 func (h *DBHandle) Migrate() {
+	if h.db.Dialect().GetName() == config.Postgres {
+		logger.Infof(context.TODO(), "Creating postgres extension uuid-ossp if it does not exist")
+		h.db.Exec("CREATE EXTENSION IF NOT EXISTS \"uuid-ossp\"")
+	}
 	h.db.AutoMigrate(&models.Dataset{})
 	h.db.AutoMigrate(&models.Artifact{})
 	h.db.AutoMigrate(&models.ArtifactData{})