53 lines
1.5 KiB
Go
53 lines
1.5 KiB
Go
package dto
|
|
|
|
import (
|
|
"google.golang.org/protobuf/proto"
|
|
"joylink.club/bj-rtsts-server/dto/data_proto"
|
|
)
|
|
|
|
type PageTrainInfoReqDto struct {
|
|
PageQueryDto
|
|
Name string `json:"name" form:"name"`
|
|
}
|
|
|
|
type TrainInfoReqDto struct {
|
|
Name string `json:"name" form:"name"`
|
|
Pid int32 `json:"pid" form:"pid"`
|
|
}
|
|
|
|
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 ConvertToTrainDto(trailInfos []*PublishedDto) []*TrainInfoDto {
|
|
var result []*TrainInfoDto
|
|
for _, t := range trailInfos {
|
|
result = append(result, ConvertDtoFromTrain(t))
|
|
}
|
|
return result
|
|
}
|
|
|
|
func ConvertDtoFromTrain(t *PublishedDto) *TrainInfoDto {
|
|
message := &data_proto.Train{}
|
|
proto.Unmarshal(t.Proto, message)
|
|
return &TrainInfoDto{
|
|
Id: t.ID,
|
|
Name: t.Name,
|
|
Description: t.Note,
|
|
TrainModel: int32(message.TrainModel),
|
|
CarriageLength: message.CarriageLength,
|
|
TotalLength: message.TotalLength,
|
|
MinDiameter: message.MinDiameter,
|
|
MaxDiameter: message.MaxDiameter,
|
|
TrainSets: message.TrainSets,
|
|
}
|
|
}
|