blog_backend_api/model/model.go

107 lines
3.1 KiB
Go

package model
import (
"background/db"
"background/logs"
"background/utils"
"crypto/md5"
"fmt"
"log"
"time"
"github.com/jinzhu/gorm"
)
type Plan struct {
ID uint32 `gorm:"column:id;primary_key" json:"id"`
StartTime utils.OrmTime `gorm:"column:start_time" json:"start_time"`
EndTime utils.OrmTime `gorm:"column:end_time" json:"end_time"`
Date string `gorm:"column:date" json:"date"`
Content string `gorm:"column:content" json:"content"`
Type uint32 `gorm:"column:type" json:"type"`
}
type Undo struct {
model gorm.Model
ID int32 `json:"id" gorm:"column:id" sql:"id"`
Content string `json:"content" gorm:"column:content" sql:"content"`
SpendTime int32 `json:"spend_time" gorm:"column:spend_time" sql:"spend_time"`
Parent int32 `json:"parent" gorm:"column:parent" sql:"parent"`
Level int32 `json:"level" gorm:"column:level" sql:"level"`
Type int32 `json:"type" gorm:"column:type" sql:"type"`
Done int `json:"done" gorm:"done" sql:"done"`
DeletedAt *time.Time `json:"deleted_at"`
CreateTime string `json:"create_time"`
FinishTime string `json:"finish_time"`
}
type PlanType struct {
ID uint32 `gorm:"column:id;primary_key" json:"id"`
TypeName string `gorm:"column:type_name;" json:"type_name"`
}
func (m *Undo) TableName() string {
return "undo"
}
func (m *PlanType) TableName() string {
return "plan_type"
}
func (m *Plan) TableName() string {
return "plan"
}
type Users struct {
ID int64 `sql:"id" json:"id"`
UserName string `sql:"user_name" json:"user_name"`
UserPwd string `sql:"user_pwd" json:"user_pwd"`
CreatedDate string `sql:"created_date" json:"created_date"`
UpdatedDate string `sql:"updated_date" json:"updated_date"`
DisplayName string `sql:"display_name" json:"display_name"`
EmailAddress string `sql:"email_address" json:"email_address"`
Tel string `sql:"tel" json:"tel"`
Avatar string `sql:"avatar" json:"avatar"`
}
func GetUsers(limit int32, offsetPage int32, name string) ([]Users, int32) {
users := []Users{}
var query string
if name != "" {
log.Println(name)
query = fmt.Sprintf("select * from users where user_name like '%s' limit %d offset %d",
"%%"+name+"%%", limit, offsetPage*limit)
log.Printf(query)
} else {
query = fmt.Sprintf("select * from users limit %d offset %d", limit, offsetPage*limit)
}
err := db.GetMysqlClient().Query2(query, &users)
if nil != err {
logs.Error(err.Error())
}
type Count struct {
Count int32 `sql:"count"`
}
cnts := []Count{}
query = fmt.Sprintf("select count(*) as count from users")
err = db.GetMysqlClient().Query2(query, &cnts)
if nil != err {
logs.Error(err.Error())
}
return users, cnts[0].Count
}
func ModyfyPassword(UserName string, Password string) error {
h := md5.New()
h.Write([]byte(Password))
query := fmt.Sprintf("update users set user_pwd = '%s' where user_name = '%s' ",
utils.ByteSliceToString(h.Sum(nil)), UserName)
n, err := db.GetMysqlClient().Update(query)
if nil != err {
logs.Error(err.Error())
return err
}
if n == 0 {
return nil
}
return nil
}