kpt-pasture/model/config_disease_type.go

package model

import (
	pasturePb "gitee.com/xuyiping_admin/go_proto/proto/go/backend/cow"
	"golang.org/x/text/language"
)

type ConfigDiseaseType struct {
	Id        int64                 `json:"id"`
	PastureId int64                 `json:"pastureId"`
	Kind      int32                 `json:"kind"`
	Zh        string                `json:"Zh"`
	En        string                `json:"En"`
	Remarks   string                `json:"remarks"`
	IsShow    pasturePb.IsShow_Kind `json:"is_show"`
	CreatedAt int64                 `json:"created_at"`
	UpdatedAt int64                 `json:"updated_at"`
}

func (c *ConfigDiseaseType) TableName() string {
	return "config_disease_type"
}

type ConfigDiseaseTypeSlice []*ConfigDiseaseType

func (c ConfigDiseaseTypeSlice) ToPB(lang string) []*pasturePb.SearchBaseConfigList {
	res := make([]*pasturePb.SearchBaseConfigList, len(c))
	for i, d := range c {
		name := d.Zh
		if lang == language.English.String() {
			name = d.En
		}
		res[i] = &pasturePb.SearchBaseConfigList{
			Id:        int32(d.Id),
			Name:      name,
			Remarks:   d.Remarks,
			IsShow:    d.IsShow,
			CreatedAt: int32(d.CreatedAt),
			UpdatedAt: int32(d.UpdatedAt),
		}
	}
	return res
}

func (c ConfigDiseaseTypeSlice) ToPB2(lang string, diseaseList []*Disease) []*pasturePb.ConfigOptionsList {
	res := make([]*pasturePb.ConfigOptionsList, len(c))
	for i, d := range c {
		children := make([]*pasturePb.ConfigOptionsList, 0)
		if len(diseaseList) > 0 {
			for _, v := range diseaseList {
				if int64(v.DiseaseType) != d.Id {
					continue
				}
				children = append(children, &pasturePb.ConfigOptionsList{
					Value:    int32(v.Id),
					Label:    v.Name,
					Disabled: true,
				})
			}
		}
		label := d.Zh
		if lang == language.English.String() {
			label = d.En
		}
		res[i] = &pasturePb.ConfigOptionsList{
			Value:    int32(d.Id),
			Label:    label,
			Disabled: true,
			Children: children,
		}
	}
	return res
}
func (c ConfigDiseaseTypeSlice) ToPB3(lang string) []*pasturePb.ConfigOptionsList {
	res := make([]*pasturePb.ConfigOptionsList, 0)
	for _, v := range c {
		label := v.Zh
		if lang == "en" {
			label = v.En
		}
		res = append(res, &pasturePb.ConfigOptionsList{
			Value:    int32(v.Kind),
			Label:    label,
			Disabled: true,
		})
	}
	return res
}