257 lines
9.6 KiB
Java
257 lines
9.6 KiB
Java
/*
|
|
|
|
Copyright 2010, Google Inc.
|
|
All rights reserved.
|
|
|
|
Redistribution and use in source and binary forms, with or without
|
|
modification, are permitted provided that the following conditions are
|
|
met:
|
|
|
|
* Redistributions of source code must retain the above copyright
|
|
notice, this list of conditions and the following disclaimer.
|
|
* Redistributions in binary form must reproduce the above
|
|
copyright notice, this list of conditions and the following disclaimer
|
|
in the documentation and/or other materials provided with the
|
|
distribution.
|
|
* Neither the name of Google Inc. nor the names of its
|
|
contributors may be used to endorse or promote products derived from
|
|
this software without specific prior written permission.
|
|
|
|
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
|
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
|
OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
|
SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
|
LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
|
THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
|
|
*/
|
|
|
|
package com.google.refine.commands.recon;
|
|
|
|
import java.io.IOException;
|
|
|
|
import javax.servlet.ServletException;
|
|
import javax.servlet.http.HttpServletRequest;
|
|
import javax.servlet.http.HttpServletResponse;
|
|
|
|
import com.google.refine.commands.Command;
|
|
import com.google.refine.expr.ExpressionUtils;
|
|
import com.google.refine.history.Change;
|
|
import com.google.refine.history.HistoryEntry;
|
|
import com.google.refine.model.Cell;
|
|
import com.google.refine.model.Column;
|
|
import com.google.refine.model.Project;
|
|
import com.google.refine.model.Recon;
|
|
import com.google.refine.model.Recon.Judgment;
|
|
import com.google.refine.model.ReconCandidate;
|
|
import com.google.refine.model.ReconStats;
|
|
import com.google.refine.model.changes.CellChange;
|
|
import com.google.refine.model.changes.ReconChange;
|
|
import com.google.refine.process.QuickHistoryEntryProcess;
|
|
import com.google.refine.util.Pool;
|
|
|
|
public class ReconJudgeOneCellCommand extends Command {
|
|
@Override
|
|
public void doPost(HttpServletRequest request, HttpServletResponse response)
|
|
throws ServletException, IOException {
|
|
|
|
try {
|
|
request.setCharacterEncoding("UTF-8");
|
|
response.setCharacterEncoding("UTF-8");
|
|
|
|
Project project = getProject(request);
|
|
|
|
int rowIndex = Integer.parseInt(request.getParameter("row"));
|
|
int cellIndex = Integer.parseInt(request.getParameter("cell"));
|
|
Judgment judgment = Recon.stringToJudgment(request.getParameter("judgment"));
|
|
|
|
ReconCandidate match = null;
|
|
String id = request.getParameter("id");
|
|
if (id != null) {
|
|
String scoreString = request.getParameter("score");
|
|
|
|
match = new ReconCandidate(
|
|
id,
|
|
request.getParameter("name"),
|
|
request.getParameter("types").split(","),
|
|
scoreString != null ? Double.parseDouble(scoreString) : 100
|
|
);
|
|
}
|
|
|
|
JudgeOneCellProcess process = new JudgeOneCellProcess(
|
|
project,
|
|
"Judge one cell's recon result",
|
|
judgment,
|
|
rowIndex,
|
|
cellIndex,
|
|
match,
|
|
request.getParameter("identifierSpace"),
|
|
request.getParameter("schemaSpace")
|
|
);
|
|
|
|
HistoryEntry historyEntry = project.processManager.queueProcess(process);
|
|
if (historyEntry != null) {
|
|
/*
|
|
* If the process is done, write back the cell's data so that the
|
|
* client side can update its UI right away.
|
|
*/
|
|
|
|
Pool pool = new Pool();
|
|
if (process.newCell != null && process.newCell.recon != null) {
|
|
pool.pool(process.newCell.recon);
|
|
}
|
|
|
|
respondJSON(response, new ReconClearOneCellCommand.CellResponse(historyEntry, process.newCell, pool));
|
|
} else {
|
|
respond(response, "{ \"code\" : \"pending\" }");
|
|
}
|
|
} catch (Exception e) {
|
|
respondException(response, e);
|
|
}
|
|
}
|
|
|
|
protected static class JudgeOneCellProcess extends QuickHistoryEntryProcess {
|
|
|
|
final int rowIndex;
|
|
final int cellIndex;
|
|
final Judgment judgment;
|
|
final ReconCandidate match;
|
|
final String identifierSpace;
|
|
final String schemaSpace;
|
|
|
|
Cell newCell;
|
|
|
|
JudgeOneCellProcess(
|
|
Project project,
|
|
String briefDescription,
|
|
Judgment judgment,
|
|
int rowIndex,
|
|
int cellIndex,
|
|
ReconCandidate match,
|
|
String identifierSpace,
|
|
String schemaSpace
|
|
) {
|
|
super(project, briefDescription);
|
|
|
|
this.judgment = judgment;
|
|
this.rowIndex = rowIndex;
|
|
this.cellIndex = cellIndex;
|
|
this.match = match;
|
|
this.identifierSpace = identifierSpace;
|
|
this.schemaSpace = schemaSpace;
|
|
}
|
|
|
|
@Override
|
|
protected HistoryEntry createHistoryEntry(long historyEntryID) throws Exception {
|
|
Cell cell = _project.rows.get(rowIndex).getCell(cellIndex);
|
|
if (cell == null || !ExpressionUtils.isNonBlankData(cell.value)) {
|
|
throw new Exception("Cell is blank or error");
|
|
}
|
|
|
|
Column column = _project.columnModel.getColumnByCellIndex(cellIndex);
|
|
if (column == null) {
|
|
throw new Exception("No such column");
|
|
}
|
|
|
|
Judgment oldJudgment = cell.recon == null ? Judgment.None : cell.recon.judgment;
|
|
|
|
Recon newRecon = null;
|
|
if (cell.recon != null) {
|
|
newRecon = cell.recon.dup(historyEntryID);
|
|
} else if (identifierSpace != null && schemaSpace != null) {
|
|
newRecon = new Recon(historyEntryID, identifierSpace, schemaSpace);
|
|
} else if (column.getReconConfig() != null) {
|
|
newRecon = column.getReconConfig().createNewRecon(historyEntryID);
|
|
} else {
|
|
// This should only happen if we are judging a cell in a column that
|
|
// has never been reconciled before.
|
|
newRecon = new Recon(historyEntryID, null, null);
|
|
}
|
|
|
|
newCell = new Cell(
|
|
cell.value,
|
|
newRecon
|
|
);
|
|
|
|
String cellDescription =
|
|
"single cell on row " + (rowIndex + 1) +
|
|
", column " + column.getName() +
|
|
", containing \"" + cell.value + "\"";
|
|
|
|
String description = null;
|
|
|
|
newCell.recon.matchRank = -1;
|
|
newCell.recon.judgmentAction = "single";
|
|
newCell.recon.judgmentBatchSize = 1;
|
|
|
|
if (judgment == Judgment.None) {
|
|
newCell.recon.judgment = Recon.Judgment.None;
|
|
newCell.recon.match = null;
|
|
|
|
description = "Discard recon judgment for " + cellDescription;
|
|
} else if (judgment == Judgment.New) {
|
|
newCell.recon.judgment = Recon.Judgment.New;
|
|
newCell.recon.match = null;
|
|
|
|
description = "Mark to create new item for " + cellDescription;
|
|
} else {
|
|
newCell.recon.judgment = Recon.Judgment.Matched;
|
|
newCell.recon.match = this.match;
|
|
if (newCell.recon.candidates != null) {
|
|
for (int m = 0; m < newCell.recon.candidates.size(); m++) {
|
|
if (newCell.recon.candidates.get(m).id.equals(this.match.id)) {
|
|
newCell.recon.matchRank = m;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
description = "Match " + this.match.name +
|
|
" (" + match.id + ") to " +
|
|
cellDescription;
|
|
}
|
|
|
|
ReconStats stats = column.getReconStats();
|
|
if (stats == null) {
|
|
stats = ReconStats.create(_project, cellIndex);
|
|
} else {
|
|
int newChange = 0;
|
|
int matchChange = 0;
|
|
|
|
if (oldJudgment == Judgment.New) {
|
|
newChange--;
|
|
}
|
|
if (oldJudgment == Judgment.Matched) {
|
|
matchChange--;
|
|
}
|
|
if (newCell.recon.judgment == Judgment.New) {
|
|
newChange++;
|
|
}
|
|
if (newCell.recon.judgment == Judgment.Matched) {
|
|
matchChange++;
|
|
}
|
|
|
|
stats = new ReconStats(
|
|
stats.nonBlanks,
|
|
stats.newTopics + newChange,
|
|
stats.matchedTopics + matchChange);
|
|
}
|
|
|
|
Change change = new ReconChange(
|
|
new CellChange(rowIndex, cellIndex, cell, newCell),
|
|
column.getName(),
|
|
column.getReconConfig(),
|
|
stats
|
|
);
|
|
|
|
return new HistoryEntry(
|
|
historyEntryID, _project, description, null, change);
|
|
}
|
|
}
|
|
}
|