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 } }