kpt-tmr-group/model/forage.go

package model

import (
	operationPb "kpt-tmr-group/proto/go/backend/operation"
	"time"
)

type Forage struct {
	Id                 int64                           `json:"int_64"`
	Name               string                          `json:"name"`
	CategoryId         int64                           `json:"category_id"`
	MaterialType       int64                           `json:"material_type"`
	UniqueEncode       string                          `json:"unique_encode"`
	ForageSourceId     operationPb.ForageSource_Kind   `json:"forage_source_id"`
	PlanTypeId         operationPb.ForagePlanType_Kind `json:"plan_type_id"`
	SmallMaterialScale string                          `json:"small_material_scale"`
	AllowError         int64                           `json:"allow_error"`
	PackageWeight      int64                           `json:"package_weight"`
	Price              int64                           `json:"price"`
	JumpWeight         int64                           `json:"jump_weight"`
	JumpDelay          operationPb.JumpDelaType_Kind   `json:"jump_delay"`
	ConfirmStart       operationPb.IsShow_Kind         `json:"confirm_start"`
	RelayLocations     int64                           `json:"relay_locations"`
	Jmp                operationPb.IsShow_Kind         `json:"jmp"`
	Backup1            string                          `json:"backup1"`
	Backup2            string                          `json:"backup2"`
	Backup3            string                          `json:"backup3"`
	IsShow             operationPb.IsShow_Kind         `json:"is_show"`
	IsDelete           operationPb.IsShow_Kind         `json:"is_delete"`
	CreatedAt          int64                           `json:"created_at"`
	UpdatedAt          int64                           `json:"updated_at"`
}

func (c *Forage) TableName() string {
	return "forage"
}

func NewForage(req *operationPb.AddForageRequest) *Forage {
	return &Forage{
		Name:               req.Name,
		CategoryId:         req.CategoryId,
		UniqueEncode:       req.UniqueEncode,
		ForageSourceId:     req.ForageSourceId,
		PlanTypeId:         req.PlanTypeId,
		SmallMaterialScale: req.SmallMaterialScale,
		AllowError:         req.AllowError,
		PackageWeight:      req.PackageWeight,
		Price:              req.Price,
		JumpWeight:         req.JumpWeight,
		JumpDelay:          req.JumpDelay,
		ConfirmStart:       req.ConfirmStart,
		RelayLocations:     req.RelayLocations,
		IsShow:             operationPb.IsShow_OK,
		IsDelete:           operationPb.IsShow_OK,
		Jmp:                req.Jmp,
		Backup1:            req.Backup1,
		Backup2:            req.Backup2,
		Backup3:            req.Backup3,
	}
}

type ForageSlice []*Forage

func (f ForageSlice) ToPB() []*operationPb.AddForageRequest {
	res := make([]*operationPb.AddForageRequest, len(f))
	for i, v := range f {
		res[i] = &operationPb.AddForageRequest{
			Id:                 v.Id,
			Name:               v.Name,
			CategoryId:         v.CategoryId,
			MaterialType:       v.MaterialType,
			UniqueEncode:       v.UniqueEncode,
			ForageSourceId:     v.ForageSourceId,
			PlanTypeId:         v.PlanTypeId,
			SmallMaterialScale: v.SmallMaterialScale,
			AllowError:         v.AllowError,
			PackageWeight:      v.PackageWeight,
			Price:              v.Price,
			JumpWeight:         v.JumpWeight,
			JumpDelay:          v.JumpDelay,
			ConfirmStart:       v.ConfirmStart,
			RelayLocations:     v.RelayLocations,
			IsShow:             v.IsShow,
			CreatedAt:          v.CreatedAt,
			CreatedAtFormat:    time.Unix(v.CreatedAt, 0).Format(LayoutTime),
		}
	}
	return res
}