org.apache.oodt.cas.pge.writers.MetadataKeyReplacerTemplateWriter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of cas-pge Show documentation

Allows data processing jobs not written in conformance with the PCS PGE (Production Generation Executive) interface to be run within the PCS.

There is a newer version: 1.9.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.oodt.cas.pge.writers;

//JDK imports
import java.io.File;
import java.util.List;
import java.util.logging.Logger;

//APACHE imports
import org.apache.commons.io.FileUtils;
import org.apache.commons.lang.StringUtils;
import org.apache.oodt.cas.metadata.Metadata;

/**
 * 
 * Simple template based replacement writer, replaces $variables inside of a
 * text file template with information computed by Apache OODT (e.g., from
 * environment variables, from input workflow or file metadata, derived
 * metadata, etc etc.)
 * 
 * Multi-valued metadata keys are expanded to joined strings delimited by the
 * 2nd varargs argument to
 * {@link #createConfigFile(String, Metadata, Object...)}. The 1st varargs
 * argument to {@link #createConfigFile(String, Metadata, Object...)} is the
 * template file to use as a basis.
 * 
 */
public class MetadataKeyReplacerTemplateWriter extends
    DynamicConfigFileWriter {

  private static final String DEFAULT_SEPARATOR = ",";

  /*
   * (non-Javadoc)
   * 
   * @see
   * org.apache.oodt.cas.pge.writers.DynamicConfigFileWriter#generateFile(java
   * .lang.String, org.apache.oodt.cas.metadata.Metadata,
   * java.util.logging.Logger, java.lang.Object[])
   */
  @Override
  public File generateFile(String filePath, Metadata metadata, Logger logger,
      Object... args) throws Exception {
    String templateFile = (String) args[0];
    String processedTemplate = FileUtils
        .readFileToString(new File(templateFile));
    String separator = args.length == 2 ? (String) args[1] : DEFAULT_SEPARATOR;

    for (String key : metadata.getAllKeys()) {
      String replaceVal = null;
      if (metadata.isMultiValued(key)) {
        List values = metadata.getAllMetadata(key);
        replaceVal = StringUtils.join(values, separator);
      } else
        replaceVal = metadata.getMetadata(key);
      processedTemplate = processedTemplate.replaceAll("\\$" + key, replaceVal);
    }

    File configFile = new File(filePath);
    FileUtils.writeStringToFile(configFile, processedTemplate);
    return configFile;
  }

}