Create & Init Project...

app/admin/live/live-admin/service/v2/userResource.go

@@ -0,0 +1,144 @@
+package v2
+
+import (
+	"context"
+	v2pb "go-common/app/admin/live/live-admin/api/http/v2"
+	"go-common/app/admin/live/live-admin/conf"
+	v2rspb "go-common/app/service/live/resource/api/grpc/v2"
+	"go-common/library/log"
+)
+
+// UserResourceService struct
+type UserResourceService struct {
+	conf *conf.Config
+	// optionally add other properties here, such as dao
+	// dao *dao.Dao
+	v2rsCli *v2rspb.Client
+}
+
+//NewUserResourceService init
+func NewUserResourceService(c *conf.Config) (s *UserResourceService) {
+	s = &UserResourceService{
+		conf: c,
+	}
+
+	var svc *v2rspb.Client
+	var err error
+
+	log.Info("ResourceServiceV2 Init: %+v", s.conf.ResourceClientV2)
+	if svc, err = v2rspb.NewClient(s.conf.ResourceClientV2); err != nil {
+		panic(err)
+	}
+	s.v2rsCli = svc
+	return s
+}
+
+// Add implementation
+// Add 添加资源接口
+// `method:"POST" internal:"true" `
+func (s *UserResourceService) Add(ctx context.Context, req *v2pb.UserResourceAddReq) (resp *v2pb.UserResourceAddResp, err error) {
+	respRPC, err := s.v2rsCli.Add(ctx, &v2rspb.AddReq{
+		ResType: req.ResType,
+		Title:   req.Title,
+		Url:     req.Url,
+		Weight:  req.Weight,
+		Creator: req.Creator,
+	})
+	if err == nil {
+		resp = &v2pb.UserResourceAddResp{
+			Id:       respRPC.Id,
+			CustomId: respRPC.CustomId,
+		}
+	}
+	return
+}
+
+// Edit implementation
+// Edit 编辑现有资源
+// `method:"POST" internal:"true" `
+func (s *UserResourceService) Edit(ctx context.Context, req *v2pb.UserResourceEditReq) (resp *v2pb.UserResourceEditResp, err error) {
+	resp = &v2pb.UserResourceEditResp{}
+	_, err = s.v2rsCli.Edit(ctx, &v2rspb.EditReq{
+		ResType:  req.ResType,
+		Title:    req.Title,
+		Url:      req.Url,
+		Weight:   req.Weight,
+		CustomId: req.CustomId,
+	})
+	return
+}
+
+// Get implementation
+// Get 获取资源列表
+// `method:"GET" internal:"true" `
+func (s *UserResourceService) Get(ctx context.Context, req *v2pb.UserResourceListReq) (resp *v2pb.UserResourceListResp, err error) {
+	respRPC, err := s.v2rsCli.List(ctx, &v2rspb.ListReq{
+		ResType:  req.ResType,
+		Page:     req.Page,
+		PageSize: req.PageSize,
+	})
+	if err == nil {
+		resp = &v2pb.UserResourceListResp{
+			CurrentPage: respRPC.CurrentPage,
+			TotalCount:  respRPC.TotalCount,
+			List:        convertRPCListRes(respRPC.List),
+		}
+	}
+	return
+}
+
+// SetStatus implementation
+// SetStatus 更改资源状态
+// `method:"POST" internal:"true" `
+func (s *UserResourceService) SetStatus(ctx context.Context, req *v2pb.UserResourceSetStatusReq) (resp *v2pb.UserResourceSetStatusResp, err error) {
+	resp = &v2pb.UserResourceSetStatusResp{}
+	_, err = s.v2rsCli.SetStatus(ctx, &v2rspb.SetStatusReq{
+		ResType:  req.ResType,
+		CustomId: req.CustomId,
+		Status:   req.Status,
+	})
+	return
+}
+
+// GetSingle implementation
+// Query 请求单个资源
+func (s *UserResourceService) GetSingle(ctx context.Context, req *v2pb.UserResourceGetSingleReq) (resp *v2pb.UserResourceGetSingleResp, err error) {
+	respRPC, err := s.v2rsCli.Query(ctx, &v2rspb.QueryReq{
+		CustomId: req.CustomId,
+		ResType:  req.ResType,
+	})
+	if err == nil {
+		resp = &v2pb.UserResourceGetSingleResp{
+			Id:       respRPC.Id,
+			ResType:  respRPC.ResType,
+			CustomId: respRPC.CustomId,
+			Title:    respRPC.Title,
+			Url:      respRPC.Url,
+			Weight:   respRPC.Weight,
+			Creator:  respRPC.Creator,
+			Status:   respRPC.Status,
+			Ctime:    respRPC.Ctime,
+			Mtime:    respRPC.Mtime,
+		}
+	}
+	return
+}
+
+func convertRPCListRes(RPCList []*v2rspb.ListResp_List) (HTTPList []*v2pb.UserResourceListResp_List) {
+	HTTPList = make([]*v2pb.UserResourceListResp_List, len(RPCList))
+	for index, RPCListItem := range RPCList {
+		HTTPList[index] = &v2pb.UserResourceListResp_List{
+			Id:       RPCListItem.Id,
+			ResType:  RPCListItem.ResType,
+			CustomId: RPCListItem.CustomId,
+			Title:    RPCListItem.Title,
+			Url:      RPCListItem.Url,
+			Weight:   RPCListItem.Weight,
+			Creator:  RPCListItem.Creator,
+			Status:   RPCListItem.Status,
+			Ctime:    RPCListItem.Ctime,
+			Mtime:    RPCListItem.Mtime,
+		}
+	}
+	return
+}