pm/models/work_type.go

package models

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

	"github.com/gin-gonic/gin"
)

type WorkType struct {
	ID        int64  `json:"id" prop:"add:false"`
	Name      string `json:"name" label:"工种名称" type:"string" prop:"add edit" search:"like"`
	OrderAt   int64  `json:"order_at" prop:"add:false select:false"`
	DeletedAt int64  `json:"deleted_at" prop:"add:false select:false"`
	CreatedAt int64  `json:"created_at" prop:"add:false select:false"`
	UpdatedAt int64  `json:"updated_at" prop:"add:false select:false"`
	db.BaseModel
}

func (WorkType) TableName() string {
	return "zy_work_type"
}

func (WorkType) ListPrivilege(c *gin.Context, data map[string]interface{}, s *db.Select) bool {
	return true
}

func (WorkType) OnePrivilege(c *gin.Context, id int64) bool {
	return true
}

func (WorkType) AddPrivilege(c *gin.Context, data map[string]interface{}, post map[string]interface{}) error {
	return nil
}

func (WorkType) EditPrivilege(c *gin.Context, id int64, data map[string]interface{}, post map[string]interface{}) error {
	return nil
}

func (WorkType) DelPrivilege(c *gin.Context, id int64) error {
	return nil
}

func (WorkType) OrderField() string {
	return "order_at"
}

func (WorkType) Page() bool {
	return false
}

func (WorkType) Count() bool {
	return true
}

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

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

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

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