/*
* 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.response.transform;
import org.apache.solr.common.SolrDocument;
import org.apache.solr.common.SolrException;
import org.apache.solr.common.SolrException.ErrorCode;
import org.apache.solr.common.params.SolrParams;
import org.apache.solr.common.util.NamedList;
import org.apache.solr.request.SolrQueryRequest;
import org.apache.solr.util.DateMathParser;
/**
*
* @since solr 4.0
*/
public class ValueAugmenterFactory extends TransformerFactory
{
protected Object value = null;
protected Object defaultValue = null;
@Override
public void init(NamedList args) {
value = args.get( "value" );
if( value == null ) {
defaultValue = args.get( "defaultValue" );
}
}
public static Object getObjectFrom( String val, String type )
{
if( type != null ) {
try {
if( "int".equals( type ) ) return Integer.valueOf( val );
if( "double".equals( type ) ) return Double.valueOf( val );
if( "float".equals( type ) ) return Float.valueOf( val );
if( "date".equals( type ) ) return DateMathParser.parseMath(null, val );
}
catch( Exception ex ) {
throw new SolrException( ErrorCode.BAD_REQUEST,
"Unable to parse "+type+"="+val, ex );
}
}
return val;
}
@Override
public DocTransformer create(String field, SolrParams params, SolrQueryRequest req) {
Object val = value;
if( val == null ) {
String v = params.get("v");
if( v == null ) {
val = defaultValue;
}
else {
val = getObjectFrom(v, params.get("t"));
}
if( val == null ) {
throw new SolrException( ErrorCode.BAD_REQUEST,
"ValueAugmenter is missing a value -- should be defined in solrconfig or inline" );
}
}
return new ValueAugmenter( field, val );
}
}
class ValueAugmenter extends DocTransformer
{
final String name;
final Object value;
public ValueAugmenter( String name, Object value )
{
this.name = name;
this.value = value;
}
@Override
public String getName()
{
return name;
}
@Override
public void transform(SolrDocument doc, int docid, float score) {
doc.setField( name, value );
}
}