init commit

service/annie-user/dao/dao.go

@@ -0,0 +1,30 @@
+package dao
+
+import (
+	"flswld.com/common/config"
+	"flswld.com/logger"
+	"github.com/jinzhu/gorm"
+	_ "github.com/jinzhu/gorm/dialects/mysql"
+)
+
+type Dao struct {
+	db *gorm.DB
+}
+
+func NewDao() (r *Dao) {
+	r = new(Dao)
+	db, err := gorm.Open("mysql", config.CONF.Database.Url)
+	if err != nil {
+		logger.LOG.Error("db open error: %v", err)
+		panic(err)
+	}
+	if config.CONF.Logger.Level == "DEBUG" {
+		db.LogMode(true)
+	}
+	r.db = db
+	return r
+}
+
+func (d *Dao) CloseDao() {
+	_ = d.db.Close()
+}

service/annie-user/dao/user_dao.go

@@ -0,0 +1,37 @@
+package dao
+
+import (
+	dbEntity "annie-user/entity/db"
+)
+
+func (d *Dao) InsertUser(user *dbEntity.User) error {
+	db := d.db
+	err := db.Create(user).Error
+	return err
+}
+
+func (d *Dao) UpdateUser(user *dbEntity.User) error {
+	db := d.db
+	db = db.Model(&user)
+	err := db.Updates(user).Error
+	return err
+}
+
+func (d *Dao) QueryUser(user *dbEntity.User) ([]dbEntity.User, error) {
+	var userList []dbEntity.User
+	db := d.db
+	if user.Uid != 0 {
+		db = db.Where("`uid` = ?", user.Uid)
+	}
+	if user.Username != "" {
+		db = db.Where("`username` = ?", user.Username)
+	}
+	if user.Password != "" {
+		db = db.Where("`password` = ?", user.Password)
+	}
+	if user.IsAdmin != false {
+		db = db.Where("`is_admin` = ?", user.IsAdmin)
+	}
+	err := db.Find(&userList).Error
+	return userList, err
+}