feat: 优化web

pkg/db/db.go

@@ -0,0 +1,27 @@
+package db
+
+import (
+	"giter.top/smart/pkg/config"
+	"gorm.io/gorm"
+)
+
+func NewDB(cfg *config.Config) *gorm.DB {
+	driver := cfg.Data.Database.Driver
+	var db *gorm.DB
+	var err error
+	switch driver {
+	case "mysql":
+		// db, err = NewMySQLDB(cfg)
+	case "postgres":
+		db, err = NewPgSQLDB(cfg)
+	case "sqlite":
+		// return NewSQLiteDB(cfg)
+	default:
+		panic("unsupported driver")
+	}
+
+	if err != nil {
+		panic(err)
+	}
+	return db
+}

pkg/db/mysql_db.go

@@ -0,0 +1 @@
+package db

pkg/db/pg_db.go

@@ -0,0 +1,18 @@
+package db
+
+import (
+	"errors"
+
+	"giter.top/smart/pkg/config"
+	"gorm.io/driver/postgres"
+	"gorm.io/gorm"
+)
+
+// pg sql 数据库连接
+func NewPgSQLDB(cfg *config.Config) (*gorm.DB , error) {	
+ 	db, err := gorm.Open(postgres.Open(cfg.Data.Database.DSN), &gorm.Config{})
+	if err != nil {
+		return nil, errors.New("failed to connect to postgres database")
+	}
+	return db, nil
+}