duoduo/models/shanghu/merchant_user.go

package shanghu

import (
	orm "duoduo/database"
	"time"
)

// 商户账号
type MerchantUser struct {
	ID          int64     `gorm:"column:id;type:bigint(20);primary_key;AUTO_INCREMENT" json:"id"` // 主键
	Code        string    `gorm:"column:code;type:varchar(255)" json:"code"`
	OpenID      string    `gorm:"column:open_id;type:varchar(255)" json:"open_id"`                   // open_id 唯一索引
	Phone       string    `gorm:"column:phone;type:varchar(24)" json:"phone"`                        // 手机号
	NickName    string    `gorm:"column:nick_name;type:varchar(255)" json:"nick_name"`               // 微信用户名
	AvatarUrl   string    `gorm:"column:avatar_url;type:varchar(255)" json:"avatar_url"`             // 头像url
	Admin       int       `gorm:"column:admin;type:int(11)" json:"admin"`                            // 1-管理员
	ActiveAdmin int       `gorm:"column:active_admin;type:int(11)" json:"active_admin"`              // 1-管理员
	CreateBy    int64     `gorm:"column:create_by;type:bigint(20)" json:"create_by"`                 // 创建者
	UpdateBy    int64     `gorm:"column:update_by;type:bigint(20)" json:"update_by"`                 // 更新者
	CreatedAt   time.Time `gorm:"column:created_at;type:datetime(3)" json:"created_at"`              // 创建时间
	UpdatedAt   time.Time `gorm:"column:updated_at;type:datetime(3)" json:"updated_at"`              // 最后更新时间
	DeletedAt   time.Time `gorm:"column:deleted_at;type:datetime(3);default:null" json:"deleted_at"` // 删除时间
}

func (m *MerchantUser) TableName() string {
	return "merchant_user"
}

func (m *MerchantUser) GetNum() int {
	var count int

	tableCount := orm.ShMysql.Table(m.TableName()).Where("open_id = ? ", m.OpenID)
	tableCount.Count(&count)
	return count

}

func (m *MerchantUser) GetCodeNum() int {
	var count int

	tableCount := orm.ShMysql.Table(m.TableName()).Where("code = ? ", m.Code)
	tableCount.Count(&count)
	return count

}

func (m *MerchantUser) GetUserInfo() (MerchantUser, error) {
	var doc MerchantUser

	table := orm.ShMysql.Table(m.TableName())
	table = table.Where("open_id = ?  ", m.OpenID)

	if err := table.Select("*").First(&doc).Error; err != nil {
		return doc, err
	}

	return doc, nil

}

func (m *MerchantUser) GetUserInfoByCode() (MerchantUser, error) {
	var doc MerchantUser

	table := orm.ShMysql.Table(m.TableName())
	table = table.Where("code = ?  ", m.Code)

	if err := table.Select("open_id").First(&doc).Error; err != nil {
		return doc, err
	}

	return doc, nil

}

func (u *MerchantUser) Create() (MerchantUser, error) {
	var doc MerchantUser
	var err error

	doc = *u
	err = orm.ShMysql.Table(u.TableName()).Create(&doc).Error
	if err != nil {
		return doc, err
	}

	return doc, nil
}