68 lines
2.0 KiB
Go
68 lines
2.0 KiB
Go
package dto
|
|
|
|
import (
|
|
"google.golang.org/protobuf/proto"
|
|
"joylink.club/bj-rtsts-server/db/model"
|
|
"joylink.club/bj-rtsts-server/ts/protos/graphicData"
|
|
)
|
|
|
|
type PageTrainInfoReqDto struct {
|
|
PageQueryDto
|
|
Name string `json:"name" form:"name"`
|
|
}
|
|
|
|
type TrainInfoReqDto struct {
|
|
Name string `json:"name" form:"name"`
|
|
}
|
|
|
|
type TrainInfoDto struct {
|
|
Id int32 `json:"id" form:"id"`
|
|
Name string `json:"name" form:"name"`
|
|
TrainModel int32 `json:"train_model" form:"train_model"`
|
|
CarriageLength int32 `json:"carriage_length" form:"carriage_length"`
|
|
TotalLength int32 `json:"total_length" form:"total_length"`
|
|
MinDiameter int32 `json:"min_diameter" form:"min_diameter"`
|
|
MaxDiameter int32 `json:"max_diameter" form:"max_diameter"`
|
|
TrainSets string `json:"train_sets" form:"train_sets"`
|
|
Description string `json:"description" form:"description"`
|
|
}
|
|
|
|
func ConvertToTrainInfoDto(trailInfos []*model.TrainInfo) []*TrainInfoDto {
|
|
var result []*TrainInfoDto
|
|
for _, t := range trailInfos {
|
|
result = append(result, ConvertDtoFromTrainInfo(t))
|
|
}
|
|
return result
|
|
}
|
|
|
|
func ConvertDtoFromTrainInfo(t *model.TrainInfo) *TrainInfoDto {
|
|
message := &graphicData.Train{}
|
|
proto.Unmarshal(t.Proto, message)
|
|
return &TrainInfoDto{
|
|
Id: t.ID,
|
|
Name: t.Name,
|
|
Description: t.Description,
|
|
TrainModel: int32(message.TrainModel),
|
|
CarriageLength: message.CarriageLength,
|
|
TotalLength: message.TotalLength,
|
|
MinDiameter: message.MinDiameter,
|
|
MaxDiameter: message.MaxDiameter,
|
|
TrainSets: message.TrainSets,
|
|
}
|
|
}
|
|
|
|
func ConvertTrainInfoFromDto(t *TrainInfoDto) *model.TrainInfo {
|
|
info := &model.TrainInfo{Name: t.Name, Description: t.Description}
|
|
message := &graphicData.Train{
|
|
TrainModel: graphicData.Train_TrainModel(t.TrainModel),
|
|
CarriageLength: t.CarriageLength,
|
|
TotalLength: t.TotalLength,
|
|
MinDiameter: t.MinDiameter,
|
|
MaxDiameter: t.MaxDiameter,
|
|
TrainSets: t.TrainSets,
|
|
}
|
|
b, _ := proto.Marshal(message)
|
|
info.Proto = b
|
|
return info
|
|
}
|