Refine the rest structure.

2019-04-26 11:43:54 +08:00
parent c55e8699b4
commit e88930e13a
49 changed files with 201 additions and 183 deletions
--- a/code/rest/user_dao.go
+++ b/code/rest/user_dao.go
@ -0,0 +1,158 @@
+package rest
+
+import (
+	"github.com/nu7hatch/gouuid"
+	"tank/code/tool/builder"
+	"time"
+)
+
+type UserDao struct {
+	BaseDao
+}
+
+//创建用户
+func (this *UserDao) Create(user *User) *User {
+
+	if user == nil {
+		panic("参数不能为nil")
+	}
+
+	timeUUID, _ := uuid.NewV4()
+	user.Uuid = string(timeUUID.String())
+	user.CreateTime = time.Now()
+	user.UpdateTime = time.Now()
+	user.LastTime = time.Now()
+	user.Sort = time.Now().UnixNano() / 1e6
+
+	db := CONTEXT.DB.Create(user)
+	this.PanicError(db.Error)
+
+	return user
+}
+
+//按照Id查询用户，找不到返回nil
+func (this *UserDao) FindByUuid(uuid string) *User {
+
+	// Read
+	var user *User = &User{}
+	db := CONTEXT.DB.Where(&User{Base: Base{Uuid: uuid}}).First(user)
+	if db.Error != nil {
+		return nil
+	}
+	return user
+}
+
+//按照Id查询用户,找不到抛panic
+func (this *UserDao) CheckByUuid(uuid string) *User {
+
+	if uuid == "" {
+		panic("uuid必须指定")
+	}
+
+	// Read
+	var user = &User{}
+	db := CONTEXT.DB.Where(&User{Base: Base{Uuid: uuid}}).First(user)
+	this.PanicError(db.Error)
+	return user
+}
+
+//按照邮箱查询用户。
+func (this *UserDao) FindByUsername(username string) *User {
+
+	var user = &User{}
+	db := CONTEXT.DB.Where(&User{Username: username}).First(user)
+	if db.Error != nil {
+		return nil
+	}
+	return user
+}
+
+//按照邮箱查询用户。
+func (this *UserDao) FindByEmail(email string) *User {
+
+	var user *User = &User{}
+	db := CONTEXT.DB.Where(&User{Email: email}).First(user)
+	if db.Error != nil {
+		return nil
+	}
+	return user
+}
+
+//显示用户列表。
+func (this *UserDao) Page(page int, pageSize int, username string, email string, phone string, status string, sortArray []builder.OrderPair) *Pager {
+
+	var wp = &builder.WherePair{}
+
+	if username != "" {
+		wp = wp.And(&builder.WherePair{Query: "username LIKE ?", Args: []interface{}{"%" + username + "%"}})
+	}
+
+	if email != "" {
+		wp = wp.And(&builder.WherePair{Query: "email LIKE ?", Args: []interface{}{"%" + email + "%"}})
+	}
+
+	if phone != "" {
+		wp = wp.And(&builder.WherePair{Query: "phone = ?", Args: []interface{}{phone}})
+	}
+
+	if status != "" {
+		wp = wp.And(&builder.WherePair{Query: "status = ?", Args: []interface{}{status}})
+	}
+
+	count := 0
+	db := CONTEXT.DB.Model(&User{}).Where(wp.Query, wp.Args...).Count(&count)
+	this.PanicError(db.Error)
+
+	var users []*User
+	orderStr := this.GetSortString(sortArray)
+	if orderStr == "" {
+		db = CONTEXT.DB.Where(wp.Query, wp.Args...).Offset(page * pageSize).Limit(pageSize).Find(&users)
+	} else {
+		db = CONTEXT.DB.Where(wp.Query, wp.Args...).Order(orderStr).Offset(page * pageSize).Limit(pageSize).Find(&users)
+	}
+
+	this.PanicError(db.Error)
+
+	pager := NewPager(page, pageSize, count, users)
+
+	return pager
+}
+
+//查询某个用户名是否已经有用户了
+func (this *UserDao) CountByUsername(username string) int {
+	var count int
+	db := CONTEXT.DB.
+		Model(&User{}).
+		Where("username = ?", username).
+		Count(&count)
+	this.PanicError(db.Error)
+	return count
+}
+
+//查询某个邮箱是否已经有用户了
+func (this *UserDao) CountByEmail(email string) int {
+	var count int
+	db := CONTEXT.DB.
+		Model(&User{}).
+		Where("email = ?", email).
+		Count(&count)
+	this.PanicError(db.Error)
+	return count
+}
+
+//保存用户
+func (this *UserDao) Save(user *User) *User {
+
+	user.UpdateTime = time.Now()
+	db := CONTEXT.DB.
+		Save(user)
+	this.PanicError(db.Error)
+	return user
+}
+
+//执行清理操作
+func (this *UserDao) Cleanup() {
+	this.logger.Info("[UserDao]执行清理：清除数据库中所有User记录。")
+	db := CONTEXT.DB.Where("uuid is not null and role != ?", USER_ROLE_ADMINISTRATOR).Delete(User{})
+	this.PanicError(db.Error)
+}