pm/models/admin.go

package models

import (
	"errors"
	"fmt"
	"zhiyuan/pkg/db"

	"github.com/gin-gonic/gin"
)

type Admin struct {
	ID                 int    `json:"id"`
	Username           string `json:"username"`
	Phone              string `json:"phone"`
	Password           string `json:"password"`
	AccessToken        string `json:"access_token"`
	AccessTokenExpired string `json:"access_token_expired"`
	RoleIds            string `json:"role_ids" search:"find_in_set"`
	DeptID             int    `json:"dept_id"`
	SiteID             int    `json:"site_id"`
	State              int    `json:"state" search:"="`
	RefreshAuth        int    `json:"refresh_auth"`
	RefreshDataAuth    int    `json:"refresh_data_auth"`
	CreatedAt          int    `json:"created_at"`
	UpdatedAt          int    `json:"updated_at"`
	OpenId             string `json:"openid"`
	UnionId            string `json:"unionid"`
	HeadImgUrl         string `json:"headimgurl"`
	Sex                int    `json:"sex"`
	NickName           string `json:"nickname"`
	ShopIds            string `json:"shop_ids"`
	EntryDate          int64  `json:"entry_date"`
	RestDay            string `json:"rest_day"`
	LogIds             string `json:"log_ids"`
	db.BaseModel
}

func (Admin) TableName() string {
	return "zy_admin"
}

func (model Admin) ListPrivilege(c *gin.Context, data map[string]interface{}, s *db.Select) bool {
	return true
}

func (Admin) DeletedField() string {
	return "deleted_at"
}

func (a Admin) GetOne(where map[string]interface{}, fields []string, retVal interface{}) (*Admin, error) {
	if retVal == nil {
		var admin *Admin
		err := db.GetOne(a.TableName(), where, fields, &admin)
		return admin, err
	} else {
		err := db.GetOne(a.TableName(), where, fields, retVal)
		return nil, err
	}
}

func (a Admin) GetMulti(where map[string]interface{}, fields []string, retVal interface{}) ([]*Admin, error) {
	if retVal == nil {
		var admin []*Admin
		err := db.GetMulti(a.TableName(), where, fields, &admin)
		return admin, err
	} else {
		err := db.GetMulti(a.TableName(), where, fields, retVal)
		return nil, err
	}
}

func (a Admin) GetOneRaw(where map[string]string, param map[string]interface{}, retVal interface{}) (*Admin, error) {
	admin, err := a.GetMultiRaw(where, param, retVal)
	if err != nil {
		return nil, err
	}
	if len(admin) == 0 {
		return nil, errors.New("empty record2")
	}
	return admin[0], nil
}

func (a Admin) GetMultiRaw(where map[string]string, param map[string]interface{}, retVal interface{}) ([]*Admin, error) {
	field := "SELECT * FROM zy_admin "
	if retVal == nil {
		var admin []*Admin
		err := db.GetMultiRaw(field, where, param, &admin)
		return admin, err
	} else {
		err := db.GetMultiRaw(field, where, param, retVal)
		return nil, err
	}
}

type Manager struct {
	ID          int    `json:"id" prop:"add:false"`
	Username    string `json:"username" label:"姓名" type:"string" prop:"edit:false" search:"like"`
	Phone       string `json:"phone" label:"电话号码" type:"string" prop:"edit:false" search:"like"`
	AccountBank string `json:"account_bank" label:"开户行" type:"string" prop:"edit" default:""`
	AccountNo   string `json:"account_no" label:"银行卡号" type:"string" prop:"edit" default:""`
	AuditId     int    `json:"audit_id" label:"预审" type:"int" prop:"edit" default:"0"`
	Areacode    string `json:"areacode" label:"区划代码" type:"string" prop:"edit" search:"rightlike"`
	DeletedAt   int64  `json:"deleted_at" prop:"add:false select:false"`
	CreatedAt   int    `json:"created_at" prop:"add:false"`
	UpdatedAt   int    `json:"updated_at" prop:"add:false"`
	db.BaseModel
}

func (Manager) TableName() string {
	return "zy_admin"
}

func (model Manager) ListPrivilege(c *gin.Context, data map[string]interface{}, s *db.Select) bool {
	s.Where = append(s.Where, fmt.Sprintf("FIND_IN_SET(%s, `%s`.`role_ids`)", s.Param(6), model.TableName()))
	s.Where = append(s.Where, fmt.Sprintf("(`%s`.`state` = %s)", model.TableName(), s.Param(1)))
	return true
}

func (Manager) AddPrivilege(c *gin.Context, data map[string]interface{}, post map[string]interface{}) error {
	return errors.New("没有权限")
}

func (Manager) EditPrivilege(c *gin.Context, id int64, data map[string]interface{}, post map[string]interface{}) error {
	return nil
}

func (Manager) Page() bool {
	return false
}

func (Manager) Count() bool {
	return true
}