forked from JointCloud/pcm-ac
360 lines
17 KiB
Go
360 lines
17 KiB
Go
// Code generated by goctl. DO NOT EDIT.
|
|
// Source: hpcAC.proto
|
|
|
|
package hpcacclient
|
|
|
|
import (
|
|
"context"
|
|
|
|
"gitlink.org.cn/jcce-pcm/pcm-ac/hpcAC"
|
|
|
|
"github.com/zeromicro/go-zero/zrpc"
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
type (
|
|
ACClusterData = hpcAC.ACClusterData
|
|
ACClusterReq = hpcAC.ACClusterReq
|
|
ACClusterResp = hpcAC.ACClusterResp
|
|
ACTokenData = hpcAC.ACTokenData
|
|
ACTokenReq = hpcAC.ACTokenReq
|
|
ACTokenResp = hpcAC.ACTokenResp
|
|
ACTokenState = hpcAC.ACTokenState
|
|
Children = hpcAC.Children
|
|
ClusterResp = hpcAC.ClusterResp
|
|
CpResp = hpcAC.CpResp
|
|
CpuCore = hpcAC.CpuCore
|
|
CpuCoreReq = hpcAC.CpuCoreReq
|
|
CpuCoreResp = hpcAC.CpuCoreResp
|
|
DeleteJobReq = hpcAC.DeleteJobReq
|
|
DeleteJobResp = hpcAC.DeleteJobResp
|
|
DeleteTaskAiReq = hpcAC.DeleteTaskAiReq
|
|
DeleteTaskAiResp = hpcAC.DeleteTaskAiResp
|
|
FileContentResp = hpcAC.FileContentResp
|
|
FileDataReq = hpcAC.FileDataReq
|
|
FileDataResp = hpcAC.FileDataResp
|
|
FileList = hpcAC.FileList
|
|
FileListData = hpcAC.FileListData
|
|
GetFileListReq = hpcAC.GetFileListReq
|
|
GetFileListResp = hpcAC.GetFileListResp
|
|
GetImageAiByIdReq = hpcAC.GetImageAiByIdReq
|
|
GetImageAiByIdResp = hpcAC.GetImageAiByIdResp
|
|
GetImageListAiReq = hpcAC.GetImageListAiReq
|
|
GetImageListAiResp = hpcAC.GetImageListAiResp
|
|
GetJobDetailResp = hpcAC.GetJobDetailResp
|
|
GetMemberJobsData = hpcAC.GetMemberJobsData
|
|
GetMemberJobsReq = hpcAC.GetMemberJobsReq
|
|
GetMemberJobsResp = hpcAC.GetMemberJobsResp
|
|
GetNodeResourcesData = hpcAC.GetNodeResourcesData
|
|
GetNodeResourcesReq = hpcAC.GetNodeResourcesReq
|
|
GetNodeResourcesResp = hpcAC.GetNodeResourcesResp
|
|
GetPytorchTaskReq = hpcAC.GetPytorchTaskReq
|
|
GetPytorchTaskResp = hpcAC.GetPytorchTaskResp
|
|
GetPytorchTaskRespParams = hpcAC.GetPytorchTaskRespParams
|
|
GetResourceSpecReq = hpcAC.GetResourceSpecReq
|
|
GetResourceSpecResp = hpcAC.GetResourceSpecResp
|
|
GetTensorflowTaskReq = hpcAC.GetTensorflowTaskReq
|
|
GetTensorflowTaskResp = hpcAC.GetTensorflowTaskResp
|
|
GetTensorflowTaskRespParams = hpcAC.GetTensorflowTaskRespParams
|
|
GetUserInfoData = hpcAC.GetUserInfoData
|
|
GetUserInfoReq = hpcAC.GetUserInfoReq
|
|
GetUserInfoResp = hpcAC.GetUserInfoResp
|
|
GiResp = hpcAC.GiResp
|
|
HistoryJobData = hpcAC.HistoryJobData
|
|
HistoryJobDetail = hpcAC.HistoryJobDetail
|
|
HistoryJobDetailReq = hpcAC.HistoryJobDetailReq
|
|
HistoryJobDetailResp = hpcAC.HistoryJobDetailResp
|
|
HistoryJobList = hpcAC.HistoryJobList
|
|
ImageAI = hpcAC.ImageAI
|
|
Job = hpcAC.Job
|
|
JobCore = hpcAC.JobCore
|
|
JobDetail = hpcAC.JobDetail
|
|
JobDetailReq = hpcAC.JobDetailReq
|
|
JobInitAttr = hpcAC.JobInitAttr
|
|
JobManager = hpcAC.JobManager
|
|
JobManagerReq = hpcAC.JobManagerReq
|
|
JobVncSessionInfo = hpcAC.JobVncSessionInfo
|
|
JobsReq = hpcAC.JobsReq
|
|
JobsResp = hpcAC.JobsResp
|
|
ListHistoryJobReq = hpcAC.ListHistoryJobReq
|
|
ListHistoryJobResp = hpcAC.ListHistoryJobResp
|
|
ListJobManagerResp = hpcAC.ListJobManagerResp
|
|
ListJobReq = hpcAC.ListJobReq
|
|
ListJobResp = hpcAC.ListJobResp
|
|
MapAppJobInfo = hpcAC.MapAppJobInfo
|
|
Metric = hpcAC.Metric
|
|
ParaStorQuotaReq = hpcAC.ParaStorQuotaReq
|
|
ParaStorQuotaResp = hpcAC.ParaStorQuotaResp
|
|
PermissionAction = hpcAC.PermissionAction
|
|
Queue = hpcAC.Queue
|
|
QueueData = hpcAC.QueueData
|
|
QueueDetailsData = hpcAC.QueueDetailsData
|
|
QueueDetailsResp = hpcAC.QueueDetailsResp
|
|
QueueJobsReq = hpcAC.QueueJobsReq
|
|
QueueJobsResp = hpcAC.QueueJobsResp
|
|
QueueReq = hpcAC.QueueReq
|
|
QueueResp = hpcAC.QueueResp
|
|
QuotaData = hpcAC.QuotaData
|
|
ResourceReq = hpcAC.ResourceReq
|
|
ResourceSpec = hpcAC.ResourceSpec
|
|
SubmitJobReq = hpcAC.SubmitJobReq
|
|
SubmitJobResp = hpcAC.SubmitJobResp
|
|
SubmitPytorchTaskParams = hpcAC.SubmitPytorchTaskParams
|
|
SubmitPytorchTaskReq = hpcAC.SubmitPytorchTaskReq
|
|
SubmitTaskAiResp = hpcAC.SubmitTaskAiResp
|
|
SubmitTensorflowTaskParams = hpcAC.SubmitTensorflowTaskParams
|
|
SubmitTensorflowTaskReq = hpcAC.SubmitTensorflowTaskReq
|
|
TokenResp = hpcAC.TokenResp
|
|
UserQuotasLimitData = hpcAC.UserQuotasLimitData
|
|
UserQuotasLimitResp = hpcAC.UserQuotasLimitResp
|
|
WallTimeReq = hpcAC.WallTimeReq
|
|
WallTimeResp = hpcAC.WallTimeResp
|
|
|
|
HpcAC interface {
|
|
// ListJob list all jobs
|
|
ListJob(ctx context.Context, in *ListJobReq, opts ...grpc.CallOption) (*ListJobResp, error)
|
|
// ListHistoryJob list all history jobs
|
|
ListHistoryJob(ctx context.Context, in *ListHistoryJobReq, opts ...grpc.CallOption) (*ListHistoryJobResp, error)
|
|
// Submit job
|
|
SubmitJob(ctx context.Context, in *SubmitJobReq, opts ...grpc.CallOption) (*SubmitJobResp, error)
|
|
// ListJobManager list all job managers
|
|
ListJobManager(ctx context.Context, in *JobManagerReq, opts ...grpc.CallOption) (*ListJobManagerResp, error)
|
|
// GetJobDetail get job detail
|
|
GetJobDetail(ctx context.Context, in *JobDetailReq, opts ...grpc.CallOption) (*GetJobDetailResp, error)
|
|
// DeleteJob delete job
|
|
DeleteJob(ctx context.Context, in *DeleteJobReq, opts ...grpc.CallOption) (*DeleteJobResp, error)
|
|
// FileContent 查看文件内容
|
|
FileContent(ctx context.Context, in *FileDataReq, opts ...grpc.CallOption) (*FileContentResp, error)
|
|
// SelectQueueByUser 查询用户可访问队列列表
|
|
SelectQueueByUser(ctx context.Context, in *QueueReq, opts ...grpc.CallOption) (*QueueResp, error)
|
|
// QueryQueueDetails 查询队列资源详细信息
|
|
QueryQueueDetails(ctx context.Context, in *QueueReq, opts ...grpc.CallOption) (*QueueDetailsResp, error)
|
|
// QueryUserQuotasLimit 查询用户资源限制信息
|
|
QueryUserQuotasLimit(ctx context.Context, in *QueueReq, opts ...grpc.CallOption) (*UserQuotasLimitResp, error)
|
|
// ParaStor Quota
|
|
ParaStorQuota(ctx context.Context, in *ParaStorQuotaReq, opts ...grpc.CallOption) (*ParaStorQuotaResp, error)
|
|
// WallTime
|
|
WallTime(ctx context.Context, in *WallTimeReq, opts ...grpc.CallOption) (*WallTimeResp, error)
|
|
// QueueJobs
|
|
QueueJobs(ctx context.Context, in *QueueJobsReq, opts ...grpc.CallOption) (*QueueJobsResp, error)
|
|
CpuCore(ctx context.Context, in *CpuCoreReq, opts ...grpc.CallOption) (*CpuCoreResp, error)
|
|
Jobs(ctx context.Context, in *JobsReq, opts ...grpc.CallOption) (*JobsResp, error)
|
|
HistoryJobDetail(ctx context.Context, in *HistoryJobDetailReq, opts ...grpc.CallOption) (*HistoryJobDetailResp, error)
|
|
// 曙光ac获取token
|
|
GetACToken(ctx context.Context, in *ACTokenReq, opts ...grpc.CallOption) (*TokenResp, error)
|
|
// 曙光ac获取clusterid
|
|
GetACClusterId(ctx context.Context, in *ACClusterReq, opts ...grpc.CallOption) (*ClusterResp, error)
|
|
// 获取曙光账号算力
|
|
GetComputingPower(ctx context.Context, in *ResourceReq, opts ...grpc.CallOption) (*CpResp, error)
|
|
// 获取曙光账号cpu,内存，存储信息
|
|
GetGeneralInfo(ctx context.Context, in *ResourceReq, opts ...grpc.CallOption) (*GiResp, error)
|
|
// 曙光智算接口
|
|
SubmitPytorchTask(ctx context.Context, in *SubmitPytorchTaskReq, opts ...grpc.CallOption) (*SubmitTaskAiResp, error)
|
|
SubmitTensorflowTask(ctx context.Context, in *SubmitTensorflowTaskReq, opts ...grpc.CallOption) (*SubmitTaskAiResp, error)
|
|
GetImageListAi(ctx context.Context, in *GetImageListAiReq, opts ...grpc.CallOption) (*GetImageListAiResp, error)
|
|
GetImageAiById(ctx context.Context, in *GetImageAiByIdReq, opts ...grpc.CallOption) (*GetImageAiByIdResp, error)
|
|
GetPytorchTask(ctx context.Context, in *GetPytorchTaskReq, opts ...grpc.CallOption) (*GetPytorchTaskResp, error)
|
|
GetTensorflowTask(ctx context.Context, in *GetTensorflowTaskReq, opts ...grpc.CallOption) (*GetTensorflowTaskResp, error)
|
|
DeleteTaskAi(ctx context.Context, in *DeleteTaskAiReq, opts ...grpc.CallOption) (*DeleteTaskAiResp, error)
|
|
GetResourceSpec(ctx context.Context, in *GetResourceSpecReq, opts ...grpc.CallOption) (*GetResourceSpecResp, error)
|
|
// 曙光文件接口
|
|
GetFileList(ctx context.Context, in *GetFileListReq, opts ...grpc.CallOption) (*GetFileListResp, error)
|
|
// 用户资源
|
|
GetUserInfo(ctx context.Context, in *GetUserInfoReq, opts ...grpc.CallOption) (*GetUserInfoResp, error)
|
|
GetMemberJobs(ctx context.Context, in *GetMemberJobsReq, opts ...grpc.CallOption) (*GetMemberJobsResp, error)
|
|
// 获取节点资源限额
|
|
GetNodeResources(ctx context.Context, in *GetNodeResourcesReq, opts ...grpc.CallOption) (*GetNodeResourcesResp, error)
|
|
}
|
|
|
|
defaultHpcAC struct {
|
|
cli zrpc.Client
|
|
}
|
|
)
|
|
|
|
func NewHpcAC(cli zrpc.Client) HpcAC {
|
|
return &defaultHpcAC{
|
|
cli: cli,
|
|
}
|
|
}
|
|
|
|
// ListJob list all jobs
|
|
func (m *defaultHpcAC) ListJob(ctx context.Context, in *ListJobReq, opts ...grpc.CallOption) (*ListJobResp, error) {
|
|
client := hpcAC.NewHpcACClient(m.cli.Conn())
|
|
return client.ListJob(ctx, in, opts...)
|
|
}
|
|
|
|
// ListHistoryJob list all history jobs
|
|
func (m *defaultHpcAC) ListHistoryJob(ctx context.Context, in *ListHistoryJobReq, opts ...grpc.CallOption) (*ListHistoryJobResp, error) {
|
|
client := hpcAC.NewHpcACClient(m.cli.Conn())
|
|
return client.ListHistoryJob(ctx, in, opts...)
|
|
}
|
|
|
|
// Submit job
|
|
func (m *defaultHpcAC) SubmitJob(ctx context.Context, in *SubmitJobReq, opts ...grpc.CallOption) (*SubmitJobResp, error) {
|
|
client := hpcAC.NewHpcACClient(m.cli.Conn())
|
|
return client.SubmitJob(ctx, in, opts...)
|
|
}
|
|
|
|
// ListJobManager list all job managers
|
|
func (m *defaultHpcAC) ListJobManager(ctx context.Context, in *JobManagerReq, opts ...grpc.CallOption) (*ListJobManagerResp, error) {
|
|
client := hpcAC.NewHpcACClient(m.cli.Conn())
|
|
return client.ListJobManager(ctx, in, opts...)
|
|
}
|
|
|
|
// GetJobDetail get job detail
|
|
func (m *defaultHpcAC) GetJobDetail(ctx context.Context, in *JobDetailReq, opts ...grpc.CallOption) (*GetJobDetailResp, error) {
|
|
client := hpcAC.NewHpcACClient(m.cli.Conn())
|
|
return client.GetJobDetail(ctx, in, opts...)
|
|
}
|
|
|
|
// DeleteJob delete job
|
|
func (m *defaultHpcAC) DeleteJob(ctx context.Context, in *DeleteJobReq, opts ...grpc.CallOption) (*DeleteJobResp, error) {
|
|
client := hpcAC.NewHpcACClient(m.cli.Conn())
|
|
return client.DeleteJob(ctx, in, opts...)
|
|
}
|
|
|
|
// FileContent 查看文件内容
|
|
func (m *defaultHpcAC) FileContent(ctx context.Context, in *FileDataReq, opts ...grpc.CallOption) (*FileContentResp, error) {
|
|
client := hpcAC.NewHpcACClient(m.cli.Conn())
|
|
return client.FileContent(ctx, in, opts...)
|
|
}
|
|
|
|
// SelectQueueByUser 查询用户可访问队列列表
|
|
func (m *defaultHpcAC) SelectQueueByUser(ctx context.Context, in *QueueReq, opts ...grpc.CallOption) (*QueueResp, error) {
|
|
client := hpcAC.NewHpcACClient(m.cli.Conn())
|
|
return client.SelectQueueByUser(ctx, in, opts...)
|
|
}
|
|
|
|
// QueryQueueDetails 查询队列资源详细信息
|
|
func (m *defaultHpcAC) QueryQueueDetails(ctx context.Context, in *QueueReq, opts ...grpc.CallOption) (*QueueDetailsResp, error) {
|
|
client := hpcAC.NewHpcACClient(m.cli.Conn())
|
|
return client.QueryQueueDetails(ctx, in, opts...)
|
|
}
|
|
|
|
// QueryUserQuotasLimit 查询用户资源限制信息
|
|
func (m *defaultHpcAC) QueryUserQuotasLimit(ctx context.Context, in *QueueReq, opts ...grpc.CallOption) (*UserQuotasLimitResp, error) {
|
|
client := hpcAC.NewHpcACClient(m.cli.Conn())
|
|
return client.QueryUserQuotasLimit(ctx, in, opts...)
|
|
}
|
|
|
|
// ParaStor Quota
|
|
func (m *defaultHpcAC) ParaStorQuota(ctx context.Context, in *ParaStorQuotaReq, opts ...grpc.CallOption) (*ParaStorQuotaResp, error) {
|
|
client := hpcAC.NewHpcACClient(m.cli.Conn())
|
|
return client.ParaStorQuota(ctx, in, opts...)
|
|
}
|
|
|
|
// WallTime
|
|
func (m *defaultHpcAC) WallTime(ctx context.Context, in *WallTimeReq, opts ...grpc.CallOption) (*WallTimeResp, error) {
|
|
client := hpcAC.NewHpcACClient(m.cli.Conn())
|
|
return client.WallTime(ctx, in, opts...)
|
|
}
|
|
|
|
// QueueJobs
|
|
func (m *defaultHpcAC) QueueJobs(ctx context.Context, in *QueueJobsReq, opts ...grpc.CallOption) (*QueueJobsResp, error) {
|
|
client := hpcAC.NewHpcACClient(m.cli.Conn())
|
|
return client.QueueJobs(ctx, in, opts...)
|
|
}
|
|
|
|
func (m *defaultHpcAC) CpuCore(ctx context.Context, in *CpuCoreReq, opts ...grpc.CallOption) (*CpuCoreResp, error) {
|
|
client := hpcAC.NewHpcACClient(m.cli.Conn())
|
|
return client.CpuCore(ctx, in, opts...)
|
|
}
|
|
|
|
func (m *defaultHpcAC) Jobs(ctx context.Context, in *JobsReq, opts ...grpc.CallOption) (*JobsResp, error) {
|
|
client := hpcAC.NewHpcACClient(m.cli.Conn())
|
|
return client.Jobs(ctx, in, opts...)
|
|
}
|
|
|
|
func (m *defaultHpcAC) HistoryJobDetail(ctx context.Context, in *HistoryJobDetailReq, opts ...grpc.CallOption) (*HistoryJobDetailResp, error) {
|
|
client := hpcAC.NewHpcACClient(m.cli.Conn())
|
|
return client.HistoryJobDetail(ctx, in, opts...)
|
|
}
|
|
|
|
// 曙光ac获取token
|
|
func (m *defaultHpcAC) GetACToken(ctx context.Context, in *ACTokenReq, opts ...grpc.CallOption) (*TokenResp, error) {
|
|
client := hpcAC.NewHpcACClient(m.cli.Conn())
|
|
return client.GetACToken(ctx, in, opts...)
|
|
}
|
|
|
|
// 曙光ac获取clusterid
|
|
func (m *defaultHpcAC) GetACClusterId(ctx context.Context, in *ACClusterReq, opts ...grpc.CallOption) (*ClusterResp, error) {
|
|
client := hpcAC.NewHpcACClient(m.cli.Conn())
|
|
return client.GetACClusterId(ctx, in, opts...)
|
|
}
|
|
|
|
// 获取曙光账号算力
|
|
func (m *defaultHpcAC) GetComputingPower(ctx context.Context, in *ResourceReq, opts ...grpc.CallOption) (*CpResp, error) {
|
|
client := hpcAC.NewHpcACClient(m.cli.Conn())
|
|
return client.GetComputingPower(ctx, in, opts...)
|
|
}
|
|
|
|
// 获取曙光账号cpu,内存，存储信息
|
|
func (m *defaultHpcAC) GetGeneralInfo(ctx context.Context, in *ResourceReq, opts ...grpc.CallOption) (*GiResp, error) {
|
|
client := hpcAC.NewHpcACClient(m.cli.Conn())
|
|
return client.GetGeneralInfo(ctx, in, opts...)
|
|
}
|
|
|
|
// 曙光智算接口
|
|
func (m *defaultHpcAC) SubmitPytorchTask(ctx context.Context, in *SubmitPytorchTaskReq, opts ...grpc.CallOption) (*SubmitTaskAiResp, error) {
|
|
client := hpcAC.NewHpcACClient(m.cli.Conn())
|
|
return client.SubmitPytorchTask(ctx, in, opts...)
|
|
}
|
|
|
|
func (m *defaultHpcAC) SubmitTensorflowTask(ctx context.Context, in *SubmitTensorflowTaskReq, opts ...grpc.CallOption) (*SubmitTaskAiResp, error) {
|
|
client := hpcAC.NewHpcACClient(m.cli.Conn())
|
|
return client.SubmitTensorflowTask(ctx, in, opts...)
|
|
}
|
|
|
|
func (m *defaultHpcAC) GetImageListAi(ctx context.Context, in *GetImageListAiReq, opts ...grpc.CallOption) (*GetImageListAiResp, error) {
|
|
client := hpcAC.NewHpcACClient(m.cli.Conn())
|
|
return client.GetImageListAi(ctx, in, opts...)
|
|
}
|
|
|
|
func (m *defaultHpcAC) GetImageAiById(ctx context.Context, in *GetImageAiByIdReq, opts ...grpc.CallOption) (*GetImageAiByIdResp, error) {
|
|
client := hpcAC.NewHpcACClient(m.cli.Conn())
|
|
return client.GetImageAiById(ctx, in, opts...)
|
|
}
|
|
|
|
func (m *defaultHpcAC) GetPytorchTask(ctx context.Context, in *GetPytorchTaskReq, opts ...grpc.CallOption) (*GetPytorchTaskResp, error) {
|
|
client := hpcAC.NewHpcACClient(m.cli.Conn())
|
|
return client.GetPytorchTask(ctx, in, opts...)
|
|
}
|
|
|
|
func (m *defaultHpcAC) GetTensorflowTask(ctx context.Context, in *GetTensorflowTaskReq, opts ...grpc.CallOption) (*GetTensorflowTaskResp, error) {
|
|
client := hpcAC.NewHpcACClient(m.cli.Conn())
|
|
return client.GetTensorflowTask(ctx, in, opts...)
|
|
}
|
|
|
|
func (m *defaultHpcAC) DeleteTaskAi(ctx context.Context, in *DeleteTaskAiReq, opts ...grpc.CallOption) (*DeleteTaskAiResp, error) {
|
|
client := hpcAC.NewHpcACClient(m.cli.Conn())
|
|
return client.DeleteTaskAi(ctx, in, opts...)
|
|
}
|
|
|
|
func (m *defaultHpcAC) GetResourceSpec(ctx context.Context, in *GetResourceSpecReq, opts ...grpc.CallOption) (*GetResourceSpecResp, error) {
|
|
client := hpcAC.NewHpcACClient(m.cli.Conn())
|
|
return client.GetResourceSpec(ctx, in, opts...)
|
|
}
|
|
|
|
// 曙光文件接口
|
|
func (m *defaultHpcAC) GetFileList(ctx context.Context, in *GetFileListReq, opts ...grpc.CallOption) (*GetFileListResp, error) {
|
|
client := hpcAC.NewHpcACClient(m.cli.Conn())
|
|
return client.GetFileList(ctx, in, opts...)
|
|
}
|
|
|
|
// 用户资源
|
|
func (m *defaultHpcAC) GetUserInfo(ctx context.Context, in *GetUserInfoReq, opts ...grpc.CallOption) (*GetUserInfoResp, error) {
|
|
client := hpcAC.NewHpcACClient(m.cli.Conn())
|
|
return client.GetUserInfo(ctx, in, opts...)
|
|
}
|
|
|
|
func (m *defaultHpcAC) GetMemberJobs(ctx context.Context, in *GetMemberJobsReq, opts ...grpc.CallOption) (*GetMemberJobsResp, error) {
|
|
client := hpcAC.NewHpcACClient(m.cli.Conn())
|
|
return client.GetMemberJobs(ctx, in, opts...)
|
|
}
|
|
|
|
// 获取节点资源限额
|
|
func (m *defaultHpcAC) GetNodeResources(ctx context.Context, in *GetNodeResourcesReq, opts ...grpc.CallOption) (*GetNodeResourcesResp, error) {
|
|
client := hpcAC.NewHpcACClient(m.cli.Conn())
|
|
return client.GetNodeResources(ctx, in, opts...)
|
|
}
|