kpt-pasture/model/prescription.go

package model

import (
	"strconv"
	"strings"

	pasturePb "gitee.com/xuyiping_admin/go_proto/proto/go/backend/cow"
)

type Prescription struct {
	Id                int32                 `json:"id"`
	Name              string                `json:"name"`
	ApplicableDisease string                `json:"applicable_disease"`
	UseDays           int32                 `json:"use_days"`
	UseCount          int32                 `json:"use_count"`
	MeatExpiredDays   int32                 `json:"meat_expired_days"`
	MilkExpiredDays   int32                 `json:"milk_expired_days"`
	IsShow            pasturePb.IsShow_Kind `json:"is_show"`
	Remarks           string                `json:"remarks"`
	OperationId       int64                 `json:"operation_id"`
	OperationName     string                `json:"operation_name"`
	CreatedAt         int64                 `json:"created_at"`
	UpdatedAt         int64                 `json:"updated_at"`
}

func (p *Prescription) TableName() string {
	return "prescription"
}

func NewPrescription(req *pasturePb.PrescriptionRequest, applicableDisease string, useDays,
	meatExpiredDays, milkExpiredDays int32, systemUser *SystemUser) *Prescription {
	return &Prescription{
		Name:              req.Name,
		ApplicableDisease: applicableDisease,
		UseDays:           useDays,
		UseCount:          0,
		MeatExpiredDays:   meatExpiredDays,
		MilkExpiredDays:   milkExpiredDays,
		IsShow:            req.IsShow,
		Remarks:           req.Remarks,
		OperationId:       systemUser.Id,
		OperationName:     systemUser.Name,
	}
}

type PrescriptionSlice []*Prescription

func (p PrescriptionSlice) ToPB(prescriptionDrugsList []*PrescriptionDrugs, diseaseMap map[int64]*Disease) []*pasturePb.SearchPrescriptionList {
	res := make([]*pasturePb.SearchPrescriptionList, len(p))
	for i, v := range p {
		drugsList := make([]*pasturePb.PrescriptionDrugsList, 0)
		for _, pl := range prescriptionDrugsList {
			if pl.PrescriptionId != v.Id {
				continue
			}
			drugsList = append(drugsList, &pasturePb.PrescriptionDrugsList{
				Id:        pl.Id,
				DrugsId:   int32(pl.DrugsId),
				DrugsName: pl.DrugsName,
				Unit:      pl.Unit,
				UnitName:  pl.UnitName,
				Dosages:   pl.Dosages,
				UseDays:   pl.UseDays,
				Specs:     pl.Specs,
			})
		}
		applicableDiseaseList := strings.Split(v.ApplicableDisease, ",")
		applicableDiseaseNames := make([]string, 0)
		applicableDiseaseIds := make([]int32, 0)
		for _, ad := range applicableDiseaseList {
			diseaseId, _ := strconv.ParseInt(ad, 10, 64)
			applicableDiseaseIds = append(applicableDiseaseIds, int32(diseaseId))
			if dis, ok := diseaseMap[diseaseId]; ok {
				applicableDiseaseNames = append(applicableDiseaseNames, dis.Name)
			}
		}

		res[i] = &pasturePb.SearchPrescriptionList{
			Id:                     int32(v.Id),
			Name:                   v.Name,
			ApplicableDisease:      v.ApplicableDisease,
			ApplicableDiseaseIds:   applicableDiseaseIds,
			ApplicableDiseaseNames: applicableDiseaseNames,
			UseDays:                v.UseDays,
			UseCount:               v.UseCount,
			MeatExpiredDays:        v.MeatExpiredDays,
			MilkExpiredDays:        v.MilkExpiredDays,
			IsShow:                 v.IsShow,
			Remarks:                v.Remarks,
			OperationId:            int32(v.OperationId),
			OperationName:          v.OperationName,
			CreatedAt:              int32(v.CreatedAt),
			UpdatedAt:              int32(v.UpdatedAt),
			DrugsList:              drugsList,
		}
	}
	return res
}

func (p PrescriptionSlice) ToPB2() []*pasturePb.ConfigOptionsList {
	res := make([]*pasturePb.ConfigOptionsList, len(p))
	for i, d := range p {
		res[i] = &pasturePb.ConfigOptionsList{
			Value:    int32(d.Id),
			Label:    d.Name,
			Disabled: true,
		}
	}
	return res
}