pm/models/work_site_check.go

package models

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

type WorkSiteCheck struct {
	ID         int    `json:"id"`
	SiteId     int    `json:"site_id"`
	SiteNodeId int    `json:"site_node_id"`
	Type       int    `json:"type"`
	AdminId    int    `json:"admin_id"`
	Days       int    `json:"days"`
	Content    string `json:"content"`
	Pictures   string `json:"pictures"`
	Explain    string `json:"explain"`
	State      int    `json:"state"`
	CreatedAt  int    `json:"created_at"`
	UpdatedAt  int    `json:"updated_at"`
	db.BaseModel
}

func (WorkSiteCheck) TableName() string {
	return "zy_work_site_check"
}

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

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

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

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