Merge pull request #9 from virtualeconomy/APIKey-feature

add API Key feature

Author: SheldonYS

src/v/systems/Blockchain.java

@@ -2,6 +2,7 @@
 
 import com.google.gson.*;
 import lombok.Getter;
+import lombok.Setter;
 import v.systems.contract.*;
 import v.systems.entity.Balance;
 import v.systems.entity.BalanceDetail;
@@ -17,14 +18,19 @@
 
 import java.io.IOException;
 import java.util.ArrayList;
+import java.util.HashMap;
 import java.util.List;
+import java.util.Map;
 
 public class Blockchain {
     public static final long V_UNITY = 100000000L;
     @Getter
     private NetworkType network;
     @Getter
     private String nodeUrl;
+    @Getter
+    @Setter
+    private String apiKey;
     private Gson gson;
     private JsonParser parser;
 
@@ -35,6 +41,11 @@ public Blockchain(NetworkType network, String nodeUrl) {
         parser = JsonHelper.getParserInstance();
     }
 
+    public Blockchain(NetworkType network, String nodeUrl, String apiKey) {
+        this(network, nodeUrl);
+        this.apiKey = apiKey;
+    }
+
     public Long getBalance(String address) throws IOException, ApiError {
         String url = String.format("%s/addresses/balance/%s", nodeUrl, address);
         Balance balance = this.callChainAPI(url, Balance.class);
@@ -48,20 +59,24 @@ public BalanceDetail getBalanceDetail(String address) throws IOException, ApiErr
 
     public List<Transaction> getActiveLeaseTransactions(String address) throws IOException, ApiError {
         String url = String.format("%s/transactions/activeLeaseList/%s", nodeUrl, address);
-        return this.getTransactionByUrl(url);
+        return this.getTransactionByUrl(url, true);
     }
 
     public List<Transaction> getTransactionHistory(String address, int num) throws IOException, ApiError {
         if (num <= 0) {
             return new ArrayList<Transaction>();
         }
         String url = String.format("%s/transactions/address/%s/limit/%d", nodeUrl, address, num);
-        return this.getTransactionByUrl(url);
+        return this.getTransactionByUrl(url, false);
     }
 
-    private List<Transaction> getTransactionByUrl(String url) throws IOException, ApiError {
+    private List<Transaction> getTransactionByUrl(String url, boolean useApiKey) throws IOException, ApiError {
         List<Transaction> result = new ArrayList<Transaction>();
-        String json = HttpClient.get(url);
+        Map<String, String> header = new HashMap<String, String>();
+        if (useApiKey && this.apiKey != null) {
+            header.put("api_key", this.apiKey);
+        }
+        String json = HttpClient.get(url, header);
         try {
             JsonElement jsonElement = parser.parse(json);
             if (!jsonElement.isJsonArray()) {

src/v/systems/utils/HttpClient.java

@@ -5,11 +5,16 @@
 import java.net.HttpURLConnection;
 import java.net.URL;
 import java.nio.charset.StandardCharsets;
+import java.util.Map;
 
 
 public class HttpClient {
 
     public static String get(String url) throws IOException {
+        return get(url, null);
+    }
+
+    public static String get(String url, Map<String, String> header) throws IOException {
 
         HttpURLConnection con = null;
 
@@ -21,6 +26,11 @@ public static String get(String url) throws IOException {
                 con = (HttpURLConnection) webURL.openConnection();
             }
             con.setRequestMethod("GET");
+            if (header != null) {
+                for (Map.Entry<String, String> entry : header.entrySet()) {
+                    con.setRequestProperty(entry.getKey(), entry.getValue());
+                }
+            }
             return getResponse(con);
         } finally {
             if (con != null) {
@@ -30,6 +40,10 @@ public static String get(String url) throws IOException {
     }
 
     public static String post(String url, String json) throws IOException {
+        return post(url, null, json);
+    }
+
+    public static String post(String url, Map<String, String> header, String json) throws IOException {
 
         HttpURLConnection con = null;
         byte[] postData = json.getBytes(StandardCharsets.UTF_8);
@@ -44,6 +58,11 @@ public static String post(String url, String json) throws IOException {
             con.setDoOutput(true);
             con.setRequestMethod("POST");
             con.setRequestProperty("Content-Type", "application/json");
+            if (header != null) {
+                for (Map.Entry<String, String> entry : header.entrySet()) {
+                    con.setRequestProperty(entry.getKey(), entry.getValue());
+                }
+            }
             try (DataOutputStream wr = new DataOutputStream(con.getOutputStream())) {
                 wr.write(postData);
             }

test/v/systems/BlockchainTest.java

@@ -200,6 +200,7 @@ public void getContractContent() throws SerializationError {
     public void getActiveLeaseTransactions() {
         String testAddress = "AUCJE7djDBbFrgeKRZ4UMREujAeqnD6L6Ph";
         List<Transaction> txList = null;
+        defaultChain.setApiKey(TestConfig.API_KEY);
         try {
             txList = defaultChain.getActiveLeaseTransactions(testAddress);
         } catch (Exception e) {