kpt-tmr-group/model/system_role.go

package model

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

type SystemRole struct {
	Id         int64                   `json:"id,omitempty"`
	Name       string                  `json:"name,omitempty"`
	Remarks    string                  `json:"remarks,omitempty"`
	IsShow     operationPb.IsShow_Kind `json:"is_show,omitempty"`
	CreateUser string                  `json:"create_user,omitempty"`
	CreatedAt  int64                   `json:"created_at,omitempty"`
	UpdatedAt  int64                   `json:"updated_at,omitempty"`
}

func (s *SystemRole) TableName() string {
	return "system_role"
}

const LayoutTime = "2006-01-02 15:04:05"

func NewSystemRole(req *operationPb.AddRoleRequest) *SystemRole {
	systemRole := &SystemRole{
		Name:       req.Name,
		Remarks:    req.Remarks,
		IsShow:     operationPb.IsShow_OK,
		CreateUser: req.CreateUser,
	}
	return systemRole
}

type SystemRoleSlice []*SystemRole

func (s SystemRoleSlice) ToPB() []*operationPb.AddRoleRequest {
	res := make([]*operationPb.AddRoleRequest, len(s))
	for i, v := range s {
		res[i] = &operationPb.AddRoleRequest{
			Id:              v.Id,
			Name:            v.Name,
			Remarks:         v.Remarks,
			CreateUser:      v.CreateUser,
			IsShow:          v.IsShow,
			CreatedAt:       v.CreatedAt,
			CreatedAtFormat: time.Unix(v.CreatedAt, 0).Format(LayoutTime),
		}
	}
	return res
}

func (s *SystemRole) ToPb() *operationPb.AddRoleRequest {
	return &operationPb.AddRoleRequest{
		Id:         s.Id,
		Name:       s.Name,
		CreateUser: s.CreateUser,
		IsShow:     s.IsShow,
		CreatedAt:  s.CreatedAt,
	}
}

type SystemRoleResponse struct {
	Page  int32                         `json:"page"`
	Total int32                         `json:"total"`
	List  []*operationPb.AddRoleRequest `json:"list"`
}

func (s *SystemRoleResponse) ToPB() *operationPb.SearchRoleResponse {
	return &operationPb.SearchRoleResponse{
		Page:  s.Page,
		Total: s.Total,
		List:  s.List,
	}
}