From 4c89be40d38fac27f321dcb0c1c896cf6b5992c7 Mon Sep 17 00:00:00 2001
From: xiaoyong931011 <15274802129@163.com>
Date: Sun, 30 Oct 2022 00:16:27 +0800
Subject: [PATCH] 20221021

---
 src/main/java/cc/mrbird/febs/common/utils/LoginUserUtil.java |   16 +++++++++++++---
 1 files changed, 13 insertions(+), 3 deletions(-)

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 6fc3971..0f4d9f4 100644
--- a/src/main/java/cc/mrbird/febs/common/utils/LoginUserUtil.java
+++ b/src/main/java/cc/mrbird/febs/common/utils/LoginUserUtil.java
@@ -20,19 +20,29 @@
 
     public static DappMemberEntity getAppUser() {
         HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.currentRequestAttributes()).getRequest();
-        return (DappMemberEntity) request.getSession().getAttribute("member");
+        return (DappMemberEntity)request.getSession().getAttribute("member");
     }
 
     // address:0x971c09aa9735eb98459b17ec8b48932d24cbb931
     // nonce:0x1d5f7444107bc02e980deda39d0fce21b06c9da4233a19cb11124cb5bfefc9ec
     // sign:0x8f92cee24906122e26c3cc6cbd72f851cfe2c9574aa03bf3371e5d506fbec68b2ad22bbbc19b00ed21d26ab5a6871507831e2c902d8ed8c33301addc2b57a7731b
     public static String getLoginKey(String address, String nonce, String sign) {
-        log.info("address:{}, nonce:{}, sign:{}", address, nonce, sign);
         StringBuilder sb = new StringBuilder();
         StringBuilder append = sb.append(address).append(":").append(nonce).append(":").append(sign);
 
+        return sha3(append.toString());
+    }
+
+    public static String getLoginTokenKey(Long memberId, String nonce) {
+        StringBuilder sb = new StringBuilder();
+        StringBuilder append = sb.append(memberId).append(":").append(nonce);
+
+        return sha3(append.toString());
+    }
+
+    public static String sha3(String str) {
         Keccak.DigestKeccak kecc = new Keccak.Digest256();
-        kecc.update(append.toString().getBytes(StandardCharsets.UTF_8));
+        kecc.update(str.getBytes(StandardCharsets.UTF_8));
         return Numeric.toHexString(kecc.digest());
     }
 }

--
Gitblit v1.9.1