gfast/template/vm/go/tree/model.template

// ==========================================================================
// 生成日期：{{.table.CreateTime}}
// 生成人：{{.table.FunctionAuthor}}
// ==========================================================================

package {{.table.BusinessName}}

import (
	"{{.table.PackageName}}/app/service/cache_service"
    "github.com/gogf/gf/container/gset"
    "github.com/gogf/gf/errors/gerror"
    "github.com/gogf/gf/frame/g"
    "github.com/gogf/gf/util/gconv"
)

{{$pk:=""}}
{{$pkGoField:=""}}
{{range $index, $column := .table.Columns}} {{if eq $column.IsPk "1"}}
    {{$pk = $column.ColumnName}}
    {{$pkGoField = $column.GoField}}
{{end}}{{end}}

// ReqAdd 用于存储新增请求的请求参数
type ReqAdd struct {
	{{range $index, $column := .table.Columns}}
	{{if and (eq $column.IsInsert "1") (ne $column.IsPk "1")}} {{$column.GoField}} 	{{if eq $column.GoType "Time"}}	*gtime.Time	{{else}}	{{$column.GoType}} {{end}}	  `p:"{{$column.HtmlField}}"  {{if eq $column.IsRequired "1"}}v:"required#{{$column.ColumnComment}}不能为空"{{end}}` {{end}} {{end}}
}

// ReqEdit 用于存储修改请求参数
type ReqEdit struct {
	{{.table.PkColumn.GoField}}    {{.table.PkColumn.GoType}}  `p:"{{.table.PkColumn.HtmlField}}" v:"required#主键ID不能为空"` {{range $index, $column := .table.Columns}} {{if eq $column.IsEdit "1"}}
	{{$column.GoField}}  {{if eq $column.GoType "Time"}}*gtime.Time{{else}}{{$column.GoType}}{{end}} `p:"{{$column.HtmlField}}" {{if eq $column.IsRequired "1"}}v:"required#{{$column.ColumnComment}}不能为空"{{end}}` {{end}} {{end}}
}

// RemoveReq 用于存储删除请求参数
type RemoveReq struct {
	Ids []int `p:"ids"` //删除id
}

// ReqSearchList 用于存储查询的请求参数
type ReqSearchList struct { {{range $index, $column := .table.Columns}} {{if eq $column.IsQuery "1"}}
	{{$column.GoField}}  {{if eq $column.GoType "Time"}}*gtime.Time{{else}}{{$column.GoType}}{{end}} `p:"{{$column.HtmlField}}"` //{{$column.ColumnComment}} {{end}} {{end}}
}

// GetByID 根据ID查询记录
func GetByID(id int64) (*Entity, error) {
	entity, err := Model.FindOne(id)
	if err != nil {
		g.Log().Error(err)
		return nil, gerror.New("根据ID查询记录出错")
	}
	if entity == nil {
		return nil, gerror.New("根据ID未能查询到记录")
	}
	return entity, nil
}

// AddSave 添加
func AddSave(req *ReqAdd) error {
	entity:= new(Entity)
    {{range $index, $column := .table.Columns}}
	    {{if and (eq $column.IsInsert "1") (ne $column.IsPk "1")}}
        entity.{{$column.GoField}} = req.{{$column.GoField}}
        {{end}}
    {{end}}

	result, err := entity.Insert()
	if err != nil {
		return err
	}
	_, err = result.LastInsertId()
	if err != nil {
		return err
	}
	return nil
}

// DeleteByIds 删除
func DeleteByIds(Ids []int) error {
	_, err := Model.Delete("{{.table.PkColumn.ColumnName}} in(?)", Ids)
	if err != nil {
		g.Log().Error(err)
		return gerror.New("删除失败")
	}
	return nil
}

// EditSave 根据ID来修改信息
func EditSave(req *ReqEdit) error {
	// 先根据ID来查询要修改的记录
	entity, err := GetByID(req.{{$pkGoField}})
	if err != nil {
		return err
	}

	// 修改实体
	{{range $index, $column := .table.Columns}} {{if eq $column.IsEdit "1"}}
	entity.{{$column.GoField}} = req.{{$column.GoField}}{{end}} {{end}}
	_, err = entity.Update()
	if err != nil {
		g.Log().Error(err)
		return gerror.New("修改失败")
	}
	return nil
}

// GetListSearch 列表查询
func GetListSearch(req *ReqSearchList) (list []*Entity, err error) {
    list, err = GetList()
	if req !=nil{
	    filterKey := gset.New(false)
	    tagWhere := true
        for key, entity := range list {
	    {{range $index, $column := .table.Columns}}
	        {{if eq $column.IsQuery "1" }}
            if tagWhere && gconv.String(req.{{$column.GoField}}) != "" && entity.{{$column.GoField}}!= req.{{$column.GoField}} {
                tagWhere = true
            }else{
                tagWhere = false
            }
            {{end}}
        {{end}}
            if tagWhere{
                filterKey.Add(key)
            }
        }
        searchList := make([]*Entity, 0, len(list))
        for key, entity := range list {
            if !filterKey.Contains(key) {
                searchList = append(searchList, entity)
            }
        }
        list = searchList
	}
	return
}

// GetList 获取列表
func GetList() (list []*Entity, err error) {
	cache := cache_service.New()
	//从缓存获取数据
	iList := cache.Get("{{.table.ModuleName}}_{{.table.BusinessName}}_{{.table.TableId}}")
	if iList != nil {
		list = iList.([]*Entity)
		return
	}
	list, err = Model.Order(Columns.{{$pkGoField}}+" ASC").All()
	if err != nil {
		g.Log().Error()
		err = gerror.New("获取数据失败")
		return
	}
	//缓存数据
	cache.Set("{{.table.ModuleName}}_{{.table.BusinessName}}_{{.table.TableId}}", list, 0, "{{.table.ModuleName}}_{{.table.BusinessName}}_tag")
	return
}