From 3ef39829da75db361716954c5a0f72e05cc2b6a9 Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Fri, 27 May 2022 09:55:20 +0800
Subject: [PATCH] complete connect

---
 src/main/java/cc/mrbird/febs/dapp/service/impl/DappMemberServiceImpl.java |   10 +++++++---
 src/main/java/cc/mrbird/febs/common/utils/LoginUserUtil.java              |    2 +-
 src/main/java/cc/mrbird/febs/common/interceptor/DappInterceptor.java      |    2 +-
 3 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/src/main/java/cc/mrbird/febs/common/interceptor/DappInterceptor.java b/src/main/java/cc/mrbird/febs/common/interceptor/DappInterceptor.java
index 2c4927e..a9df61c 100644
--- a/src/main/java/cc/mrbird/febs/common/interceptor/DappInterceptor.java
+++ b/src/main/java/cc/mrbird/febs/common/interceptor/DappInterceptor.java
@@ -37,7 +37,7 @@
         String headAddress = request.getHeader("address");
         String key = request.getHeader("key");
         String path = request.getServletPath().replace("/dapi", "");
-        log.info("进入拦截:{}, {}", headAddress, path);
+        log.info("进入拦截:{}, {}, {}", headAddress, path, key);
 
         String chain = request.getHeader("chain");
         if (StrUtil.isBlank(headAddress) || StrUtil.isBlank(key)) {
diff --git a/src/main/java/cc/mrbird/febs/common/utils/LoginUserUtil.java b/src/main/java/cc/mrbird/febs/common/utils/LoginUserUtil.java
index d1dbba0..b54a8c5 100644
--- a/src/main/java/cc/mrbird/febs/common/utils/LoginUserUtil.java
+++ b/src/main/java/cc/mrbird/febs/common/utils/LoginUserUtil.java
@@ -22,6 +22,6 @@
         StringBuilder sb = new StringBuilder();
         StringBuilder append = sb.append(address).append(":").append(nonce).append(":").append(sign);
 
-        return append.toString();
+        return Hash.sha3(append.toString());
     }
 }
diff --git a/src/main/java/cc/mrbird/febs/dapp/service/impl/DappMemberServiceImpl.java b/src/main/java/cc/mrbird/febs/dapp/service/impl/DappMemberServiceImpl.java
index 6efb1dc..d891f48 100644
--- a/src/main/java/cc/mrbird/febs/dapp/service/impl/DappMemberServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/dapp/service/impl/DappMemberServiceImpl.java
@@ -130,9 +130,13 @@
         String key = LoginUserUtil.getLoginKey(connectDto.getAddress(), connectDto.getNonce(), connectDto.getSign());
         redisUtils.set(connectDto.getAddress(), member);
 
-        HashMap<String, String> map = new HashMap<>();
-        map.put(connectDto.getAddress(), key);
-        redisUtils.set(AppContants.REDIS_KEY_SIGN, map);
+        Map<Object, Object> keys = redisUtils.hmget(AppContants.REDIS_KEY_SIGN);
+        if (keys == null) {
+            keys = new HashMap<>();
+        }
+
+        keys.put(connectDto.getAddress(), key);
+        redisUtils.set(AppContants.REDIS_KEY_SIGN, keys);
     }
 
     @Override

--
Gitblit v1.9.1