/**
* 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.avro.mojo;
import java.io.File;
import java.io.IOException;
import java.net.URL;
import java.net.URLClassLoader;
import java.util.List;
import org.apache.avro.Protocol;
import org.apache.avro.compiler.idl.Idl;
import org.apache.avro.compiler.idl.ParseException;
import org.apache.avro.compiler.specific.SpecificCompiler;
import org.apache.avro.generic.GenericData;
import org.apache.maven.artifact.DependencyResolutionRequiredException;
/**
* Generate Java classes and interfaces from AvroIDL files (.avdl)
*
* @goal idl-protocol
* @requiresDependencyResolution runtime
* @phase generate-sources
*/
public class IDLProtocolMojo extends AbstractAvroMojo {
/**
* A set of Ant-like inclusion patterns used to select files from the source
* directory for processing. By default, the pattern
* <code>**/*.avdl</code> is used to select IDL files.
*
* @parameter
*/
private String[] includes = new String[] { "**/*.avdl" };
/**
* A set of Ant-like inclusion patterns used to select files from the source
* directory for processing. By default, the pattern
* <code>**/*.avdl</code> is used to select IDL files.
*
* @parameter
*/
private String[] testIncludes = new String[] { "**/*.avdl" };
@Override
protected void doCompile(String filename, File sourceDirectory, File outputDirectory) throws IOException {
try {
@SuppressWarnings("rawtypes")
List runtimeClasspathElements = project.getRuntimeClasspathElements();
Idl parser;
// If runtimeClasspathElements is not empty values add its values to Idl path.
if(runtimeClasspathElements == null || runtimeClasspathElements.isEmpty()) {
parser = new Idl(new File(sourceDirectory, filename));
} else {
URL[] runtimeUrls = new URL[runtimeClasspathElements.size()];
for (int i = 0; i < runtimeClasspathElements.size(); i++) {
String element = (String) runtimeClasspathElements.get(i);
runtimeUrls[i] = new File(element).toURI().toURL();
}
URLClassLoader projPathLoader = new URLClassLoader
(runtimeUrls, Thread.currentThread().getContextClassLoader());
parser = new Idl(new File(sourceDirectory, filename), projPathLoader);
}
Protocol p = parser.CompilationUnit();
String json = p.toString(true);
Protocol protocol = Protocol.parse(json);
SpecificCompiler compiler = new SpecificCompiler(protocol);
compiler.setStringType(GenericData.StringType.valueOf(stringType));
compiler.setTemplateDir(templateDirectory);
compiler.setFieldVisibility(getFieldVisibility());
compiler.setCreateSetters(createSetters);
compiler.compileToDestination(null, outputDirectory);
} catch (ParseException e) {
throw new IOException(e);
} catch (DependencyResolutionRequiredException drre) {
throw new IOException(drre);
}
}
@Override
protected String[] getIncludes() {
return includes;
}
@Override
protected String[] getTestIncludes() {
return testIncludes;
}
}