Adding missing headers

[ealt-edge.git] / mecm / mepm / applcm / broker / pkg / plugin / grpcclient.go

diff --git a/mecm/mepm/applcm/broker/pkg/plugin/grpcclient.go b/mecm/mepm/applcm/broker/pkg/plugin/grpcclient.go
index 27402ed..d55098d 100644 (file)
--- a/mecm/mepm/applcm/broker/pkg/plugin/grpcclient.go
+++ b/mecm/mepm/applcm/broker/pkg/plugin/grpcclient.go
@@ -1,3 +1,19 @@
+/*
+ * Copyright 2020 Huawei Technologies Co., Ltd.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
 package plugin
 
 import (
@@ -68,7 +84,7 @@ func NewClientGRPC(cfg ClientGRPCConfig) (c ClientGRPC, err error) {
        return
 }
 
-func (c *ClientGRPC) Instantiate(ctx context.Context, f string, hostIP string) (workloadId string, status string) {
+func (c *ClientGRPC) Instantiate(ctx context.Context, f string, hostIP string) (workloadId string, status string, error error) {
        var (
                writing = true
                buf     []byte
@@ -142,9 +158,10 @@ func (c *ClientGRPC) Instantiate(ctx context.Context, f string, hostIP string) (
        res, err := stream.CloseAndRecv()
        if err != nil {
                logger.Fatalf("failed to receive upstream status response: ", err)
+               return "", "", err
        }
        log.Printf("response", res)
-       return res.WorkloadId, res.Status
+       return res.WorkloadId, res.Status, err
 }
 
 func (c *ClientGRPC) Query(ctx context.Context, hostIP string, workloadId string) (status string) {