Skip to content

Commit cfbfec5

Browse files
committed
Merge branch 'tronprotocol:release_v4.7.2' into release_v4.7.2
2 parents b605617 + 0a3f2aa commit cfbfec5

4 files changed

Lines changed: 40 additions & 41 deletions

File tree

framework/src/main/java/org/tron/core/net/service/nodepersist/NodePersistService.java

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -4,34 +4,32 @@
44
import java.util.ArrayList;
55
import java.util.Comparator;
66
import java.util.List;
7+
import java.util.Objects;
78
import java.util.Timer;
89
import java.util.TimerTask;
910
import lombok.extern.slf4j.Slf4j;
1011
import org.springframework.stereotype.Component;
12+
import org.tron.common.parameter.CommonParameter;
1113
import org.tron.common.utils.ByteArray;
1214
import org.tron.common.utils.JsonUtil;
1315
import org.tron.core.ChainBaseManager;
1416
import org.tron.core.capsule.BytesCapsule;
15-
import org.tron.core.config.args.Args;
1617
import org.tron.core.net.TronNetService;
1718
import org.tron.p2p.discover.Node;
1819

1920
@Slf4j(topic = "net")
2021
@Component
2122
public class NodePersistService {
2223
private static final byte[] DB_KEY_PEERS = "peers".getBytes();
23-
private static final long DB_COMMIT_RATE = 1 * 60 * 1000L;
24+
private static final long DB_COMMIT_RATE = 60 * 1000L;
2425
private static final int MAX_NODES_WRITE_TO_DB = 30;
25-
26-
private boolean isNodePersist = Args.getInstance().isNodeDiscoveryPersist();
27-
28-
private ChainBaseManager chainBaseManager = ChainBaseManager.getInstance();
29-
26+
private final boolean isNodePersist = CommonParameter.getInstance().isNodeDiscoveryPersist();
27+
private final ChainBaseManager chainBaseManager = ChainBaseManager.getInstance();
3028
private Timer nodePersistTaskTimer;
3129

3230
public void init() {
33-
nodePersistTaskTimer = new Timer("NodePersistTaskTimer");
3431
if (isNodePersist) {
32+
nodePersistTaskTimer = new Timer("NodePersistTaskTimer");
3533
nodePersistTaskTimer.scheduleAtFixedRate(new TimerTask() {
3634
@Override
3735
public void run() {
@@ -42,6 +40,9 @@ public void run() {
4240
}
4341

4442
public void close() {
43+
if (Objects.isNull(nodePersistTaskTimer)) {
44+
return;
45+
}
4546
try {
4647
nodePersistTaskTimer.cancel();
4748
} catch (Exception e) {

framework/src/test/java/org/tron/common/utils/Sha256HashTest.java

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,8 @@
33
import java.util.Arrays;
44
import java.util.concurrent.atomic.AtomicLong;
55
import java.util.stream.IntStream;
6-
import org.testng.Assert;
7-
import org.testng.annotations.Test;
6+
import org.junit.Assert;
7+
import org.junit.Test;
88
import org.tron.common.parameter.CommonParameter;
99

1010
public class Sha256HashTest {
@@ -17,10 +17,10 @@ public void testHash() {
1717
.getInstance().isECKeyCryptoEngine(), input);
1818
byte[] hash1 = Sha256Hash.hash(CommonParameter
1919
.getInstance().isECKeyCryptoEngine(), hash0);
20-
Assert.assertEquals(hash0, ByteArray
21-
.fromHexString("CD5D4A7E8BE869C00E17F8F7712F41DBE2DDBD4D8EC36A7280CD578863717084"));
22-
Assert.assertEquals(hash1, ByteArray
23-
.fromHexString("10AE21E887E8FE30C591A22A5F8BB20EB32B2A739486DC5F3810E00BBDB58C5C"));
20+
Assert.assertEquals(Arrays.toString(hash0), Arrays.toString(ByteArray
21+
.fromHexString("CD5D4A7E8BE869C00E17F8F7712F41DBE2DDBD4D8EC36A7280CD578863717084")));
22+
Assert.assertEquals(Arrays.toString(hash1), Arrays.toString(ByteArray
23+
.fromHexString("10AE21E887E8FE30C591A22A5F8BB20EB32B2A739486DC5F3810E00BBDB58C5C")));
2424

2525
}
2626

@@ -40,7 +40,7 @@ public void testMultiThreadingHash() {
4040
countAll.incrementAndGet();
4141
if (!Arrays.equals(hash, hash0)) {
4242
countFailed.incrementAndGet();
43-
Assert.assertTrue(false);
43+
Assert.fail();
4444
}
4545
}
4646
});

framework/src/test/java/org/tron/core/services/http/BroadcastServletTest.java

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -57,10 +57,9 @@ public static void init() {
5757
/**
5858
* set up.
5959
*
60-
* @throws InterruptedException .
6160
*/
6261
@Before
63-
public void setUp() throws InterruptedException {
62+
public void setUp() {
6463
broadcastServlet = new BroadcastServlet();
6564
this.request = mock(HttpServletRequest.class);
6665
this.response = mock(HttpServletResponse.class);
@@ -115,7 +114,7 @@ public void testDoPost() throws IOException {
115114
+ "eapis.com/protocol.TransferContract\"},\"type\":\"TransferCon"
116115
+ "tract\"}],\"ref_block_bytes\":\"267e\",\"ref_block_hash\":\"9a447d222e8"
117116
+ "de9f2\",\"expiration\":1530893064000,\"timestamp\":1530893006233}}";
118-
httpUrlConnection.setRequestProperty("Content-Length", "" + postData.length());
117+
httpUrlConnection.setRequestProperty("Content-Length", String.valueOf(postData.length()));
119118

120119
when(httpUrlConnection.getOutputStream()).thenReturn(outContent);
121120
OutputStreamWriter out = new OutputStreamWriter(httpUrlConnection.getOutputStream(),
@@ -146,8 +145,8 @@ public void testDoPost() throws IOException {
146145
InputStreamReader inputStreamReader = new InputStreamReader(fileInputStream);
147146
BufferedReader bufferedReader = new BufferedReader(inputStreamReader);
148147

149-
StringBuffer sb = new StringBuffer();
150-
String text = null;
148+
StringBuilder sb = new StringBuilder();
149+
String text;
151150
while ((text = bufferedReader.readLine()) != null) {
152151
sb.append(text);
153152
}

framework/src/test/java/org/tron/core/services/ratelimiter/adaptor/AdaptorTest.java

Lines changed: 20 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
import java.util.concurrent.CountDownLatch;
66
import java.util.concurrent.Semaphore;
77
import org.junit.Assert;
8-
import org.testng.annotations.Test;
8+
import org.junit.Test;
99
import org.tron.common.utils.ReflectUtils;
1010
import org.tron.core.services.ratelimiter.adapter.GlobalPreemptibleAdapter;
1111
import org.tron.core.services.ratelimiter.adapter.IPQPSRateLimiterAdapter;
@@ -23,20 +23,19 @@ public void testStrategy() {
2323
IPQpsStrategy strategy1 = (IPQpsStrategy) ReflectUtils.getFieldObject(adapter1,
2424
"strategy");
2525

26-
Assert.assertTrue(Double.valueOf(
26+
Assert.assertEquals(5.0d, Double.parseDouble(
2727
ReflectUtils.getFieldValue(strategy1.getMapParams().get("qps"),
28-
"value").toString())
29-
== 5.0d);
30-
Assert.assertTrue(strategy1.getMapParams().get("notExist") == null);
28+
"value").toString()), 0.0);
29+
Assert.assertNull(strategy1.getMapParams().get("notExist"));
3130

3231
String paramString2 = "qps=5xyz";
3332
IPQPSRateLimiterAdapter adapter2 = new IPQPSRateLimiterAdapter(paramString2);
3433
IPQpsStrategy strategy2 = (IPQpsStrategy) ReflectUtils.getFieldObject(adapter2,
3534
"strategy");
3635

37-
Assert.assertTrue(Double.valueOf(
36+
Assert.assertEquals(IPQpsStrategy.DEFAULT_IPQPS, Double.valueOf(
3837
ReflectUtils.getFieldValue(strategy2.getMapParams().get("qps"),
39-
"value").toString()).equals(IPQpsStrategy.DEFAULT_IPQPS));
38+
"value").toString()));
4039
}
4140

4241
@Test
@@ -46,9 +45,9 @@ public void testIPQPSRateLimiterAdapter() {
4645

4746
IPQpsStrategy strategy = (IPQpsStrategy) ReflectUtils.getFieldObject(adapter,
4847
"strategy");
49-
Assert.assertTrue(Double
50-
.valueOf(ReflectUtils.getFieldValue(strategy.getMapParams().get("qps"),
51-
"value").toString()) == 5.0d);
48+
Assert.assertEquals(5.0d, Double
49+
.parseDouble(ReflectUtils.getFieldValue(strategy.getMapParams().get("qps"),
50+
"value").toString()), 0.0);
5251

5352
long t0 = System.currentTimeMillis();
5453
for (int i = 0; i < 20; i++) {
@@ -61,15 +60,15 @@ public void testIPQPSRateLimiterAdapter() {
6160
for (int i = 0; i < 20; i++) {
6261
if (i % 2 == 0) {
6362
strategy.acquire("1.2.3.4");
64-
} else if (i % 2 == 1) {
63+
} else {
6564
strategy.acquire("4.3.2.1");
6665
}
6766
}
6867
t1 = System.currentTimeMillis();
6968
Assert.assertTrue(t1 - t0 > 1500);
7069
Cache<String, RateLimiter> ipLimiter = (Cache<String, RateLimiter>) ReflectUtils
7170
.getFieldObject(strategy, "ipLimiter");
72-
Assert.assertTrue(ipLimiter.size() == 2);
71+
Assert.assertEquals(2, ipLimiter.size());
7372
}
7473

7574
@Test
@@ -78,9 +77,9 @@ public void testGlobalPreemptibleAdapter() {
7877
GlobalPreemptibleAdapter adapter1 = new GlobalPreemptibleAdapter(paramString1);
7978
GlobalPreemptibleStrategy strategy1 = (GlobalPreemptibleStrategy) ReflectUtils
8079
.getFieldObject(adapter1, "strategy");
81-
Assert.assertTrue(Integer.valueOf(
80+
Assert.assertEquals(1, Integer.parseInt(
8281
ReflectUtils.getFieldValue(strategy1.getMapParams().get("permit"),
83-
"value").toString()) == 1);
82+
"value").toString()));
8483
boolean first = strategy1.acquire();
8584
Assert.assertTrue(first);
8685

@@ -95,9 +94,9 @@ public void testGlobalPreemptibleAdapter() {
9594
GlobalPreemptibleAdapter adapter2 = new GlobalPreemptibleAdapter(paramString2);
9695
GlobalPreemptibleStrategy strategy2 = (GlobalPreemptibleStrategy) ReflectUtils
9796
.getFieldObject(adapter2, "strategy");
98-
Assert.assertTrue(Integer.valueOf(
97+
Assert.assertEquals(3, Integer.parseInt(
9998
ReflectUtils.getFieldValue(strategy2.getMapParams().get("permit"),
100-
"value").toString()) == 3);
99+
"value").toString()));
101100

102101
first = strategy2.acquire();
103102
Assert.assertTrue(first);
@@ -114,11 +113,11 @@ public void testGlobalPreemptibleAdapter() {
114113
Assert.assertTrue(fourAfterOneRelease);
115114

116115
Semaphore sp = (Semaphore) ReflectUtils.getFieldObject(strategy2, "sp");
117-
Assert.assertTrue(sp.availablePermits() == 0);
116+
Assert.assertEquals(0, sp.availablePermits());
118117
strategy2.release();
119118
strategy2.release();
120119
strategy2.release();
121-
Assert.assertTrue(sp.availablePermits() == 3);
120+
Assert.assertEquals(3, sp.availablePermits());
122121

123122
}
124123

@@ -128,9 +127,9 @@ public void testQpsRateLimiterAdapter() {
128127
QpsRateLimiterAdapter adapter = new QpsRateLimiterAdapter(paramString);
129128

130129
QpsStrategy strategy = (QpsStrategy) ReflectUtils.getFieldObject(adapter, "strategy");
131-
Assert.assertTrue(Double
132-
.valueOf(ReflectUtils.getFieldValue(strategy.getMapParams().get("qps"),
133-
"value").toString()) == 5);
130+
Assert.assertEquals(5, Double
131+
.parseDouble(ReflectUtils.getFieldValue(strategy.getMapParams().get("qps"),
132+
"value").toString()), 0.0);
134133
strategy.acquire();
135134

136135
long t0 = System.currentTimeMillis();

0 commit comments

Comments
 (0)