create postgres extension (#18)

this extension is required for `datasets` because of UUID.
eapolinario · Nov 19, 2019 · 65025b4 · 65025b4
1 parent 2a1c2d2
commit 65025b4
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pkg/repositories/handle.go b/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{})