Initial commit

server/framework/database/database.go

@@ -0,0 +1,74 @@
+/**
+ ******************************************************************************
+ * @file    database.go
+ * @author  MakerYang
+ ******************************************************************************
+ */
+
+package Database
+
+import (
+	"Game/framework/config"
+	"fmt"
+	"github.com/gookit/color"
+	"github.com/jinzhu/gorm"
+	_ "github.com/jinzhu/gorm/dialects/mysql"
+	"log"
+	"time"
+)
+
+var Get *gorm.DB
+
+type DefaultField struct {
+	CreateAt int `gorm:"Column:create_at" json:"create_at"`
+	UpdateAt int `gorm:"Column:update_at" json:"update_at"`
+	DeleteAt int `gorm:"Column:delete_at" json:"delete_at"`
+}
+
+func Init() {
+
+	var err error
+
+	Get, err = gorm.Open(Config.Get.Database.Type, fmt.Sprintf("%s:%s@tcp(%s)/%s?charset=utf8mb4&parseTime=True&loc=Local", Config.Get.Database.User, Config.Get.Database.Password, Config.Get.Database.Host, Config.Get.Database.Name))
+	if err != nil {
+		log.Println("[database]", color.Red.Text(err.Error()))
+	}
+
+	if Config.Get.Service.Mode == "release" {
+		Get.LogMode(false)
+	} else {
+		Get.LogMode(true)
+	}
+
+	gorm.DefaultTableNameHandler = func(db *gorm.DB, defaultTableName string) string {
+		return defaultTableName
+	}
+
+	Get.SingularTable(true)
+
+	Get.Callback().Create().Replace("gorm:update_time_stamp", func(scope *gorm.Scope) {
+		if !scope.HasError() {
+			nowTime := time.Now().Unix()
+			if createTimeField, ok := scope.FieldByName("CreateAt"); ok {
+				if createTimeField.IsBlank {
+					err := createTimeField.Set(nowTime)
+					if err != nil {
+					}
+				}
+			}
+
+			if modifyTimeField, ok := scope.FieldByName("UpdateAt"); ok {
+				if modifyTimeField.IsBlank {
+					err := modifyTimeField.Set(nowTime)
+					if err != nil {
+					}
+				}
+			}
+		}
+	})
+
+	Get.DB().SetMaxIdleConns(1000)
+	Get.DB().SetMaxOpenConns(10000)
+
+	Get.DB().SetConnMaxLifetime(time.Second * 45)
+}

server/framework/database/interface.go

@@ -0,0 +1,65 @@
+/**
+ ******************************************************************************
+ * @file    interface.go
+ * @author  MakerYang
+ ******************************************************************************
+ */
+
+package Database
+
+import (
+	"github.com/jinzhu/gorm"
+	"time"
+)
+
+type Base struct {
+	TableName string
+}
+
+func New(table string) *Base {
+	return &Base{
+		TableName: table,
+	}
+}
+
+func (base *Base) CreateData(data interface{}) error {
+	err := Get.Table(base.TableName).Create(data).Error
+	return err
+}
+
+func (base *Base) UpdateData(query interface{}, data map[string]interface{}) error {
+	data["update_at"] = time.Now().Unix()
+	err := Get.Table(base.TableName).Where(query).Updates(data).Error
+	return err
+}
+
+func (base *Base) ExprData(query interface{}, field string, operation string, data int) error {
+	err := Get.Table(base.TableName).Where(query).Update(field, gorm.Expr(field+" "+operation+" ?", data)).Error
+	return err
+}
+
+func (base *Base) GetData(dataStruct interface{}, query interface{}, order string) error {
+	err := Get.Table(base.TableName).Where(query).Order(order).First(dataStruct).Error
+	return err
+}
+
+func (base *Base) ListData(dataStruct interface{}, query interface{}, order string, limit int) error {
+	err := Get.Table(base.TableName).Where(query).Order(order).Limit(limit).Find(dataStruct).Error
+	return err
+}
+
+func (base *Base) PageData(dataStruct interface{}, query interface{}, order string, limit int, page int) error {
+	err := Get.Table(base.TableName).Where(query).Order(order).Limit(limit).Offset(page * limit).Find(dataStruct).Error
+	return err
+}
+
+func (base *Base) CountData(query interface{}) (int, error) {
+	count := 0
+	err := Get.Table(base.TableName).Where(query).Count(&count).Error
+	return count, err
+}
+
+func (base *Base) DeleteData(dataStruct interface{}, query interface{}) error {
+	err := Get.Table(base.TableName).Where(query).Delete(dataStruct).Error
+	return err
+}