JsonSaver.java

package com.osroyale.util.parser;
 
import com.google.common.base.Preconditions;
import com.google.gson.Gson;
import com.google.gson.GsonBuilder;
import com.google.gson.JsonArray;
import com.google.gson.JsonObject;
 
import java.io.FileWriter;

 
* A util class used for constructing and writing {@code JSON} files.
 * <p>
 * <p>
 * And an example of usage:
 * 
 * <pre>
 * JsonSaver json = new JsonSaver();
 * 
 * for (Player player : players) {
 *  json.current().addProperty(&quot;name&quot;, player.getUsername());
 *  json.current().addProperty(&quot;value1&quot;, 1);
 *  json.current().addProperty(&quot;value2&quot;, true);
 *  json.split();
 * }
 * 
 * json.publish(&quot;./data/some_player_database.json&quot;);
 * </pre>
 *
 * @author lare96 <http://github.org/lare96>
 */
public final class JsonSaver {

    private final Gson serializer = new GsonBuilder().setPrettyPrinting().disableHtmlEscaping().create();

    private final JsonArray array = new JsonArray();

    private final boolean singletonTable;

    private JsonObject currentWriter = new JsonObject();

    public JsonSaver() {
        this(false);
    }

    public JsonSaver(boolean singletonTable) {
        this.singletonTable = singletonTable;
    }
 
    /***
     * Gets the current {@code JsonObject} that is writing data.
     * 
     * @return the current writer.
     */
    public JsonObject current() {
        return currentWriter;
    }

    public void publish(String path) {
        try (FileWriter fw = new FileWriter(path)) {
            fw.write(toString());
        } catch (final Exception e) {
            e.printStackTrace();
        }
    }

    public Gson serializer() {
        return serializer;
    }

    public void split() {
        Preconditions.checkState(!singletonTable, "JsonSaver instance is a singleton table!");
        array.add(currentWriter);
        currentWriter = new JsonObject();
    }

    @Override
    public String toString() {
        if (singletonTable) {
            return serializer.toJson(currentWriter);
        }
        if (currentWriter.entrySet().size() > 0) {
            split();
        }
        return serializer.toJson(array);
    }
}