From ef7fd5b97a125df1d90990239fd5c3037bf7df03 Mon Sep 17 00:00:00 2001
From: Olgun Cengiz <olgun.cengiz@vereign.com>
Date: Thu, 26 Sep 2019 16:23:04 +0300
Subject: [PATCH] replaced GetData with GetString

---
 handler/handler.go | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

diff --git a/handler/handler.go b/handler/handler.go
index 14e1963..3032b53 100644
--- a/handler/handler.go
+++ b/handler/handler.go
@@ -28,7 +28,6 @@ import (
 	"code.vereign.com/code/viam-apis/authentication"
 	"code.vereign.com/code/viam-apis/key-storage-agent/api"
 	"code.vereign.com/code/viam-apis/utils"
-	"code.vereign.com/code/viam-apis/system"
 	"golang.org/x/net/context"
 	"google.golang.org/grpc/metadata"
 )
@@ -237,11 +236,9 @@ func (s *KeyStorageServerImpl) GetKeyId(ctx context.Context, in *api.GetKeyIdByK
 
 	keyIdResponse := &api.GetKeyIdByKeyResponse{}
 
-	stringMessage := &system.String{}
-	
-	hasData, errorsString, err := dataStorageClient.GetData(keyToKeyIdTable, in.PublicKey, stringMessage)
-	if err == nil && errorsString == "" && hasData {
-		keyIdResponse.KeyId = stringMessage.Value
+	stringMessage, errorsString, err := dataStorageClient.GetString(keyToKeyIdTable, in.PublicKey)
+	if err == nil && errorsString == "" {
+		keyIdResponse.KeyId = stringMessage
 		return keyIdResponse, nil
 	} else {
 		if err != nil {
-- 
GitLab