You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

storeLink.go 24 kB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856
  1. /*
  2. Copyright (c) [2023] [pcm]
  3. [pcm-coordinator] is licensed under Mulan PSL v2.
  4. You can use this software according to the terms and conditions of the Mulan PSL v2.
  5. You may obtain a copy of Mulan PSL v2 at:
  6. http://license.coscl.org.cn/MulanPSL2
  7. THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
  8. EITHER EXPaRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
  9. MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
  10. See the Mulan PSL v2 for more details.
  11. */
  12. package storeLink
  13. import (
  14. "context"
  15. "github.com/pkg/errors"
  16. "gitlink.org.cn/jcce-pcm/pcm-ac/hpcAC"
  17. "gitlink.org.cn/jcce-pcm/pcm-coordinator/api/internal/svc"
  18. "gitlink.org.cn/jcce-pcm/pcm-coordinator/api/internal/types"
  19. "gitlink.org.cn/jcce-pcm/pcm-coordinator/pkg/models"
  20. "gitlink.org.cn/jcce-pcm/pcm-coordinator/pkg/utils/timeutils"
  21. "gitlink.org.cn/jcce-pcm/pcm-participant-modelarts/modelarts"
  22. "gitlink.org.cn/jcce-pcm/pcm-participant-octopus/octopus"
  23. "gorm.io/gorm"
  24. )
  25. type Linkage interface {
  26. UploadImage(path string) (interface{}, error)
  27. DeleteImage(imageId string) (interface{}, error)
  28. QueryImageList() (interface{}, error)
  29. SubmitTask(imageId string, cmd string, envs []string, params []string, resourceId string) (interface{}, error)
  30. QueryTask(taskId string) (interface{}, error)
  31. QuerySpecs() (interface{}, error)
  32. DeleteTask(taskId string) (interface{}, error)
  33. }
  34. const (
  35. PY_PARAM_PREFIX = "--"
  36. SPACE = " "
  37. UNDERSCORE = "_"
  38. EQUAL = "="
  39. COMMA = ","
  40. TYPE_OCTOPUS = "1"
  41. TYPE_MODELARTS = "2"
  42. TYPE_SHUGUANGAI = "3"
  43. TYPE_SHUGUANGHPC = "4"
  44. OCTOPUS = "Octopus"
  45. MODELARTS = "Modelarts"
  46. SHUGUANGAI = "ShuguangAi"
  47. SHUGUANGHPC = "ShuguangHpc"
  48. )
  49. var (
  50. OctImgStatus = map[int32]string{
  51. 1: "未上传",
  52. 3: "制作完成",
  53. 4: "制作失败",
  54. }
  55. AITYPE = map[string]string{
  56. "1": OCTOPUS,
  57. "2": MODELARTS,
  58. "3": SHUGUANGAI,
  59. "4": SHUGUANGHPC,
  60. }
  61. ERROR_RESP_EMPTY = errors.New("resp empty error")
  62. ERROR_CONVERT_EMPTY = errors.New("convert empty error")
  63. )
  64. type StoreLink struct {
  65. ILinkage Linkage
  66. }
  67. func NewStoreLink(ctx context.Context, svcCtx *svc.ServiceContext, participant *models.StorelinkCenter) *StoreLink {
  68. switch participant.Type {
  69. case TYPE_OCTOPUS:
  70. linkStruct := NewOctopusLink(ctx, svcCtx, participant.Name, participant.Id)
  71. return &StoreLink{ILinkage: linkStruct}
  72. case TYPE_MODELARTS:
  73. linkStruct := NewModelArtsLink(ctx, svcCtx, participant.Name, participant.Id)
  74. return &StoreLink{ILinkage: linkStruct}
  75. case TYPE_SHUGUANGAI:
  76. linkStruct := NewShuguangAi(ctx, svcCtx, participant.Name, participant.Id)
  77. return &StoreLink{ILinkage: linkStruct}
  78. case TYPE_SHUGUANGHPC:
  79. linkStruct := NewShuguangHpc(ctx, svcCtx, participant.Name, participant.Id)
  80. return &StoreLink{ILinkage: linkStruct}
  81. default:
  82. return nil
  83. }
  84. }
  85. func GetParticipants(dbEngin *gorm.DB) []*models.StorelinkCenter {
  86. var participants []*models.StorelinkCenter
  87. dbEngin.Raw("select * from storelink_center").Scan(&participants)
  88. return participants
  89. }
  90. func GetParticipantById(partId int64, dbEngin *gorm.DB) *models.StorelinkCenter {
  91. var participant models.StorelinkCenter
  92. dbEngin.Raw("select * from storelink_center where id = ?", partId).Scan(&participant)
  93. return &participant
  94. }
  95. func ConvertType(in interface{}, out interface{}, participant *models.StorelinkCenter) (interface{}, error) {
  96. switch (interface{})(in).(type) {
  97. case *octopus.UploadImageResp:
  98. inresp := (interface{})(in).(*octopus.UploadImageResp)
  99. switch (interface{})(out).(type) {
  100. case *types.UploadLinkImageResp:
  101. resp := (interface{})(out).(*types.UploadLinkImageResp)
  102. resp.Success = inresp.Success
  103. if !resp.Success {
  104. resp.ErrorMsg = inresp.Error.Message
  105. return resp, nil
  106. }
  107. return resp, nil
  108. }
  109. return nil, nil
  110. case *octopus.DeleteImageResp:
  111. inresp := (interface{})(in).(*octopus.DeleteImageResp)
  112. switch (interface{})(out).(type) {
  113. case *types.DeleteLinkImageResp:
  114. resp := (interface{})(out).(*types.DeleteLinkImageResp)
  115. resp.Success = inresp.Success
  116. if !resp.Success {
  117. resp.ErrorMsg = inresp.Error.Message
  118. return resp, nil
  119. }
  120. return resp, nil
  121. }
  122. return nil, nil
  123. case *octopus.GetUserImageListResp:
  124. inresp := (interface{})(in).(*octopus.GetUserImageListResp)
  125. switch (interface{})(out).(type) {
  126. case *types.GetLinkImageListResp:
  127. resp := (interface{})(out).(*types.GetLinkImageListResp)
  128. resp.Success = inresp.Success
  129. if !resp.Success {
  130. resp.ErrorMsg = inresp.Error.Message
  131. resp.Images = nil
  132. return resp, nil
  133. }
  134. for _, v := range inresp.Payload.Images {
  135. var image types.ImageSl
  136. image.ImageId = v.Image.Id
  137. image.ImageName = v.Image.ImageName
  138. image.ImageStatus = OctImgStatus[v.Image.ImageStatus]
  139. resp.Images = append(resp.Images, &image)
  140. }
  141. return resp, nil
  142. }
  143. return nil, nil
  144. case *modelarts.ListReposDetailsResp:
  145. inresp := (interface{})(in).(*modelarts.ListReposDetailsResp)
  146. switch (interface{})(out).(type) {
  147. case *types.GetLinkImageListResp:
  148. resp := (interface{})(out).(*types.GetLinkImageListResp)
  149. if inresp.Errors != nil {
  150. resp.Success = false
  151. resp.ErrorMsg = inresp.Errors[0].ErrorMessage
  152. resp.Images = nil
  153. return resp, nil
  154. }
  155. resp.Success = true
  156. for _, v := range inresp.Items {
  157. for _, r := range v.Tags {
  158. var image types.ImageSl
  159. image.ImageId = v.Namespace + "/" + v.Name + ":" + r
  160. image.ImageName = v.Name
  161. image.ImageStatus = "created"
  162. resp.Images = append(resp.Images, &image)
  163. }
  164. }
  165. return resp, nil
  166. }
  167. return nil, nil
  168. case *hpcAC.GetImageListAiResp:
  169. inresp := (interface{})(in).(*hpcAC.GetImageListAiResp)
  170. switch (interface{})(out).(type) {
  171. case *types.GetLinkImageListResp:
  172. resp := (interface{})(out).(*types.GetLinkImageListResp)
  173. if inresp.Code == "0" {
  174. resp.Success = true
  175. for _, img := range inresp.Data {
  176. var image types.ImageSl
  177. image.ImageId = img.ImageId
  178. image.ImageName = img.Version
  179. image.ImageStatus = "created"
  180. resp.Images = append(resp.Images, &image)
  181. }
  182. } else {
  183. resp.Success = false
  184. resp.ErrorMsg = inresp.Msg
  185. resp.Images = nil
  186. }
  187. return resp, nil
  188. }
  189. return nil, nil
  190. case *octopus.CreateTrainJobResp:
  191. inresp := (interface{})(in).(*octopus.CreateTrainJobResp)
  192. switch (interface{})(out).(type) {
  193. case *types.SubmitLinkTaskResp:
  194. resp := (interface{})(out).(*types.SubmitLinkTaskResp)
  195. resp.Success = inresp.Success
  196. if !resp.Success {
  197. resp.ErrorMsg = inresp.Error.Message
  198. return resp, nil
  199. }
  200. resp.TaskId = inresp.Payload.JobId
  201. return resp, nil
  202. }
  203. return nil, nil
  204. case *modelarts.CreateTrainingJobResp:
  205. inresp := (interface{})(in).(*modelarts.CreateTrainingJobResp)
  206. switch (interface{})(out).(type) {
  207. case *types.SubmitLinkTaskResp:
  208. resp := (interface{})(out).(*types.SubmitLinkTaskResp)
  209. if inresp.ErrorMsg != "" {
  210. resp.ErrorMsg = inresp.ErrorMsg
  211. resp.Success = false
  212. return resp, nil
  213. }
  214. resp.Success = true
  215. resp.TaskId = inresp.Metadata.Id
  216. return resp, nil
  217. }
  218. return nil, nil
  219. case *hpcAC.SubmitTaskAiResp:
  220. inresp := (interface{})(in).(*hpcAC.SubmitTaskAiResp)
  221. switch (interface{})(out).(type) {
  222. case *types.SubmitLinkTaskResp:
  223. resp := (interface{})(out).(*types.SubmitLinkTaskResp)
  224. if inresp.Code == "0" {
  225. resp.Success = true
  226. resp.TaskId = inresp.Data
  227. } else {
  228. resp.Success = false
  229. resp.ErrorMsg = inresp.Msg
  230. }
  231. return resp, nil
  232. }
  233. return nil, nil
  234. case *hpcAC.SubmitJobResp:
  235. inresp := (interface{})(in).(*hpcAC.SubmitJobResp)
  236. switch (interface{})(out).(type) {
  237. case *types.SubmitLinkTaskResp:
  238. resp := (interface{})(out).(*types.SubmitLinkTaskResp)
  239. if inresp.Code == "0" {
  240. resp.Success = true
  241. resp.TaskId = inresp.Data
  242. } else {
  243. resp.Success = false
  244. resp.ErrorMsg = inresp.Msg
  245. }
  246. return resp, nil
  247. }
  248. return nil, nil
  249. case *octopus.GetTrainJobResp:
  250. inresp := (interface{})(in).(*octopus.GetTrainJobResp)
  251. switch (interface{})(out).(type) {
  252. case *types.GetLinkTaskResp:
  253. resp := (interface{})(out).(*types.GetLinkTaskResp)
  254. resp.Success = inresp.Success
  255. if !resp.Success {
  256. resp.ErrorMsg = inresp.Error.Message
  257. return resp, nil
  258. }
  259. var task types.TaskSl
  260. task.TaskId = inresp.Payload.TrainJob.Id
  261. task.TaskName = inresp.Payload.TrainJob.Name
  262. task.StartedAt = inresp.Payload.TrainJob.StartedAt
  263. task.CompletedAt = inresp.Payload.TrainJob.CompletedAt
  264. task.TaskStatus = inresp.Payload.TrainJob.Status
  265. resp.Task = &task
  266. return resp, nil
  267. }
  268. return nil, nil
  269. case *modelarts.JobResponse:
  270. inresp := (interface{})(in).(*modelarts.JobResponse)
  271. switch (interface{})(out).(type) {
  272. case *types.GetLinkTaskResp:
  273. resp := (interface{})(out).(*types.GetLinkTaskResp)
  274. if inresp.ErrorMsg != "" {
  275. resp.ErrorMsg = inresp.ErrorMsg
  276. resp.Success = false
  277. return resp, nil
  278. }
  279. resp.Success = true
  280. resp.Task = &types.TaskSl{}
  281. resp.Task.TaskId = inresp.Metadata.Id
  282. resp.Task.TaskName = inresp.Metadata.Name
  283. resp.Task.StartedAt = int64(inresp.Status.StartTime)
  284. resp.Task.CompletedAt = int64(inresp.Status.Duration)
  285. resp.Task.TaskStatus = inresp.Status.Phase
  286. return resp, nil
  287. }
  288. return nil, nil
  289. case *hpcAC.GetPytorchTaskResp:
  290. inresp := (interface{})(in).(*hpcAC.GetPytorchTaskResp)
  291. switch (interface{})(out).(type) {
  292. case *types.GetLinkTaskResp:
  293. resp := (interface{})(out).(*types.GetLinkTaskResp)
  294. if inresp.Code == "0" {
  295. resp.Success = true
  296. var task types.TaskSl
  297. task.TaskId = inresp.Data.Id
  298. task.TaskName = inresp.Data.TaskName
  299. task.TaskStatus = inresp.Data.Status
  300. task.StartedAt = timeutils.StringToUnixTime(inresp.Data.StartTime)
  301. task.CompletedAt = timeutils.StringToUnixTime(inresp.Data.EndTime)
  302. resp.Task = &task
  303. } else {
  304. resp.Success = false
  305. resp.ErrorMsg = inresp.Msg
  306. resp.Task = nil
  307. }
  308. return resp, nil
  309. }
  310. return nil, nil
  311. case *hpcAC.GetJobDetailResp:
  312. inresp := (interface{})(in).(*hpcAC.GetJobDetailResp)
  313. switch (interface{})(out).(type) {
  314. case *types.GetLinkTaskResp:
  315. resp := (interface{})(out).(*types.GetLinkTaskResp)
  316. if inresp.Code == "0" {
  317. resp.Success = true
  318. var task types.TaskSl
  319. task.TaskId = inresp.Data.JobId
  320. task.TaskName = inresp.Data.JobName
  321. task.TaskStatus = AcStatus[inresp.Data.JobStatus]
  322. task.StartedAt = timeutils.StringToUnixTime(inresp.Data.JobStartTime)
  323. task.CompletedAt = timeutils.StringToUnixTime(inresp.Data.JobEndTime)
  324. resp.Task = &task
  325. } else {
  326. resp.Success = false
  327. resp.ErrorMsg = inresp.Msg
  328. resp.Task = nil
  329. }
  330. return resp, nil
  331. }
  332. return nil, nil
  333. case *hpcAC.HistoryJobDetailResp:
  334. inresp := (interface{})(in).(*hpcAC.HistoryJobDetailResp)
  335. switch (interface{})(out).(type) {
  336. case *types.GetLinkTaskResp:
  337. resp := (interface{})(out).(*types.GetLinkTaskResp)
  338. if inresp.Code == "0" {
  339. resp.Success = true
  340. var task types.TaskSl
  341. task.TaskId = inresp.Data.JobId
  342. task.TaskName = inresp.Data.JobName
  343. task.TaskStatus = AcStatus[inresp.Data.JobState]
  344. task.StartedAt = timeutils.StringToUnixTime(inresp.Data.JobStartTime)
  345. task.CompletedAt = timeutils.StringToUnixTime(inresp.Data.JobEndTime)
  346. resp.Task = &task
  347. } else {
  348. resp.Success = false
  349. resp.ErrorMsg = inresp.Msg
  350. resp.Task = nil
  351. }
  352. return resp, nil
  353. }
  354. return nil, nil
  355. case *octopus.DeleteTrainJobResp:
  356. inresp := (interface{})(in).(*octopus.DeleteTrainJobResp)
  357. switch (interface{})(out).(type) {
  358. case *types.DeleteLinkTaskResp:
  359. resp := (interface{})(out).(*types.DeleteLinkTaskResp)
  360. resp.Success = inresp.Success
  361. if !resp.Success {
  362. resp.ErrorMsg = inresp.Error.Message
  363. return resp, nil
  364. }
  365. return resp, nil
  366. }
  367. return nil, nil
  368. case *modelarts.DeleteTrainingJobResp:
  369. inresp := (interface{})(in).(*modelarts.DeleteTrainingJobResp)
  370. switch (interface{})(out).(type) {
  371. case *types.DeleteLinkTaskResp:
  372. resp := (interface{})(out).(*types.DeleteLinkTaskResp)
  373. if inresp.ErrorMsg != "" {
  374. resp.ErrorMsg = inresp.ErrorMsg
  375. resp.Success = false
  376. return resp, nil
  377. }
  378. resp.Success = true
  379. return resp, nil
  380. }
  381. return nil, nil
  382. case *hpcAC.DeleteTaskAiResp:
  383. inresp := (interface{})(in).(*hpcAC.DeleteTaskAiResp)
  384. switch (interface{})(out).(type) {
  385. case *types.DeleteLinkTaskResp:
  386. resp := (interface{})(out).(*types.DeleteLinkTaskResp)
  387. if inresp.Code == "0" {
  388. resp.Success = true
  389. } else {
  390. resp.Success = false
  391. resp.ErrorMsg = inresp.Msg
  392. }
  393. return resp, nil
  394. }
  395. return nil, nil
  396. case *hpcAC.DeleteJobResp:
  397. inresp := (interface{})(in).(*hpcAC.DeleteJobResp)
  398. switch (interface{})(out).(type) {
  399. case *types.DeleteLinkTaskResp:
  400. resp := (interface{})(out).(*types.DeleteLinkTaskResp)
  401. if inresp.Code == "0" {
  402. resp.Success = true
  403. } else {
  404. resp.Success = false
  405. resp.ErrorMsg = inresp.Msg
  406. }
  407. return resp, nil
  408. }
  409. return nil, nil
  410. case *octopus.GetResourceSpecsResp:
  411. inresp := (interface{})(in).(*octopus.GetResourceSpecsResp)
  412. switch (interface{})(out).(type) {
  413. case *types.GetResourceSpecsResp:
  414. resp := (interface{})(out).(*types.GetResourceSpecsResp)
  415. resp.Success = inresp.Success
  416. if !resp.Success {
  417. resp.ResourceSpecs = nil
  418. return resp, nil
  419. }
  420. for _, spec := range inresp.TrainResourceSpecs {
  421. var respec types.ResourceSpecSl
  422. respec.SpecId = spec.Id
  423. respec.SpecName = spec.Name
  424. respec.ParticipantId = participant.Id
  425. respec.ParticipantName = participant.Name
  426. respec.SpecPrice = spec.Price
  427. resp.ResourceSpecs = append(resp.ResourceSpecs, &respec)
  428. }
  429. return resp, nil
  430. }
  431. return nil, nil
  432. case *hpcAC.GetResourceSpecResp:
  433. inresp := (interface{})(in).(*hpcAC.GetResourceSpecResp)
  434. switch (interface{})(out).(type) {
  435. case *types.GetResourceSpecsResp:
  436. resp := (interface{})(out).(*types.GetResourceSpecsResp)
  437. if inresp.Code != "0" {
  438. resp.Success = false
  439. resp.ResourceSpecs = nil
  440. } else {
  441. var spec types.ResourceSpecSl
  442. resp.Success = true
  443. spec.ParticipantName = participant.Name
  444. spec.ParticipantId = participant.Id
  445. spec.SpecName = SHUGUANGAI_CUSTOM_RESOURCE_NAME
  446. spec.SpecId = SHUGUANGAI_CUSTOM_RESOURCE_ID
  447. resp.ResourceSpecs = append(resp.ResourceSpecs, &spec)
  448. }
  449. return resp, nil
  450. }
  451. return nil, nil
  452. case *modelarts.TrainingJobFlavorsResp:
  453. inresp := (interface{})(in).(*modelarts.TrainingJobFlavorsResp)
  454. switch (interface{})(out).(type) {
  455. case *types.GetResourceSpecsResp:
  456. resp := (interface{})(out).(*types.GetResourceSpecsResp)
  457. resp.Success = true
  458. if inresp.Flavors == nil {
  459. resp.Success = false
  460. resp.ResourceSpecs = nil
  461. return resp, nil
  462. }
  463. for _, spec := range inresp.Flavors {
  464. var respec types.ResourceSpecSl
  465. respec.SpecId = spec.FlavorId
  466. respec.SpecName = spec.FlavorName
  467. respec.ParticipantId = participant.Id
  468. respec.ParticipantName = participant.Name
  469. respec.SpecPrice = 0
  470. resp.ResourceSpecs = append(resp.ResourceSpecs, &respec)
  471. }
  472. return resp, nil
  473. }
  474. return nil, nil
  475. default:
  476. return nil, errors.New("type convert fail")
  477. }
  478. }
  479. func ConvertTypeOld[T any](in *T, participant *models.StorelinkCenter) (interface{}, error) {
  480. switch (interface{})(in).(type) {
  481. case *octopus.UploadImageResp:
  482. var resp types.UploadLinkImageResp
  483. inresp := (interface{})(in).(*octopus.UploadImageResp)
  484. resp.Success = inresp.Success
  485. if !resp.Success {
  486. resp.ErrorMsg = inresp.Error.Message
  487. return resp, nil
  488. }
  489. return resp, nil
  490. case *octopus.DeleteImageResp:
  491. var resp types.DeleteLinkImageResp
  492. inresp := (interface{})(in).(*octopus.DeleteImageResp)
  493. resp.Success = inresp.Success
  494. if !resp.Success {
  495. resp.ErrorMsg = inresp.Error.Message
  496. return resp, nil
  497. }
  498. return resp, nil
  499. case *octopus.GetUserImageListResp:
  500. var resp types.GetLinkImageListResp
  501. inresp := (interface{})(in).(*octopus.GetUserImageListResp)
  502. resp.Success = inresp.Success
  503. if !resp.Success {
  504. resp.ErrorMsg = inresp.Error.Message
  505. resp.Images = nil
  506. return resp, nil
  507. }
  508. for _, v := range inresp.Payload.Images {
  509. var image types.ImageSl
  510. image.ImageId = v.Image.Id
  511. image.ImageName = v.Image.ImageName
  512. image.ImageStatus = OctImgStatus[v.Image.ImageStatus]
  513. resp.Images = append(resp.Images, &image)
  514. }
  515. return resp, nil
  516. case *modelarts.ListReposDetailsResp:
  517. var resp types.GetLinkImageListResp
  518. inresp := (interface{})(in).(*modelarts.ListReposDetailsResp)
  519. if inresp.Errors != nil {
  520. resp.Success = false
  521. resp.ErrorMsg = inresp.Errors[0].ErrorMessage
  522. resp.Images = nil
  523. return resp, nil
  524. }
  525. resp.Success = true
  526. for _, v := range inresp.Items {
  527. for _, r := range v.Tags {
  528. var image types.ImageSl
  529. image.ImageId = v.Namespace + "/" + v.Name + ":" + r
  530. image.ImageName = v.Name
  531. image.ImageStatus = "created"
  532. resp.Images = append(resp.Images, &image)
  533. }
  534. }
  535. return resp, nil
  536. case *hpcAC.GetImageListAiResp:
  537. var resp types.GetLinkImageListResp
  538. inresp := (interface{})(in).(*hpcAC.GetImageListAiResp)
  539. if inresp.Code == "0" {
  540. resp.Success = true
  541. for _, img := range inresp.Data {
  542. var image types.ImageSl
  543. image.ImageId = img.ImageId
  544. image.ImageName = img.Version
  545. image.ImageStatus = "created"
  546. resp.Images = append(resp.Images, &image)
  547. }
  548. } else {
  549. resp.Success = false
  550. resp.ErrorMsg = inresp.Msg
  551. resp.Images = nil
  552. }
  553. return resp, nil
  554. case *octopus.CreateTrainJobResp:
  555. var resp types.SubmitLinkTaskResp
  556. inresp := (interface{})(in).(*octopus.CreateTrainJobResp)
  557. resp.Success = inresp.Success
  558. if !resp.Success {
  559. resp.ErrorMsg = inresp.Error.Message
  560. return resp, nil
  561. }
  562. resp.TaskId = inresp.Payload.JobId
  563. return resp, nil
  564. case *modelarts.CreateTrainingJobResp:
  565. var resp types.SubmitLinkTaskResp
  566. inresp := (interface{})(in).(*modelarts.CreateTrainingJobResp)
  567. if inresp.ErrorMsg != "" {
  568. resp.ErrorMsg = inresp.ErrorMsg
  569. resp.Success = false
  570. return resp, nil
  571. }
  572. resp.Success = true
  573. resp.TaskId = inresp.Metadata.Id
  574. return resp, nil
  575. case *hpcAC.SubmitTaskAiResp:
  576. var resp types.SubmitLinkTaskResp
  577. inresp := (interface{})(in).(*hpcAC.SubmitTaskAiResp)
  578. if inresp.Code == "0" {
  579. resp.Success = true
  580. resp.TaskId = inresp.Data
  581. } else {
  582. resp.Success = false
  583. resp.ErrorMsg = inresp.Msg
  584. }
  585. return resp, nil
  586. case *hpcAC.SubmitJobResp:
  587. var resp types.SubmitLinkTaskResp
  588. inresp := (interface{})(in).(*hpcAC.SubmitJobResp)
  589. if inresp.Code == "0" {
  590. resp.Success = true
  591. resp.TaskId = inresp.Data
  592. } else {
  593. resp.Success = false
  594. resp.ErrorMsg = inresp.Msg
  595. }
  596. return resp, nil
  597. case *octopus.GetTrainJobResp:
  598. var resp types.GetLinkTaskResp
  599. inresp := (interface{})(in).(*octopus.GetTrainJobResp)
  600. resp.Success = inresp.Success
  601. if !resp.Success {
  602. resp.ErrorMsg = inresp.Error.Message
  603. return resp, nil
  604. }
  605. var task types.TaskSl
  606. task.TaskId = inresp.Payload.TrainJob.Id
  607. task.TaskName = inresp.Payload.TrainJob.Name
  608. task.StartedAt = inresp.Payload.TrainJob.StartedAt
  609. task.CompletedAt = inresp.Payload.TrainJob.CompletedAt
  610. task.TaskStatus = inresp.Payload.TrainJob.Status
  611. resp.Task = &task
  612. return resp, nil
  613. case *modelarts.JobResponse:
  614. var resp types.GetLinkTaskResp
  615. inresp := (interface{})(in).(*modelarts.JobResponse)
  616. if inresp.ErrorMsg != "" {
  617. resp.ErrorMsg = inresp.ErrorMsg
  618. resp.Success = false
  619. return resp, nil
  620. }
  621. resp.Success = true
  622. resp.Task = &types.TaskSl{}
  623. resp.Task.TaskId = inresp.Metadata.Id
  624. resp.Task.TaskName = inresp.Metadata.Name
  625. resp.Task.StartedAt = int64(inresp.Status.StartTime)
  626. resp.Task.CompletedAt = int64(inresp.Status.Duration)
  627. resp.Task.TaskStatus = inresp.Status.Phase
  628. return resp, nil
  629. case *hpcAC.GetPytorchTaskResp:
  630. var resp types.GetLinkTaskResp
  631. inresp := (interface{})(in).(*hpcAC.GetPytorchTaskResp)
  632. if inresp.Code == "0" {
  633. resp.Success = true
  634. var task types.TaskSl
  635. task.TaskId = inresp.Data.Id
  636. task.TaskName = inresp.Data.TaskName
  637. task.TaskStatus = inresp.Data.Status
  638. task.StartedAt = timeutils.StringToUnixTime(inresp.Data.StartTime)
  639. task.CompletedAt = timeutils.StringToUnixTime(inresp.Data.EndTime)
  640. resp.Task = &task
  641. } else {
  642. resp.Success = false
  643. resp.ErrorMsg = inresp.Msg
  644. resp.Task = nil
  645. }
  646. return resp, nil
  647. case *hpcAC.GetJobDetailResp:
  648. var resp types.GetLinkTaskResp
  649. inresp := (interface{})(in).(*hpcAC.GetJobDetailResp)
  650. if inresp.Code == "0" {
  651. resp.Success = true
  652. var task types.TaskSl
  653. task.TaskId = inresp.Data.JobId
  654. task.TaskName = inresp.Data.JobName
  655. task.TaskStatus = AcStatus[inresp.Data.JobStatus]
  656. task.StartedAt = timeutils.StringToUnixTime(inresp.Data.JobStartTime)
  657. task.CompletedAt = timeutils.StringToUnixTime(inresp.Data.JobEndTime)
  658. resp.Task = &task
  659. } else {
  660. resp.Success = false
  661. resp.ErrorMsg = inresp.Msg
  662. resp.Task = nil
  663. }
  664. return resp, nil
  665. case *hpcAC.HistoryJobDetailResp:
  666. var resp types.GetLinkTaskResp
  667. inresp := (interface{})(in).(*hpcAC.HistoryJobDetailResp)
  668. if inresp.Code == "0" {
  669. resp.Success = true
  670. var task types.TaskSl
  671. task.TaskId = inresp.Data.JobId
  672. task.TaskName = inresp.Data.JobName
  673. task.TaskStatus = AcStatus[inresp.Data.JobState]
  674. task.StartedAt = timeutils.StringToUnixTime(inresp.Data.JobStartTime)
  675. task.CompletedAt = timeutils.StringToUnixTime(inresp.Data.JobEndTime)
  676. resp.Task = &task
  677. } else {
  678. resp.Success = false
  679. resp.ErrorMsg = inresp.Msg
  680. resp.Task = nil
  681. }
  682. return resp, nil
  683. case *octopus.DeleteTrainJobResp:
  684. var resp types.DeleteLinkTaskResp
  685. inresp := (interface{})(in).(*octopus.DeleteTrainJobResp)
  686. resp.Success = inresp.Success
  687. if !resp.Success {
  688. resp.ErrorMsg = inresp.Error.Message
  689. return resp, nil
  690. }
  691. return resp, nil
  692. case *modelarts.DeleteTrainingJobResp:
  693. var resp types.DeleteLinkTaskResp
  694. inresp := (interface{})(in).(*modelarts.DeleteTrainingJobResp)
  695. if inresp.ErrorMsg != "" {
  696. resp.ErrorMsg = inresp.ErrorMsg
  697. resp.Success = false
  698. return resp, nil
  699. }
  700. resp.Success = true
  701. return resp, nil
  702. case *hpcAC.DeleteTaskAiResp:
  703. var resp types.DeleteLinkTaskResp
  704. inresp := (interface{})(in).(*hpcAC.DeleteTaskAiResp)
  705. if inresp.Code == "0" {
  706. resp.Success = true
  707. } else {
  708. resp.Success = false
  709. resp.ErrorMsg = inresp.Msg
  710. }
  711. return resp, nil
  712. case *hpcAC.DeleteJobResp:
  713. var resp types.DeleteLinkTaskResp
  714. inresp := (interface{})(in).(*hpcAC.DeleteJobResp)
  715. if inresp.Code == "0" {
  716. resp.Success = true
  717. } else {
  718. resp.Success = false
  719. resp.ErrorMsg = inresp.Msg
  720. }
  721. return resp, nil
  722. case *octopus.GetResourceSpecsResp:
  723. var resp types.GetResourceSpecsResp
  724. inresp := (interface{})(in).(*octopus.GetResourceSpecsResp)
  725. resp.Success = inresp.Success
  726. if !resp.Success {
  727. resp.ResourceSpecs = nil
  728. return resp, nil
  729. }
  730. for _, spec := range inresp.TrainResourceSpecs {
  731. var respec types.ResourceSpecSl
  732. respec.SpecId = spec.Id
  733. respec.SpecName = spec.Name
  734. respec.ParticipantId = participant.Id
  735. respec.ParticipantName = participant.Name
  736. respec.SpecPrice = spec.Price
  737. resp.ResourceSpecs = append(resp.ResourceSpecs, &respec)
  738. }
  739. return resp, nil
  740. case *hpcAC.GetResourceSpecResp:
  741. var resp types.GetResourceSpecsResp
  742. inresp := (interface{})(in).(*hpcAC.GetResourceSpecResp)
  743. if inresp.Code != "0" {
  744. resp.Success = false
  745. resp.ResourceSpecs = nil
  746. } else {
  747. var spec types.ResourceSpecSl
  748. resp.Success = true
  749. spec.ParticipantName = participant.Name
  750. spec.ParticipantId = participant.Id
  751. spec.SpecName = SHUGUANGAI_CUSTOM_RESOURCE_NAME
  752. spec.SpecId = SHUGUANGAI_CUSTOM_RESOURCE_ID
  753. resp.ResourceSpecs = append(resp.ResourceSpecs, &spec)
  754. }
  755. return resp, nil
  756. case *modelarts.TrainingJobFlavorsResp:
  757. var resp types.GetResourceSpecsResp
  758. resp.Success = true
  759. inresp := (interface{})(in).(*modelarts.TrainingJobFlavorsResp)
  760. if inresp.Flavors == nil {
  761. resp.Success = false
  762. resp.ResourceSpecs = nil
  763. return resp, nil
  764. }
  765. for _, spec := range inresp.Flavors {
  766. var respec types.ResourceSpecSl
  767. respec.SpecId = spec.FlavorId
  768. respec.SpecName = spec.FlavorName
  769. respec.ParticipantId = participant.Id
  770. respec.ParticipantName = participant.Name
  771. respec.SpecPrice = 0
  772. resp.ResourceSpecs = append(resp.ResourceSpecs, &respec)
  773. }
  774. return resp, nil
  775. default:
  776. return nil, errors.New("type convert fail")
  777. }
  778. }

PCM is positioned as Software stack over Cloud, aiming to build the standards and ecology of heterogeneous cloud collaboration for JCC in a non intrusive and autonomous peer-to-peer manner.