From 424ad35dcd1fafb51a0ba77523b11e6871f2858c Mon Sep 17 00:00:00 2001
From: KKSU <15274802129@163.com>
Date: Thu, 06 Jun 2024 11:36:25 +0800
Subject: [PATCH] 购买节点

---
 src/main/java/cc/mrbird/febs/job/SystemInit.java |   34 ++++++++++++++++++++++++++++------
 1 files changed, 28 insertions(+), 6 deletions(-)

diff --git a/src/main/java/cc/mrbird/febs/job/SystemInit.java b/src/main/java/cc/mrbird/febs/job/SystemInit.java
index 9a55dd6..21ee0b5 100644
--- a/src/main/java/cc/mrbird/febs/job/SystemInit.java
+++ b/src/main/java/cc/mrbird/febs/job/SystemInit.java
@@ -1,15 +1,21 @@
 package cc.mrbird.febs.job;
 
+import cc.mrbird.febs.common.annotation.EncryptEnable;
 import cc.mrbird.febs.common.contants.AppContants;
 import cc.mrbird.febs.common.utils.RedisUtils;
+import cc.mrbird.febs.common.utils.SpringContextUtil;
 import cc.mrbird.febs.dapp.chain.ChainEnum;
 import cc.mrbird.febs.dapp.chain.ChainService;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.CommandLineRunner;
+import org.springframework.boot.autoconfigure.SpringBootApplication;
 import org.springframework.stereotype.Component;
 
 import javax.annotation.PostConstruct;
+import java.lang.reflect.Method;
 import java.math.BigDecimal;
+import java.util.Map;
 
 /**
  * @author wzy
@@ -17,13 +23,29 @@
  **/
 @Slf4j
 @Component
-public class SystemInit {
+public class SystemInit implements CommandLineRunner {
 
-    @Autowired
-    private RedisUtils redisUtils;
+//    @PostConstruct
+//    public void init() {
+//
+//    }
 
-    @PostConstruct
-    public void init() {
-        log.info("系统参数初始化");
+    @Override
+    public void run(String... args) throws Exception {
+        Map<String, Object> encryptBeans = SpringContextUtil.getBeansWithAnnotation(EncryptEnable.class);
+        if (encryptBeans.isEmpty()) {
+            return;
+        }
+
+        for (Map.Entry<String, Object> entry : encryptBeans.entrySet()) {
+            Method[] methods = entry.getValue().getClass().getMethods();
+            if (methods.length == 0) {
+                continue;
+            }
+
+            for (Method method : methods) {
+                AppContants.ENCRYPT_METHOD.add(method.getName());
+            }
+        }
     }
 }

--
Gitblit v1.9.1