Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
add MotanAsync annotation for async call
- Loading branch information
1 parent
3c7aac1
commit 6a007a4
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,24 @@ | ||
| /* | ||
| * Copyright 2009-2016 Weibo, Inc. | ||
| * | ||
| * Licensed 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 com.weibo.api.motan.rpc; | ||
|
|
||
| public interface ResponseFuture extends Future, Response{ | ||
| public void onSuccess(Response response); | ||
This comment has been minimized.
Sorry, something went wrong.
This comment has been minimized.
Sorry, something went wrong. |
||
|
|
||
| public void onFailure(Response response) ; | ||
|
|
||
| public long getCreateTime(); | ||
| } | ||
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,34 @@ | ||
| /* | ||
| * Copyright 2009-2016 Weibo, Inc. | ||
| * | ||
| * Licensed 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 com.weibo.api.motan.transport.async; | ||
|
|
||
| import java.lang.annotation.Documented; | ||
| import java.lang.annotation.ElementType; | ||
| import java.lang.annotation.Retention; | ||
| import java.lang.annotation.RetentionPolicy; | ||
| import java.lang.annotation.Target; | ||
|
|
||
| /** | ||
| * | ||
| * @Description async request annotation | ||
| * @author zhanglei | ||
| * @date Feb 21, 2017 | ||
| * | ||
| */ | ||
| @Documented | ||
| @Target({ElementType.TYPE}) | ||
| @Retention(RetentionPolicy.SOURCE) | ||
| public @interface MotanAsync { | ||
|
|
||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,168 @@ | ||
| /* | ||
| * Copyright 2009-2016 Weibo, Inc. | ||
| * | ||
| * Licensed 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 com.weibo.api.motan.transport.async; | ||
|
|
||
| import java.io.File; | ||
| import java.io.IOException; | ||
| import java.util.ArrayList; | ||
| import java.util.List; | ||
| import java.util.Set; | ||
|
|
||
| import javax.annotation.processing.AbstractProcessor; | ||
| import javax.annotation.processing.ProcessingEnvironment; | ||
| import javax.annotation.processing.RoundEnvironment; | ||
| import javax.annotation.processing.SupportedAnnotationTypes; | ||
| import javax.annotation.processing.SupportedSourceVersion; | ||
| import javax.lang.model.SourceVersion; | ||
| import javax.lang.model.element.Element; | ||
| import javax.lang.model.element.ElementKind; | ||
| import javax.lang.model.element.ExecutableElement; | ||
| import javax.lang.model.element.Modifier; | ||
| import javax.lang.model.element.TypeElement; | ||
| import javax.lang.model.element.TypeParameterElement; | ||
| import javax.lang.model.element.VariableElement; | ||
| import javax.lang.model.type.DeclaredType; | ||
| import javax.lang.model.type.TypeKind; | ||
| import javax.lang.model.type.TypeMirror; | ||
| import javax.tools.Diagnostic; | ||
|
|
||
| import com.squareup.javapoet.JavaFile; | ||
| import com.squareup.javapoet.MethodSpec; | ||
| import com.squareup.javapoet.ParameterSpec; | ||
| import com.squareup.javapoet.TypeName; | ||
| import com.squareup.javapoet.TypeSpec; | ||
| import com.squareup.javapoet.TypeVariableName; | ||
| import com.weibo.api.motan.common.MotanConstants; | ||
| import com.weibo.api.motan.rpc.ResponseFuture; | ||
|
|
||
| /** | ||
| * | ||
| * @Description create a {interfaceName}Async class with all method add a 'Async' suffix. TODO | ||
| * support methods inherited from superinterfaces | ||
| * @author zhanglei | ||
| * @date Feb 24, 2017 | ||
| * | ||
| */ | ||
| @SupportedSourceVersion(SourceVersion.RELEASE_6) | ||
| @SupportedAnnotationTypes({"com.weibo.api.motan.transport.async.MotanAsync"}) | ||
| public class MotanAsyncProcessor extends AbstractProcessor { | ||
| protected static String ASYNC = MotanConstants.ASYNC_SUFFIX; | ||
| protected static String SRC_DIR = "src/main/java/"; | ||
| protected static String TARGET_DIR = "target/generated-sources/annotations/"; | ||
|
|
||
| public synchronized void init(ProcessingEnvironment processingEnv) { | ||
| super.init(processingEnv); | ||
| } | ||
|
|
||
| @Override | ||
| public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment roundEnv) { | ||
| if (roundEnv.processingOver()) { | ||
| return true; | ||
| } | ||
| for (Element elem : roundEnv.getElementsAnnotatedWith(MotanAsync.class)) { | ||
| processingEnv.getMessager().printMessage(Diagnostic.Kind.NOTE, "MotanAsyncProcessor will process " + elem.toString()); | ||
| try { | ||
| writeAsyncClass(elem); | ||
| processingEnv.getMessager().printMessage(Diagnostic.Kind.NOTE, "MotanAsyncProcessor done for " + elem.toString()); | ||
| } catch (Exception e) { | ||
| processingEnv.getMessager().printMessage(Diagnostic.Kind.WARNING, | ||
| "MotanAsyncProcessor process " + elem.toString() + " fail. exception:" + e.getMessage()); | ||
| e.printStackTrace(); | ||
| } | ||
| } | ||
| return true; | ||
| } | ||
|
|
||
| private void writeAsyncClass(Element elem) throws ClassNotFoundException, IOException, Exception { | ||
|
|
||
| if (elem.getKind().isInterface()) { | ||
| TypeElement interfaceClazz = (TypeElement) elem; | ||
| String className = interfaceClazz.getSimpleName().toString(); | ||
| TypeSpec.Builder classBuilder = | ||
| TypeSpec.interfaceBuilder(className + ASYNC).addModifiers(Modifier.PUBLIC) | ||
| .addSuperinterface(TypeName.get(elem.asType())); | ||
|
|
||
| // add class generic type | ||
| classBuilder.addTypeVariables(getTypeNames(interfaceClazz.getTypeParameters())); | ||
|
|
||
| // add direct method | ||
| addMethods(interfaceClazz, classBuilder); | ||
|
|
||
| // add method form superinterface | ||
| addSuperInterfaceMethods(interfaceClazz.getInterfaces(), classBuilder); | ||
|
|
||
| // write class | ||
| JavaFile javaFile = | ||
| JavaFile.builder(processingEnv.getElementUtils().getPackageOf(interfaceClazz).getQualifiedName().toString(), | ||
| classBuilder.build()).build(); | ||
|
|
||
| javaFile.writeTo(new File(System.getProperty("basedir"), TARGET_DIR)); | ||
|
|
||
| } else { | ||
| processingEnv.getMessager().printMessage(Diagnostic.Kind.NOTE, | ||
| "MotanAsyncProcessor not process, because " + elem.toString() + " not a interface."); | ||
| } | ||
| } | ||
|
|
||
| private void addMethods(TypeElement interfaceClazz, TypeSpec.Builder classBuilder) { | ||
| List<? extends Element> elements = interfaceClazz.getEnclosedElements(); | ||
| if (elements != null && !elements.isEmpty()) { | ||
| for (Element e : elements) { | ||
| if (ElementKind.METHOD.equals(e.getKind())) { | ||
| ExecutableElement method = (ExecutableElement) e; | ||
| MethodSpec.Builder methodBuilder = | ||
| MethodSpec.methodBuilder(method.getSimpleName().toString() + ASYNC) | ||
| .addModifiers(Modifier.PUBLIC, Modifier.ABSTRACT).returns(ResponseFuture.class) | ||
| .addTypeVariables(getTypeNames(method.getTypeParameters())); | ||
| // add method params | ||
| List<? extends VariableElement> vars = method.getParameters(); | ||
| for (VariableElement var : vars) { | ||
| methodBuilder.addParameter(ParameterSpec.builder(TypeName.get(var.asType()), var.getSimpleName().toString()) | ||
| .build()); | ||
| } | ||
| classBuilder.addMethod(methodBuilder.build()); | ||
| } | ||
| } | ||
| } | ||
| } | ||
|
|
||
| private List<TypeVariableName> getTypeNames(List<? extends TypeParameterElement> types) { | ||
| List<TypeVariableName> result = new ArrayList<TypeVariableName>(); | ||
| if (types != null && !types.isEmpty()) { | ||
| for (TypeParameterElement type : types) { | ||
| result.add(TypeVariableName.get(type)); | ||
| } | ||
| } | ||
| return result; | ||
| } | ||
|
|
||
| private void addSuperInterfaceMethods(List<? extends TypeMirror> superInterfaces, TypeSpec.Builder classBuilder) { | ||
| if (superInterfaces != null && !superInterfaces.isEmpty()) { | ||
| for (TypeMirror tm : superInterfaces) { | ||
| try { | ||
| if (tm.getKind().equals(TypeKind.DECLARED)) { | ||
| TypeElement de = (TypeElement) ((DeclaredType) tm).asElement(); | ||
| addMethods(de, classBuilder); | ||
| addSuperInterfaceMethods(de.getInterfaces(), classBuilder); | ||
| } | ||
| } catch (Exception e) { | ||
| processingEnv.getMessager().printMessage(Diagnostic.Kind.WARNING, | ||
| "MotanAsyncProcessor process superinterface " + tm.toString() + " fail. exception:" + e.getMessage()); | ||
| e.printStackTrace(); | ||
| } | ||
| } | ||
| } | ||
| } | ||
|
|
||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1 @@ | ||
| com.weibo.api.motan.transport.async.MotanAsyncProcessor |
Oops, something went wrong.
'public' is redundant for interface methods