All Downloads are FREE. Search and download functionalities are using the official Maven repository.

org.apache.lucene.benchmark.byTask.tasks.AddFacetedDocTask Maven / Gradle / Ivy

There is a newer version: 9.11.1
Show newest version
/*
 * 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.lucene.benchmark.byTask.tasks;


import java.util.ArrayList;
import java.util.List;

import org.apache.lucene.benchmark.byTask.PerfRunData;
import org.apache.lucene.benchmark.byTask.feeds.FacetSource;
import org.apache.lucene.facet.FacetField;
import org.apache.lucene.facet.FacetsConfig;

/**
 * Add a faceted document.
 * 

* Config properties: *

    *
  • with.facets=<tells whether to actually add any facets to the * document| Default: true>
    * This config property allows to easily compare the performance of adding docs * with and without facets. Note that facets are created even when this is * false, just that they are not added to the document (nor to the taxonomy). *
*

* See {@link AddDocTask} for general document parameters and configuration. *

* Makes use of the {@link FacetSource} in effect - see {@link PerfRunData} for * facet source settings. */ public class AddFacetedDocTask extends AddDocTask { private FacetsConfig config; public AddFacetedDocTask(PerfRunData runData) { super(runData); } @Override public void setup() throws Exception { super.setup(); if (config == null) { boolean withFacets = getRunData().getConfig().get("with.facets", true); if (withFacets) { FacetSource facetsSource = getRunData().getFacetSource(); config = new FacetsConfig(); facetsSource.configure(config); } } } @Override protected String getLogMessage(int recsCount) { if (config == null) { return super.getLogMessage(recsCount); } return super.getLogMessage(recsCount)+ " with facets"; } @Override public int doLogic() throws Exception { if (config != null) { List facets = new ArrayList<>(); getRunData().getFacetSource().getNextFacets(facets); for(FacetField ff : facets) { doc.add(ff); } doc = config.build(getRunData().getTaxonomyWriter(), doc); } return super.doLogic(); } }





© 2015 - 2024 Weber Informatics LLC | Privacy Policy