codes

codes/main.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"database/sql"
 	"time"
 
 	"github.com/micro/services/codes/handler"
@@ -9,8 +10,8 @@ import (
 	"github.com/micro/micro/v3/service"
 	"github.com/micro/micro/v3/service/config"
 	"github.com/micro/micro/v3/service/logger"
-	"gorm.io/driver/postgres"
-	"gorm.io/gorm"
+
+	_ "github.com/jackc/pgx/v4/stdlib"
 )
 
 var dbAddress = "postgresql://postgres:postgres@localhost:5432/codes?sslmode=disable"
@@ -28,16 +29,15 @@ func main() {
 		logger.Fatalf("Error loading config: %v", err)
 	}
 	addr := cfg.String(dbAddress)
-	db, err := gorm.Open(postgres.Open(addr), &gorm.Config{})
+	sqlDB, err := sql.Open("pgx", addr)
 	if err != nil {
-		logger.Fatalf("Error connecting to database: %v", err)
-	}
-	if err := db.AutoMigrate(&handler.Code{}); err != nil {
-		logger.Fatalf("Error migrating database: %v", err)
+		logger.Fatalf("Failed to open connection to DB %s", err)
 	}
 
+	h := &handler.Codes{Time: time.Now}
+	h.DBConn(sqlDB).Migrations(&handler.Code{})
 	// Register handler
-	pb.RegisterCodesHandler(srv.Server(), &handler.Codes{DB: db, Time: time.Now})
+	pb.RegisterCodesHandler(srv.Server(), h)
 
 	// Run service
 	if err := srv.Run(); err != nil {