pm/models/user_check.go

package models

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

type UserCheck struct {
	ID        int `json:"id"`
	UserId    int `json:"user_id"`
	Total     int `json:"total"`
	Progress  int `json:"progress"`
	Right     int `json:"right"`
	State     int `json:"state"`
	CreatedAt int `json:"created_at"`
	UpdatedAt int `json:"updated_at"`
}

func (UserCheck) TableName() string {
	return "zy_user_check"
}

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

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

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

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