Github user dsmiley commented on a diff in the pull request: https://github.com/apache/lucene-solr/pull/410#discussion_r200372947 --- Diff: solr/core/src/java/org/apache/solr/update/processor/NestedUpdateProcessorFactory.java --- @@ -0,0 +1,126 @@ +/* + * 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.update.processor; + +import java.io.IOException; +import java.util.Collection; + +import org.apache.solr.common.SolrException; +import org.apache.solr.common.SolrInputDocument; +import org.apache.solr.common.SolrInputField; +import org.apache.solr.request.SolrQueryRequest; +import org.apache.solr.response.SolrQueryResponse; +import org.apache.solr.schema.IndexSchema; +import org.apache.solr.update.AddUpdateCommand; + +public class NestedUpdateProcessorFactory extends UpdateRequestProcessorFactory { + + public UpdateRequestProcessor getInstance(SolrQueryRequest req, SolrQueryResponse rsp, UpdateRequestProcessor next ) { + boolean storeParent = shouldStoreDocParent(req.getSchema()); + boolean storePath = shouldStoreDocPath(req.getSchema()); + if(!(storeParent || storePath)) { + return next; + } + return new NestedUpdateProcessor(req, rsp, shouldStoreDocParent(req.getSchema()), shouldStoreDocPath(req.getSchema()), next); + } + + private static boolean shouldStoreDocParent(IndexSchema schema) { + return schema.getFields().containsKey(IndexSchema.NEST_PARENT_FIELD_NAME); + } + + private static boolean shouldStoreDocPath(IndexSchema schema) { + return schema.getFields().containsKey(IndexSchema.NEST_PATH_FIELD_NAME); + } +} + +class NestedUpdateProcessor extends UpdateRequestProcessor { + private static final String PATH_SEP_CHAR = "/"; + private static final String NUM_SEP_CHAR = ","; + private static final String SINGLE_VALUE_CHAR = "s"; + private boolean storePath; + private boolean storeParent; + private String uniqueKeyFieldName; + + + protected NestedUpdateProcessor(SolrQueryRequest req, SolrQueryResponse rsp, boolean storeParent, boolean storePath, UpdateRequestProcessor next) { + super(next); + this.storeParent = storeParent; + this.storePath = storePath; + this.uniqueKeyFieldName = req.getSchema().getUniqueKeyField().getName(); + } + + @Override + public void processAdd(AddUpdateCommand cmd) throws IOException { + SolrInputDocument doc = cmd.getSolrInputDocument(); + processDocChildren(doc, null); + super.processAdd(cmd); + } + + private void processDocChildren(SolrInputDocument doc, String fullPath) { + int childNum = 0; + for(SolrInputField field: doc.values()) { + boolean isSingleVal = !(field.getValue() instanceof Collection); + for(Object val: field) { + if(!(val instanceof SolrInputDocument)) { + // either all collection items are child docs or none are. + break; + } + final String fieldName = field.getName(); + + if(fieldName.contains(PATH_SEP_CHAR)) { + throw new SolrException(SolrException.ErrorCode.BAD_REQUEST, "Field name: '" + fieldName + + "' contains: '" + PATH_SEP_CHAR + "' , which is reserved for the nested URP"); + } + final String sChildNum = isSingleVal ? SINGLE_VALUE_CHAR: String.valueOf(childNum); + final String lastPath = fieldName + NUM_SEP_CHAR + sChildNum + NUM_SEP_CHAR; --- End diff -- I'm a bit confused why SINGLE_VALUE_CHAR was introduced. Is this so that a DocTransformer can differentiates between a one-ary array and a single doc? I suppose so. The choice of 's' was confusing as I thought it was an attempt to pluralize something at first but now I think you intended an abbreviation for "singular". Could we simply use the empty string in this case? Also, please define lastPath & jointPath later closer to when they are used, which is immediately prior to calling processChildDoc. Why the trailing NUM_SEP_CHAR in lastPath?
--- --------------------------------------------------------------------- To unsubscribe, e-mail: dev-unsubscr...@lucene.apache.org For additional commands, e-mail: dev-h...@lucene.apache.org