/**
* 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.pig.impl.util;
import java.net.URI;
import java.util.Arrays;
import java.util.List;
import java.util.Locale;
import javax.tools.Diagnostic;
import javax.tools.DiagnosticCollector;
import javax.tools.JavaCompiler;
import javax.tools.JavaFileManager;
import javax.tools.JavaFileObject;
import javax.tools.SimpleJavaFileObject;
import javax.tools.ToolProvider;
import com.google.common.collect.Lists;
import org.apache.hadoop.util.Shell;
import org.apache.log4j.Logger;
public class JavaCompilerHelper {
private static final Logger LOG = Logger.getLogger(JavaCompilerHelper.class);
/**
* This class allows code to be generated directly from a String, instead of having to be
* on disk.
*/
public static class JavaSourceFromString extends SimpleJavaFileObject {
final String code;
public JavaSourceFromString(String name, String code) {
super(URI.create("string:///" + name.replace('.','/') + Kind.SOURCE.extension), Kind.SOURCE);
this.code = code;
}
@Override
public CharSequence getCharContent(boolean ignoreEncodingErrors) {
return code;
}
}
private String classPath;
public JavaCompilerHelper() {
this.classPath = System.getProperty("java.class.path");
}
public void compile(String target, JavaSourceFromString... sources) {
LOG.info("compiling java classes to "+target);
JavaCompiler compiler = ToolProvider.getSystemJavaCompiler();
JavaFileManager fileManager = compiler.getStandardFileManager(null, null, null);
DiagnosticCollector<JavaFileObject> diagnostics = new DiagnosticCollector<JavaFileObject>();
List<String> optionList = Lists.newArrayList();
// Adds the current classpath to the compiler along with our generated code
optionList.add("-classpath");
optionList.add(classPath);
optionList.add("-d");
optionList.add(target);
if (!compiler.getTask(null, fileManager, diagnostics, optionList, null, Arrays.asList(sources)).call()) {
LOG.warn("Error compiling Printing compilation errors and shutting down.");
for (Diagnostic<? extends JavaFileObject> diagnostic : diagnostics.getDiagnostics()) {
LOG.warn("Error on line " + diagnostic.getLineNumber() + ": " + diagnostic.getMessage(Locale.US));
}
throw new RuntimeException("Unable to compile");
}
}
public void addToClassPath(String path) {
this.classPath = this.classPath + System.getProperty("path.separator") + path;
}
public String getClassPath() {
return classPath;
}
}