use DAO

database/basedao.go

@@ -0,0 +1,89 @@
+package database
+
+import (
+	"errors"
+
+	"gorm.io/gorm"
+)
+
+type BaseDAO[T any] struct{}
+
+func (b BaseDAO[T]) Migrate(db *gorm.DB, e T) error {
+	return db.Set("gorm:table_options", "CHARSET=utf8mb4").AutoMigrate(e)
+}
+
+func (BaseDAO[T]) GetAll(db *gorm.DB, e T) ([]T, error) {
+	var r []T
+	if err := db.Find(&r, e).Error; err != nil {
+		return nil, err
+	}
+	return r, nil
+}
+
+func (BaseDAO[T]) GetOne(db *gorm.DB, e T) (T, error) {
+	var r T
+	if err := db.First(&r, e).Error; err != nil {
+		return r, err
+	}
+	return r, nil
+}
+
+func (BaseDAO[T]) GetSome(db *gorm.DB, e T, limit, offset int) ([]T, error) {
+	var r []T
+	if err := db.Find(&r, e).Limit(limit).Offset(offset).Error; err != nil {
+		return nil, err
+	}
+	return r, nil
+}
+
+func (BaseDAO[T]) Create(db *gorm.DB, e T) (T, error) {
+	if err := db.Create(&e).Error; err != nil {
+		return e, err
+	}
+	return e, nil
+}
+
+func (BaseDAO[T]) FirstOrCreate(db *gorm.DB, e T) (T, error) {
+	if err := db.FirstOrCreate(&e).Error; err != nil {
+		return e, err
+	}
+	return e, nil
+}
+
+func (BaseDAO[T]) Update(db *gorm.DB, e T) (T, error) {
+	if err := db.Updates(&e).Error; err != nil {
+		return e, err
+	}
+	return e, nil
+}
+
+func (b BaseDAO[T]) UpdateOrCreate(db *gorm.DB, e T) (T, error) {
+	e, err := b.Update(db, e)
+	if errors.Is(err, gorm.ErrRecordNotFound) {
+		return b.Create(db, e)
+	}
+	return e, err
+}
+
+func (BaseDAO[T]) Delete(db *gorm.DB, e T) error {
+	if err := db.Delete(e).Error; err != nil {
+		return err
+	}
+	return nil
+}
+
+type IBaseDAO interface {
+	Migrate()
+
+	GetAll()
+	GetOne()
+	GetSome()
+
+	Create()
+	FirstOrCreate()
+
+	Update()
+	UpdateOrCreate()
+
+	Delete()
+}

database/basemapper.go

@@ -1,76 +0,0 @@
-package database
-
-import (
-	"errors"
-
-	"gorm.io/gorm"
-)
-
-type BaseDAO[T any] struct {
-	db *gorm.DB
-}
-
-func NewDAO[T any](db *gorm.DB) *BaseDAO[T] {
-	return &BaseDAO[T]{
-		db: db,
-	}
-}
-
-/* Single Table Operations */
-
-func (b *BaseDAO[T]) GetAll(e T) ([]T, error) {
-	var r []T
-	if err := b.db.Find(&r, e).Error; err != nil {
-		return nil, err
-	}
-	return r, nil
-}
-
-func (b *BaseDAO[T]) GetOne(e T) (T, error) {
-	var r T
-	if err := b.db.First(&r, e).Error; err != nil {
-		return r, err
-	}
-	return r, nil
-}
-
-func (b *BaseDAO[T]) GetSome(e T, limit, offset int) ([]T, error) {
-	var r []T
-	if err := b.db.Find(&r, e).Limit(limit).Offset(offset).Error; err != nil {
-		return nil, err
-	}
-	return r, nil
-}
-
-func (b *BaseDAO[T]) Create(e T) (T, error) {
-	if err := b.db.Create(&e).Error; err != nil {
-		return e, err
-	}
-	return e, nil
-}
-
-func (b *BaseDAO[T]) Update(e T) (T, error) {
-	if err := b.db.Updates(&e).Error; err != nil {
-		return e, err
-	}
-	return e, nil
-}
-
-func (b *BaseDAO[T]) UpdateOrCreate(e T) (T, error) {
-	var r T
-	err := b.db.First(&r, e).Error
-	if err != nil {
-		if errors.Is(err, gorm.ErrRecordNotFound) {
-			return b.Create(e)
-		}
-		return e, err
-	}
-	return b.Update(e)
-}
-
-func (b *BaseDAO[T]) Delete(e T) error {
-	if err := b.db.Delete(e).Error; err != nil {
-		return err
-	}
-	return nil
-}

database/database.go

@@ -5,6 +5,13 @@ import (
 	"gorm.io/gorm"
 )
 
-func Connect(DSN string) (*gorm.DB, error) {
-	return gorm.Open(mysql.Open(DSN), &gorm.Config{})
+var Client *gorm.DB
+
+func Connect(DSN string) error {
+	var err error
+	Client, err = gorm.Open(mysql.Open(DSN), &gorm.Config{})
+	if err != nil {
+		return err
+	}
+	return nil
 }