46 lines
1.1 KiB
Go
46 lines
1.1 KiB
Go
// Package server generate by warden_gen
|
|
package server
|
|
|
|
import (
|
|
"context"
|
|
|
|
v1 "go-common/app/service/main/vipinfo/api"
|
|
service "go-common/app/service/main/vipinfo/service"
|
|
"go-common/library/net/rpc/warden"
|
|
)
|
|
|
|
// New VipInfo warden rpc server
|
|
func New(c *warden.ServerConfig, svr *service.Service) *warden.Server {
|
|
ws := warden.NewServer(c)
|
|
v1.RegisterVipInfoServer(ws.Server(), &server{svr})
|
|
ws, err := ws.Start()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return ws
|
|
}
|
|
|
|
type server struct {
|
|
svr *service.Service
|
|
}
|
|
|
|
var _ v1.VipInfoServer = &server{}
|
|
|
|
// Info get vipinfo by mid.
|
|
func (s *server) Info(ctx context.Context, req *v1.InfoReq) (res *v1.InfoReply, err error) {
|
|
var info *v1.ModelInfo
|
|
if info, err = s.svr.Info(ctx, req.Mid); err != nil {
|
|
return
|
|
}
|
|
return &v1.InfoReply{Res: info}, nil
|
|
}
|
|
|
|
// Infos get vipinfos by mids
|
|
func (s *server) Infos(ctx context.Context, req *v1.InfosReq) (res *v1.InfosReply, err error) {
|
|
var infos map[int64]*v1.ModelInfo
|
|
if infos, err = s.svr.Infos(ctx, req.Mids); err != nil {
|
|
return
|
|
}
|
|
return &v1.InfosReply{Res: infos}, nil
|
|
}
|