pm/models/user.go

package models

import (
	"zhiyuan/pkg/db"
)

type User struct {
	ID                 int    `json:"id"`
	Name               string `json:"name"`
	Phone              string `json:"phone"`
	OpenId             string `json:"openid"`
	UnionId            string `json:"unionid"`
	HeadImgUrl         string `json:"headimgurl"`
	Sex                int    `json:"sex"`
	HasIssue           int    `json:"has_issue"`
	Issue              string `json:"issue"`
	SiteID             int    `json:"site_id"`
	NickName           string `json:"nickname"`
	AccessToken        string `json:"access_token"`
	AccessTokenExipred int    `json:"access_token_expired"`
	InWarranty         int    `json:"in_warranty"`
	ShowIds            string `json:"show_ids"`
	CreatedAt          int    `json:"created_at"`
	UpdatedAt          int    `json:"updated_at"`
}

func (User) TableName() string {
	return "zy_user"
}

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

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