Coverage Summary for Class: NodeChallengeManager (co.rsk.net.discovery)
Class |
Class, %
|
Method, %
|
Line, %
|
NodeChallengeManager |
0%
(0/1)
|
0%
(0/4)
|
0%
(0/9)
|
1 /*
2 * This file is part of RskJ
3 * Copyright (C) 2017 RSK Labs Ltd.
4 *
5 * This program is free software: you can redistribute it and/or modify
6 * it under the terms of the GNU Lesser General Public License as published by
7 * the Free Software Foundation, either version 3 of the License, or
8 * (at your option) any later version.
9 *
10 * This program is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 * GNU Lesser General Public License for more details.
14 *
15 * You should have received a copy of the GNU Lesser General Public License
16 * along with this program. If not, see <http://www.gnu.org/licenses/>.
17 */
18
19 package co.rsk.net.discovery;
20
21 import co.rsk.net.discovery.message.PingPeerMessage;
22 import com.google.common.annotations.VisibleForTesting;
23 import org.ethereum.net.rlpx.Node;
24
25 import java.util.Map;
26 import java.util.concurrent.ConcurrentHashMap;
27
28 /**
29 * Created by mario on 22/02/17.
30 */
31 public class NodeChallengeManager {
32 private Map<String, NodeChallenge> activeChallenges = new ConcurrentHashMap<>();
33
34 public NodeChallenge startChallenge(Node challengedNode, Node challenger, PeerExplorer explorer) {
35 PingPeerMessage pingMessage = explorer.sendPing(challengedNode.getAddress(), 1, challengedNode);
36 String messageId = pingMessage.getMessageId();
37 NodeChallenge challenge = new NodeChallenge(challengedNode, challenger, messageId);
38 activeChallenges.put(messageId, challenge);
39 return challenge;
40 }
41
42 public NodeChallenge removeChallenge(String challengeId) {
43 return activeChallenges.remove(challengeId);
44 }
45
46 @VisibleForTesting
47 public int activeChallengesCount() {
48 return activeChallenges.size();
49 }
50 }