multi create resp

Signed-off-by: zhouqunjie <450705171@qq.com>
This commit is contained in:
zhouqunjie 2022-04-13 17:34:31 +08:00
parent 5b1bacb38f
commit 8c234e947c
6 changed files with 24 additions and 16 deletions

View File

@ -17,28 +17,36 @@ import (
func CreatePods(ctx context.Context, req *pbpod.CreatePodsReq) (*pbpod.CreatePodsResp, error) { func CreatePods(ctx context.Context, req *pbpod.CreatePodsReq) (*pbpod.CreatePodsResp, error) {
var ( var (
wg sync.WaitGroup wg sync.WaitGroup
pods []*pbpod.PodInstance requestIds = make([]string, 0)
requestIds = make([]string, len(req.CreatePodReq))
) )
wg.Add(len(req.CreatePodReq)) wg.Add(len(req.CreatePodReq))
c := make(chan string)
for k := range req.CreatePodReq { for k := range req.CreatePodReq {
reqPod := req.CreatePodReq[k] reqPod := req.CreatePodReq[k]
go func() { go func() {
defer wg.Done() defer wg.Done()
resp, err := CreatePod(ctx, reqPod) resp, err := CreatePod(ctx, reqPod)
if err != nil { if err != nil || resp == nil {
fmt.Println(errors.Wrap(err, "Batch pod creation error")) fmt.Println(errors.Wrap(err, "Batch pod creation error"))
return
} }
requestIds[k] = resp.RequestId c <- resp.RequestId
}() }()
} }
wg.Wait() go func() {
defer close(c)
wg.Wait()
}()
isFinished := false isFinished := false
if len(pods) > 0 { if len(requestIds) > 0 {
isFinished = true isFinished = true
} }
for v := range c {
requestIds = append(requestIds, v)
}
return &pbpod.CreatePodsResp{ return &pbpod.CreatePodsResp{
Finished: isFinished, Finished: isFinished,
RequestId: requestIds, RequestId: requestIds,

View File

@ -73,7 +73,7 @@ func (eci *AliEci) CreatePod(ctx context.Context, req *pbpod.CreatePodReq) (*pbp
return &pbpod.CreatePodResp{ return &pbpod.CreatePodResp{
Pods: nil, Pods: nil,
Finished: isFinished, Finished: isFinished,
RequestId: resp.RequestId, RequestId: "ali pod ID:" + resp.ContainerGroupId,
}, nil }, nil
} }

View File

@ -118,9 +118,9 @@ func (cci *HuaweiCci) CreatePod(ctx context.Context, req *pbpod.CreatePodReq) (*
} }
return &pbpod.CreatePodResp{ return &pbpod.CreatePodResp{
Pods: nil, Pods: nil,
Finished: isFinished, Finished: isFinished,
//RequestId: resp.RequestId, RequestId: "huawei pod Name:" + resp.Name,
}, nil }, nil
} }

View File

@ -134,6 +134,7 @@ func (k *K8SPoder) CreatePod(ctx context.Context, req *pbpod.CreatePodReq) (*pbp
} }
resp, err := k.cli.CoreV1().Pods(req.Namespace).Create(&pod) resp, err := k.cli.CoreV1().Pods(req.Namespace).Create(&pod)
glog.Info("K8S create pod resp", resp) glog.Info("K8S create pod resp", resp)
if err != nil { if err != nil {
return nil, errors.Wrap(err, "K8S CreatePod error") return nil, errors.Wrap(err, "K8S CreatePod error")
@ -145,9 +146,9 @@ func (k *K8SPoder) CreatePod(ctx context.Context, req *pbpod.CreatePodReq) (*pbp
} }
return &pbpod.CreatePodResp{ return &pbpod.CreatePodResp{
Pods: nil, Pods: nil,
Finished: isFinished, Finished: isFinished,
//RequestId: resp.RequestId, RequestId: "K8S pod Name:" + resp.Name,
}, nil }, nil
} }
@ -162,7 +163,7 @@ func (k K8SPoder) DeletePod(ctx context.Context, req *pbpod.DeletePodReq) (*pbpo
isFinished := true isFinished := true
if err != nil { if err != nil {
isFinished = false isFinished = false
return nil, errors.Wrap(err, "Huaweiyun DeletePod error") return nil, errors.Wrap(err, "K8S DeletePod error")
} }
return &pbpod.DeletePodResp{ return &pbpod.DeletePodResp{

View File

@ -149,12 +149,11 @@ func (eks *TencentEks) UpdatePod(ctx context.Context, req *pbpod.UpdatePodReq) (
glog.Errorf("Tencent UpdatePod error: %v", err) glog.Errorf("Tencent UpdatePod error: %v", err)
return nil, errors.Wrap(err, "Tencent UpdatePod error") return nil, errors.Wrap(err, "Tencent UpdatePod error")
} }
requestId := resp.Response.RequestId
return &pbpod.UpdatePodResp{ return &pbpod.UpdatePodResp{
Pod: nil, Pod: nil,
Finished: isFinished, Finished: isFinished,
RequestId: *requestId, RequestId: "tencent pod ID:" + *resp.Response.EksCiId,
}, nil }, nil
} }