Login with vereign

handler/handler.go

@@ -243,9 +243,15 @@ func (s *KeyStorageServerImpl) GetKeyId(ctx context.Context, in *api.GetKeyIdByK
 		keyID := string(response.Data.Data)
 		keyIdResponse.KeyId = keyID
 		return keyIdResponse, nil
+	} else {
+		if err != nil {
+			return nil, err
+		} else {
+			return nil, errors.New("Can not get public key")
+		}
 	}
 
-	entity, err := clientutils.GetLastEntity(entitiesMagamentClient, auth.Uuid)
+	entity, err := clientutils.GetLastEntity(entitiesMagamentClient, in.EntityUUID)
 
 	if err != nil {
 		return nil, err