<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.1//EN"
"http://www.w3.org/TR/xhtml11/DTD/xhtml11.dtd">
<html xmlns="http://www.w3.org/1999/xhtml">
<head><meta http-equiv="content-type" content="text/html; charset=utf-8" /><style type="text/css"><!--
#msg dl { border: 1px #006 solid; background: #369; padding: 6px; color: #fff; }
#msg dt { float: left; width: 6em; font-weight: bold; }
#msg dt:after { content:':';}
#msg dl, #msg dt, #msg ul, #msg li, #header, #footer { font-family: verdana,arial,helvetica,sans-serif; font-size: 10pt; }
#msg dl a { font-weight: bold}
#msg dl a:link { color:#fc3; }
#msg dl a:active { color:#ff0; }
#msg dl a:visited { color:#cc6; }
h3 { font-family: verdana,arial,helvetica,sans-serif; font-size: 10pt; font-weight: bold; }
#msg pre, #msg p { overflow: auto; background: #ffc; border: 1px #fc0 solid; padding: 6px; }
#msg ul { overflow: auto; }
#header, #footer { color: #fff; background: #636; border: 1px #300 solid; padding: 6px; }
#patch { width: 100%; }
#patch h4 {font-family: verdana,arial,helvetica,sans-serif;font-size:10pt;padding:8px;background:#369;color:#fff;margin:0;}
#patch .propset h4, #patch .binary h4 {margin:0;}
#patch pre {padding:0;line-height:1.2em;margin:0;}
#patch .diff {width:100%;background:#eee;padding: 0 0 10px 0;overflow:auto;}
#patch .propset .diff, #patch .binary .diff {padding:10px 0;}
#patch span {display:block;padding:0 10px;}
#patch .modfile, #patch .addfile, #patch .delfile, #patch .propset, #patch .binary, #patch .copfile {border:1px solid #ccc;margin:10px 0;}
#patch ins {background:#dfd;text-decoration:none;display:block;padding:0 10px;}
#patch del {background:#fdd;text-decoration:none;display:block;padding:0 10px;}
#patch .lines, .info {color:#888;background:#fff;}
--></style>
<title>[576] blacklight_importer/src/MarcUpdateRequestHandler.java:
First draft of a Solr/MARC update handler</title>
</head>
<body>
<div id="msg">
<dl>
<dt>Revision</dt> <dd>576</dd>
<dt>Author</dt> <dd>erikhatcher</dd>
<dt>Date</dt> <dd>2008-06-01 21:38:20 -0400 (Sun, 01 Jun 2008)</dd>
</dl>
<h3>Log Message</h3>
<pre>First draft of a Solr/MARC update handler</pre>
<h3>Added Paths</h3>
<ul>
<li><a href="#blacklight_importersrcMarcUpdateRequestHandlerjava">blacklight_importer/src/MarcUpdateRequestHandler.java</a></li>
</ul>
</div>
<div id="patch">
<h3>Diff</h3>
<a id="blacklight_importersrcMarcUpdateRequestHandlerjava"></a>
<div class="addfile"><h4>Added: blacklight_importer/src/MarcUpdateRequestHandler.java (0 => 576)</h4>
<pre class="diff"><span>
<span class="info">--- blacklight_importer/src/MarcUpdateRequestHandler.java (rev 0)
+++ blacklight_importer/src/MarcUpdateRequestHandler.java 2008-06-02 01:38:20 UTC (rev 576)
</span><span class="lines">@@ -0,0 +1,93 @@
</span><ins>+import org.apache.solr.handler.RequestHandlerBase;
+import org.apache.solr.handler.RequestHandlerUtils;
+import org.apache.solr.request.SolrQueryRequest;
+import org.apache.solr.request.SolrQueryResponse;
+import org.apache.solr.common.params.SolrParams;
+import org.apache.solr.common.params.UpdateParams;
+import org.apache.solr.common.util.ContentStream;
+import org.apache.solr.common.SolrException;
+import org.apache.solr.common.SolrInputDocument;
+import org.apache.solr.update.processor.UpdateRequestProcessorFactory;
+import org.apache.solr.update.processor.UpdateRequestProcessor;
+import org.apache.solr.update.AddUpdateCommand;
+import org.apache.commons.io.IOUtils;
+import org.marc4j.MarcReader;
+import org.marc4j.MarcStreamReader;
+import org.marc4j.marc.Record;
+import org.marc4j.marc.ControlField;
+
+import java.io.InputStream;
+
+public class MarcUpdateRequestHandler extends RequestHandlerBase {
+ public void handleRequestBody(SolrQueryRequest req, SolrQueryResponse rsp) throws Exception {
+ SolrParams params = req.getParams();
+ UpdateRequestProcessorFactory processorFactory =
+ req.getCore().getUpdateProcessorFactory(params.get(UpdateParams.UPDATE_PROCESSOR));
+
+ UpdateRequestProcessor processor = processorFactory.getInstance(req, rsp, null);
+
+ try {
+ AddUpdateCommand addCommand = new AddUpdateCommand();
+ if (params.getBool(UpdateParams.OVERWRITE, true)) {
+ addCommand.allowDups = false;
+ addCommand.overwriteCommitted = true;
+ addCommand.overwritePending = true;
+ } else {
+ addCommand.allowDups = true;
+ addCommand.overwriteCommitted = false;
+ addCommand.overwritePending = false;
+ }
+
+ Iterable<ContentStream> streams = req.getContentStreams();
+
+ if( streams == null ) {
+ if( !RequestHandlerUtils.handleCommit(processor, params, false) ) {
+ throw new SolrException( SolrException.ErrorCode.BAD_REQUEST, "missing content stream" );
+ }
+ } else {
+ for(ContentStream stream : streams) {
+ InputStream is = stream.getStream();
+ try {
+ MarcReader reader = new MarcStreamReader(is);
+ while (reader.hasNext()) {
+ Record record = reader.next();
+
+ addCommand.clear();
+
+ SolrInputDocument doc = new SolrInputDocument(); // TODO: plug in SolrIndexer (rename it though)
+ doc.setField("id", ((ControlField) record.getVariableField("001")).getData());
+ doc.setField("marc_t", record.toString());
+
+ addCommand.solrDoc = doc;
+ processor.processAdd(addCommand);
+ }
+ }
+ finally {
+ IOUtils.closeQuietly(is);
+ }
+
+ }
+ RequestHandlerUtils.handleCommit( processor, params, false );
+ }
+ } finally {
+ // finish the request
+ processor.finish();
+ }
+ }
+
+ public String getDescription() {
+ return null; //To change body of implemented methods use File | Settings | File Templates.
+ }
+
+ public String getSourceId() {
+ return null; //To change body of implemented methods use File | Settings | File Templates.
+ }
+
+ public String getSource() {
+ return null; //To change body of implemented methods use File | Settings | File Templates.
+ }
+
+ public String getVersion() {
+ return null; //To change body of implemented methods use File | Settings | File Templates.
+ }
+}
</ins></span></pre>
</div>
</div>
</body>
</html>