| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You under the Apache License, Version 2.0 |
| * (the "License"); you may not use this file except in compliance with |
| * the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| package org.apache.solr.handler; |
| import org.apache.solr.common.SolrException; |
| import org.apache.solr.common.params.SolrParams; |
| import org.apache.solr.common.util.ContentStream; |
| import org.apache.solr.common.util.NamedList; |
| import org.apache.solr.handler.loader.ContentStreamLoader; |
| import org.apache.solr.request.SolrQueryRequest; |
| import org.apache.solr.response.SolrQueryResponse; |
| import org.apache.solr.update.SolrCoreState; |
| import org.apache.solr.update.processor.UpdateRequestProcessor; |
| import org.apache.solr.update.processor.UpdateRequestProcessorChain; |
| |
| /** |
| * Shares common code between various handlers that manipulate |
| * {@link org.apache.solr.common.util.ContentStream} objects. |
| */ |
| public abstract class ContentStreamHandlerBase extends RequestHandlerBase { |
| |
| @Override |
| public void init(@SuppressWarnings({"rawtypes"})NamedList args) { |
| super.init(args); |
| |
| // Caching off by default |
| httpCaching = false; |
| if (args != null) { |
| Object caching = args.get("httpCaching"); |
| if(caching!=null) { |
| httpCaching = Boolean.parseBoolean(caching.toString()); |
| } |
| } |
| } |
| |
| @Override |
| public void handleRequestBody(SolrQueryRequest req, SolrQueryResponse rsp) throws Exception { |
| /* |
| We track update requests so that we can preserve consistency by waiting for them to complete |
| on a node shutdown and then immediately trigger a leader election without waiting for the core to close. |
| See how the SolrCoreState#pauseUpdatesAndAwaitInflightRequests() method is used in CoreContainer#shutdown() |
| |
| Also see https://issues.apache.org/jira/browse/SOLR-14942 for details on why we do not care for |
| other kinds of requests. |
| */ |
| SolrCoreState solrCoreState = req.getCore().getSolrCoreState(); |
| if (!solrCoreState.registerInFlightUpdate()) { |
| throw new SolrException(SolrException.ErrorCode.SERVER_ERROR, "Updates are temporarily paused for core: " + req.getCore().getName()); |
| } |
| try { |
| SolrParams params = req.getParams(); |
| UpdateRequestProcessorChain processorChain = |
| req.getCore().getUpdateProcessorChain(params); |
| |
| UpdateRequestProcessor processor = processorChain.createProcessor(req, rsp); |
| |
| try { |
| ContentStreamLoader documentLoader = newLoader(req, processor); |
| |
| |
| Iterable<ContentStream> streams = req.getContentStreams(); |
| if (streams == null) { |
| if (!RequestHandlerUtils.handleCommit(req, processor, params, false) && !RequestHandlerUtils.handleRollback(req, processor, params, false)) { |
| throw new SolrException(SolrException.ErrorCode.BAD_REQUEST, "missing content stream"); |
| } |
| } else { |
| |
| for (ContentStream stream : streams) { |
| documentLoader.load(req, rsp, stream, processor); |
| } |
| |
| // Perhaps commit from the parameters |
| RequestHandlerUtils.handleCommit(req, processor, params, false); |
| RequestHandlerUtils.handleRollback(req, processor, params, false); |
| } |
| } finally { |
| // finish the request |
| try { |
| processor.finish(); |
| } finally { |
| processor.close(); |
| } |
| } |
| } finally { |
| solrCoreState.deregisterInFlightUpdate(); |
| } |
| } |
| |
| protected abstract ContentStreamLoader newLoader(SolrQueryRequest req, UpdateRequestProcessor processor); |
| } |