org.btrplace.json.model.ModelConverter Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of scheduler-json Show documentation
Show all versions of scheduler-json Show documentation
Read and write model according to the JSON format
/*
* Copyright (c) 2016 University Nice Sophia Antipolis
*
* This file is part of btrplace.
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 3 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this program. If not, see {
/**
* Key to indicate the mapping.
*/
public static final String MAPPING_LABEL = "mapping";
/**
* Key to indicate the decorators.
*/
public static final String VIEWS_LABEL = "views";
/**
* Key to indicate the attributes.
*/
public static final String ATTRS_LABEL = "attributes";
private ModelViewsConverter viewsConverter;
/**
* Make a new converter.
*/
public ModelConverter() {
viewsConverter = ModelViewsConverter.newBundle();
}
/**
* Get the converter that manage the decorators.
*
* @return the used converter
*/
public ModelViewsConverter getViewsConverter() {
return viewsConverter;
}
/**
* set the converter that manage the decorators.
*
* @param c the converter to use
*/
public void setModelViewConverters(ModelViewsConverter c) {
this.viewsConverter = c;
}
@Override
public JSONObject toJSON(Model i) throws JSONConverterException {
JSONArray rcs = new JSONArray();
for (ModelView v : i.getViews()) {
rcs.add(viewsConverter.toJSON(v));
}
JSONObject o = new JSONObject();
o.put(MAPPING_LABEL, toJSON(i.getMapping()));
o.put(ATTRS_LABEL, AttributesConverter.toJSON(i.getAttributes()));
o.put(VIEWS_LABEL, rcs);
return o;
}
@Override
public Model fromJSON(JSONObject o) throws JSONConverterException {
checkKeys(o, MAPPING_LABEL, ATTRS_LABEL, VIEWS_LABEL);
Model i = new DefaultModel();
fillMapping(i, (JSONObject) o.get(MAPPING_LABEL));
i.setAttributes(AttributesConverter.fromJSON(i, (JSONObject) o.get(ATTRS_LABEL)));
for (Object view : (JSONArray) o.get(VIEWS_LABEL)) {
i.attach(viewsConverter.fromJSON(i, (JSONObject) view));
}
return i;
}
/**
* Serialise the mapping.
*
* @param c the mapping
* @return the resulting JSONObject
*/
private static JSONObject toJSON(Mapping c) {
JSONObject o = new JSONObject();
o.put("offlineNodes", nodesToJSON(c.getOfflineNodes()));
o.put("readyVMs", vmsToJSON(c.getReadyVMs()));
JSONObject ons = new JSONObject();
for (Node n : c.getOnlineNodes()) {
JSONObject w = new JSONObject();
w.put("runningVMs", vmsToJSON(c.getRunningVMs(n)));
w.put("sleepingVMs", vmsToJSON(c.getSleepingVMs(n)));
ons.put(Integer.toString(n.id()), w);
}
o.put("onlineNodes", ons);
return o;
}
/**
* Create the elements inside the model and fill the mapping.
*
* @param mo the model where to attach the elements
* @param o the json describing the mapping
* @throws JSONConverterException if a error occurred during the conversion
*/
public void fillMapping(Model mo, JSONObject o) throws JSONConverterException {
Mapping c = mo.getMapping();
for (Node u : newNodes(mo, o, "offlineNodes")) {
c.addOfflineNode(u);
}
for (VM u : newVMs(mo, o, "readyVMs")) {
c.addReadyVM(u);
}
JSONObject ons = (JSONObject) o.get("onlineNodes");
for (Map.Entry e : ons.entrySet()) {
int id = Integer.parseInt(e.getKey());
Node u = mo.newNode(id);
if (u == null) {
throw JSONConverterException.nodeAlreadyDeclared(id);
}
JSONObject on = (JSONObject) e.getValue();
c.addOnlineNode(u);
for (VM vm : newVMs(mo, on, "runningVMs")) {
c.addRunningVM(vm, u);
}
for (VM vm : newVMs(mo, on, "sleepingVMs")) {
c.addSleepingVM(vm, u);
}
}
}
/**
* Build nodes from a key.
*
* @param mo the model to build
* @param o the object that contains the node
* @param key the key associated to the nodes
* @return the resulting set of nodes
* @throws JSONConverterException if at least one of the parsed node already exists
*/
private static Set newNodes(Model mo, JSONObject o, String key) throws JSONConverterException {
checkKeys(o, key);
Object x = o.get(key);
if (!(x instanceof JSONArray)) {
throw new JSONConverterException("array expected at key '" + key + "'");
}
Set s = new HashSet<>(((JSONArray) x).size());
for (Object i : (JSONArray) x) {
int id = (Integer) i;
Node n = mo.newNode(id);
if (n == null) {
throw JSONConverterException.nodeAlreadyDeclared(id);
}
s.add(n);
}
return s;
}
/**
* Build VMs from a key.
*
* @param mo the model to build
* @param o the object that contains the vm
* @param key the key associated to the VMs
* @return the resulting set of VMs
* @throws JSONConverterException if at least one of the parsed VM already exists
*/
private static Set newVMs(Model mo, JSONObject o, String key) throws JSONConverterException {
checkKeys(o, key);
Object x = o.get(key);
if (!(x instanceof JSONArray)) {
throw new JSONConverterException("array expected at key '" + key + "'");
}
Set s = new HashSet<>(((JSONArray) x).size());
for (Object i : (JSONArray) x) {
int id = (Integer) i;
VM vm = mo.newVM(id);
if (vm == null) {
throw JSONConverterException.vmAlreadyDeclared(id);
}
s.add(vm);
}
return s;
}
}