Remove remaining uses of org.json.JSONWriter in the core backend

This commit is contained in:
Antonin Delpeuch 2018-10-07 14:59:45 +01:00
parent 9219ef36aa
commit 7156c49c4d
5 changed files with 137 additions and 122 deletions

View File

@ -34,13 +34,13 @@ OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
package com.google.refine.exporters;
import java.io.IOException;
import java.io.StringWriter;
import java.io.Writer;
import java.util.Properties;
import org.json.JSONException;
import org.json.JSONObject;
import org.json.JSONWriter;
import com.fasterxml.jackson.annotation.JsonProperty;
import com.google.refine.browsing.Engine;
import com.google.refine.browsing.Engine.Mode;
@ -64,6 +64,26 @@ public class TemplatingExporter implements WriterExporter {
return "application/x-unknown";
}
protected static class TemplateConfig {
@JsonProperty("template")
protected String template;
@JsonProperty("prefix")
protected String prefix;
@JsonProperty("suffix")
protected String suffix;
@JsonProperty("separator")
protected String separator;
protected TemplateConfig(
String template, String prefix,
String suffix, String separator) {
this.template = template;
this.prefix = prefix;
this.suffix = suffix;
this.separator = separator;
}
}
@Override
public void export(Project project, Properties options, Engine engine, Writer writer) throws IOException {
String limitString = options.getProperty("limit");
@ -94,20 +114,10 @@ public class TemplatingExporter implements WriterExporter {
template.setSeparator(separatorString);
if (!"true".equals(options.getProperty("preview"))) {
StringWriter stringWriter = new StringWriter();
JSONWriter jsonWriter = new JSONWriter(stringWriter);
try {
jsonWriter.object();
jsonWriter.key("template"); jsonWriter.value(templateString);
jsonWriter.key("prefix"); jsonWriter.value(prefixString);
jsonWriter.key("suffix"); jsonWriter.value(suffixString);
jsonWriter.key("separator"); jsonWriter.value(separatorString);
jsonWriter.endObject();
} catch (JSONException e) {
// ignore
}
project.getMetadata().getPreferenceStore().put("exporters.templating.template", stringWriter.toString());
TemplateConfig config = new TemplateConfig(templateString, prefixString,
suffixString, separatorString);
project.getMetadata().getPreferenceStore().put("exporters.templating.template",
ParsingUtilities.defaultWriter.writeValueAsString(config));
}
if (engine.getMode() == Mode.RowBased) {

View File

@ -48,9 +48,9 @@ import javax.servlet.http.HttpServletResponse;
import org.json.JSONArray;
import org.json.JSONException;
import org.json.JSONObject;
import org.json.JSONWriter;
import com.fasterxml.jackson.annotation.JsonProperty;
import com.fasterxml.jackson.core.JsonGenerator;
import com.google.refine.RefineServlet;
import com.google.refine.commands.HttpUtilities;
@ -187,22 +187,21 @@ public class DefaultImportingController implements ImportingController {
ImportingUtilities.previewParse(job, format, optionObj, exceptions);
Writer w = response.getWriter();
JSONWriter writer = new JSONWriter(w);
JsonGenerator writer = ParsingUtilities.mapper.getFactory().createGenerator(w);
try {
writer.object();
writer.writeStartObject();
if (exceptions.size() == 0) {
job.project.update(); // update all internal models, indexes, caches, etc.
writer.key("status"); writer.value("ok");
writer.writeStringField("status", "ok");
} else {
writer.key("status"); writer.value("error");
writer.key("errors");
writer.array();
writer.writeStringField("status", "error");
writer.writeArrayFieldStart("errors");
writeErrors(writer, exceptions);
writer.endArray();
writer.writeEndArray();
}
writer.endObject();
} catch (JSONException e) {
writer.writeEndObject();
} catch (IOException e) {
throw new ServletException(e);
} finally {
w.flush();
@ -304,17 +303,15 @@ public class DefaultImportingController implements ImportingController {
w.close();
}
static public void writeErrors(JSONWriter writer, List<Exception> exceptions) throws JSONException {
static public void writeErrors(JsonGenerator writer, List<Exception> exceptions) throws IOException {
for (Exception e : exceptions) {
StringWriter sw = new StringWriter();
e.printStackTrace(new PrintWriter(sw));
writer.object();
writer.key("message");
writer.value(e.getLocalizedMessage());
writer.key("stack");
writer.value(sw.toString());
writer.endObject();
writer.writeStartObject();
writer.writeStringField("message", e.getLocalizedMessage());
writer.writeStringField("stack", sw.toString());
writer.writeEndObject();
}
}

View File

@ -43,15 +43,12 @@ import java.util.HashSet;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.Properties;
import java.util.Set;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.TimeUnit;
import org.apache.commons.io.FileUtils;
import org.json.JSONException;
import org.json.JSONWriter;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

View File

@ -40,15 +40,17 @@ import java.net.HttpURLConnection;
import java.net.URL;
import java.time.OffsetDateTime;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import org.apache.commons.lang3.StringUtils;
import org.json.JSONArray;
import org.json.JSONException;
import org.json.JSONObject;
import org.json.JSONWriter;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
@ -56,6 +58,7 @@ import com.fasterxml.jackson.annotation.JsonIgnore;
import com.fasterxml.jackson.annotation.JsonInclude;
import com.fasterxml.jackson.annotation.JsonInclude.Include;
import com.fasterxml.jackson.annotation.JsonProperty;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.google.refine.expr.ExpressionUtils;
import com.google.refine.model.Cell;
@ -234,40 +237,78 @@ public class StandardReconConfig extends ReconConfig {
*/
StandardReconJob job = new StandardReconJob();
try {
StringWriter stringWriter = new StringWriter();
JSONWriter jsonWriter = new JSONWriter(stringWriter);
jsonWriter.object();
jsonWriter.key("query");
jsonWriter.value(query);
jsonWriter.endObject();
String queryJson = ParsingUtilities.defaultWriter.writeValueAsString(
Collections.singletonMap("query", query));
job.text = query;
job.code = stringWriter.toString();
job.code = queryJson;
return job;
} catch (JSONException je) {
} catch (JsonProcessingException e) {
e.printStackTrace();
return null;
}
}
protected static class QueryProperty {
@JsonProperty("pid")
String pid;
@JsonProperty("v")
Object v;
protected QueryProperty(
String pid,
Object v) {
this.pid = pid;
this.v = v;
}
}
protected static class ReconQuery {
@JsonProperty("query")
protected String query;
@JsonProperty("type")
@JsonInclude(Include.NON_NULL)
protected String typeID;
@JsonProperty("type_strict")
@JsonInclude(Include.NON_NULL)
public String isTypeStrict() {
if(typeID != null) {
return "should";
}
return null;
}
@JsonProperty("properties")
@JsonInclude(Include.NON_EMPTY)
protected List<QueryProperty> properties;
// Only send limit if it's non-default to preserve backward compatibility with
// services which might choke on this
@JsonProperty("limit")
@JsonInclude(Include.NON_DEFAULT)
protected int limit;
public ReconQuery(
String query,
String typeID,
List<QueryProperty> properties,
int limit) {
this.query = query;
this.typeID = typeID;
this.properties = properties;
this.limit = limit;
}
}
@Override
public ReconJob createJob(Project project, int rowIndex, Row row,
String columnName, Cell cell) {
StandardReconJob job = new StandardReconJob();
try {
StringWriter stringWriter = new StringWriter();
JSONWriter jsonWriter = new JSONWriter(stringWriter);
jsonWriter.object();
jsonWriter.key("query"); jsonWriter.value(cell.value.toString());
if (typeID != null) {
jsonWriter.key("type"); jsonWriter.value(typeID);
jsonWriter.key("type_strict"); jsonWriter.value("should");
}
if (columnDetails.size() > 0) {
jsonWriter.key("properties");
jsonWriter.array();
List<QueryProperty> properties = new ArrayList<>();
for (ColumnDetail c : columnDetails) {
int detailCellIndex = project.columnModel.getColumnByName(c.columnName).getCellIndex();
@ -288,40 +329,27 @@ public class StandardReconConfig extends ReconConfig {
}
if (cell2 != null && ExpressionUtils.isNonBlankData(cell2.value)) {
jsonWriter.object();
jsonWriter.key("pid"); jsonWriter.value(c.propertyID);
jsonWriter.key("v");
Object v = null;
if (cell2.recon != null && cell2.recon.match != null) {
jsonWriter.object();
jsonWriter.key("id"); jsonWriter.value(cell2.recon.match.id);
jsonWriter.key("name"); jsonWriter.value(cell2.recon.match.name);
jsonWriter.endObject();
} else if (cell2.value instanceof OffsetDateTime) {
jsonWriter.value(ParsingUtilities.dateToString((OffsetDateTime) cell2.value));
Map<String, String> recon = new HashMap<>();
recon.put("id", cell2.recon.match.id);
recon.put("name", cell2.recon.match.name);
v = recon;
} else {
jsonWriter.value(cell2.value.toString());
v = cell2.value;
}
properties.add(new QueryProperty(c.propertyID, v));
}
jsonWriter.endObject();
}
}
jsonWriter.endArray();
}
// Only send limit if it's non-default to preserve backward compatibility with
// services which might choke on this
if (limit != 0) {
jsonWriter.key("limit"); jsonWriter.value(limit);
}
jsonWriter.endObject();
ReconQuery query = new ReconQuery(cell.value.toString(), typeID, properties, limit);
job.text = cell.value.toString();
job.code = stringWriter.toString();
} catch (JSONException e) {
//
try {
job.code = ParsingUtilities.defaultWriter.writeValueAsString(query);
} catch (JsonProcessingException e) {
e.printStackTrace();
}
return job;
}

View File

@ -46,7 +46,6 @@ import java.util.Map;
import org.json.JSONArray;
import org.json.JSONException;
import org.json.JSONObject;
import org.json.JSONWriter;
public class JSONUtilities {
@ -172,22 +171,6 @@ public class JSONUtilities {
}
}
static public void writeStringList(JSONWriter writer, List<String> list) throws JSONException {
writer.array();
for (String s : list) {
writer.value(s);
}
writer.endArray();
}
static public void writeStringArray(JSONWriter writer, String[] strings) throws JSONException {
writer.array();
for (String s : strings) {
writer.value(s);
}
writer.endArray();
}
static public void putField(JSONObject obj, String key, Object value) throws JSONException {
if (value instanceof Integer) {
obj.put(key, ((Integer) value).intValue());