de.ingrid.iplug.ckan.index.mapper.ScriptedDocumentMapper.java Source code

Introduction

Here is the source code for de.ingrid.iplug.ckan.index.mapper.ScriptedDocumentMapper.java

Source

/*
 * **************************************************-
 * InGrid-iPlug CKAN
 * ==================================================
 * Copyright (C) 2014 - 2016 wemove digital solutions GmbH
 * ==================================================
 * Licensed under the EUPL, Version 1.1 or  as soon they will be
 * approved by the European Commission - subsequent versions of the
 * EUPL (the "Licence");
 * 
 * You may not use this work except in compliance with the Licence.
 * You may obtain a copy of the Licence at:
 * 
 * http://ec.europa.eu/idabc/eupl5
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the Licence is distributed on an "AS IS" basis,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the Licence for the specific language governing permissions and
 * limitations under the Licence.
 * **************************************************#
 */
/**
 * 
 */
package de.ingrid.iplug.ckan.index.mapper;

import java.util.Hashtable;
import java.util.Map;

import org.apache.log4j.Logger;
import org.springframework.core.annotation.Order;
import org.springframework.core.io.Resource;

import de.ingrid.iplug.ckan.om.SourceRecord;
import de.ingrid.iplug.ckan.utils.IndexUtils;
import de.ingrid.iplug.ckan.utils.ScriptEngine;
import de.ingrid.utils.ElasticDocument;

/**
 * Script based source record to lucene document mapping. This class takes a
 * {@link Resource} as parameter to specify the mapping script. The scripting
 * engine will be automatically determined from the extension of the mapping
 * script.
 * <p />
 * If the {@link compile} parameter is set to true, the script is compiled, if
 * the ScriptEngine supports compilation.
 * 
 * @author joachim@wemove.com
 * 
 */
@Order(1)
public class ScriptedDocumentMapper implements IRecordMapper {

    private Resource[] mappingScripts;
    private boolean compile = false;

    private static final Logger log = Logger.getLogger(ScriptedDocumentMapper.class);

    @Override
    public synchronized void map(SourceRecord record, ElasticDocument doc) throws Exception {
        if (mappingScripts == null) {
            log.error("Mapping script(s) not set!");
            throw new IllegalArgumentException("Mapping script(s) not set!");
        }
        try {
            // create utils for script
            IndexUtils idxUtils = new IndexUtils(doc);

            Map<String, Object> parameters = new Hashtable<String, Object>();
            parameters.put("sourceRecord", record.get("json"));
            parameters.put("luceneDoc", doc);
            parameters.put("log", log);
            parameters.put("IDX", idxUtils);
            parameters.put("javaVersion", System.getProperty("java.version"));

            ScriptEngine.execute(this.mappingScripts, parameters, compile);
        } catch (Exception e) {
            log.error("Error mapping source record to lucene document.", e);
            throw e;
        }
    }

    public Resource[] getMappingScripts() {
        return mappingScripts;
    }

    public void setMappingScripts(Resource[] mappingScripts) {
        this.mappingScripts = mappingScripts;
    }

    public boolean isCompile() {
        return compile;
    }

    public void setCompile(boolean compile) {
        this.compile = compile;
    }

}