启智章鱼接口调整

Former-commit-id: e9129ab518f09b5a77e9820249897d6f0d26f038
This commit is contained in:
tzwang 2023-05-22 17:30:48 +08:00
parent 3ce935fc4d
commit a6ab1fd7da
11 changed files with 3530 additions and 682 deletions

View File

@ -40,4 +40,9 @@ type OctopusApi struct {
DeleteDataSetVersion string
GetDatasetVersionList string
DeleteNotebook string
StartNotebook string
StopNotebook string
CreateNotebook string
GetNotebook string
CreateTrainJob string
}

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 CreateNotebookLogic struct {
ctx context.Context
svcCtx *svc.ServiceContext
logx.Logger
}
func NewCreateNotebookLogic(ctx context.Context, svcCtx *svc.ServiceContext) *CreateNotebookLogic {
return &CreateNotebookLogic{
ctx: ctx,
svcCtx: svcCtx,
Logger: logx.WithContext(ctx),
}
}
func (l *CreateNotebookLogic) CreateNotebook(in *octopus.CreateNotebookReq) (*octopus.CreateNotebookResp, error) {
resp := &octopus.CreateNotebookResp{}
var url_prefix = common.OctopusUrls[in.Platform]
var reqUrl = url_prefix + l.svcCtx.Config.OctopusApi.CreateNotebook
token := common.GetToken(in.Platform)
req := tool.GetACHttpRequest()
_, err := req.
SetHeader("Authorization", "Bearer "+token).
SetBody(in.Param).
SetResult(resp).
Post(reqUrl)
if err != nil {
return nil, err
}
return resp, nil
}

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 CreateTrainJobLogic struct {
ctx context.Context
svcCtx *svc.ServiceContext
logx.Logger
}
func NewCreateTrainJobLogic(ctx context.Context, svcCtx *svc.ServiceContext) *CreateTrainJobLogic {
return &CreateTrainJobLogic{
ctx: ctx,
svcCtx: svcCtx,
Logger: logx.WithContext(ctx),
}
}
// TrainJobService
func (l *CreateTrainJobLogic) CreateTrainJob(in *octopus.CreateTrainJobReq) (*octopus.CreateTrainJobResp, error) {
resp := &octopus.CreateTrainJobResp{}
var url_prefix = common.OctopusUrls[in.Platform]
var reqUrl = url_prefix + l.svcCtx.Config.OctopusApi.CreateTrainJob
token := common.GetToken(in.Platform)
req := tool.GetACHttpRequest()
_, err := req.
SetHeader("Authorization", "Bearer "+token).
SetBody(in.Param).
SetResult(resp).
Post(reqUrl)
if err != nil {
return nil, err
}
return resp, nil
}

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 GetNotebookLogic struct {
ctx context.Context
svcCtx *svc.ServiceContext
logx.Logger
}
func NewGetNotebookLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetNotebookLogic {
return &GetNotebookLogic{
ctx: ctx,
svcCtx: svcCtx,
Logger: logx.WithContext(ctx),
}
}
func (l *GetNotebookLogic) GetNotebook(in *octopus.GetNotebookReq) (*octopus.GetNotebookResp, error) {
resp := &octopus.GetNotebookResp{}
var url_prefix = common.OctopusUrls[in.Platform]
var reqUrl = url_prefix + l.svcCtx.Config.OctopusApi.GetNotebook
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
}

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 StartNotebookLogic struct {
ctx context.Context
svcCtx *svc.ServiceContext
logx.Logger
}
func NewStartNotebookLogic(ctx context.Context, svcCtx *svc.ServiceContext) *StartNotebookLogic {
return &StartNotebookLogic{
ctx: ctx,
svcCtx: svcCtx,
Logger: logx.WithContext(ctx),
}
}
func (l *StartNotebookLogic) StartNotebook(in *octopus.StartNotebookReq) (*octopus.StartNotebookResp, error) {
resp := &octopus.StartNotebookResp{}
var url_prefix = common.OctopusUrls[in.Platform]
var reqUrl = url_prefix + l.svcCtx.Config.OctopusApi.StartNotebook
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
}

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 StopNotebookLogic struct {
ctx context.Context
svcCtx *svc.ServiceContext
logx.Logger
}
func NewStopNotebookLogic(ctx context.Context, svcCtx *svc.ServiceContext) *StopNotebookLogic {
return &StopNotebookLogic{
ctx: ctx,
svcCtx: svcCtx,
Logger: logx.WithContext(ctx),
}
}
func (l *StopNotebookLogic) StopNotebook(in *octopus.StopNotebookReq) (*octopus.StopNotebookResp, error) {
resp := &octopus.StopNotebookResp{}
var url_prefix = common.OctopusUrls[in.Platform]
var reqUrl = url_prefix + l.svcCtx.Config.OctopusApi.StopNotebook
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
}

View File

@ -124,17 +124,37 @@ func (s *OctopusServer) DeleteDataSetVersion(ctx context.Context, in *octopus.De
return l.DeleteDataSetVersion(in)
}
// ModelDeployService
// Develop
func (s *OctopusServer) GetNotebookList(ctx context.Context, in *octopus.GetNotebookListReq) (*octopus.GetNotebookListResp, error) {
l := logic.NewGetNotebookListLogic(ctx, s.svcCtx)
return l.GetNotebookList(in)
}
func (s *OctopusServer) GetNotebook(ctx context.Context, in *octopus.GetNotebookReq) (*octopus.GetNotebookResp, error) {
l := logic.NewGetNotebookLogic(ctx, s.svcCtx)
return l.GetNotebook(in)
}
func (s *OctopusServer) DeleteNotebook(ctx context.Context, in *octopus.DeleteNotebookReq) (*octopus.DeleteNotebookResp, error) {
l := logic.NewDeleteNotebookLogic(ctx, s.svcCtx)
return l.DeleteNotebook(in)
}
func (s *OctopusServer) CreateNotebook(ctx context.Context, in *octopus.CreateNotebookReq) (*octopus.CreateNotebookResp, error) {
l := logic.NewCreateNotebookLogic(ctx, s.svcCtx)
return l.CreateNotebook(in)
}
func (s *OctopusServer) StartNotebook(ctx context.Context, in *octopus.StartNotebookReq) (*octopus.StartNotebookResp, error) {
l := logic.NewStartNotebookLogic(ctx, s.svcCtx)
return l.StartNotebook(in)
}
func (s *OctopusServer) StopNotebook(ctx context.Context, in *octopus.StopNotebookReq) (*octopus.StopNotebookResp, error) {
l := logic.NewStopNotebookLogic(ctx, s.svcCtx)
return l.StopNotebook(in)
}
// ImageService
func (s *OctopusServer) GetUserImageList(ctx context.Context, in *octopus.GetUserImageListReq) (*octopus.GetUserImageListResp, error) {
l := logic.NewGetUserImageListLogic(ctx, s.svcCtx)
@ -160,3 +180,9 @@ func (s *OctopusServer) UploadImageConfirm(ctx context.Context, in *octopus.Uplo
l := logic.NewUploadImageConfirmLogic(ctx, s.svcCtx)
return l.UploadImageConfirm(in)
}
// TrainJobService
func (s *OctopusServer) CreateTrainJob(ctx context.Context, in *octopus.CreateTrainJobReq) (*octopus.CreateTrainJobResp, error) {
l := logic.NewCreateTrainJobLogic(ctx, s.svcCtx)
return l.CreateTrainJob(in)
}

File diff suppressed because it is too large Load Diff

View File

@ -44,16 +44,21 @@ type OctopusClient interface {
UploadDataSetConfirm(ctx context.Context, in *UploadDataSetConfirmReq, opts ...grpc.CallOption) (*UploadDataSetConfirmResp, error)
CreateDataSetVersion(ctx context.Context, in *CreateDataSetVersionReq, opts ...grpc.CallOption) (*CreateDataSetVersionResp, error)
DeleteDataSetVersion(ctx context.Context, in *DeleteDataSetVersionReq, opts ...grpc.CallOption) (*DeleteDataSetVersionResp, error)
// ModelDeployService
// Develop
GetNotebookList(ctx context.Context, in *GetNotebookListReq, opts ...grpc.CallOption) (*GetNotebookListResp, error)
GetNotebook(ctx context.Context, in *GetNotebookReq, opts ...grpc.CallOption) (*GetNotebookResp, error)
DeleteNotebook(ctx context.Context, in *DeleteNotebookReq, opts ...grpc.CallOption) (*DeleteNotebookResp, error)
CreateNotebook(ctx context.Context, in *CreateNotebookReq, opts ...grpc.CallOption) (*CreateNotebookResp, error)
StartNotebook(ctx context.Context, in *StartNotebookReq, opts ...grpc.CallOption) (*StartNotebookResp, error)
StopNotebook(ctx context.Context, in *StopNotebookReq, opts ...grpc.CallOption) (*StopNotebookResp, error)
// ImageService
GetUserImageList(ctx context.Context, in *GetUserImageListReq, opts ...grpc.CallOption) (*GetUserImageListResp, error)
CreateImage(ctx context.Context, in *CreateImageReq, opts ...grpc.CallOption) (*CreateImageResp, error)
DeleteImage(ctx context.Context, in *DeleteImageReq, opts ...grpc.CallOption) (*DeleteImageResp, error)
UploadImage(ctx context.Context, in *UploadImageReq, opts ...grpc.CallOption) (*UploadImageResp, error)
UploadImageConfirm(ctx context.Context, in *UploadImageConfirmReq, opts ...grpc.CallOption) (*UploadImageConfirmResp, error)
// TrainJobService
CreateTrainJob(ctx context.Context, in *CreateTrainJobReq, opts ...grpc.CallOption) (*CreateTrainJobResp, error)
}
type octopusClient struct {
@ -253,6 +258,15 @@ func (c *octopusClient) GetNotebookList(ctx context.Context, in *GetNotebookList
return out, nil
}
func (c *octopusClient) GetNotebook(ctx context.Context, in *GetNotebookReq, opts ...grpc.CallOption) (*GetNotebookResp, error) {
out := new(GetNotebookResp)
err := c.cc.Invoke(ctx, "/octopus.Octopus/GetNotebook", in, out, opts...)
if err != nil {
return nil, err
}
return out, nil
}
func (c *octopusClient) DeleteNotebook(ctx context.Context, in *DeleteNotebookReq, opts ...grpc.CallOption) (*DeleteNotebookResp, error) {
out := new(DeleteNotebookResp)
err := c.cc.Invoke(ctx, "/octopus.Octopus/DeleteNotebook", in, out, opts...)
@ -262,6 +276,33 @@ func (c *octopusClient) DeleteNotebook(ctx context.Context, in *DeleteNotebookRe
return out, nil
}
func (c *octopusClient) CreateNotebook(ctx context.Context, in *CreateNotebookReq, opts ...grpc.CallOption) (*CreateNotebookResp, error) {
out := new(CreateNotebookResp)
err := c.cc.Invoke(ctx, "/octopus.Octopus/CreateNotebook", in, out, opts...)
if err != nil {
return nil, err
}
return out, nil
}
func (c *octopusClient) StartNotebook(ctx context.Context, in *StartNotebookReq, opts ...grpc.CallOption) (*StartNotebookResp, error) {
out := new(StartNotebookResp)
err := c.cc.Invoke(ctx, "/octopus.Octopus/StartNotebook", in, out, opts...)
if err != nil {
return nil, err
}
return out, nil
}
func (c *octopusClient) StopNotebook(ctx context.Context, in *StopNotebookReq, opts ...grpc.CallOption) (*StopNotebookResp, error) {
out := new(StopNotebookResp)
err := c.cc.Invoke(ctx, "/octopus.Octopus/StopNotebook", in, out, opts...)
if err != nil {
return nil, err
}
return out, nil
}
func (c *octopusClient) GetUserImageList(ctx context.Context, in *GetUserImageListReq, opts ...grpc.CallOption) (*GetUserImageListResp, error) {
out := new(GetUserImageListResp)
err := c.cc.Invoke(ctx, "/octopus.Octopus/GetUserImageList", in, out, opts...)
@ -307,6 +348,15 @@ func (c *octopusClient) UploadImageConfirm(ctx context.Context, in *UploadImageC
return out, nil
}
func (c *octopusClient) CreateTrainJob(ctx context.Context, in *CreateTrainJobReq, opts ...grpc.CallOption) (*CreateTrainJobResp, error) {
out := new(CreateTrainJobResp)
err := c.cc.Invoke(ctx, "/octopus.Octopus/CreateTrainJob", in, out, opts...)
if err != nil {
return nil, err
}
return out, nil
}
// OctopusServer is the server API for Octopus service.
// All implementations must embed UnimplementedOctopusServer
// for forward compatibility
@ -333,16 +383,21 @@ type OctopusServer interface {
UploadDataSetConfirm(context.Context, *UploadDataSetConfirmReq) (*UploadDataSetConfirmResp, error)
CreateDataSetVersion(context.Context, *CreateDataSetVersionReq) (*CreateDataSetVersionResp, error)
DeleteDataSetVersion(context.Context, *DeleteDataSetVersionReq) (*DeleteDataSetVersionResp, error)
// ModelDeployService
// Develop
GetNotebookList(context.Context, *GetNotebookListReq) (*GetNotebookListResp, error)
GetNotebook(context.Context, *GetNotebookReq) (*GetNotebookResp, error)
DeleteNotebook(context.Context, *DeleteNotebookReq) (*DeleteNotebookResp, error)
CreateNotebook(context.Context, *CreateNotebookReq) (*CreateNotebookResp, error)
StartNotebook(context.Context, *StartNotebookReq) (*StartNotebookResp, error)
StopNotebook(context.Context, *StopNotebookReq) (*StopNotebookResp, error)
// ImageService
GetUserImageList(context.Context, *GetUserImageListReq) (*GetUserImageListResp, error)
CreateImage(context.Context, *CreateImageReq) (*CreateImageResp, error)
DeleteImage(context.Context, *DeleteImageReq) (*DeleteImageResp, error)
UploadImage(context.Context, *UploadImageReq) (*UploadImageResp, error)
UploadImageConfirm(context.Context, *UploadImageConfirmReq) (*UploadImageConfirmResp, error)
// TrainJobService
CreateTrainJob(context.Context, *CreateTrainJobReq) (*CreateTrainJobResp, error)
mustEmbedUnimplementedOctopusServer()
}
@ -413,9 +468,21 @@ func (UnimplementedOctopusServer) DeleteDataSetVersion(context.Context, *DeleteD
func (UnimplementedOctopusServer) GetNotebookList(context.Context, *GetNotebookListReq) (*GetNotebookListResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method GetNotebookList not implemented")
}
func (UnimplementedOctopusServer) GetNotebook(context.Context, *GetNotebookReq) (*GetNotebookResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method GetNotebook not implemented")
}
func (UnimplementedOctopusServer) DeleteNotebook(context.Context, *DeleteNotebookReq) (*DeleteNotebookResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method DeleteNotebook not implemented")
}
func (UnimplementedOctopusServer) CreateNotebook(context.Context, *CreateNotebookReq) (*CreateNotebookResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method CreateNotebook not implemented")
}
func (UnimplementedOctopusServer) StartNotebook(context.Context, *StartNotebookReq) (*StartNotebookResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method StartNotebook not implemented")
}
func (UnimplementedOctopusServer) StopNotebook(context.Context, *StopNotebookReq) (*StopNotebookResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method StopNotebook not implemented")
}
func (UnimplementedOctopusServer) GetUserImageList(context.Context, *GetUserImageListReq) (*GetUserImageListResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method GetUserImageList not implemented")
}
@ -431,6 +498,9 @@ func (UnimplementedOctopusServer) UploadImage(context.Context, *UploadImageReq)
func (UnimplementedOctopusServer) UploadImageConfirm(context.Context, *UploadImageConfirmReq) (*UploadImageConfirmResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method UploadImageConfirm not implemented")
}
func (UnimplementedOctopusServer) CreateTrainJob(context.Context, *CreateTrainJobReq) (*CreateTrainJobResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method CreateTrainJob not implemented")
}
func (UnimplementedOctopusServer) mustEmbedUnimplementedOctopusServer() {}
// UnsafeOctopusServer may be embedded to opt out of forward compatibility for this service.
@ -822,6 +892,24 @@ func _Octopus_GetNotebookList_Handler(srv interface{}, ctx context.Context, dec
return interceptor(ctx, in, info, handler)
}
func _Octopus_GetNotebook_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(GetNotebookReq)
if err := dec(in); err != nil {
return nil, err
}
if interceptor == nil {
return srv.(OctopusServer).GetNotebook(ctx, in)
}
info := &grpc.UnaryServerInfo{
Server: srv,
FullMethod: "/octopus.Octopus/GetNotebook",
}
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(OctopusServer).GetNotebook(ctx, req.(*GetNotebookReq))
}
return interceptor(ctx, in, info, handler)
}
func _Octopus_DeleteNotebook_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(DeleteNotebookReq)
if err := dec(in); err != nil {
@ -840,6 +928,60 @@ func _Octopus_DeleteNotebook_Handler(srv interface{}, ctx context.Context, dec f
return interceptor(ctx, in, info, handler)
}
func _Octopus_CreateNotebook_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(CreateNotebookReq)
if err := dec(in); err != nil {
return nil, err
}
if interceptor == nil {
return srv.(OctopusServer).CreateNotebook(ctx, in)
}
info := &grpc.UnaryServerInfo{
Server: srv,
FullMethod: "/octopus.Octopus/CreateNotebook",
}
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(OctopusServer).CreateNotebook(ctx, req.(*CreateNotebookReq))
}
return interceptor(ctx, in, info, handler)
}
func _Octopus_StartNotebook_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(StartNotebookReq)
if err := dec(in); err != nil {
return nil, err
}
if interceptor == nil {
return srv.(OctopusServer).StartNotebook(ctx, in)
}
info := &grpc.UnaryServerInfo{
Server: srv,
FullMethod: "/octopus.Octopus/StartNotebook",
}
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(OctopusServer).StartNotebook(ctx, req.(*StartNotebookReq))
}
return interceptor(ctx, in, info, handler)
}
func _Octopus_StopNotebook_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(StopNotebookReq)
if err := dec(in); err != nil {
return nil, err
}
if interceptor == nil {
return srv.(OctopusServer).StopNotebook(ctx, in)
}
info := &grpc.UnaryServerInfo{
Server: srv,
FullMethod: "/octopus.Octopus/StopNotebook",
}
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(OctopusServer).StopNotebook(ctx, req.(*StopNotebookReq))
}
return interceptor(ctx, in, info, handler)
}
func _Octopus_GetUserImageList_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(GetUserImageListReq)
if err := dec(in); err != nil {
@ -930,6 +1072,24 @@ func _Octopus_UploadImageConfirm_Handler(srv interface{}, ctx context.Context, d
return interceptor(ctx, in, info, handler)
}
func _Octopus_CreateTrainJob_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(CreateTrainJobReq)
if err := dec(in); err != nil {
return nil, err
}
if interceptor == nil {
return srv.(OctopusServer).CreateTrainJob(ctx, in)
}
info := &grpc.UnaryServerInfo{
Server: srv,
FullMethod: "/octopus.Octopus/CreateTrainJob",
}
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(OctopusServer).CreateTrainJob(ctx, req.(*CreateTrainJobReq))
}
return interceptor(ctx, in, info, handler)
}
// Octopus_ServiceDesc is the grpc.ServiceDesc for Octopus service.
// It's only intended for direct use with grpc.RegisterService,
// and not to be introspected or modified (even as a copy)
@ -1021,10 +1181,26 @@ var Octopus_ServiceDesc = grpc.ServiceDesc{
MethodName: "GetNotebookList",
Handler: _Octopus_GetNotebookList_Handler,
},
{
MethodName: "GetNotebook",
Handler: _Octopus_GetNotebook_Handler,
},
{
MethodName: "DeleteNotebook",
Handler: _Octopus_DeleteNotebook_Handler,
},
{
MethodName: "CreateNotebook",
Handler: _Octopus_CreateNotebook_Handler,
},
{
MethodName: "StartNotebook",
Handler: _Octopus_StartNotebook_Handler,
},
{
MethodName: "StopNotebook",
Handler: _Octopus_StopNotebook_Handler,
},
{
MethodName: "GetUserImageList",
Handler: _Octopus_GetUserImageList_Handler,
@ -1045,6 +1221,10 @@ var Octopus_ServiceDesc = grpc.ServiceDesc{
MethodName: "UploadImageConfirm",
Handler: _Octopus_UploadImageConfirm_Handler,
},
{
MethodName: "CreateTrainJob",
Handler: _Octopus_CreateTrainJob_Handler,
},
},
Streams: []grpc.StreamDesc{},
Metadata: "octopus.proto",

View File

@ -16,6 +16,7 @@ type (
AlgorithmDetail = octopus.AlgorithmDetail
Algorithms = octopus.Algorithms
Applies = octopus.Applies
Config = octopus.Config
CpResp = octopus.CpResp
CreateDataSet = octopus.CreateDataSet
CreateDataSetReq = octopus.CreateDataSetReq
@ -28,6 +29,12 @@ type (
CreateMyAlgorithm = octopus.CreateMyAlgorithm
CreateMyAlgorithmReq = octopus.CreateMyAlgorithmReq
CreateMyAlgorithmResp = octopus.CreateMyAlgorithmResp
CreateNotebookParam = octopus.CreateNotebookParam
CreateNotebookReq = octopus.CreateNotebookReq
CreateNotebookResp = octopus.CreateNotebookResp
CreateTrainJobParam = octopus.CreateTrainJobParam
CreateTrainJobReq = octopus.CreateTrainJobReq
CreateTrainJobResp = octopus.CreateTrainJobResp
DatasetVersion = octopus.DatasetVersion
Datasets = octopus.Datasets
DeleteDataSetReq = octopus.DeleteDataSetReq
@ -42,6 +49,7 @@ type (
DeleteNotebookResp = octopus.DeleteNotebookResp
DownloadAlgorithmReq = octopus.DownloadAlgorithmReq
DownloadAlgorithmResp = octopus.DownloadAlgorithmResp
Envs = octopus.Envs
Error = octopus.Error
GetAlgorithmApplyListReq = octopus.GetAlgorithmApplyListReq
GetAlgorithmApplyListResp = octopus.GetAlgorithmApplyListResp
@ -59,6 +67,8 @@ type (
GetMyDatasetListResp = octopus.GetMyDatasetListResp
GetNotebookListReq = octopus.GetNotebookListReq
GetNotebookListResp = octopus.GetNotebookListResp
GetNotebookReq = octopus.GetNotebookReq
GetNotebookResp = octopus.GetNotebookResp
GetUserImageListReq = octopus.GetUserImageListReq
GetUserImageListResp = octopus.GetUserImageListResp
GiResp = octopus.GiResp
@ -66,13 +76,18 @@ type (
Image = octopus.Image
Images = octopus.Images
Lables = octopus.Lables
Notebooks = octopus.Notebooks
Mounts = octopus.Mounts
Nfs = octopus.Nfs
Notebook = octopus.Notebook
Parameters = octopus.Parameters
PayloadAlgorithmFrameworkList = octopus.PayloadAlgorithmFrameworkList
PayloadAlgorithmList = octopus.PayloadAlgorithmList
PayloadCreateDataSet = octopus.PayloadCreateDataSet
PayloadCreateDataSetVersion = octopus.PayloadCreateDataSetVersion
PayloadCreateImage = octopus.PayloadCreateImage
PayloadCreateMyAlgorithm = octopus.PayloadCreateMyAlgorithm
PayloadCreateNotebook = octopus.PayloadCreateNotebook
PayloadCreateTrainJob = octopus.PayloadCreateTrainJob
PayloadDeleteDataSet = octopus.PayloadDeleteDataSet
PayloadDeleteDataSetVersion = octopus.PayloadDeleteDataSetVersion
PayloadDeleteImage = octopus.PayloadDeleteImage
@ -82,9 +97,12 @@ type (
PayloadGetAlgorithm = octopus.PayloadGetAlgorithm
PayloadGetAlgorithmApplyList = octopus.PayloadGetAlgorithmApplyList
PayloadGetDatasetVersion = octopus.PayloadGetDatasetVersion
PayloadGetNotebook = octopus.PayloadGetNotebook
PayloadMyAlgorithmList = octopus.PayloadMyAlgorithmList
PayloadMyDatasetList = octopus.PayloadMyDatasetList
PayloadNotebookList = octopus.PayloadNotebookList
PayloadStartNotebook = octopus.PayloadStartNotebook
PayloadStopNotebook = octopus.PayloadStopNotebook
PayloadUploadAlgorithm = octopus.PayloadUploadAlgorithm
PayloadUploadAlgorithmConfirm = octopus.PayloadUploadAlgorithmConfirm
PayloadUploadDataSet = octopus.PayloadUploadDataSet
@ -92,8 +110,14 @@ type (
PayloadUploadImage = octopus.PayloadUploadImage
PayloadUploadImageConfirm = octopus.PayloadUploadImageConfirm
PayloadUserImageList = octopus.PayloadUserImageList
ReplicaStates = octopus.ReplicaStates
ResourceReq = octopus.ResourceReq
StartNotebookReq = octopus.StartNotebookReq
StartNotebookResp = octopus.StartNotebookResp
StopNotebookReq = octopus.StopNotebookReq
StopNotebookResp = octopus.StopNotebookResp
Tasks = octopus.Tasks
TrainJobOctopus = octopus.TrainJobOctopus
UploadAlgorithmConfirmReq = octopus.UploadAlgorithmConfirmReq
UploadAlgorithmConfirmResp = octopus.UploadAlgorithmConfirmResp
UploadAlgorithmParam = octopus.UploadAlgorithmParam
@ -134,15 +158,21 @@ type (
UploadDataSetConfirm(ctx context.Context, in *UploadDataSetConfirmReq, opts ...grpc.CallOption) (*UploadDataSetConfirmResp, error)
CreateDataSetVersion(ctx context.Context, in *CreateDataSetVersionReq, opts ...grpc.CallOption) (*CreateDataSetVersionResp, error)
DeleteDataSetVersion(ctx context.Context, in *DeleteDataSetVersionReq, opts ...grpc.CallOption) (*DeleteDataSetVersionResp, error)
// ModelDeployService
// Develop
GetNotebookList(ctx context.Context, in *GetNotebookListReq, opts ...grpc.CallOption) (*GetNotebookListResp, error)
GetNotebook(ctx context.Context, in *GetNotebookReq, opts ...grpc.CallOption) (*GetNotebookResp, error)
DeleteNotebook(ctx context.Context, in *DeleteNotebookReq, opts ...grpc.CallOption) (*DeleteNotebookResp, error)
CreateNotebook(ctx context.Context, in *CreateNotebookReq, opts ...grpc.CallOption) (*CreateNotebookResp, error)
StartNotebook(ctx context.Context, in *StartNotebookReq, opts ...grpc.CallOption) (*StartNotebookResp, error)
StopNotebook(ctx context.Context, in *StopNotebookReq, opts ...grpc.CallOption) (*StopNotebookResp, error)
// ImageService
GetUserImageList(ctx context.Context, in *GetUserImageListReq, opts ...grpc.CallOption) (*GetUserImageListResp, error)
CreateImage(ctx context.Context, in *CreateImageReq, opts ...grpc.CallOption) (*CreateImageResp, error)
DeleteImage(ctx context.Context, in *DeleteImageReq, opts ...grpc.CallOption) (*DeleteImageResp, error)
UploadImage(ctx context.Context, in *UploadImageReq, opts ...grpc.CallOption) (*UploadImageResp, error)
UploadImageConfirm(ctx context.Context, in *UploadImageConfirmReq, opts ...grpc.CallOption) (*UploadImageConfirmResp, error)
// TrainJobService
CreateTrainJob(ctx context.Context, in *CreateTrainJobReq, opts ...grpc.CallOption) (*CreateTrainJobResp, error)
}
defaultOctopus struct {
@ -258,17 +288,37 @@ func (m *defaultOctopus) DeleteDataSetVersion(ctx context.Context, in *DeleteDat
return client.DeleteDataSetVersion(ctx, in, opts...)
}
// ModelDeployService
// Develop
func (m *defaultOctopus) GetNotebookList(ctx context.Context, in *GetNotebookListReq, opts ...grpc.CallOption) (*GetNotebookListResp, error) {
client := octopus.NewOctopusClient(m.cli.Conn())
return client.GetNotebookList(ctx, in, opts...)
}
func (m *defaultOctopus) GetNotebook(ctx context.Context, in *GetNotebookReq, opts ...grpc.CallOption) (*GetNotebookResp, error) {
client := octopus.NewOctopusClient(m.cli.Conn())
return client.GetNotebook(ctx, in, opts...)
}
func (m *defaultOctopus) DeleteNotebook(ctx context.Context, in *DeleteNotebookReq, opts ...grpc.CallOption) (*DeleteNotebookResp, error) {
client := octopus.NewOctopusClient(m.cli.Conn())
return client.DeleteNotebook(ctx, in, opts...)
}
func (m *defaultOctopus) CreateNotebook(ctx context.Context, in *CreateNotebookReq, opts ...grpc.CallOption) (*CreateNotebookResp, error) {
client := octopus.NewOctopusClient(m.cli.Conn())
return client.CreateNotebook(ctx, in, opts...)
}
func (m *defaultOctopus) StartNotebook(ctx context.Context, in *StartNotebookReq, opts ...grpc.CallOption) (*StartNotebookResp, error) {
client := octopus.NewOctopusClient(m.cli.Conn())
return client.StartNotebook(ctx, in, opts...)
}
func (m *defaultOctopus) StopNotebook(ctx context.Context, in *StopNotebookReq, opts ...grpc.CallOption) (*StopNotebookResp, error) {
client := octopus.NewOctopusClient(m.cli.Conn())
return client.StopNotebook(ctx, in, opts...)
}
// ImageService
func (m *defaultOctopus) GetUserImageList(ctx context.Context, in *GetUserImageListReq, opts ...grpc.CallOption) (*GetUserImageListResp, error) {
client := octopus.NewOctopusClient(m.cli.Conn())
@ -294,3 +344,9 @@ func (m *defaultOctopus) UploadImageConfirm(ctx context.Context, in *UploadImage
client := octopus.NewOctopusClient(m.cli.Conn())
return client.UploadImageConfirm(ctx, in, opts...)
}
// TrainJobService
func (m *defaultOctopus) CreateTrainJob(ctx context.Context, in *CreateTrainJobReq, opts ...grpc.CallOption) (*CreateTrainJobResp, error) {
client := octopus.NewOctopusClient(m.cli.Conn())
return client.CreateTrainJob(ctx, in, opts...)
}

View File

@ -417,6 +417,53 @@ message PayloadDeleteDataSetVersion{
/******************ModelDeployService End*************************/
/******************Develop Start*************************/
message CreateNotebookReq{
string platform =1;
CreateNotebookParam param = 2;
}
message CreateNotebookParam{
string algorithmId = 1;
string algorithmVersion = 2;
string command = 3;
string datasetId = 4;
string datasetVersion = 5;
string desc = 6;
Envs envs = 7;
string imageId = 8;
string imageUrl = 9;
repeated Mounts mounts = 10;
string name = 11;
string resourcePool = 12;
string resourceSpecId = 13;
int32 taskNumber = 14;
}
message CreateNotebookResp{
bool success =1;
PayloadCreateNotebook payload =2;
Error error = 3;
}
message PayloadCreateNotebook{
string id = 1;
}
message GetNotebookReq{
string platform =1;
string id =2;
}
message GetNotebookResp{
bool success =1;
PayloadGetNotebook payload =2;
Error error = 3;
}
message PayloadGetNotebook{
Notebook notebook = 1;
}
message DeleteNotebookReq{
string platform =1;
string id =2;
@ -446,10 +493,10 @@ message GetNotebookListResp{
message PayloadNotebookList{
int32 totalSize =1;
repeated Notebooks notebooks = 2;
repeated Notebook notebooks = 2;
}
message Notebooks{
message Notebook{
int64 createdAt =1;
int64 updatedAt =2;
string id = 3;
@ -479,6 +526,38 @@ message Tasks{
string url =1;
string name= 2;
}
message StartNotebookReq{
string platform =1;
string id =2;
}
message StartNotebookResp{
bool success =1;
PayloadStartNotebook payload =2;
Error error = 3;
}
message PayloadStartNotebook{
string id = 1;
}
message StopNotebookReq{
string platform =1;
string id =2;
}
message StopNotebookResp{
bool success =1;
PayloadStopNotebook payload =2;
Error error = 3;
}
message PayloadStopNotebook{
string id = 1;
}
/******************Develop End*************************/
/******************ImageService Start*************************/
@ -608,6 +687,85 @@ message PayloadUploadImageConfirm{
/******************Model End*************************/
/******************TrainJobService Start*************************/
message CreateTrainJobReq{
string platform =1;
CreateTrainJobParam param = 2;
}
message CreateTrainJobResp{
bool success =1;
PayloadCreateTrainJob payload =2;
Error error = 3;
}
message CreateTrainJobParam{
string algorithmId = 1;
string algorithmVersion = 2;
repeated Config config = 3;
string dataSetId = 4;
string dataSetVersion = 5;
string desc = 6;
string imageId = 7;
string imageUrl = 8;
bool isDistributed = 9;
repeated Mounts mounts = 10;
string name = 11;
string resourcePool = 12;
}
message Config{
string command = 1;
Envs envs = 2;
bool isMainRole = 3;
int32 minFailedTaskCount = 4;
int32 minSucceededTaskCount =5;
string name = 6;
repeated Parameters parameters = 7;
repeated ReplicaStates replicaStates = 8;
string resourceSpecId = 9;
string resourceSpecName = 10;
int32 resourceSpecPrice = 11;
string subTaskState = 12;
int32 taskNumber = 13;
}
message Envs{
string additionalProp1 = 1;
string additionalProp2 = 2;
string additionalProp3 = 3;
}
message Parameters{
string key = 1;
string value = 2;
}
message ReplicaStates{
string key = 1;
string state = 2;
}
message Mounts{
string containerPath = 1;
Nfs nfs = 2;
TrainJobOctopus octopus = 3;
bool readOnly = 4;
}
message PayloadCreateTrainJob{
string jobId = 1;
}
message Nfs{
string path = 1;
string server = 2;
}
message TrainJobOctopus{
string bucket = 1;
string object = 2;
}
/******************TrainJobService End*************************/
message Error{
@ -638,9 +796,6 @@ service Octopus {
//DatasetService
rpc GetMyDatasetList(GetMyDatasetListReq) returns (GetMyDatasetListResp);
rpc GetDatasetVersionList(GetDatasetVersionListReq) returns (GetDatasetVersionListResp);
@ -653,12 +808,17 @@ service Octopus {
//ModelDeployService
//Develop
rpc GetNotebookList(GetNotebookListReq) returns (GetNotebookListResp);
rpc GetNotebook(GetNotebookReq) returns (GetNotebookResp);
rpc DeleteNotebook(DeleteNotebookReq) returns (DeleteNotebookResp);
rpc CreateNotebook(CreateNotebookReq) returns (CreateNotebookResp);
rpc StartNotebook(StartNotebookReq) returns (StartNotebookResp);
rpc StopNotebook(StopNotebookReq) returns (StopNotebookResp);
//ImageService
@ -669,9 +829,12 @@ service Octopus {
rpc UploadImageConfirm(UploadImageConfirmReq) returns (UploadImageConfirmResp);
//Model
//TrainJobService
rpc CreateTrainJob(CreateTrainJobReq) returns (CreateTrainJobResp);
}