启智章鱼接口调整

Former-commit-id: e9129ab518f09b5a77e9820249897d6f0d26f038

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

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

adaptor/PCM-AI/PCM-OCTOPUS/rpc/internal/logic/createnotebooklogic.go

@@ -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
+}

adaptor/PCM-AI/PCM-OCTOPUS/rpc/internal/logic/createtrainjoblogic.go

@@ -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
+}

adaptor/PCM-AI/PCM-OCTOPUS/rpc/internal/logic/getnotebooklogic.go

@@ -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
+}

adaptor/PCM-AI/PCM-OCTOPUS/rpc/internal/logic/startnotebooklogic.go

@@ -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
+}

adaptor/PCM-AI/PCM-OCTOPUS/rpc/internal/logic/stopnotebooklogic.go

@@ -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
+}

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

@@ -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)
+}

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

@@ -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",

adaptor/PCM-AI/PCM-OCTOPUS/rpc/octopusclient/octopus.go

@@ -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...)
+}

adaptor/PCM-AI/PCM-OCTOPUS/rpc/pb/octopus.proto

@@ -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);
+
 
 }