From 29e6848e925fda2b51dc7e6e3ce1ac3e7eb3b1d6 Mon Sep 17 00:00:00 2001 From: Alexander Holodov <alexander.holodov@vereign.com> Date: Tue, 12 Mar 2019 09:59:11 +0300 Subject: [PATCH] 11 fix panic --- handler/handler.go | 15 +++++++++++++-- 1 file changed, 13 insertions(+), 2 deletions(-) diff --git a/handler/handler.go b/handler/handler.go index 5babaa4..c64f5b9 100644 --- a/handler/handler.go +++ b/handler/handler.go @@ -93,7 +93,13 @@ func (s *KeyStorageServerImpl) GetKey(ctx context.Context, in *api.GetKeyRequest return getKeyResponse, nil } - data, _ := client.DoGetDataCall("keys", in.Uuid+"/"+api.KeyType.String(in.KeyType)) + data, err := client.DoGetDataCall("keys", in.Uuid+"/"+api.KeyType.String(in.KeyType)) + if err != nil { + getKeyResponse.Key = nil + getKeyResponse.StatusList = utils.AddStatus(getKeyResponse.StatusList, + "500", api.StatusType_ERROR, err.Error()) + return getKeyResponse, nil + } if data.Errors != "" { getKeyResponse.Key = nil @@ -128,7 +134,12 @@ func (s *KeyStorageServerImpl) SetKey(ctx context.Context, in *api.SetKeyRequest return setKeyResponse, nil } - data, _ := client.DoGetDataCall("keys", in.Uuid+"/"+api.KeyType.String(in.KeyType)) + data, err := client.DoGetDataCall("keys", in.Uuid+"/"+api.KeyType.String(in.KeyType)) + if err != nil { + setKeyResponse.StatusList = utils.AddStatus(setKeyResponse.StatusList, + "500", api.StatusType_ERROR, err.Error()) + return setKeyResponse, nil + } if data.Errors != "" { setKeyResponse.StatusList = utils.AddStatus(setKeyResponse.StatusList, -- GitLab