floraachy
/
pcm-coordinator
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

启智章鱼modeldeploy相关接口 

Former-commit-id: 1ee2df3200319364caba3a5b0e2000d86142de3f
This commit is contained in:
tzwang 2023-05-27 17:13:02 +08:00
parent a833ccf404
commit d0bfea9fd4
13 changed files with 4515 additions and 1507 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
adaptor/PCM-AI/PCM-OCTOPUS/rpc/internal/config/octopusConfig.go
View File

@ -57,4 +57,11 @@ type OctopusApi struct {
StopTrainJob string
GetTrainJobEvent string
GetTrainJobMetric string
CreateModelDeploy string
GetModelDeployList string
GetModelDeploy string
GetModelDeployEvent string
StopModelDeploy string
DeleteModelDeploy string
InferModelDeploy string
}

48
adaptor/PCM-AI/PCM-OCTOPUS/rpc/internal/logic/createmodeldeploylogic.go Normal file
View File

@ -0,0 +1,48 @@
package logic
import (
"PCM/adaptor/PCM-AI/PCM-OCTOPUS/rpc/internal/common"
"PCM/common/tool"
"context"
"PCM/adaptor/PCM-AI/PCM-OCTOPUS/rpc/internal/svc"
"PCM/adaptor/PCM-AI/PCM-OCTOPUS/rpc/octopus"
"github.com/zeromicro/go-zero/core/logx"
)
type CreateModelDeployLogic struct {
ctx context.Context
svcCtx *svc.ServiceContext
logx.Logger
}
func NewCreateModelDeployLogic(ctx context.Context, svcCtx *svc.ServiceContext) *CreateModelDeployLogic {
return &CreateModelDeployLogic{
ctx: ctx,
svcCtx: svcCtx,
Logger: logx.WithContext(ctx),
}
}
// ModelDeployService
func (l *CreateModelDeployLogic) CreateModelDeploy(in *octopus.CreateModelDeployReq) (*octopus.CreateModelDeployResp, error) {
resp := &octopus.CreateModelDeployResp{}
var url_prefix = common.OctopusUrls[in.Platform]
var reqUrl = url_prefix + l.svcCtx.Config.OctopusApi.CreateModelDeploy
token := common.GetToken(in.Platform)
req := tool.GetACHttpRequest()
_, err := req.
SetHeader("Authorization", "Bearer "+token).
SetBody(in.Params).
SetResult(resp).
Post(reqUrl)
if err != nil {
return nil, err
}
return resp, nil
}

52
adaptor/PCM-AI/PCM-OCTOPUS/rpc/internal/logic/deletemodeldeploylogic.go Normal file
View File

@ -0,0 +1,52 @@
package logic
import (
"PCM/adaptor/PCM-AI/PCM-OCTOPUS/rpc/internal/common"
"PCM/common/tool"
"context"
"PCM/adaptor/PCM-AI/PCM-OCTOPUS/rpc/internal/svc"
"PCM/adaptor/PCM-AI/PCM-OCTOPUS/rpc/octopus"
"github.com/zeromicro/go-zero/core/logx"
)
type DeleteModelDeployLogic struct {
ctx context.Context
svcCtx *svc.ServiceContext
logx.Logger
}
func NewDeleteModelDeployLogic(ctx context.Context, svcCtx *svc.ServiceContext) *DeleteModelDeployLogic {
return &DeleteModelDeployLogic{
ctx: ctx,
svcCtx: svcCtx,
Logger: logx.WithContext(ctx),
}
}
func (l *DeleteModelDeployLogic) DeleteModelDeploy(in *octopus.DeleteModelDeployReq) (*octopus.DeleteModelDeployResp, error) {
resp := &octopus.DeleteModelDeployResp{}
var url_prefix = common.OctopusUrls[in.Platform]
var reqUrl = url_prefix + l.svcCtx.Config.OctopusApi.DeleteModelDeploy
token := common.GetToken(in.Platform)
var queryStr string
for _, s := range in.GetJobIds() {
queryStr += "jobIds=" + s + "&"
}
req := tool.GetACHttpRequest()
_, err := req.
SetHeader("Authorization", "Bearer "+token).
SetQueryString(queryStr).
SetResult(resp).
Delete(reqUrl)
if err != nil {
return nil, err
}
return resp, nil
}

52
adaptor/PCM-AI/PCM-OCTOPUS/rpc/internal/logic/getmodeldeployeventlogic.go Normal file
View File

@ -0,0 +1,52 @@
package logic
import (
"PCM/adaptor/PCM-AI/PCM-OCTOPUS/rpc/internal/common"
"PCM/common/tool"
"context"
"strconv"
"PCM/adaptor/PCM-AI/PCM-OCTOPUS/rpc/internal/svc"
"PCM/adaptor/PCM-AI/PCM-OCTOPUS/rpc/octopus"
"github.com/zeromicro/go-zero/core/logx"
)
type GetModelDeployEventLogic struct {
ctx context.Context
svcCtx *svc.ServiceContext
logx.Logger
}
func NewGetModelDeployEventLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetModelDeployEventLogic {
return &GetModelDeployEventLogic{
ctx: ctx,
svcCtx: svcCtx,
Logger: logx.WithContext(ctx),
}
}
func (l *GetModelDeployEventLogic) GetModelDeployEvent(in *octopus.GetModelDeployEventReq) (*octopus.GetModelDeployEventResp, error) {
resp := &octopus.GetModelDeployEventResp{}
var url_prefix = common.OctopusUrls[in.Platform]
var reqUrl = url_prefix + l.svcCtx.Config.OctopusApi.GetModelDeployEvent
token := common.GetToken(in.Platform)
req := tool.GetACHttpRequest()
_, err := req.
SetHeader("Authorization", "Bearer "+token).
SetQueryString("id=" + in.Id).
SetQueryString("pageIndex=" + strconv.Itoa(int(in.PageIndex))).
SetQueryString("pageSize=" + strconv.Itoa(int(in.PageSize))).
SetQueryString("isMain=" + strconv.FormatBool(in.IsMain)).
SetResult(resp).
Get(reqUrl)
if err != nil {
return nil, err
}
return resp, nil
}

50
adaptor/PCM-AI/PCM-OCTOPUS/rpc/internal/logic/getmodeldeploylistlogic.go Normal file
View File

@ -0,0 +1,50 @@
package logic
import (
"PCM/adaptor/PCM-AI/PCM-OCTOPUS/rpc/internal/common"
"PCM/common/tool"
"context"
"strconv"
"PCM/adaptor/PCM-AI/PCM-OCTOPUS/rpc/internal/svc"
"PCM/adaptor/PCM-AI/PCM-OCTOPUS/rpc/octopus"
"github.com/zeromicro/go-zero/core/logx"
)
type GetModelDeployListLogic struct {
ctx context.Context
svcCtx *svc.ServiceContext
logx.Logger
}
func NewGetModelDeployListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetModelDeployListLogic {
return &GetModelDeployListLogic{
ctx: ctx,
svcCtx: svcCtx,
Logger: logx.WithContext(ctx),
}
}
func (l *GetModelDeployListLogic) GetModelDeployList(in *octopus.GetModelDeployListReq) (*octopus.GetModelDeployListResp, error) {
resp := &octopus.GetModelDeployListResp{}
var url_prefix = common.OctopusUrls[in.Platform]
var reqUrl = url_prefix + l.svcCtx.Config.OctopusApi.GetModelDeployList
token := common.GetToken(in.Platform)
req := tool.GetACHttpRequest()
_, err := req.
SetHeader("Authorization", "Bearer "+token).
SetQueryString("pageIndex=" + strconv.Itoa(int(in.PageIndex))).
SetQueryString("pageSize=" + strconv.Itoa(int(in.PageSize))).
SetResult(resp).
Get(reqUrl)
if err != nil {
return nil, err
}
return resp, nil
}

47
adaptor/PCM-AI/PCM-OCTOPUS/rpc/internal/logic/getmodeldeploylogic.go Normal file
View File

@ -0,0 +1,47 @@
package logic
import (
"PCM/adaptor/PCM-AI/PCM-OCTOPUS/rpc/internal/common"
"PCM/adaptor/PCM-AI/PCM-OCTOPUS/rpc/internal/svc"
"PCM/adaptor/PCM-AI/PCM-OCTOPUS/rpc/octopus"
"PCM/common/tool"
"context"
"github.com/zeromicro/go-zero/core/logx"
)
type GetModelDeployLogic struct {
ctx context.Context
svcCtx *svc.ServiceContext
logx.Logger
}
func NewGetModelDeployLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetModelDeployLogic {
return &GetModelDeployLogic{
ctx: ctx,
svcCtx: svcCtx,
Logger: logx.WithContext(ctx),
}
}
func (l *GetModelDeployLogic) GetModelDeploy(in *octopus.GetModelDeployReq) (*octopus.GetModelDeployResp, error) {
resp := &octopus.GetModelDeployResp{}
var url_prefix = common.OctopusUrls[in.Platform]
var reqUrl = url_prefix + l.svcCtx.Config.OctopusApi.GetModelDeploy
token := common.GetToken(in.Platform)
req := tool.GetACHttpRequest()
_, err := req.
SetHeader("Authorization", "Bearer "+token).
SetPathParam("id", in.Id).
SetResult(resp).
Get(reqUrl)
if err != nil {
return nil, err
}
return resp, nil
}

47
adaptor/PCM-AI/PCM-OCTOPUS/rpc/internal/logic/infermodeldeploylogic.go Normal file
View File

@ -0,0 +1,47 @@
package logic
import (
"PCM/adaptor/PCM-AI/PCM-OCTOPUS/rpc/internal/common"
"PCM/common/tool"
"context"
"PCM/adaptor/PCM-AI/PCM-OCTOPUS/rpc/internal/svc"
"PCM/adaptor/PCM-AI/PCM-OCTOPUS/rpc/octopus"
"github.com/zeromicro/go-zero/core/logx"
)
type InferModelDeployLogic struct {
ctx context.Context
svcCtx *svc.ServiceContext
logx.Logger
}
func NewInferModelDeployLogic(ctx context.Context, svcCtx *svc.ServiceContext) *InferModelDeployLogic {
return &InferModelDeployLogic{
ctx: ctx,
svcCtx: svcCtx,
Logger: logx.WithContext(ctx),
}
}
func (l *InferModelDeployLogic) InferModelDeploy(in *octopus.InferModelDeployReq) (*octopus.InferModelDeployResp, error) {
resp := &octopus.InferModelDeployResp{}
var url_prefix = common.OctopusUrls[in.Platform]
var reqUrl = url_prefix + l.svcCtx.Config.OctopusApi.InferModelDeploy
token := common.GetToken(in.Platform)
req := tool.GetACHttpRequest()
_, err := req.
SetHeader("Authorization", "Bearer "+token).
SetResult(resp).
Post(reqUrl)
if err != nil {
return nil, err
}
return resp, nil
}

49
adaptor/PCM-AI/PCM-OCTOPUS/rpc/internal/logic/stopmodeldeploylogic.go Normal file
View File

@ -0,0 +1,49 @@
package logic
import (
"PCM/adaptor/PCM-AI/PCM-OCTOPUS/rpc/internal/common"
"PCM/common/tool"
"context"
"PCM/adaptor/PCM-AI/PCM-OCTOPUS/rpc/internal/svc"
"PCM/adaptor/PCM-AI/PCM-OCTOPUS/rpc/octopus"
"github.com/zeromicro/go-zero/core/logx"
)
type StopModelDeployLogic struct {
ctx context.Context
svcCtx *svc.ServiceContext
logx.Logger
}
func NewStopModelDeployLogic(ctx context.Context, svcCtx *svc.ServiceContext) *StopModelDeployLogic {
return &StopModelDeployLogic{
ctx: ctx,
svcCtx: svcCtx,
Logger: logx.WithContext(ctx),
}
}
func (l *StopModelDeployLogic) StopModelDeploy(in *octopus.StopModelDeployReq) (*octopus.StopModelDeployResp, error) {
resp := &octopus.StopModelDeployResp{}
var url_prefix = common.OctopusUrls[in.Platform]
var reqUrl = url_prefix + l.svcCtx.Config.OctopusApi.StopModelDeploy
token := common.GetToken(in.Platform)
req := tool.GetACHttpRequest()
_, err := req.
SetHeader("Authorization", "Bearer "+token).
SetPathParam("id", in.Id).
SetBody("{}").
SetResult(resp).
Post(reqUrl)
if err != nil {
return nil, err
}
return resp, nil
}

36
adaptor/PCM-AI/PCM-OCTOPUS/rpc/internal/server/octopusserver.go
View File

@ -134,6 +134,42 @@ func (s *OctopusServer) GetDatasetTypeList(ctx context.Context, in *octopus.GetD
return l.GetDatasetTypeList(in)
}
// ModelDeployService
func (s *OctopusServer) CreateModelDeploy(ctx context.Context, in *octopus.CreateModelDeployReq) (*octopus.CreateModelDeployResp, error) {
l := logic.NewCreateModelDeployLogic(ctx, s.svcCtx)
return l.CreateModelDeploy(in)
}
func (s *OctopusServer) GetModelDeployList(ctx context.Context, in *octopus.GetModelDeployListReq) (*octopus.GetModelDeployListResp, error) {
l := logic.NewGetModelDeployListLogic(ctx, s.svcCtx)
return l.GetModelDeployList(in)
}
func (s *OctopusServer) GetModelDeploy(ctx context.Context, in *octopus.GetModelDeployReq) (*octopus.GetModelDeployResp, error) {
l := logic.NewGetModelDeployLogic(ctx, s.svcCtx)
return l.GetModelDeploy(in)
}
func (s *OctopusServer) GetModelDeployEvent(ctx context.Context, in *octopus.GetModelDeployEventReq) (*octopus.GetModelDeployEventResp, error) {
l := logic.NewGetModelDeployEventLogic(ctx, s.svcCtx)
return l.GetModelDeployEvent(in)
}
func (s *OctopusServer) StopModelDeploy(ctx context.Context, in *octopus.StopModelDeployReq) (*octopus.StopModelDeployResp, error) {
l := logic.NewStopModelDeployLogic(ctx, s.svcCtx)
return l.StopModelDeploy(in)
}
func (s *OctopusServer) DeleteModelDeploy(ctx context.Context, in *octopus.DeleteModelDeployReq) (*octopus.DeleteModelDeployResp, error) {
l := logic.NewDeleteModelDeployLogic(ctx, s.svcCtx)
return l.DeleteModelDeploy(in)
}
func (s *OctopusServer) InferModelDeploy(ctx context.Context, in *octopus.InferModelDeployReq) (*octopus.InferModelDeployResp, error) {
l := logic.NewInferModelDeployLogic(ctx, s.svcCtx)
return l.InferModelDeploy(in)
}
// Develop
func (s *OctopusServer) GetNotebookList(ctx context.Context, in *octopus.GetNotebookListReq) (*octopus.GetNotebookListResp, error) {
l := logic.NewGetNotebookListLogic(ctx, s.svcCtx)

5150
adaptor/PCM-AI/PCM-OCTOPUS/rpc/octopus/octopus.pb.go
View File

File diff suppressed because it is too large Load Diff

254
adaptor/PCM-AI/PCM-OCTOPUS/rpc/octopus/octopus_grpc.pb.go
View File

@ -46,6 +46,14 @@ type OctopusClient interface {
DeleteDataSetVersion(ctx context.Context, in *DeleteDataSetVersionReq, opts ...grpc.CallOption) (*DeleteDataSetVersionResp, error)
GetDatasetApplyList(ctx context.Context, in *GetDatasetApplyListReq, opts ...grpc.CallOption) (*GetDatasetApplyListResp, error)
GetDatasetTypeList(ctx context.Context, in *GetDatasetTypeListRep, opts ...grpc.CallOption) (*GetDatasetTypeListResp, error)
// ModelDeployService
CreateModelDeploy(ctx context.Context, in *CreateModelDeployReq, opts ...grpc.CallOption) (*CreateModelDeployResp, error)
GetModelDeployList(ctx context.Context, in *GetModelDeployListReq, opts ...grpc.CallOption) (*GetModelDeployListResp, error)
GetModelDeploy(ctx context.Context, in *GetModelDeployReq, opts ...grpc.CallOption) (*GetModelDeployResp, error)
GetModelDeployEvent(ctx context.Context, in *GetModelDeployEventReq, opts ...grpc.CallOption) (*GetModelDeployEventResp, error)
StopModelDeploy(ctx context.Context, in *StopModelDeployReq, opts ...grpc.CallOption) (*StopModelDeployResp, error)
DeleteModelDeploy(ctx context.Context, in *DeleteModelDeployReq, opts ...grpc.CallOption) (*DeleteModelDeployResp, error)
InferModelDeploy(ctx context.Context, in *InferModelDeployReq, opts ...grpc.CallOption) (*InferModelDeployResp, error)
// Develop
GetNotebookList(ctx context.Context, in *GetNotebookListReq, opts ...grpc.CallOption) (*GetNotebookListResp, error)
GetNotebook(ctx context.Context, in *GetNotebookReq, opts ...grpc.CallOption) (*GetNotebookResp, error)
@ -281,6 +289,69 @@ func (c *octopusClient) GetDatasetTypeList(ctx context.Context, in *GetDatasetTy
return out, nil
}
func (c *octopusClient) CreateModelDeploy(ctx context.Context, in *CreateModelDeployReq, opts ...grpc.CallOption) (*CreateModelDeployResp, error) {
out := new(CreateModelDeployResp)
err := c.cc.Invoke(ctx, "/octopus.Octopus/CreateModelDeploy", in, out, opts...)
if err != nil {
return nil, err
}
return out, nil
}
func (c *octopusClient) GetModelDeployList(ctx context.Context, in *GetModelDeployListReq, opts ...grpc.CallOption) (*GetModelDeployListResp, error) {
out := new(GetModelDeployListResp)
err := c.cc.Invoke(ctx, "/octopus.Octopus/GetModelDeployList", in, out, opts...)
if err != nil {
return nil, err
}
return out, nil
}
func (c *octopusClient) GetModelDeploy(ctx context.Context, in *GetModelDeployReq, opts ...grpc.CallOption) (*GetModelDeployResp, error) {
out := new(GetModelDeployResp)
err := c.cc.Invoke(ctx, "/octopus.Octopus/GetModelDeploy", in, out, opts...)
if err != nil {
return nil, err
}
return out, nil
}
func (c *octopusClient) GetModelDeployEvent(ctx context.Context, in *GetModelDeployEventReq, opts ...grpc.CallOption) (*GetModelDeployEventResp, error) {
out := new(GetModelDeployEventResp)
err := c.cc.Invoke(ctx, "/octopus.Octopus/GetModelDeployEvent", in, out, opts...)
if err != nil {
return nil, err
}
return out, nil
}
func (c *octopusClient) StopModelDeploy(ctx context.Context, in *StopModelDeployReq, opts ...grpc.CallOption) (*StopModelDeployResp, error) {
out := new(StopModelDeployResp)
err := c.cc.Invoke(ctx, "/octopus.Octopus/StopModelDeploy", in, out, opts...)
if err != nil {
return nil, err
}
return out, nil
}
func (c *octopusClient) DeleteModelDeploy(ctx context.Context, in *DeleteModelDeployReq, opts ...grpc.CallOption) (*DeleteModelDeployResp, error) {
out := new(DeleteModelDeployResp)
err := c.cc.Invoke(ctx, "/octopus.Octopus/DeleteModelDeploy", in, out, opts...)
if err != nil {
return nil, err
}
return out, nil
}
func (c *octopusClient) InferModelDeploy(ctx context.Context, in *InferModelDeployReq, opts ...grpc.CallOption) (*InferModelDeployResp, error) {
out := new(InferModelDeployResp)
err := c.cc.Invoke(ctx, "/octopus.Octopus/InferModelDeploy", in, out, opts...)
if err != nil {
return nil, err
}
return out, nil
}
func (c *octopusClient) GetNotebookList(ctx context.Context, in *GetNotebookListReq, opts ...grpc.CallOption) (*GetNotebookListResp, error) {
out := new(GetNotebookListResp)
err := c.cc.Invoke(ctx, "/octopus.Octopus/GetNotebookList", in, out, opts...)
@ -516,6 +587,14 @@ type OctopusServer interface {
DeleteDataSetVersion(context.Context, *DeleteDataSetVersionReq) (*DeleteDataSetVersionResp, error)
GetDatasetApplyList(context.Context, *GetDatasetApplyListReq) (*GetDatasetApplyListResp, error)
GetDatasetTypeList(context.Context, *GetDatasetTypeListRep) (*GetDatasetTypeListResp, error)
// ModelDeployService
CreateModelDeploy(context.Context, *CreateModelDeployReq) (*CreateModelDeployResp, error)
GetModelDeployList(context.Context, *GetModelDeployListReq) (*GetModelDeployListResp, error)
GetModelDeploy(context.Context, *GetModelDeployReq) (*GetModelDeployResp, error)
GetModelDeployEvent(context.Context, *GetModelDeployEventReq) (*GetModelDeployEventResp, error)
StopModelDeploy(context.Context, *StopModelDeployReq) (*StopModelDeployResp, error)
DeleteModelDeploy(context.Context, *DeleteModelDeployReq) (*DeleteModelDeployResp, error)
InferModelDeploy(context.Context, *InferModelDeployReq) (*InferModelDeployResp, error)
// Develop
GetNotebookList(context.Context, *GetNotebookListReq) (*GetNotebookListResp, error)
GetNotebook(context.Context, *GetNotebookReq) (*GetNotebookResp, error)
@ -616,6 +695,27 @@ func (UnimplementedOctopusServer) GetDatasetApplyList(context.Context, *GetDatas
func (UnimplementedOctopusServer) GetDatasetTypeList(context.Context, *GetDatasetTypeListRep) (*GetDatasetTypeListResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method GetDatasetTypeList not implemented")
}
func (UnimplementedOctopusServer) CreateModelDeploy(context.Context, *CreateModelDeployReq) (*CreateModelDeployResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method CreateModelDeploy not implemented")
}
func (UnimplementedOctopusServer) GetModelDeployList(context.Context, *GetModelDeployListReq) (*GetModelDeployListResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method GetModelDeployList not implemented")
}
func (UnimplementedOctopusServer) GetModelDeploy(context.Context, *GetModelDeployReq) (*GetModelDeployResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method GetModelDeploy not implemented")
}
func (UnimplementedOctopusServer) GetModelDeployEvent(context.Context, *GetModelDeployEventReq) (*GetModelDeployEventResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method GetModelDeployEvent not implemented")
}
func (UnimplementedOctopusServer) StopModelDeploy(context.Context, *StopModelDeployReq) (*StopModelDeployResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method StopModelDeploy not implemented")
}
func (UnimplementedOctopusServer) DeleteModelDeploy(context.Context, *DeleteModelDeployReq) (*DeleteModelDeployResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method DeleteModelDeploy not implemented")
}
func (UnimplementedOctopusServer) InferModelDeploy(context.Context, *InferModelDeployReq) (*InferModelDeployResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method InferModelDeploy not implemented")
}
func (UnimplementedOctopusServer) GetNotebookList(context.Context, *GetNotebookListReq) (*GetNotebookListResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method GetNotebookList not implemented")
}
@ -1094,6 +1194,132 @@ func _Octopus_GetDatasetTypeList_Handler(srv interface{}, ctx context.Context, d
return interceptor(ctx, in, info, handler)
}
func _Octopus_CreateModelDeploy_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(CreateModelDeployReq)
if err := dec(in); err != nil {
return nil, err
}
if interceptor == nil {
return srv.(OctopusServer).CreateModelDeploy(ctx, in)
}
info := &grpc.UnaryServerInfo{
Server: srv,
FullMethod: "/octopus.Octopus/CreateModelDeploy",
}
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(OctopusServer).CreateModelDeploy(ctx, req.(*CreateModelDeployReq))
}
return interceptor(ctx, in, info, handler)
}
func _Octopus_GetModelDeployList_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(GetModelDeployListReq)
if err := dec(in); err != nil {
return nil, err
}
if interceptor == nil {
return srv.(OctopusServer).GetModelDeployList(ctx, in)
}
info := &grpc.UnaryServerInfo{
Server: srv,
FullMethod: "/octopus.Octopus/GetModelDeployList",
}
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(OctopusServer).GetModelDeployList(ctx, req.(*GetModelDeployListReq))
}
return interceptor(ctx, in, info, handler)
}
func _Octopus_GetModelDeploy_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(GetModelDeployReq)
if err := dec(in); err != nil {
return nil, err
}
if interceptor == nil {
return srv.(OctopusServer).GetModelDeploy(ctx, in)
}
info := &grpc.UnaryServerInfo{
Server: srv,
FullMethod: "/octopus.Octopus/GetModelDeploy",
}
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(OctopusServer).GetModelDeploy(ctx, req.(*GetModelDeployReq))
}
return interceptor(ctx, in, info, handler)
}
func _Octopus_GetModelDeployEvent_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(GetModelDeployEventReq)
if err := dec(in); err != nil {
return nil, err
}
if interceptor == nil {
return srv.(OctopusServer).GetModelDeployEvent(ctx, in)
}
info := &grpc.UnaryServerInfo{
Server: srv,
FullMethod: "/octopus.Octopus/GetModelDeployEvent",
}
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(OctopusServer).GetModelDeployEvent(ctx, req.(*GetModelDeployEventReq))
}
return interceptor(ctx, in, info, handler)
}
func _Octopus_StopModelDeploy_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(StopModelDeployReq)
if err := dec(in); err != nil {
return nil, err
}
if interceptor == nil {
return srv.(OctopusServer).StopModelDeploy(ctx, in)
}
info := &grpc.UnaryServerInfo{
Server: srv,
FullMethod: "/octopus.Octopus/StopModelDeploy",
}
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(OctopusServer).StopModelDeploy(ctx, req.(*StopModelDeployReq))
}
return interceptor(ctx, in, info, handler)
}
func _Octopus_DeleteModelDeploy_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(DeleteModelDeployReq)
if err := dec(in); err != nil {
return nil, err
}
if interceptor == nil {
return srv.(OctopusServer).DeleteModelDeploy(ctx, in)
}
info := &grpc.UnaryServerInfo{
Server: srv,
FullMethod: "/octopus.Octopus/DeleteModelDeploy",
}
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(OctopusServer).DeleteModelDeploy(ctx, req.(*DeleteModelDeployReq))
}
return interceptor(ctx, in, info, handler)
}
func _Octopus_InferModelDeploy_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(InferModelDeployReq)
if err := dec(in); err != nil {
return nil, err
}
if interceptor == nil {
return srv.(OctopusServer).InferModelDeploy(ctx, in)
}
info := &grpc.UnaryServerInfo{
Server: srv,
FullMethod: "/octopus.Octopus/InferModelDeploy",
}
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(OctopusServer).InferModelDeploy(ctx, req.(*InferModelDeployReq))
}
return interceptor(ctx, in, info, handler)
}
func _Octopus_GetNotebookList_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(GetNotebookListReq)
if err := dec(in); err != nil {
@ -1603,6 +1829,34 @@ var Octopus_ServiceDesc = grpc.ServiceDesc{
MethodName: "GetDatasetTypeList",
Handler: _Octopus_GetDatasetTypeList_Handler,
},
{
MethodName: "CreateModelDeploy",
Handler: _Octopus_CreateModelDeploy_Handler,
},
{
MethodName: "GetModelDeployList",
Handler: _Octopus_GetModelDeployList_Handler,
},
{
MethodName: "GetModelDeploy",
Handler: _Octopus_GetModelDeploy_Handler,
},
{
MethodName: "GetModelDeployEvent",
Handler: _Octopus_GetModelDeployEvent_Handler,
},
{
MethodName: "StopModelDeploy",
Handler: _Octopus_StopModelDeploy_Handler,
},
{
MethodName: "DeleteModelDeploy",
Handler: _Octopus_DeleteModelDeploy_Handler,
},
{
MethodName: "InferModelDeploy",
Handler: _Octopus_InferModelDeploy_Handler,
},
{
MethodName: "GetNotebookList",
Handler: _Octopus_GetNotebookList_Handler,

68
adaptor/PCM-AI/PCM-OCTOPUS/rpc/octopusclient/octopus.go
View File

@ -27,6 +27,9 @@ type (
CreateImage = octopus.CreateImage
CreateImageReq = octopus.CreateImageReq
CreateImageResp = octopus.CreateImageResp
CreateModelDeployParam = octopus.CreateModelDeployParam
CreateModelDeployReq = octopus.CreateModelDeployReq
CreateModelDeployResp = octopus.CreateModelDeployResp
CreateMyAlgorithm = octopus.CreateMyAlgorithm
CreateMyAlgorithmReq = octopus.CreateMyAlgorithmReq
CreateMyAlgorithmResp = octopus.CreateMyAlgorithmResp
@ -44,6 +47,8 @@ type (
DeleteDataSetVersionResp = octopus.DeleteDataSetVersionResp
DeleteImageReq = octopus.DeleteImageReq
DeleteImageResp = octopus.DeleteImageResp
DeleteModelDeployReq = octopus.DeleteModelDeployReq
DeleteModelDeployResp = octopus.DeleteModelDeployResp
DeleteModelVersionReq = octopus.DeleteModelVersionReq
DeleteModelVersionResp = octopus.DeleteModelVersionResp
DeleteMyAlgorithmReq = octopus.DeleteMyAlgorithmReq
@ -54,6 +59,8 @@ type (
DeleteNotebookResp = octopus.DeleteNotebookResp
DeleteTrainJobReq = octopus.DeleteTrainJobReq
DeleteTrainJobResp = octopus.DeleteTrainJobResp
DepEvent = octopus.DepEvent
DepInfo = octopus.DepInfo
DownloadAlgorithmReq = octopus.DownloadAlgorithmReq
DownloadAlgorithmResp = octopus.DownloadAlgorithmResp
DownloadModelVersionReq = octopus.DownloadModelVersionReq
@ -74,6 +81,12 @@ type (
GetDatasetTypeListResp = octopus.GetDatasetTypeListResp
GetDatasetVersionListReq = octopus.GetDatasetVersionListReq
GetDatasetVersionListResp = octopus.GetDatasetVersionListResp
GetModelDeployEventReq = octopus.GetModelDeployEventReq
GetModelDeployEventResp = octopus.GetModelDeployEventResp
GetModelDeployListReq = octopus.GetModelDeployListReq
GetModelDeployListResp = octopus.GetModelDeployListResp
GetModelDeployReq = octopus.GetModelDeployReq
GetModelDeployResp = octopus.GetModelDeployResp
GetModelVersionListReq = octopus.GetModelVersionListReq
GetModelVersionListResp = octopus.GetModelVersionListResp
GetMyAlgorithmListReq = octopus.GetMyAlgorithmListReq
@ -100,6 +113,8 @@ type (
Headers = octopus.Headers
Image = octopus.Image
Images = octopus.Images
InferModelDeployReq = octopus.InferModelDeployReq
InferModelDeployResp = octopus.InferModelDeployResp
JobEvent = octopus.JobEvent
Lables = octopus.Lables
Model = octopus.Model
@ -113,12 +128,14 @@ type (
PayloadCreateDataSet = octopus.PayloadCreateDataSet
PayloadCreateDataSetVersion = octopus.PayloadCreateDataSetVersion
PayloadCreateImage = octopus.PayloadCreateImage
PayloadCreateModelDeploy = octopus.PayloadCreateModelDeploy
PayloadCreateMyAlgorithm = octopus.PayloadCreateMyAlgorithm
PayloadCreateNotebook = octopus.PayloadCreateNotebook
PayloadCreateTrainJob = octopus.PayloadCreateTrainJob
PayloadDeleteDataSet = octopus.PayloadDeleteDataSet
PayloadDeleteDataSetVersion = octopus.PayloadDeleteDataSetVersion
PayloadDeleteImage = octopus.PayloadDeleteImage
PayloadDeleteModelDeploy = octopus.PayloadDeleteModelDeploy
PayloadDeleteModelVersion = octopus.PayloadDeleteModelVersion
PayloadDeleteMyAlgorithm = octopus.PayloadDeleteMyAlgorithm
PayloadDeleteMyModel = octopus.PayloadDeleteMyModel
@ -131,6 +148,9 @@ type (
PayloadGetDatasetApplyList = octopus.PayloadGetDatasetApplyList
PayloadGetDatasetTypeList = octopus.PayloadGetDatasetTypeList
PayloadGetDatasetVersion = octopus.PayloadGetDatasetVersion
PayloadGetModelDeploy = octopus.PayloadGetModelDeploy
PayloadGetModelDeployEvent = octopus.PayloadGetModelDeployEvent
PayloadGetModelDeployList = octopus.PayloadGetModelDeployList
PayloadGetModelVersionList = octopus.PayloadGetModelVersionList
PayloadGetMyModelList = octopus.PayloadGetMyModelList
PayloadGetNotebook = octopus.PayloadGetNotebook
@ -138,10 +158,12 @@ type (
PayloadGetTrainJobEvent = octopus.PayloadGetTrainJobEvent
PayloadGetTrainJobList = octopus.PayloadGetTrainJobList
PayloadGetTrainJobMetric = octopus.PayloadGetTrainJobMetric
PayloadInferModelDeploy = octopus.PayloadInferModelDeploy
PayloadMyAlgorithmList = octopus.PayloadMyAlgorithmList
PayloadMyDatasetList = octopus.PayloadMyDatasetList
PayloadNotebookList = octopus.PayloadNotebookList
PayloadStartNotebook = octopus.PayloadStartNotebook
PayloadStopModelDeploy = octopus.PayloadStopModelDeploy
PayloadStopNotebook = octopus.PayloadStopNotebook
PayloadStopTrainJob = octopus.PayloadStopTrainJob
PayloadUploadAlgorithm = octopus.PayloadUploadAlgorithm
@ -155,6 +177,8 @@ type (
ResourceReq = octopus.ResourceReq
StartNotebookReq = octopus.StartNotebookReq
StartNotebookResp = octopus.StartNotebookResp
StopModelDeployReq = octopus.StopModelDeployReq
StopModelDeployResp = octopus.StopModelDeployResp
StopNotebookReq = octopus.StopNotebookReq
StopNotebookResp = octopus.StopNotebookResp
StopTrainJobReq = octopus.StopTrainJobReq
@ -207,6 +231,14 @@ type (
DeleteDataSetVersion(ctx context.Context, in *DeleteDataSetVersionReq, opts ...grpc.CallOption) (*DeleteDataSetVersionResp, error)
GetDatasetApplyList(ctx context.Context, in *GetDatasetApplyListReq, opts ...grpc.CallOption) (*GetDatasetApplyListResp, error)
GetDatasetTypeList(ctx context.Context, in *GetDatasetTypeListRep, opts ...grpc.CallOption) (*GetDatasetTypeListResp, error)
// ModelDeployService
CreateModelDeploy(ctx context.Context, in *CreateModelDeployReq, opts ...grpc.CallOption) (*CreateModelDeployResp, error)
GetModelDeployList(ctx context.Context, in *GetModelDeployListReq, opts ...grpc.CallOption) (*GetModelDeployListResp, error)
GetModelDeploy(ctx context.Context, in *GetModelDeployReq, opts ...grpc.CallOption) (*GetModelDeployResp, error)
GetModelDeployEvent(ctx context.Context, in *GetModelDeployEventReq, opts ...grpc.CallOption) (*GetModelDeployEventResp, error)
StopModelDeploy(ctx context.Context, in *StopModelDeployReq, opts ...grpc.CallOption) (*StopModelDeployResp, error)
DeleteModelDeploy(ctx context.Context, in *DeleteModelDeployReq, opts ...grpc.CallOption) (*DeleteModelDeployResp, error)
InferModelDeploy(ctx context.Context, in *InferModelDeployReq, opts ...grpc.CallOption) (*InferModelDeployResp, error)
// Develop
GetNotebookList(ctx context.Context, in *GetNotebookListReq, opts ...grpc.CallOption) (*GetNotebookListResp, error)
GetNotebook(ctx context.Context, in *GetNotebookReq, opts ...grpc.CallOption) (*GetNotebookResp, error)
@ -359,6 +391,42 @@ func (m *defaultOctopus) GetDatasetTypeList(ctx context.Context, in *GetDatasetT
return client.GetDatasetTypeList(ctx, in, opts...)
}
// ModelDeployService
func (m *defaultOctopus) CreateModelDeploy(ctx context.Context, in *CreateModelDeployReq, opts ...grpc.CallOption) (*CreateModelDeployResp, error) {
client := octopus.NewOctopusClient(m.cli.Conn())
return client.CreateModelDeploy(ctx, in, opts...)
}
func (m *defaultOctopus) GetModelDeployList(ctx context.Context, in *GetModelDeployListReq, opts ...grpc.CallOption) (*GetModelDeployListResp, error) {
client := octopus.NewOctopusClient(m.cli.Conn())
return client.GetModelDeployList(ctx, in, opts...)
}
func (m *defaultOctopus) GetModelDeploy(ctx context.Context, in *GetModelDeployReq, opts ...grpc.CallOption) (*GetModelDeployResp, error) {
client := octopus.NewOctopusClient(m.cli.Conn())
return client.GetModelDeploy(ctx, in, opts...)
}
func (m *defaultOctopus) GetModelDeployEvent(ctx context.Context, in *GetModelDeployEventReq, opts ...grpc.CallOption) (*GetModelDeployEventResp, error) {
client := octopus.NewOctopusClient(m.cli.Conn())
return client.GetModelDeployEvent(ctx, in, opts...)
}
func (m *defaultOctopus) StopModelDeploy(ctx context.Context, in *StopModelDeployReq, opts ...grpc.CallOption) (*StopModelDeployResp, error) {
client := octopus.NewOctopusClient(m.cli.Conn())
return client.StopModelDeploy(ctx, in, opts...)
}
func (m *defaultOctopus) DeleteModelDeploy(ctx context.Context, in *DeleteModelDeployReq, opts ...grpc.CallOption) (*DeleteModelDeployResp, error) {
client := octopus.NewOctopusClient(m.cli.Conn())
return client.DeleteModelDeploy(ctx, in, opts...)
}
func (m *defaultOctopus) InferModelDeploy(ctx context.Context, in *InferModelDeployReq, opts ...grpc.CallOption) (*InferModelDeployResp, error) {
client := octopus.NewOctopusClient(m.cli.Conn())
return client.InferModelDeploy(ctx, in, opts...)
}
// Develop
func (m *defaultOctopus) GetNotebookList(ctx context.Context, in *GetNotebookListReq, opts ...grpc.CallOption) (*GetNotebookListResp, error) {
client := octopus.NewOctopusClient(m.cli.Conn())

162
adaptor/PCM-AI/PCM-OCTOPUS/rpc/pb/octopus.proto
View File

@ -464,6 +464,156 @@ message PayloadGetDatasetTypeList{
/******************DatasetService End*************************/
/******************ModelDeployService Start*************************/
message CreateModelDeployReq{
string platform =1;
CreateModelDeployParam params = 2;
}
message CreateModelDeployParam{
string desc = 1;
string domain = 2;
string modelFrame = 3;
string modelId = 4;
string modelVersion = 5;
string name = 6;
string resourcePool = 7;
string resourceSpecId = 8;
string resourceType = 9;
string serviceType = 10;
}
message CreateModelDeployResp{
bool success =1;
PayloadCreateModelDeploy payload =2;
Error error = 3;
}
message PayloadCreateModelDeploy{
string message = 1;
string serviceId = 2;
string serviceUrl = 3;
}
message GetModelDeployListReq{
string platform =1;
int32 pageIndex =2;
int32 pageSize =3;
}
message GetModelDeployListResp{
bool success =1;
PayloadGetModelDeployList payload =2;
Error error = 3;
}
message PayloadGetModelDeployList{
int32 totalSize = 1;
repeated DepInfo depInfos = 2;
}
message DepInfo{
int64 completedAt = 1;
int64 createdAt = 2;
string desc = 3;
string id = 4;
string modelFrame = 5;
string modelId = 6;
string modelName = 7;
string modelVersion = 8;
string name = 9;
int64 runSec = 10;
string serviceUrl = 11;
int64 startedAt = 12;
string status = 13;
int64 updatedAt = 14;
string userId = 15;
string workspaceId = 16;
}
message GetModelDeployReq{
string platform =1;
string id =2;
}
message GetModelDeployResp{
bool success =1;
PayloadGetModelDeploy payload =2;
Error error = 3;
}
message PayloadGetModelDeploy{
DepInfo depInfo = 1;
}
message GetModelDeployEventReq{
string platform =1;
string id = 2;
bool isMain = 3;
int32 pageIndex =4;
int32 pageSize =5;
}
message GetModelDeployEventResp{
bool success =1;
PayloadGetModelDeployEvent payload =2;
Error error = 3;
}
message PayloadGetModelDeployEvent{
int32 totalSize =1;
repeated DepEvent depEvents = 2;
}
message DepEvent{
string message = 1;
string name = 2;
string reason = 3;
string timestamp = 4;
}
message StopModelDeployReq{
string platform =1;
string id = 2;
}
message StopModelDeployResp{
bool success =1;
PayloadStopModelDeploy payload =2;
Error error = 3;
}
message PayloadStopModelDeploy{
int64 stoppedAt = 1;
}
message DeleteModelDeployReq{
string platform =1;
repeated string jobIds = 2;
}
message DeleteModelDeployResp{
bool success =1;
PayloadDeleteModelDeploy payload =2;
Error error = 3;
}
message PayloadDeleteModelDeploy{
int64 deletedAt = 1;
}
message InferModelDeployReq{
string platform =1;
}
message InferModelDeployResp{
bool success =1;
PayloadInferModelDeploy payload =2;
Error error = 3;
}
message PayloadInferModelDeploy{
int64 stoppedAt = 1;
}
/******************ModelDeployService End*************************/
@ -1091,7 +1241,6 @@ service Octopus {
rpc UploadAlgorithmConfirm(UploadAlgorithmConfirmReq) returns (UploadAlgorithmConfirmResp);
//DatasetService
rpc GetMyDatasetList(GetMyDatasetListReq) returns (GetMyDatasetListResp);
rpc GetDatasetVersionList(GetDatasetVersionListReq) returns (GetDatasetVersionListResp);
@ -1106,6 +1255,13 @@ service Octopus {
//ModelDeployService
rpc CreateModelDeploy(CreateModelDeployReq) returns (CreateModelDeployResp);
rpc GetModelDeployList(GetModelDeployListReq) returns (GetModelDeployListResp);
rpc GetModelDeploy(GetModelDeployReq) returns (GetModelDeployResp);
rpc GetModelDeployEvent(GetModelDeployEventReq) returns (GetModelDeployEventResp);
rpc StopModelDeploy(StopModelDeployReq) returns (StopModelDeployResp);
rpc DeleteModelDeploy(DeleteModelDeployReq) returns (DeleteModelDeployResp);
rpc InferModelDeploy(InferModelDeployReq) returns (InferModelDeployResp);
//Develop
@ -1117,7 +1273,6 @@ service Octopus {
rpc StopNotebook(StopNotebookReq) returns (StopNotebookResp);
//ImageService
rpc GetUserImageList(GetUserImageListReq) returns (GetUserImageListResp);
rpc CreateImage(CreateImageReq) returns (CreateImageResp);
@ -1144,7 +1299,4 @@ service Octopus {
rpc GetTrainJobMetric(GetTrainJobMetricReq) returns (GetTrainJobMetricResp);
}