Java tutorial
/* * 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.drools.compiler.commons.jci.compilers; import java.io.ByteArrayOutputStream; import java.io.IOException; import java.io.InputStream; import java.util.ArrayList; import java.util.Collection; import java.util.Locale; import java.util.Map; import java.util.StringTokenizer; import org.drools.compiler.commons.jci.problems.CompilationProblem; import org.drools.compiler.commons.jci.readers.ResourceReader; import org.drools.compiler.commons.jci.stores.ResourceStore; import org.drools.core.util.ClassUtils; import org.eclipse.jdt.core.compiler.IProblem; import org.eclipse.jdt.internal.compiler.ClassFile; import org.eclipse.jdt.internal.compiler.CompilationResult; import org.eclipse.jdt.internal.compiler.Compiler; import org.eclipse.jdt.internal.compiler.DefaultErrorHandlingPolicies; import org.eclipse.jdt.internal.compiler.ICompilerRequestor; import org.eclipse.jdt.internal.compiler.IErrorHandlingPolicy; import org.eclipse.jdt.internal.compiler.IProblemFactory; import org.eclipse.jdt.internal.compiler.classfmt.ClassFileReader; import org.eclipse.jdt.internal.compiler.classfmt.ClassFormatException; import org.eclipse.jdt.internal.compiler.env.ICompilationUnit; import org.eclipse.jdt.internal.compiler.env.INameEnvironment; import org.eclipse.jdt.internal.compiler.env.NameEnvironmentAnswer; import org.eclipse.jdt.internal.compiler.problem.DefaultProblemFactory; /** * Eclipse compiler implementation */ public final class EclipseJavaCompiler extends AbstractJavaCompiler { private String prefix = ""; private final EclipseJavaCompilerSettings defaultSettings; public EclipseJavaCompiler() { this(new EclipseJavaCompilerSettings(), ""); } public EclipseJavaCompiler(final Map pSettings) { defaultSettings = new EclipseJavaCompilerSettings(pSettings); } public EclipseJavaCompiler(final EclipseJavaCompilerSettings pSettings, String prefix) { defaultSettings = pSettings; this.prefix = prefix; } public String getPathName(String fullPath) { if (prefix.length() == 0) { return fullPath; } if (fullPath.charAt(0) == '/') { return fullPath.substring(prefix.length() + 1); } else { return fullPath.substring(prefix.length()); } } final class CompilationUnit implements ICompilationUnit { final private String clazzName; final private String fileName; final private char[] typeName; final private char[][] packageName; final private ResourceReader reader; CompilationUnit(final ResourceReader pReader, final String pSourceFile) { reader = pReader; clazzName = ClassUtils.convertResourceToClassName(getPathName(pSourceFile)); fileName = pSourceFile; int dot = clazzName.lastIndexOf('.'); if (dot > 0) { typeName = clazzName.substring(dot + 1).toCharArray(); } else { typeName = clazzName.toCharArray(); } final StringTokenizer izer = new StringTokenizer(clazzName, "."); packageName = new char[izer.countTokens() - 1][]; for (int i = 0; i < packageName.length; i++) { packageName[i] = izer.nextToken().toCharArray(); } } public char[] getFileName() { return fileName.toCharArray(); } public char[] getContents() { final byte[] content = reader.getBytes(fileName); if (content == null) { return null; //throw new RuntimeException("resource " + fileName + " could not be found"); } return new String(content).toCharArray(); } public char[] getMainTypeName() { return typeName; } public char[][] getPackageName() { return packageName; } public boolean ignoreOptionalProblems() { return true; } } public org.drools.compiler.commons.jci.compilers.CompilationResult compile(final String[] pSourceFiles, final ResourceReader pReader, final ResourceStore pStore, final ClassLoader pClassLoader, final JavaCompilerSettings pSettings) { final Map settingsMap = new EclipseJavaCompilerSettings(pSettings).toNativeSettings(); final Collection problems = new ArrayList(); final ICompilationUnit[] compilationUnits = new ICompilationUnit[pSourceFiles.length]; for (int i = 0; i < compilationUnits.length; i++) { final String sourceFile = pSourceFiles[i]; if (pReader.isAvailable(sourceFile)) { compilationUnits[i] = new CompilationUnit(pReader, sourceFile); } else { // log.error("source not found " + sourceFile); final CompilationProblem problem = new CompilationProblem() { public int getEndColumn() { return 0; } public int getEndLine() { return 0; } public String getFileName() { return sourceFile; } public String getMessage() { return "Source " + sourceFile + " could not be found"; } public int getStartColumn() { return 0; } public int getStartLine() { return 0; } public boolean isError() { return true; } public String toString() { return getMessage(); } }; if (problemHandler != null) { problemHandler.handle(problem); } problems.add(problem); } } if (problems.size() > 0) { final CompilationProblem[] result = new CompilationProblem[problems.size()]; problems.toArray(result); return new org.drools.compiler.commons.jci.compilers.CompilationResult(result); } final IErrorHandlingPolicy policy = DefaultErrorHandlingPolicies.proceedWithAllProblems(); final IProblemFactory problemFactory = new DefaultProblemFactory(Locale.getDefault()); final INameEnvironment nameEnvironment = new INameEnvironment() { public NameEnvironmentAnswer findType(final char[][] pCompoundTypeName) { final StringBuilder result = new StringBuilder(); for (int i = 0; i < pCompoundTypeName.length; i++) { if (i != 0) { result.append('.'); } result.append(pCompoundTypeName[i]); } //log.debug("finding compoundTypeName=" + result.toString()); return findType(result.toString()); } public NameEnvironmentAnswer findType(final char[] pTypeName, final char[][] pPackageName) { final StringBuilder result = new StringBuilder(); for (int i = 0; i < pPackageName.length; i++) { result.append(pPackageName[i]); result.append('.'); } // log.debug("finding typeName=" + new String(typeName) + " packageName=" + result.toString()); result.append(pTypeName); return findType(result.toString()); } private NameEnvironmentAnswer findType(final String pClazzName) { final String resourceName = ClassUtils.convertClassToResourcePath(pClazzName); final byte[] clazzBytes = pStore.read(resourceName); if (clazzBytes != null) { try { return createNameEnvironmentAnswer(pClazzName, clazzBytes); } catch (final ClassFormatException e) { throw new RuntimeException( "ClassFormatException in loading class '" + pClazzName + "' with JCI."); } } InputStream is = null; ByteArrayOutputStream baos = null; try { is = pClassLoader.getResourceAsStream(resourceName); if (is == null) { return null; } if (ClassUtils.isWindows() || ClassUtils.isOSX()) { // check it really is a class, this issue is due to windows case sensitivity issues for the class org.kie.Process and path org/droosl/process try { pClassLoader.loadClass(pClazzName); } catch (ClassNotFoundException e) { return null; } catch (NoClassDefFoundError e) { return null; } } final byte[] buffer = new byte[8192]; baos = new ByteArrayOutputStream(buffer.length); int count; while ((count = is.read(buffer, 0, buffer.length)) > 0) { baos.write(buffer, 0, count); } baos.flush(); return createNameEnvironmentAnswer(pClazzName, baos.toByteArray()); } catch (final IOException e) { throw new RuntimeException("could not read class", e); } catch (final ClassFormatException e) { throw new RuntimeException("wrong class format", e); } finally { try { if (baos != null) { baos.close(); } } catch (final IOException oe) { throw new RuntimeException("could not close output stream", oe); } try { if (is != null) { is.close(); } } catch (final IOException ie) { throw new RuntimeException("could not close input stream", ie); } } } private NameEnvironmentAnswer createNameEnvironmentAnswer(final String pClazzName, final byte[] clazzBytes) throws ClassFormatException { final char[] fileName = pClazzName.toCharArray(); final ClassFileReader classFileReader = new ClassFileReader(clazzBytes, fileName, true); return new NameEnvironmentAnswer(classFileReader, null); } private boolean isSourceAvailable(final String pClazzName, final ResourceReader pReader) { // FIXME: this should not be tied to the extension final String javaSource = pClazzName.replace('.', '/') + ".java"; final String classSource = pClazzName.replace('.', '/') + ".class"; return pReader.isAvailable(prefix + javaSource) || pReader.isAvailable(prefix + classSource); } private boolean isPackage(final String pClazzName) { InputStream is = null; try { is = pClassLoader.getResourceAsStream(ClassUtils.convertClassToResourcePath(pClazzName)); if (is != null) { if (ClassUtils.isWindows() || ClassUtils.isOSX()) { // check it really is a class, this issue is due to windows case sensitivity issues for the class org.kie.Process and path org/droosl/process try { Class cls = pClassLoader.loadClass(pClazzName); if (cls != null) { return true; } } catch (ClassNotFoundException e) { return true; } catch (NoClassDefFoundError e) { return true; } } } boolean result = is == null && !isSourceAvailable(pClazzName, pReader); return result; } finally { if (is != null) { try { is.close(); } catch (IOException e) { throw new RuntimeException("Unable to close stream for resource: " + pClazzName); } } } } public boolean isPackage(char[][] parentPackageName, char[] pPackageName) { final StringBuilder result = new StringBuilder(); if (parentPackageName != null) { for (int i = 0; i < parentPackageName.length; i++) { if (i != 0) { result.append('.'); } result.append(parentPackageName[i]); } } // log.debug("isPackage parentPackageName=" + result.toString() + " packageName=" + new String(packageName)); if (parentPackageName != null && parentPackageName.length > 0) { result.append('.'); } result.append(pPackageName); return isPackage(result.toString()); } public void cleanup() { } }; final ICompilerRequestor compilerRequestor = new ICompilerRequestor() { public void acceptResult(final CompilationResult pResult) { if (pResult.hasProblems()) { final IProblem[] iproblems = pResult.getProblems(); for (int i = 0; i < iproblems.length; i++) { final IProblem iproblem = iproblems[i]; final CompilationProblem problem = new EclipseCompilationProblem(iproblem); if (problemHandler != null) { problemHandler.handle(problem); } problems.add(problem); } } if (!pResult.hasErrors()) { final ClassFile[] clazzFiles = pResult.getClassFiles(); for (int i = 0; i < clazzFiles.length; i++) { final ClassFile clazzFile = clazzFiles[i]; final char[][] compoundName = clazzFile.getCompoundName(); final StringBuilder clazzName = new StringBuilder(); for (int j = 0; j < compoundName.length; j++) { if (j != 0) { clazzName.append('.'); } clazzName.append(compoundName[j]); } pStore.write(clazzName.toString().replace('.', '/') + ".class", clazzFile.getBytes()); } } } }; final Compiler compiler = new Compiler(nameEnvironment, policy, settingsMap, compilerRequestor, problemFactory, false); compiler.compile(compilationUnits); final CompilationProblem[] result = new CompilationProblem[problems.size()]; problems.toArray(result); return new org.drools.compiler.commons.jci.compilers.CompilationResult(result); } public JavaCompilerSettings createDefaultSettings() { return this.defaultSettings; } }