Merge remote-tracking branch 'origin/release_v4.7.2' into release_v4.7.2

lxcmyf · lxcmyf · commit e27c862ac9e1 · 2023-04-19T13:55:38.000+08:00
diff --git a/framework/src/main/java/org/tron/core/net/service/nodepersist/NodePersistService.java b/framework/src/main/java/org/tron/core/net/service/nodepersist/NodePersistService.java
@@ -8,12 +8,13 @@
 import java.util.Timer;
 import java.util.TimerTask;
 import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Component;
 import org.tron.common.parameter.CommonParameter;
 import org.tron.common.utils.ByteArray;
 import org.tron.common.utils.JsonUtil;
-import org.tron.core.ChainBaseManager;
 import org.tron.core.capsule.BytesCapsule;
+import org.tron.core.db.CommonStore;
 import org.tron.core.net.TronNetService;
 import org.tron.p2p.discover.Node;
 
@@ -24,7 +25,8 @@ public class NodePersistService {
   private static final long DB_COMMIT_RATE = 60 * 1000L;
   private static final int MAX_NODES_WRITE_TO_DB = 30;
   private final boolean isNodePersist = CommonParameter.getInstance().isNodeDiscoveryPersist();
-  private final ChainBaseManager chainBaseManager = ChainBaseManager.getInstance();
+  @Autowired
+  private CommonStore commonStore;
   private Timer nodePersistTaskTimer;
 
   public void init() {
@@ -53,7 +55,7 @@ public void close() {
   public List<InetSocketAddress> dbRead() {
     List<InetSocketAddress> nodes = new ArrayList<>();
     try {
-      byte[] nodeBytes = chainBaseManager.getCommonStore().get(DB_KEY_PEERS).getData();
+      byte[] nodeBytes = commonStore.get(DB_KEY_PEERS).getData();
       if (ByteArray.isEmpty(nodeBytes)) {
         return nodes;
       }
@@ -85,8 +87,7 @@ private void dbWrite() {
 
       logger.info("Write nodes to store: {}/{} nodes", batch.size(), tableNodes.size());
 
-      chainBaseManager.getCommonStore()
-              .put(DB_KEY_PEERS, new BytesCapsule(JsonUtil.obj2Json(dbNodes).getBytes()));
+      commonStore.put(DB_KEY_PEERS, new BytesCapsule(JsonUtil.obj2Json(dbNodes).getBytes()));
     } catch (Exception e) {
       logger.warn("DB write nodes failed, {}", e.getMessage());
     }
diff --git a/framework/src/test/java/org/tron/common/runtime/InheritanceTest.java b/framework/src/test/java/org/tron/common/runtime/InheritanceTest.java
@@ -22,6 +22,7 @@ public class InheritanceTest extends BaseTest {
 
   private static final String OWNER_ADDRESS;
   private RepositoryImpl repository;
+  private static boolean init;
 
   static {
     dbPath = "output_InheritanceTest";
@@ -34,9 +35,13 @@ public class InheritanceTest extends BaseTest {
    */
   @Before
   public void init() {
+    if (init) {
+      return;
+    }
     repository = RepositoryImpl.createRoot(StoreFactory.getInstance());
     repository.createAccount(Hex.decode(OWNER_ADDRESS), AccountType.Normal);
     repository.addBalance(Hex.decode(OWNER_ADDRESS), 100000000);
+    init = true;
   }
 
   /**
diff --git a/framework/src/test/java/org/tron/common/runtime/InternalTransactionComplexTest.java b/framework/src/test/java/org/tron/common/runtime/InternalTransactionComplexTest.java
@@ -24,6 +24,7 @@ public class InternalTransactionComplexTest extends BaseTest {
   private static final String OWNER_ADDRESS;
   private static Runtime runtime;
   private static RepositoryImpl repository;
+  private static boolean init;
 
   static {
     dbPath = "output_InternalTransactionComplexTest";
@@ -37,9 +38,13 @@ public class InternalTransactionComplexTest extends BaseTest {
    */
   @Before
   public void init() {
+    if (init) {
+      return;
+    }
     repository = RepositoryImpl.createRoot(StoreFactory.getInstance());
     repository.createAccount(Hex.decode(OWNER_ADDRESS), AccountType.Normal);
     repository.addBalance(Hex.decode(OWNER_ADDRESS), 100000000);
+    init = true;
   }
 
   /**
diff --git a/framework/src/test/java/org/tron/common/runtime/ProgramResultTest.java b/framework/src/test/java/org/tron/common/runtime/ProgramResultTest.java
@@ -41,6 +41,7 @@ public class ProgramResultTest extends BaseTest {
   private static final String TRANSFER_TO;
   private static Runtime runtime;
   private static RepositoryImpl repository;
+  private static boolean init;
 
   static {
     dbPath = "output_InternalTransactionComplexTest";
@@ -55,12 +56,16 @@ public class ProgramResultTest extends BaseTest {
    */
   @Before
   public void init() {
+    if (init) {
+      return;
+    }
     repository = RepositoryImpl.createRoot(StoreFactory.getInstance());
     repository.createAccount(Hex.decode(OWNER_ADDRESS), AccountType.Normal);
     repository.addBalance(Hex.decode(OWNER_ADDRESS), 100000000);
     repository.createAccount(Hex.decode(TRANSFER_TO), AccountType.Normal);
     repository.addBalance(Hex.decode(TRANSFER_TO), 0);
     repository.commit();
+    init = true;
   }
 
   /**
diff --git a/framework/src/test/java/org/tron/common/runtime/RuntimeTransferComplexTest.java b/framework/src/test/java/org/tron/common/runtime/RuntimeTransferComplexTest.java
@@ -29,6 +29,7 @@ public class RuntimeTransferComplexTest extends BaseTest {
   private static final String TRANSFER_TO;
   private static Runtime runtime;
   private static RepositoryImpl repository;
+  private static boolean init;
 
   static {
     dbPath = "output_RuntimeTransferComplexTest";
@@ -42,12 +43,16 @@ public class RuntimeTransferComplexTest extends BaseTest {
    */
   @Before
   public void init() {
+    if (init) {
+      return;
+    }
     repository = RepositoryImpl.createRoot(StoreFactory.getInstance());
     repository.createAccount(Hex.decode(OWNER_ADDRESS), AccountType.Normal);
     repository.addBalance(Hex.decode(OWNER_ADDRESS), 1000000000);
     repository.createAccount(Hex.decode(TRANSFER_TO), AccountType.Normal);
     repository.addBalance(Hex.decode(TRANSFER_TO), 10);
     repository.commit();
+    init = true;
   }
 
   /**
diff --git a/framework/src/test/java/org/tron/common/runtime/vm/BandWidthRuntimeOutOfTimeTest.java b/framework/src/test/java/org/tron/common/runtime/vm/BandWidthRuntimeOutOfTimeTest.java
@@ -64,6 +64,7 @@ public class BandWidthRuntimeOutOfTimeTest extends BaseTest {
 
   private static final String OwnerAddress = "TCWHANtDDdkZCTo2T2peyEq3Eg9c2XB7ut";
   private static final String TriggerOwnerAddress = "TCSgeWapPJhCqgWRxXCKb6jJ5AgNWSGjPA";
+  private static boolean init;
 
   static {
     dbPath = "output_bandwidth_runtime_out_of_time_test";
@@ -84,6 +85,9 @@ public class BandWidthRuntimeOutOfTimeTest extends BaseTest {
    */
   @Before
   public void init() {
+    if (init) {
+      return;
+    }
     //init energy
     dbManager.getDynamicPropertiesStore().saveLatestBlockHeaderTimestamp(1526647828000L);
     dbManager.getDynamicPropertiesStore().saveTotalEnergyWeight(10_000_000L);
@@ -107,6 +111,7 @@ public void init() {
         .put(Commons.decodeFromBase58Check(TriggerOwnerAddress), accountCapsule2);
     dbManager.getDynamicPropertiesStore()
         .saveLatestBlockHeaderTimestamp(System.currentTimeMillis() / 1000);
+    init = true;
   }
 
   @Test
diff --git a/framework/src/test/java/org/tron/common/runtime/vm/BandWidthRuntimeOutOfTimeWithCheckTest.java b/framework/src/test/java/org/tron/common/runtime/vm/BandWidthRuntimeOutOfTimeWithCheckTest.java
@@ -65,6 +65,7 @@ public class BandWidthRuntimeOutOfTimeWithCheckTest extends BaseTest {
   private static final String indexDirectory = "index_BandWidthRuntimeOutOfTimeTest_test";
   private static final String OwnerAddress = "TCWHANtDDdkZCTo2T2peyEq3Eg9c2XB7ut";
   private static final String TriggerOwnerAddress = "TCSgeWapPJhCqgWRxXCKb6jJ5AgNWSGjPA";
+  private static boolean init;
 
   static {
     dbPath = "output_bandwidth_runtime_out_of_time_with_check_test";
@@ -85,6 +86,9 @@ public class BandWidthRuntimeOutOfTimeWithCheckTest extends BaseTest {
    */
   @Before
   public void init() {
+    if (init) {
+      return;
+    }
     //init energy
     dbManager.getDynamicPropertiesStore().saveLatestBlockHeaderTimestamp(1526647837000L);
     dbManager.getDynamicPropertiesStore().saveTotalEnergyWeight(10_000_000L);
@@ -108,6 +112,7 @@ public void init() {
         .put(Commons.decodeFromBase58Check(TriggerOwnerAddress), accountCapsule2);
     dbManager.getDynamicPropertiesStore()
         .saveLatestBlockHeaderTimestamp(System.currentTimeMillis() / 1000);
+    init = true;
   }
 
   @Test
diff --git a/framework/src/test/java/org/tron/core/WalletTest.java b/framework/src/test/java/org/tron/core/WalletTest.java
@@ -131,6 +131,7 @@ public class WalletTest extends BaseTest {
   private static final String OWNER_ADDRESS;
   private static final String RECEIVER_ADDRESS;
   private static final long initBalance = 43_200_000_000L;
+  private static boolean init;
 
   static {
     dbPath = "output_wallet_test";
@@ -141,11 +142,15 @@ public class WalletTest extends BaseTest {
 
   @Before
   public void before() {
+    initAccountCapsule();
+    if (init) {
+      return;
+    }
     initTransaction();
     initBlock();
-    initAccountCapsule();
     chainBaseManager.getDynamicPropertiesStore().saveLatestBlockHeaderNumber(5);
     chainBaseManager.getDelegatedResourceStore().reset();
+    init = true;
   }
 
   /**
diff --git a/framework/src/test/java/org/tron/core/db/AccountStoreTest.java b/framework/src/test/java/org/tron/core/db/AccountStoreTest.java
@@ -33,6 +33,7 @@ public class AccountStoreTest extends BaseTest {
   private AssetIssueStore assetIssueStore;
   private static byte[] address = TransactionStoreTest.randomBytes(32);
   private static byte[] accountName = TransactionStoreTest.randomBytes(32);
+  private static boolean init;
 
   static {
     dbPath = "output_AccountStore_test";
@@ -48,13 +49,17 @@ public class AccountStoreTest extends BaseTest {
 
   @Before
   public void init() {
+    if (init) {
+      return;
+    }
     assetIssueStore = chainBaseManager.getAssetIssueStore();
     dynamicPropertiesStore.saveAllowBlackHoleOptimization(1);
     AccountCapsule accountCapsule = new AccountCapsule(ByteString.copyFrom(address),
         ByteString.copyFrom(accountName),
         AccountType.forNumber(1));
 
     accountStore.put(data, accountCapsule);
+    init = true;
   }
 
   @Test
diff --git a/framework/src/test/java/org/tron/core/db/CommonStoreTest.java b/framework/src/test/java/org/tron/core/db/CommonStoreTest.java
diff --git a/framework/src/test/java/org/tron/core/db/TransactionTraceTest.java b/framework/src/test/java/org/tron/core/db/TransactionTraceTest.java
@@ -57,6 +57,7 @@ public class TransactionTraceTest extends BaseTest {
   private static ByteString contractAddress = ByteString.copyFrom(ByteArray.fromInt(2));
   private static String OwnerAddress = "TCWHANtDDdkZCTo2T2peyEq3Eg9c2XB7ut";
   private static String TriggerOwnerAddress = "TCSgeWapPJhCqgWRxXCKb6jJ5AgNWSGjPA";
+  private static boolean init;
 
   static {
     dbPath = "output_TransactionTrace_test";
@@ -77,12 +78,15 @@ public class TransactionTraceTest extends BaseTest {
    */
   @Before
   public void init() {
+    if (init) {
+      return;
+    }
     //init energy
     dbManager.getDynamicPropertiesStore().saveLatestBlockHeaderTimestamp(1526647838000L);
     dbManager.getDynamicPropertiesStore().saveTotalEnergyWeight(100_000L);
     dbManager.getDynamicPropertiesStore().saveLatestBlockHeaderTimestamp(0);
     VMConfig.initVmHardFork(false);
-
+    init = true;
   }
 
   @Test
diff --git a/framework/src/test/java/org/tron/core/db/api/AssetUpdateHelperTest.java b/framework/src/test/java/org/tron/core/db/api/AssetUpdateHelperTest.java
@@ -24,6 +24,7 @@
 public class AssetUpdateHelperTest extends BaseTest {
 
   private static final ByteString assetName = ByteString.copyFrom("assetIssueName".getBytes());
+  private static boolean init;
 
   static {
     dbPath = "output_AssetUpdateHelperTest_test";
@@ -33,7 +34,9 @@ public class AssetUpdateHelperTest extends BaseTest {
 
   @Before
   public void init() {
-
+    if (init) {
+      return;
+    }
     AssetIssueContract contract =
         AssetIssueContract.newBuilder().setName(assetName).setNum(12581).setPrecision(5).build();
     AssetIssueCapsule assetIssueCapsule = new AssetIssueCapsule(contract);
@@ -70,6 +73,7 @@ public void init() {
                 .build());
     chainBaseManager.getAccountStore().put(ByteArray.fromHexString("121212abc"),
         accountCapsule);
+    init = true;
   }
 
   @Test
diff --git a/framework/src/test/java/org/tron/core/jsonrpc/WalletCursorTest.java b/framework/src/test/java/org/tron/core/jsonrpc/WalletCursorTest.java
@@ -27,6 +27,7 @@ public class WalletCursorTest extends BaseTest {
   private Wallet wallet;
   @Resource
   private NodeInfoService nodeInfoService;
+  private static boolean init;
 
   static {
     dbPath = "output_wallet_cursor_test";
@@ -38,13 +39,17 @@ public class WalletCursorTest extends BaseTest {
 
   @Before
   public void init() {
+    if (init) {
+      return;
+    }
     AccountCapsule accountCapsule =
         new AccountCapsule(
             ByteString.copyFromUtf8(OWNER_ADDRESS_ACCOUNT_NAME),
             ByteString.copyFrom(ByteArray.fromHexString(OWNER_ADDRESS)),
             Protocol.AccountType.Normal,
             10000_000_000L);
     dbManager.getAccountStore().put(accountCapsule.getAddress().toByteArray(), accountCapsule);
+    init = true;
   }
 
   @Test
diff --git a/framework/src/test/java/org/tron/core/services/ProposalServiceTest.java b/framework/src/test/java/org/tron/core/services/ProposalServiceTest.java
@@ -22,6 +22,7 @@
 @Slf4j
 public class ProposalServiceTest extends BaseTest {
 
+  private static boolean init;
 
   @BeforeClass
   public static void init() {
@@ -32,7 +33,11 @@ public static void init() {
 
   @Before
   public void before() {
+    if (init) {
+      return;
+    }
     dbManager.getDynamicPropertiesStore().saveLatestBlockHeaderNumber(5);
+    init = true;
   }
 
   @Test
diff --git a/framework/src/test/java/org/tron/core/zksnark/MerkleTreeTest.java b/framework/src/test/java/org/tron/core/zksnark/MerkleTreeTest.java
@@ -28,6 +28,7 @@ public class MerkleTreeTest extends BaseTest {
   public static final long totalBalance = 1000_0000_000_000L;
   private static final String dbDirectory = "db_ShieldedTransaction_test";
   private static final String indexDirectory = "index_ShieldedTransaction_test";
+  private static boolean init;
 
   static {
     dbPath = "output_ShieldedTransaction_test";
@@ -48,10 +49,14 @@ public class MerkleTreeTest extends BaseTest {
    */
   @Before
   public void init() {
+    if (init) {
+      return;
+    }
     //init energy
     dbManager.getDynamicPropertiesStore().saveLatestBlockHeaderTimestamp(1526647838000L);
     dbManager.getDynamicPropertiesStore().saveTotalEnergyWeight(100_000L);
     dbManager.getDynamicPropertiesStore().saveLatestBlockHeaderTimestamp(0);
+    init = true;
   }
 
   private JSONArray readFile(String fileName) throws Exception {
diff --git a/framework/src/test/java/org/tron/core/zksnark/NoteEncDecryTest.java b/framework/src/test/java/org/tron/core/zksnark/NoteEncDecryTest.java
@@ -36,6 +36,7 @@ public class NoteEncDecryTest extends BaseTest {
   private static final int VOTE_SCORE = 2;
   private static final String DESCRIPTION = "TRX";
   private static final String URL = "https://tron.network";
+  private static boolean init;
   @Resource
   private Wallet wallet;
 
@@ -50,9 +51,13 @@ public class NoteEncDecryTest extends BaseTest {
    */
   @Before
   public void init() {
+    if (init) {
+      return;
+    }
     //give a big value for pool, avoid for
     dbManager.getDynamicPropertiesStore().saveTotalShieldedPoolValue(10_000_000_000L);
     // Args.getInstance().setAllowShieldedTransaction(1);
+    init = true;
   }
 
   /**
