kpt-tmr-group/model/forage_category.go

package model

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

type ForageCategory struct {
	Id         int64                                 `json:"id"`
	ParentId   operationPb.ForageCategoryParent_Kind `json:"parent_id"`
	ParentName string                                `json:"parent_name"`
	Name       string                                `json:"name"`
	Number     string                                `json:"number"`
	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 (s *ForageCategory) TableName() string {
	return "forage_category"
}

func NewForageCategory(req *operationPb.AddForageCategoryRequest) *ForageCategory {
	return &ForageCategory{
		ParentId:   req.ParentId,
		ParentName: req.ParentName,
		Name:       req.Name,
		Number:     req.Number,
		IsShow:     operationPb.IsShow_OK,
		IsDelete:   operationPb.IsShow_OK,
	}
}

type ForageCategorySlice []*ForageCategory

func (f ForageCategorySlice) ToPB() []*operationPb.AddForageCategoryRequest {
	res := make([]*operationPb.AddForageCategoryRequest, len(f))
	for i, v := range f {
		res[i] = &operationPb.AddForageCategoryRequest{
			Id:              v.Id,
			Name:            v.Name,
			ParentId:        v.ParentId,
			ParentName:      v.ParentName,
			Number:          v.Number,
			IsShow:          v.IsShow,
			CreatedAt:       v.CreatedAt,
			CreatedAtFormat: time.Unix(v.CreatedAt, 0).Format(LayoutTime),
		}
	}
	return res
}

func (c *ForageCategory) ToPb() *operationPb.AddForageCategoryRequest {
	return &operationPb.AddForageCategoryRequest{
		Id:         c.Id,
		Name:       c.Name,
		Number:     c.Number,
		ParentId:   c.ParentId,
		ParentName: c.ParentName,
		IsShow:     c.IsShow,
		CreatedAt:  c.CreatedAt,
	}
}