ProgressInfo: Add creation from json object and json array
This will allow to creat ProgressInfo object from json object and json Array it needed to report stream file information. Signed-off-by: Amnon Heiman <amnon@scylladb.com>
This commit is contained in:
parent
d589f3a3a3
commit
afd49d7bd4
@ -26,6 +26,11 @@ package org.apache.cassandra.streaming;
|
|||||||
|
|
||||||
import java.io.Serializable;
|
import java.io.Serializable;
|
||||||
import java.net.InetAddress;
|
import java.net.InetAddress;
|
||||||
|
import java.net.UnknownHostException;
|
||||||
|
import java.util.HashMap;
|
||||||
|
import java.util.Map;
|
||||||
|
import javax.json.JsonArray;
|
||||||
|
import javax.json.JsonObject;
|
||||||
|
|
||||||
import com.google.common.base.Objects;
|
import com.google.common.base.Objects;
|
||||||
|
|
||||||
@ -74,6 +79,31 @@ public class ProgressInfo implements Serializable
|
|||||||
this.totalBytes = totalBytes;
|
this.totalBytes = totalBytes;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
static public ProgressInfo fromJsonObject(JsonObject obj) {
|
||||||
|
try {
|
||||||
|
return new ProgressInfo(InetAddress.getByName(obj.getString("peer")),
|
||||||
|
obj.getInt("session_index"),
|
||||||
|
obj.getString("file_name"),
|
||||||
|
Direction.valueOf(obj.getString("direction")),
|
||||||
|
obj.getJsonNumber("current_bytes").longValue(),
|
||||||
|
obj.getJsonNumber("total_bytes").longValue());
|
||||||
|
} catch (UnknownHostException e) {
|
||||||
|
// Not suppose to get here
|
||||||
|
}
|
||||||
|
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
static public Map<String, ProgressInfo> fromJArrray(JsonArray arr) {
|
||||||
|
Map<String, ProgressInfo> res = new HashMap<String, ProgressInfo>();
|
||||||
|
if (arr != null) {
|
||||||
|
for (int i = 0; i < arr.size(); i++) {
|
||||||
|
ProgressInfo obj = fromJsonObject(arr.getJsonObject(i).getJsonObject("value"));
|
||||||
|
res.put(obj.fileName, obj);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return res;
|
||||||
|
}
|
||||||
/**
|
/**
|
||||||
* @return true if file transfer is completed
|
* @return true if file transfer is completed
|
||||||
*/
|
*/
|
||||||
|
Loading…
Reference in New Issue
Block a user