compiler 重构
This commit is contained in:
AriaLyy
@@ -16,6 +16,7 @@
|
||||
package com.arialyy.compiler;
|
||||
|
||||
import com.arialyy.annotations.Download;
|
||||
import com.arialyy.annotations.DownloadGroup;
|
||||
import com.arialyy.annotations.Upload;
|
||||
import com.google.auto.service.AutoService;
|
||||
import java.util.LinkedHashSet;
|
||||
@@ -53,17 +54,24 @@ import javax.lang.model.element.TypeElement;
|
||||
annotataions.add(Download.onTaskRunning.class.getCanonicalName());
|
||||
annotataions.add(Download.onTaskStart.class.getCanonicalName());
|
||||
annotataions.add(Download.onTaskStop.class.getCanonicalName());
|
||||
//下载任务的注解
|
||||
annotataions.add(Download.onPre.class.getCanonicalName());
|
||||
annotataions.add(Download.onNoSupportBreakPoint.class.getCanonicalName());
|
||||
annotataions.add(Download.onTaskCancel.class.getCanonicalName());
|
||||
annotataions.add(Download.onTaskComplete.class.getCanonicalName());
|
||||
annotataions.add(Download.onTaskFail.class.getCanonicalName());
|
||||
annotataions.add(Download.onTaskPre.class.getCanonicalName());
|
||||
annotataions.add(Download.onTaskResume.class.getCanonicalName());
|
||||
annotataions.add(Download.onTaskRunning.class.getCanonicalName());
|
||||
annotataions.add(Download.onTaskStart.class.getCanonicalName());
|
||||
annotataions.add(Download.onTaskStop.class.getCanonicalName());
|
||||
//下载任务组的注解
|
||||
annotataions.add(DownloadGroup.onPre.class.getCanonicalName());
|
||||
annotataions.add(DownloadGroup.onTaskCancel.class.getCanonicalName());
|
||||
annotataions.add(DownloadGroup.onTaskComplete.class.getCanonicalName());
|
||||
annotataions.add(DownloadGroup.onTaskFail.class.getCanonicalName());
|
||||
annotataions.add(DownloadGroup.onTaskPre.class.getCanonicalName());
|
||||
annotataions.add(DownloadGroup.onTaskResume.class.getCanonicalName());
|
||||
annotataions.add(DownloadGroup.onTaskRunning.class.getCanonicalName());
|
||||
annotataions.add(DownloadGroup.onTaskStart.class.getCanonicalName());
|
||||
annotataions.add(DownloadGroup.onTaskStop.class.getCanonicalName());
|
||||
//任务组子任务的注解
|
||||
annotataions.add(DownloadGroup.onSubTaskPre.class.getCanonicalName());
|
||||
annotataions.add(DownloadGroup.onSubTaskCancel.class.getCanonicalName());
|
||||
annotataions.add(DownloadGroup.onSubTaskComplete.class.getCanonicalName());
|
||||
annotataions.add(DownloadGroup.onSubTaskFail.class.getCanonicalName());
|
||||
annotataions.add(DownloadGroup.onSubTaskRunning.class.getCanonicalName());
|
||||
annotataions.add(DownloadGroup.onSubTaskStart.class.getCanonicalName());
|
||||
annotataions.add(DownloadGroup.onSubTaskStop.class.getCanonicalName());
|
||||
//上传任务的注解
|
||||
annotataions.add(Upload.onPre.class.getCanonicalName());
|
||||
annotataions.add(Upload.onNoSupportBreakPoint.class.getCanonicalName());
|
||||
@@ -86,6 +94,7 @@ import javax.lang.model.element.TypeElement;
|
||||
mHandler.clean();
|
||||
mHandler.handleDownload(roundEnv);
|
||||
mHandler.handleDownloadGroup(roundEnv);
|
||||
mHandler.handleDownloadGroupSub(roundEnv);
|
||||
mHandler.handleUpload(roundEnv);
|
||||
mHandler.createProxyFile();
|
||||
return true;
|
||||
|
||||
123
AriaCompiler/src/main/java/com/arialyy/compiler/CountFiler.java
Normal file
123
AriaCompiler/src/main/java/com/arialyy/compiler/CountFiler.java
Normal file
@@ -0,0 +1,123 @@
|
||||
/*
|
||||
* Copyright (C) 2016 AriaLyy(https://github.com/AriaLyy/Aria)
|
||||
*
|
||||
* 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.arialyy.compiler;
|
||||
|
||||
import com.squareup.javapoet.ClassName;
|
||||
import com.squareup.javapoet.CodeBlock;
|
||||
import com.squareup.javapoet.FieldSpec;
|
||||
import com.squareup.javapoet.JavaFile;
|
||||
import com.squareup.javapoet.MethodSpec;
|
||||
import com.squareup.javapoet.ParameterizedTypeName;
|
||||
import com.squareup.javapoet.TypeSpec;
|
||||
import java.io.IOException;
|
||||
import java.util.HashMap;
|
||||
import java.util.HashSet;
|
||||
import java.util.Map;
|
||||
import java.util.Set;
|
||||
import javax.annotation.processing.Filer;
|
||||
import javax.lang.model.element.Modifier;
|
||||
|
||||
/**
|
||||
* Created by lyy on 2017/9/6.
|
||||
* 各类注解统计技术类
|
||||
*/
|
||||
final class CountFiler {
|
||||
private Filer mFiler;
|
||||
private ParamObtainUtil mPbUtil;
|
||||
|
||||
CountFiler(Filer filer, ParamObtainUtil pbUtil) {
|
||||
mFiler = filer;
|
||||
mPbUtil = pbUtil;
|
||||
}
|
||||
|
||||
/**
|
||||
* 每一种注解对应的统计类
|
||||
*/
|
||||
void createCountFile() throws IOException {
|
||||
Set<String> keys = mPbUtil.getListenerClass().keySet();
|
||||
TypeSpec.Builder builder = TypeSpec.classBuilder(ProxyConstance.PROXY_COUNTER_NAME)
|
||||
.addModifiers(Modifier.PUBLIC, Modifier.FINAL);
|
||||
|
||||
FieldSpec mappingField = FieldSpec.builder(
|
||||
ParameterizedTypeName.get(ClassName.get(Map.class), ClassName.get(String.class),
|
||||
ParameterizedTypeName.get(ClassName.get(Set.class), ClassName.get(String.class))),
|
||||
ProxyConstance.PROXY_COUNTER_MAP)
|
||||
.addModifiers(Modifier.PRIVATE)
|
||||
.initializer("new $T()", HashMap.class)
|
||||
.build();
|
||||
builder.addField(mappingField);
|
||||
|
||||
//增加构造函数
|
||||
CodeBlock.Builder cb = CodeBlock.builder();
|
||||
cb.add("Set<String> set = null;\n");
|
||||
for (String key : keys) {
|
||||
addTypeData(key, mPbUtil.getListenerClass().get(key), cb);
|
||||
}
|
||||
MethodSpec structure =
|
||||
MethodSpec.constructorBuilder().addModifiers(Modifier.PUBLIC).addCode(cb.build()).build();
|
||||
builder.addMethod(structure);
|
||||
|
||||
builder.addMethod(
|
||||
createMethod(ProxyConstance.COUNT_METHOD_DOWNLOAD, ProxyConstance.COUNT_DOWNLOAD));
|
||||
builder.addMethod(
|
||||
createMethod(ProxyConstance.COUNT_METHOD_UPLOAD, ProxyConstance.COUNT_UPLOAD));
|
||||
builder.addMethod(createMethod(ProxyConstance.COUNT_METHOD_DOWNLOAD_GROUP,
|
||||
ProxyConstance.COUNT_DOWNLOAD_GROUP));
|
||||
|
||||
JavaFile jf = JavaFile.builder(ProxyConstance.PROXY_COUNTER_PACKAGE, builder.build()).build();
|
||||
createFile(jf);
|
||||
}
|
||||
|
||||
/**
|
||||
* 创建不同任务类型的代理类集合
|
||||
*
|
||||
* @param key {@link ParamObtainUtil#addListenerMapping(String, String)}
|
||||
*/
|
||||
private MethodSpec createMethod(String methodName, String key) {
|
||||
MethodSpec.Builder builder = MethodSpec.methodBuilder(methodName);
|
||||
ParameterizedTypeName returnName =
|
||||
ParameterizedTypeName.get(ClassName.get(Set.class), ClassName.get(String.class));
|
||||
builder.addModifiers(Modifier.PUBLIC, Modifier.FINAL)
|
||||
.returns(returnName)
|
||||
.addCode("return " + ProxyConstance.PROXY_COUNTER_MAP + ".get(\"" + key + "\");\n");
|
||||
return builder.build();
|
||||
}
|
||||
|
||||
/**
|
||||
* 添加每一种注解对应类
|
||||
*
|
||||
* @param type {@link ParamObtainUtil#addListenerMapping(String, String)}
|
||||
*/
|
||||
private void addTypeData(String type, Set<String> clsNames, CodeBlock.Builder cb) {
|
||||
if (clsNames == null || clsNames.isEmpty()) return;
|
||||
StringBuilder sb = new StringBuilder();
|
||||
sb.append("set = new $T();\n");
|
||||
for (String clsName : clsNames) {
|
||||
sb.append("set.add(\"").append(clsName).append("\");\n");
|
||||
}
|
||||
sb.append("typeMapping.put(\"").append(type).append("\", ").append("set);\n");
|
||||
cb.add(sb.toString(), ClassName.get(HashSet.class));
|
||||
}
|
||||
|
||||
private void createFile(JavaFile jf) throws IOException {
|
||||
if (ProxyConstance.DEBUG) {
|
||||
// 如果需要在控制台打印生成的文件，则去掉下面的注释
|
||||
jf.writeTo(System.out);
|
||||
} else {
|
||||
jf.writeTo(mFiler);
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -18,31 +18,9 @@ package com.arialyy.compiler;
|
||||
import com.arialyy.annotations.Download;
|
||||
import com.arialyy.annotations.DownloadGroup;
|
||||
import com.arialyy.annotations.Upload;
|
||||
import com.squareup.javapoet.ClassName;
|
||||
import com.squareup.javapoet.CodeBlock;
|
||||
import com.squareup.javapoet.FieldSpec;
|
||||
import com.squareup.javapoet.JavaFile;
|
||||
import com.squareup.javapoet.MethodSpec;
|
||||
import com.squareup.javapoet.ParameterSpec;
|
||||
import com.squareup.javapoet.TypeSpec;
|
||||
import com.squareup.javapoet.ParameterizedTypeName;
|
||||
import java.io.IOException;
|
||||
import java.lang.annotation.Annotation;
|
||||
import java.util.Collections;
|
||||
import java.util.HashMap;
|
||||
import java.util.HashSet;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
import java.util.Set;
|
||||
import javax.annotation.processing.Filer;
|
||||
import javax.annotation.processing.RoundEnvironment;
|
||||
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.PackageElement;
|
||||
import javax.lang.model.element.TypeElement;
|
||||
import javax.lang.model.element.VariableElement;
|
||||
import javax.lang.model.util.Elements;
|
||||
|
||||
/**
|
||||
@@ -50,16 +28,13 @@ import javax.lang.model.util.Elements;
|
||||
* 元素处理
|
||||
*/
|
||||
class ElementHandler {
|
||||
private static final boolean DEBUG = false;
|
||||
|
||||
private Filer mFiler;
|
||||
private Elements mElementUtil;
|
||||
private Map<String, ProxyMethodParam> mMethods = new HashMap<>();
|
||||
private Map<String, Set<String>> mListenerClass = new HashMap<>();
|
||||
private ParamObtainUtil mPbUtil;
|
||||
|
||||
ElementHandler(Filer filer, Elements elements) {
|
||||
mFiler = filer;
|
||||
mElementUtil = elements;
|
||||
mPbUtil = new ParamObtainUtil(elements);
|
||||
}
|
||||
|
||||
/**
|
||||
@@ -69,43 +44,68 @@ class ElementHandler {
|
||||
* PackageElement 一般代表Package
|
||||
*/
|
||||
void handleDownload(RoundEnvironment roundEnv) {
|
||||
saveMethod(TaskEnum.DOWNLOAD, roundEnv, Download.onNoSupportBreakPoint.class,
|
||||
mPbUtil.saveMethod(TaskEnum.DOWNLOAD, roundEnv, Download.onNoSupportBreakPoint.class,
|
||||
ProxyConstance.TASK_NO_SUPPORT_BREAKPOINT);
|
||||
saveMethod(TaskEnum.DOWNLOAD, roundEnv, Download.onPre.class, ProxyConstance.PRE);
|
||||
saveMethod(TaskEnum.DOWNLOAD, roundEnv, Download.onTaskCancel.class,
|
||||
mPbUtil.saveMethod(TaskEnum.DOWNLOAD, roundEnv, Download.onPre.class, ProxyConstance.PRE);
|
||||
mPbUtil.saveMethod(TaskEnum.DOWNLOAD, roundEnv, Download.onTaskCancel.class,
|
||||
ProxyConstance.TASK_CANCEL);
|
||||
saveMethod(TaskEnum.DOWNLOAD, roundEnv, Download.onTaskComplete.class,
|
||||
mPbUtil.saveMethod(TaskEnum.DOWNLOAD, roundEnv, Download.onTaskComplete.class,
|
||||
ProxyConstance.TASK_COMPLETE);
|
||||
saveMethod(TaskEnum.DOWNLOAD, roundEnv, Download.onTaskFail.class, ProxyConstance.TASK_FAIL);
|
||||
saveMethod(TaskEnum.DOWNLOAD, roundEnv, Download.onTaskPre.class, ProxyConstance.TASK_PRE);
|
||||
saveMethod(TaskEnum.DOWNLOAD, roundEnv, Download.onTaskResume.class,
|
||||
mPbUtil.saveMethod(TaskEnum.DOWNLOAD, roundEnv, Download.onTaskFail.class,
|
||||
ProxyConstance.TASK_FAIL);
|
||||
mPbUtil.saveMethod(TaskEnum.DOWNLOAD, roundEnv, Download.onTaskPre.class,
|
||||
ProxyConstance.TASK_PRE);
|
||||
mPbUtil.saveMethod(TaskEnum.DOWNLOAD, roundEnv, Download.onTaskResume.class,
|
||||
ProxyConstance.TASK_RESUME);
|
||||
saveMethod(TaskEnum.DOWNLOAD, roundEnv, Download.onTaskRunning.class,
|
||||
mPbUtil.saveMethod(TaskEnum.DOWNLOAD, roundEnv, Download.onTaskRunning.class,
|
||||
ProxyConstance.TASK_RUNNING);
|
||||
saveMethod(TaskEnum.DOWNLOAD, roundEnv, Download.onTaskStart.class, ProxyConstance.TASK_START);
|
||||
saveMethod(TaskEnum.DOWNLOAD, roundEnv, Download.onTaskStop.class, ProxyConstance.TASK_STOP);
|
||||
mPbUtil.saveMethod(TaskEnum.DOWNLOAD, roundEnv, Download.onTaskStart.class,
|
||||
ProxyConstance.TASK_START);
|
||||
mPbUtil.saveMethod(TaskEnum.DOWNLOAD, roundEnv, Download.onTaskStop.class,
|
||||
ProxyConstance.TASK_STOP);
|
||||
}
|
||||
|
||||
/**
|
||||
* 处理搜索到的下载任务组注解
|
||||
*/
|
||||
void handleDownloadGroup(RoundEnvironment roundEnv) {
|
||||
saveMethod(TaskEnum.DOWNLOAD_GROUP, roundEnv, DownloadGroup.onPre.class, ProxyConstance.PRE);
|
||||
saveMethod(TaskEnum.DOWNLOAD_GROUP, roundEnv, DownloadGroup.onTaskCancel.class,
|
||||
mPbUtil.saveMethod(TaskEnum.DOWNLOAD_GROUP, roundEnv, DownloadGroup.onPre.class,
|
||||
ProxyConstance.PRE);
|
||||
mPbUtil.saveMethod(TaskEnum.DOWNLOAD_GROUP, roundEnv, DownloadGroup.onTaskCancel.class,
|
||||
ProxyConstance.TASK_CANCEL);
|
||||
saveMethod(TaskEnum.DOWNLOAD_GROUP, roundEnv, DownloadGroup.onTaskComplete.class,
|
||||
mPbUtil.saveMethod(TaskEnum.DOWNLOAD_GROUP, roundEnv, DownloadGroup.onTaskComplete.class,
|
||||
ProxyConstance.TASK_COMPLETE);
|
||||
saveMethod(TaskEnum.DOWNLOAD_GROUP, roundEnv, DownloadGroup.onTaskFail.class,
|
||||
mPbUtil.saveMethod(TaskEnum.DOWNLOAD_GROUP, roundEnv, DownloadGroup.onTaskFail.class,
|
||||
ProxyConstance.TASK_FAIL);
|
||||
saveMethod(TaskEnum.DOWNLOAD_GROUP, roundEnv, DownloadGroup.onTaskPre.class,
|
||||
mPbUtil.saveMethod(TaskEnum.DOWNLOAD_GROUP, roundEnv, DownloadGroup.onTaskPre.class,
|
||||
ProxyConstance.TASK_PRE);
|
||||
saveMethod(TaskEnum.DOWNLOAD_GROUP, roundEnv, DownloadGroup.onTaskResume.class,
|
||||
mPbUtil.saveMethod(TaskEnum.DOWNLOAD_GROUP, roundEnv, DownloadGroup.onTaskResume.class,
|
||||
ProxyConstance.TASK_RESUME);
|
||||
saveMethod(TaskEnum.DOWNLOAD_GROUP, roundEnv, DownloadGroup.onTaskRunning.class,
|
||||
mPbUtil.saveMethod(TaskEnum.DOWNLOAD_GROUP, roundEnv, DownloadGroup.onTaskRunning.class,
|
||||
ProxyConstance.TASK_RUNNING);
|
||||
saveMethod(TaskEnum.DOWNLOAD_GROUP, roundEnv, DownloadGroup.onTaskStart.class,
|
||||
mPbUtil.saveMethod(TaskEnum.DOWNLOAD_GROUP, roundEnv, DownloadGroup.onTaskStart.class,
|
||||
ProxyConstance.TASK_START);
|
||||
saveMethod(TaskEnum.DOWNLOAD_GROUP, roundEnv, DownloadGroup.onTaskStop.class,
|
||||
mPbUtil.saveMethod(TaskEnum.DOWNLOAD_GROUP, roundEnv, DownloadGroup.onTaskStop.class,
|
||||
ProxyConstance.TASK_STOP);
|
||||
}
|
||||
|
||||
/**
|
||||
* 处理搜索到的下载任务组子任务注解
|
||||
*/
|
||||
void handleDownloadGroupSub(RoundEnvironment roundEnv) {
|
||||
mPbUtil.saveMethod(TaskEnum.DOWNLOAD_GROUP_SUB, roundEnv, DownloadGroup.onSubTaskPre.class,
|
||||
ProxyConstance.TASK_PRE);
|
||||
mPbUtil.saveMethod(TaskEnum.DOWNLOAD_GROUP_SUB, roundEnv, DownloadGroup.onSubTaskCancel.class,
|
||||
ProxyConstance.TASK_CANCEL);
|
||||
mPbUtil.saveMethod(TaskEnum.DOWNLOAD_GROUP_SUB, roundEnv, DownloadGroup.onSubTaskComplete.class,
|
||||
ProxyConstance.TASK_COMPLETE);
|
||||
mPbUtil.saveMethod(TaskEnum.DOWNLOAD_GROUP_SUB, roundEnv, DownloadGroup.onSubTaskFail.class,
|
||||
ProxyConstance.TASK_FAIL);
|
||||
mPbUtil.saveMethod(TaskEnum.DOWNLOAD_GROUP_SUB, roundEnv, DownloadGroup.onSubTaskRunning.class,
|
||||
ProxyConstance.TASK_RUNNING);
|
||||
mPbUtil.saveMethod(TaskEnum.DOWNLOAD_GROUP_SUB, roundEnv, DownloadGroup.onSubTaskStart.class,
|
||||
ProxyConstance.TASK_START);
|
||||
mPbUtil.saveMethod(TaskEnum.DOWNLOAD_GROUP_SUB, roundEnv, DownloadGroup.onSubTaskStop.class,
|
||||
ProxyConstance.TASK_STOP);
|
||||
}
|
||||
|
||||
@@ -113,488 +113,38 @@ class ElementHandler {
|
||||
* 处理搜索到的上传注解F
|
||||
*/
|
||||
void handleUpload(RoundEnvironment roundEnv) {
|
||||
saveMethod(TaskEnum.UPLOAD, roundEnv, Upload.onNoSupportBreakPoint.class,
|
||||
mPbUtil.saveMethod(TaskEnum.UPLOAD, roundEnv, Upload.onNoSupportBreakPoint.class,
|
||||
ProxyConstance.TASK_NO_SUPPORT_BREAKPOINT);
|
||||
saveMethod(TaskEnum.UPLOAD, roundEnv, Upload.onPre.class, ProxyConstance.PRE);
|
||||
saveMethod(TaskEnum.UPLOAD, roundEnv, Upload.onTaskCancel.class, ProxyConstance.TASK_CANCEL);
|
||||
saveMethod(TaskEnum.UPLOAD, roundEnv, Upload.onTaskComplete.class,
|
||||
mPbUtil.saveMethod(TaskEnum.UPLOAD, roundEnv, Upload.onPre.class, ProxyConstance.PRE);
|
||||
mPbUtil.saveMethod(TaskEnum.UPLOAD, roundEnv, Upload.onTaskCancel.class,
|
||||
ProxyConstance.TASK_CANCEL);
|
||||
mPbUtil.saveMethod(TaskEnum.UPLOAD, roundEnv, Upload.onTaskComplete.class,
|
||||
ProxyConstance.TASK_COMPLETE);
|
||||
saveMethod(TaskEnum.UPLOAD, roundEnv, Upload.onTaskFail.class, ProxyConstance.TASK_FAIL);
|
||||
saveMethod(TaskEnum.UPLOAD, roundEnv, Upload.onTaskResume.class, ProxyConstance.TASK_RESUME);
|
||||
saveMethod(TaskEnum.UPLOAD, roundEnv, Upload.onTaskRunning.class, ProxyConstance.TASK_RUNNING);
|
||||
saveMethod(TaskEnum.UPLOAD, roundEnv, Upload.onTaskStart.class, ProxyConstance.TASK_START);
|
||||
saveMethod(TaskEnum.UPLOAD, roundEnv, Upload.onTaskStop.class, ProxyConstance.TASK_STOP);
|
||||
mPbUtil.saveMethod(TaskEnum.UPLOAD, roundEnv, Upload.onTaskFail.class,
|
||||
ProxyConstance.TASK_FAIL);
|
||||
mPbUtil.saveMethod(TaskEnum.UPLOAD, roundEnv, Upload.onTaskResume.class,
|
||||
ProxyConstance.TASK_RESUME);
|
||||
mPbUtil.saveMethod(TaskEnum.UPLOAD, roundEnv, Upload.onTaskRunning.class,
|
||||
ProxyConstance.TASK_RUNNING);
|
||||
mPbUtil.saveMethod(TaskEnum.UPLOAD, roundEnv, Upload.onTaskStart.class,
|
||||
ProxyConstance.TASK_START);
|
||||
mPbUtil.saveMethod(TaskEnum.UPLOAD, roundEnv, Upload.onTaskStop.class,
|
||||
ProxyConstance.TASK_STOP);
|
||||
}
|
||||
|
||||
/**
|
||||
* 在build文件夹中生成如下代码的文件
|
||||
* <pre>
|
||||
* <code>
|
||||
* package com.arialyy.simple.download;
|
||||
*
|
||||
* import com.arialyy.aria.core.download.DownloadTask;
|
||||
* import com.arialyy.aria.core.scheduler.AbsSchedulerListener;
|
||||
*
|
||||
* public final class SingleTaskActivity$$DownloadListenerProxy extends
|
||||
* AbsSchedulerListener<DownloadTask> {
|
||||
* private SingleTaskActivity obj;
|
||||
*
|
||||
* public void onPre(final DownloadTask task) {
|
||||
* obj.onPre((DownloadTask)task);
|
||||
* }
|
||||
*
|
||||
* public void onTaskStart(final DownloadTask task) {
|
||||
* obj.onStart((DownloadTask)task);
|
||||
* }
|
||||
*
|
||||
* public void setListener(final Object obj) {
|
||||
* this.obj = (SingleTaskActivity)obj;
|
||||
* }
|
||||
* }
|
||||
* </code>
|
||||
* </pre>
|
||||
* 在build文件夹中生成代理文件
|
||||
*/
|
||||
void createProxyFile() {
|
||||
try {
|
||||
createProxyListenerFile();
|
||||
createProxyClassFile();
|
||||
new EventProxyFiler(mFiler, mPbUtil).createEventProxyFile();
|
||||
//new CountFiler(mFiler, mPbUtil).createCountFile();
|
||||
} catch (IOException e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 创建事件代理文件
|
||||
*/
|
||||
private void createProxyListenerFile() throws IOException {
|
||||
Set<String> keys = mMethods.keySet();
|
||||
for (String key : keys) {
|
||||
ProxyMethodParam entity = mMethods.get(key);
|
||||
for (TaskEnum taskEnum : entity.taskEnums) {
|
||||
JavaFile jf =
|
||||
JavaFile.builder(entity.packageName, createProxyClass(entity, taskEnum)).build();
|
||||
createFile(jf);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private void createFile(JavaFile jf) throws IOException {
|
||||
if (DEBUG) {
|
||||
// 如果需要在控制台打印生成的文件，则去掉下面的注释
|
||||
jf.writeTo(System.out);
|
||||
} else {
|
||||
jf.writeTo(mFiler);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 每一种注解对应的类集合
|
||||
*/
|
||||
private void createProxyClassFile() throws IOException {
|
||||
Set<String> keys = mListenerClass.keySet();
|
||||
TypeSpec.Builder builder = TypeSpec.classBuilder(ProxyConstance.PROXY_COUNTER_NAME)
|
||||
.addModifiers(Modifier.PUBLIC, Modifier.FINAL);
|
||||
|
||||
FieldSpec mappingField = FieldSpec.builder(
|
||||
ParameterizedTypeName.get(ClassName.get(Map.class), ClassName.get(String.class),
|
||||
ParameterizedTypeName.get(ClassName.get(Set.class), ClassName.get(String.class))),
|
||||
ProxyConstance.PROXY_COUNTER_MAP)
|
||||
.addModifiers(Modifier.PRIVATE)
|
||||
.initializer("new $T()", HashMap.class)
|
||||
.build();
|
||||
builder.addField(mappingField);
|
||||
|
||||
//增加构造函数
|
||||
CodeBlock.Builder cb = CodeBlock.builder();
|
||||
cb.add("Set<String> set = null;\n");
|
||||
for (String key : keys) {
|
||||
addTypeData(key, mListenerClass.get(key), cb);
|
||||
}
|
||||
MethodSpec structure =
|
||||
MethodSpec.constructorBuilder().addModifiers(Modifier.PUBLIC).addCode(cb.build()).build();
|
||||
builder.addMethod(structure);
|
||||
|
||||
builder.addMethod(
|
||||
creatMethod(ProxyConstance.COUNT_METHOD_DOWNLOAD, ProxyConstance.COUNT_DOWNLOAD));
|
||||
builder.addMethod(creatMethod(ProxyConstance.COUNT_METHOD_UPLOAD, ProxyConstance.COUNT_UPLOAD));
|
||||
builder.addMethod(creatMethod(ProxyConstance.COUNT_METHOD_DOWNLOAD_GROUP,
|
||||
ProxyConstance.COUNT_DOWNLOAD_GROUP));
|
||||
|
||||
JavaFile jf = JavaFile.builder(ProxyConstance.PROXY_COUNTER_PACKAGE, builder.build()).build();
|
||||
createFile(jf);
|
||||
}
|
||||
|
||||
/**
|
||||
* 创建不同任务类型的代理类集合
|
||||
*
|
||||
* @param key {@link #addListenerMapping(String, String)}
|
||||
*/
|
||||
private MethodSpec creatMethod(String methodName, String key) {
|
||||
MethodSpec.Builder builder = MethodSpec.methodBuilder(methodName);
|
||||
ParameterizedTypeName returnName =
|
||||
ParameterizedTypeName.get(ClassName.get(Set.class), ClassName.get(String.class));
|
||||
builder.addModifiers(Modifier.PUBLIC, Modifier.FINAL)
|
||||
.returns(returnName)
|
||||
.addCode("return " + ProxyConstance.PROXY_COUNTER_MAP + ".get(\"" + key + "\");\n");
|
||||
return builder.build();
|
||||
}
|
||||
|
||||
/**
|
||||
* 添加每一种注解对应类
|
||||
*
|
||||
* @param type {@link #addListenerMapping(String, String)}
|
||||
*/
|
||||
private void addTypeData(String type, Set<String> clsNames, CodeBlock.Builder cb) {
|
||||
if (clsNames == null || clsNames.isEmpty()) return;
|
||||
StringBuilder sb = new StringBuilder();
|
||||
sb.append("set = new $T();\n");
|
||||
for (String clsName : clsNames) {
|
||||
sb.append("set.add(\"").append(clsName).append("\");\n");
|
||||
}
|
||||
sb.append("typeMapping.put(\"").append(type).append("\", ").append("set);\n");
|
||||
cb.add(sb.toString(), ClassName.get(HashSet.class));
|
||||
}
|
||||
|
||||
/**
|
||||
* 创建代理方法
|
||||
*
|
||||
* @param taskEnum 任务类型枚举{@link TaskEnum}
|
||||
* @param annotation {@link Download}、{@link Upload}
|
||||
* @param methodName 被代理类注解的方法名
|
||||
*/
|
||||
private MethodSpec createProxyMethod(TaskEnum taskEnum, Class<? extends Annotation> annotation,
|
||||
String methodName) {
|
||||
ClassName task = ClassName.get(taskEnum.getPkg(), taskEnum.getClassName());
|
||||
|
||||
ParameterSpec parameterSpec =
|
||||
ParameterSpec.builder(task, "task").addModifiers(Modifier.FINAL).build();
|
||||
StringBuilder sb = new StringBuilder();
|
||||
sb.append("Set<String> keys = keyMapping.get(\"").append(methodName).append("\");\n");
|
||||
sb.append("if (keys != null) {\n\tif (keys.contains(task.getKey())) {\n")
|
||||
.append("\t\tobj.")
|
||||
.append(methodName)
|
||||
.append("((")
|
||||
.append(taskEnum.getClassName())
|
||||
.append(")task);\n")
|
||||
.append("\t}\n} else {\n")
|
||||
.append("\tobj.")
|
||||
.append(methodName)
|
||||
.append("((")
|
||||
.append(taskEnum.getClassName())
|
||||
.append(")task);\n}\n");
|
||||
|
||||
return MethodSpec.methodBuilder(annotation.getSimpleName())
|
||||
.addModifiers(Modifier.PUBLIC)
|
||||
.returns(void.class)
|
||||
.addParameter(parameterSpec)
|
||||
.addAnnotation(Override.class)
|
||||
.addCode(sb.toString())
|
||||
.build();
|
||||
}
|
||||
|
||||
/**
|
||||
* 创建代理类
|
||||
*/
|
||||
private TypeSpec createProxyClass(ProxyMethodParam entity, TaskEnum taskEnum) {
|
||||
TypeSpec.Builder builder = TypeSpec.classBuilder(entity.className + taskEnum.getProxySuffix())
|
||||
.addModifiers(Modifier.PUBLIC, Modifier.FINAL);
|
||||
|
||||
//添加被代理的类的字段
|
||||
ClassName obj = ClassName.get(entity.packageName, entity.className);
|
||||
FieldSpec observerField = FieldSpec.builder(obj, "obj").addModifiers(Modifier.PRIVATE).build();
|
||||
builder.addField(observerField);
|
||||
|
||||
//添加url映射表
|
||||
FieldSpec mappingField = FieldSpec.builder(
|
||||
ParameterizedTypeName.get(ClassName.get(Map.class), ClassName.get(String.class),
|
||||
ParameterizedTypeName.get(ClassName.get(Set.class), ClassName.get(String.class))),
|
||||
"keyMapping").addModifiers(Modifier.PRIVATE).initializer("new $T()", HashMap.class).build();
|
||||
builder.addField(mappingField);
|
||||
|
||||
//添加注解方法
|
||||
Map<Class<? extends Annotation>, String> temp = entity.methods.get(taskEnum);
|
||||
if (temp != null) {
|
||||
for (Class<? extends Annotation> annotation : temp.keySet()) {
|
||||
MethodSpec method = createProxyMethod(taskEnum, annotation, temp.get(annotation));
|
||||
builder.addMethod(method);
|
||||
}
|
||||
}
|
||||
|
||||
//增加构造函数
|
||||
CodeBlock.Builder cb = CodeBlock.builder();
|
||||
cb.add("Set<String> set = null;\n");
|
||||
for (String methodName : entity.keyMappings.keySet()) {
|
||||
Set<String> keys = entity.keyMappings.get(methodName);
|
||||
if (keys == null || keys.size() == 0) continue;
|
||||
StringBuilder sb = new StringBuilder();
|
||||
sb.append("set = new $T();\n");
|
||||
for (String key : keys) {
|
||||
if (key.isEmpty()) continue;
|
||||
sb.append("set.add(\"").append(key).append("\");\n");
|
||||
}
|
||||
|
||||
sb.append("keyMapping.put(\"").append(methodName).append("\", ").append("set);\n");
|
||||
cb.add(sb.toString(), ClassName.get(HashSet.class));
|
||||
}
|
||||
MethodSpec structure =
|
||||
MethodSpec.constructorBuilder().addModifiers(Modifier.PUBLIC).addCode(cb.build()).build();
|
||||
builder.addMethod(structure);
|
||||
|
||||
//添加设置代理的类
|
||||
ParameterSpec parameterSpec =
|
||||
ParameterSpec.builder(Object.class, "obj").addModifiers(Modifier.FINAL).build();
|
||||
MethodSpec listener = MethodSpec.methodBuilder(ProxyConstance.SET_LISTENER)
|
||||
.addModifiers(Modifier.PUBLIC)
|
||||
.returns(void.class)
|
||||
.addParameter(parameterSpec)
|
||||
.addAnnotation(Override.class)
|
||||
.addCode("this.obj = (" + entity.className + ")obj;\n")
|
||||
.build();
|
||||
builder.addJavadoc("该文件为Aria自动生成的代理文件，请不要修改该文件的任何代码！\n");
|
||||
|
||||
//创建父类参数
|
||||
ClassName superClass = ClassName.get("com.arialyy.aria.core.scheduler", "AbsSchedulerListener");
|
||||
//创建泛型
|
||||
ClassName typeVariableName = ClassName.get(taskEnum.getPkg(), taskEnum.getClassName());
|
||||
builder.superclass(ParameterizedTypeName.get(superClass, typeVariableName));
|
||||
builder.addMethod(listener);
|
||||
return builder.build();
|
||||
}
|
||||
|
||||
void clean() {
|
||||
mMethods.clear();
|
||||
}
|
||||
|
||||
/**
|
||||
* 查找并保存扫描到的方法
|
||||
*/
|
||||
private void saveMethod(TaskEnum taskEnum, RoundEnvironment roundEnv,
|
||||
Class<? extends Annotation> annotationClazz, int annotationType) {
|
||||
for (Element element : roundEnv.getElementsAnnotatedWith(annotationClazz)) {
|
||||
ElementKind kind = element.getKind();
|
||||
if (kind == ElementKind.METHOD) {
|
||||
ExecutableElement method = (ExecutableElement) element;
|
||||
TypeElement classElement = (TypeElement) method.getEnclosingElement();
|
||||
PackageElement packageElement = mElementUtil.getPackageOf(classElement);
|
||||
checkDownloadMethod(taskEnum, method);
|
||||
String methodName = method.getSimpleName().toString();
|
||||
String className = method.getEnclosingElement().toString(); //全类名
|
||||
ProxyMethodParam proxyEntity = mMethods.get(className);
|
||||
if (proxyEntity == null) {
|
||||
proxyEntity = new ProxyMethodParam();
|
||||
proxyEntity.taskEnums = new HashSet<>();
|
||||
proxyEntity.packageName = packageElement.getQualifiedName().toString();
|
||||
proxyEntity.className = classElement.getSimpleName().toString();
|
||||
mMethods.put(className, proxyEntity);
|
||||
}
|
||||
proxyEntity.taskEnums.add(taskEnum);
|
||||
if (proxyEntity.methods.get(taskEnum) == null) {
|
||||
proxyEntity.methods.put(taskEnum, new HashMap<Class<? extends Annotation>, String>());
|
||||
}
|
||||
proxyEntity.methods.get(taskEnum).put(annotationClazz, methodName);
|
||||
proxyEntity.keyMappings.put(methodName, getValues(taskEnum, method, annotationType));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取注解的内容
|
||||
*/
|
||||
private Set<String> getValues(TaskEnum taskEnum, ExecutableElement method, int annotationType) {
|
||||
String clsName = method.getEnclosingElement().toString();
|
||||
String[] keys = null;
|
||||
switch (taskEnum) {
|
||||
case DOWNLOAD:
|
||||
keys = getDownloadValues(method, annotationType);
|
||||
addListenerMapping(clsName, ProxyConstance.COUNT_DOWNLOAD);
|
||||
break;
|
||||
case UPLOAD:
|
||||
keys = getUploadValues(method, annotationType);
|
||||
addListenerMapping(clsName, ProxyConstance.COUNT_UPLOAD);
|
||||
break;
|
||||
case DOWNLOAD_GROUP:
|
||||
keys = getDownloadGroupValues(method, annotationType);
|
||||
addListenerMapping(clsName, ProxyConstance.COUNT_DOWNLOAD_GROUP);
|
||||
break;
|
||||
}
|
||||
return keys == null ? null : convertSet(keys);
|
||||
}
|
||||
|
||||
/**
|
||||
* 添加方法映射
|
||||
*
|
||||
* @param clsName 注解事件的类
|
||||
* @param key {@link ProxyConstance#COUNT_DOWNLOAD}、{@link ProxyConstance#COUNT_UPLOAD}、{@link
|
||||
* ProxyConstance#COUNT_DOWNLOAD_GROUP}
|
||||
*/
|
||||
private void addListenerMapping(String clsName, String key) {
|
||||
Set<String> cls = mListenerClass.get(key);
|
||||
if (cls == null) {
|
||||
cls = new HashSet<>();
|
||||
mListenerClass.put(key, cls);
|
||||
}
|
||||
cls.add(clsName);
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取下载任务组的注解数据
|
||||
*/
|
||||
private String[] getDownloadGroupValues(ExecutableElement method, int annotationType) {
|
||||
String[] values = null;
|
||||
switch (annotationType) {
|
||||
case ProxyConstance.PRE:
|
||||
values = method.getAnnotation(DownloadGroup.onPre.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_PRE:
|
||||
values = method.getAnnotation(DownloadGroup.onTaskPre.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_RESUME:
|
||||
values = method.getAnnotation(DownloadGroup.onTaskResume.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_START:
|
||||
values = method.getAnnotation(DownloadGroup.onTaskStart.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_RUNNING:
|
||||
values = method.getAnnotation(DownloadGroup.onTaskRunning.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_STOP:
|
||||
values = method.getAnnotation(DownloadGroup.onTaskStop.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_COMPLETE:
|
||||
values = method.getAnnotation(DownloadGroup.onTaskComplete.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_CANCEL:
|
||||
values = method.getAnnotation(DownloadGroup.onTaskCancel.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_FAIL:
|
||||
values = method.getAnnotation(DownloadGroup.onTaskFail.class).value();
|
||||
break;
|
||||
}
|
||||
return values;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取上传的注解数据
|
||||
*/
|
||||
private String[] getUploadValues(ExecutableElement method, int annotationType) {
|
||||
String[] values = null;
|
||||
switch (annotationType) {
|
||||
case ProxyConstance.PRE:
|
||||
values = method.getAnnotation(Upload.onPre.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_PRE:
|
||||
//values = method.getAnnotation(Upload.onTaskPre.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_RESUME:
|
||||
values = method.getAnnotation(Upload.onTaskResume.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_START:
|
||||
values = method.getAnnotation(Upload.onTaskStart.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_RUNNING:
|
||||
values = method.getAnnotation(Upload.onTaskRunning.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_STOP:
|
||||
values = method.getAnnotation(Upload.onTaskStop.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_COMPLETE:
|
||||
values = method.getAnnotation(Upload.onTaskComplete.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_CANCEL:
|
||||
values = method.getAnnotation(Upload.onTaskCancel.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_FAIL:
|
||||
values = method.getAnnotation(Upload.onTaskFail.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_NO_SUPPORT_BREAKPOINT:
|
||||
//values = method.getAnnotation(Upload.onNoSupportBreakPoint.class).value();
|
||||
break;
|
||||
}
|
||||
return values;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取下载的注解数据
|
||||
*/
|
||||
private String[] getDownloadValues(ExecutableElement method, int annotationType) {
|
||||
String[] values = null;
|
||||
switch (annotationType) {
|
||||
case ProxyConstance.PRE:
|
||||
values = method.getAnnotation(Download.onPre.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_PRE:
|
||||
values = method.getAnnotation(Download.onTaskPre.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_RESUME:
|
||||
values = method.getAnnotation(Download.onTaskResume.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_START:
|
||||
values = method.getAnnotation(Download.onTaskStart.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_RUNNING:
|
||||
values = method.getAnnotation(Download.onTaskRunning.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_STOP:
|
||||
values = method.getAnnotation(Download.onTaskStop.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_COMPLETE:
|
||||
values = method.getAnnotation(Download.onTaskComplete.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_CANCEL:
|
||||
values = method.getAnnotation(Download.onTaskCancel.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_FAIL:
|
||||
values = method.getAnnotation(Download.onTaskFail.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_NO_SUPPORT_BREAKPOINT:
|
||||
values = method.getAnnotation(Download.onNoSupportBreakPoint.class).value();
|
||||
break;
|
||||
}
|
||||
return values;
|
||||
}
|
||||
|
||||
/**
|
||||
* 检查和下载相关的方法，如果被注解的方法为private或参数不合法，则抛异常
|
||||
*/
|
||||
private void checkDownloadMethod(TaskEnum taskEnum, ExecutableElement method) {
|
||||
String methodName = method.getSimpleName().toString();
|
||||
String className = method.getEnclosingElement().toString();
|
||||
Set<Modifier> modifiers = method.getModifiers();
|
||||
if (modifiers.contains(Modifier.PRIVATE)) {
|
||||
throw new IllegalAccessError(className + "." + methodName + "不能为private方法");
|
||||
}
|
||||
List<VariableElement> params = (List<VariableElement>) method.getParameters();
|
||||
if (params.size() > 1) {
|
||||
throw new IllegalArgumentException(
|
||||
className + "." + methodName + "参数错误, 参数只有一个，且参数必须是" + getCheckParams(taskEnum));
|
||||
}
|
||||
if (!params.get(0).asType().toString().equals(getCheckParams(taskEnum))) {
|
||||
throw new IllegalArgumentException(className
|
||||
+ "."
|
||||
+ methodName
|
||||
+ "参数【"
|
||||
+ params.get(0).getSimpleName()
|
||||
+ "】类型错误，参数必须是"
|
||||
+ getCheckParams(taskEnum));
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 字符串数组转set
|
||||
*
|
||||
* @param keys 注解中查到的key
|
||||
*/
|
||||
private Set<String> convertSet(final String[] keys) {
|
||||
if (keys == null || keys.length == 0) {
|
||||
return null;
|
||||
}
|
||||
if (keys[0].isEmpty()) return null;
|
||||
Set<String> set = new HashSet<>();
|
||||
Collections.addAll(set, keys);
|
||||
return set;
|
||||
}
|
||||
|
||||
private String getCheckParams(TaskEnum taskEnum) {
|
||||
return taskEnum.pkg + "." + taskEnum.getClassName();
|
||||
mPbUtil.getMethodParams().clear();
|
||||
}
|
||||
}
|
||||
|
||||
@@ -0,0 +1,244 @@
|
||||
/*
|
||||
* Copyright (C) 2016 AriaLyy(https://github.com/AriaLyy/Aria)
|
||||
*
|
||||
* 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.arialyy.compiler;
|
||||
|
||||
import com.arialyy.annotations.Download;
|
||||
import com.arialyy.annotations.Upload;
|
||||
import com.squareup.javapoet.ClassName;
|
||||
import com.squareup.javapoet.CodeBlock;
|
||||
import com.squareup.javapoet.FieldSpec;
|
||||
import com.squareup.javapoet.JavaFile;
|
||||
import com.squareup.javapoet.MethodSpec;
|
||||
import com.squareup.javapoet.ParameterSpec;
|
||||
import com.squareup.javapoet.ParameterizedTypeName;
|
||||
import com.squareup.javapoet.TypeSpec;
|
||||
import java.io.IOException;
|
||||
import java.lang.annotation.Annotation;
|
||||
import java.util.HashMap;
|
||||
import java.util.HashSet;
|
||||
import java.util.Map;
|
||||
import java.util.Set;
|
||||
import javax.annotation.processing.Filer;
|
||||
import javax.lang.model.element.Modifier;
|
||||
|
||||
/**
|
||||
* Created by lyy on 2017/9/6.
|
||||
* 任务事件代理文件
|
||||
*
|
||||
* <pre>
|
||||
* <code>
|
||||
* package com.arialyy.simple.download;
|
||||
*
|
||||
* import com.arialyy.aria.core.download.DownloadTask;
|
||||
* import com.arialyy.aria.core.scheduler.AbsSchedulerListener;
|
||||
*
|
||||
* public final class SingleTaskActivity$$DownloadListenerProxy extends
|
||||
* AbsSchedulerListener<DownloadTask> {
|
||||
* private SingleTaskActivity obj;
|
||||
*
|
||||
* public void onPre(final DownloadTask task) {
|
||||
* obj.onPre((DownloadTask)task);
|
||||
* }
|
||||
*
|
||||
* public void onTaskStart(final DownloadTask task) {
|
||||
* obj.onStart((DownloadTask)task);
|
||||
* }
|
||||
*
|
||||
* public void setListener(final Object obj) {
|
||||
* this.obj = (SingleTaskActivity)obj;
|
||||
* }
|
||||
* }
|
||||
* </code>
|
||||
* </pre>
|
||||
*/
|
||||
final class EventProxyFiler {
|
||||
|
||||
private Filer mFiler;
|
||||
private ParamObtainUtil mPbUtil;
|
||||
|
||||
EventProxyFiler(Filer filer, ParamObtainUtil pbUtil) {
|
||||
mFiler = filer;
|
||||
mPbUtil = pbUtil;
|
||||
}
|
||||
|
||||
/**
|
||||
* 创建任务事件代理文件
|
||||
*/
|
||||
void createEventProxyFile() throws IOException {
|
||||
Map<String, ProxyClassParam> map = mPbUtil.getMethodParams();
|
||||
Set<String> keys = mPbUtil.getMethodParams().keySet();
|
||||
for (String key : keys) {
|
||||
//if (key.equals(
|
||||
// "com.arialyy.simple.download.SingleTaskActivity$$DownloadGroupListenerProxy")) {
|
||||
//
|
||||
// ProxyClassParam entity = mPbUtil.getMethodParams().get(key);
|
||||
// JavaFile jf = JavaFile.builder(entity.packageName, createProxyClass(entity)).build();
|
||||
// createFile(jf);
|
||||
//}
|
||||
|
||||
ProxyClassParam entity = mPbUtil.getMethodParams().get(key);
|
||||
JavaFile jf = JavaFile.builder(entity.packageName, createProxyClass(entity)).build();
|
||||
createFile(jf);
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 创建代理方法
|
||||
*
|
||||
* @param taskEnum 任务类型枚举{@link TaskEnum}
|
||||
* @param annotation {@link Download}、{@link Upload}
|
||||
* @param methodName 被代理类注解的方法名
|
||||
*/
|
||||
private MethodSpec createProxyMethod(TaskEnum taskEnum, Class<? extends Annotation> annotation,
|
||||
String methodName) {
|
||||
ClassName task = ClassName.get(taskEnum.getPkg(), taskEnum.getClassName());
|
||||
|
||||
ParameterSpec taskParam =
|
||||
ParameterSpec.builder(task, "task").addModifiers(Modifier.FINAL).build();
|
||||
|
||||
String callCode;
|
||||
if (taskEnum == TaskEnum.DOWNLOAD_GROUP_SUB) {
|
||||
callCode = "task, subTask";
|
||||
} else {
|
||||
callCode = "task";
|
||||
}
|
||||
StringBuilder sb = new StringBuilder();
|
||||
sb.append("Set<String> keys = keyMapping.get(\"").append(methodName).append("\");\n");
|
||||
sb.append("if (keys != null) {\n\tif (keys.contains(task.getKey())) {\n")
|
||||
.append("\t\tobj.")
|
||||
.append(methodName)
|
||||
.append("((")
|
||||
.append(taskEnum.getClassName())
|
||||
.append(")")
|
||||
.append(callCode)
|
||||
.append(");\n\t}\n} else {\n")
|
||||
.append("\tobj.")
|
||||
.append(methodName)
|
||||
.append("((")
|
||||
.append(taskEnum.getClassName())
|
||||
.append(")")
|
||||
.append(callCode)
|
||||
.append(");\n}\n");
|
||||
|
||||
MethodSpec.Builder builder = MethodSpec.methodBuilder(annotation.getSimpleName())
|
||||
.addModifiers(Modifier.PUBLIC)
|
||||
.returns(void.class)
|
||||
.addParameter(taskParam)
|
||||
.addAnnotation(Override.class)
|
||||
.addCode(sb.toString());
|
||||
|
||||
//任务组接口
|
||||
if (taskEnum == TaskEnum.DOWNLOAD_GROUP_SUB) {
|
||||
ClassName subTask = ClassName.get(TaskEnum.DOWNLOAD.pkg, TaskEnum.DOWNLOAD.className);
|
||||
ParameterSpec subTaskParam =
|
||||
ParameterSpec.builder(subTask, "subTask").addModifiers(Modifier.FINAL).build();
|
||||
|
||||
builder.addParameter(subTaskParam);
|
||||
}
|
||||
|
||||
//PrintLog.getInstance().info(builder.build().toString());
|
||||
|
||||
return builder.build();
|
||||
}
|
||||
|
||||
/**
|
||||
* 创建代理类
|
||||
*/
|
||||
private TypeSpec createProxyClass(ProxyClassParam entity) {
|
||||
TypeSpec.Builder builder =
|
||||
TypeSpec.classBuilder(entity.proxyClassName).addModifiers(Modifier.PUBLIC, Modifier.FINAL);
|
||||
|
||||
//添加被代理的类的字段
|
||||
ClassName obj = ClassName.get(entity.packageName, entity.className);
|
||||
FieldSpec observerField = FieldSpec.builder(obj, "obj").addModifiers(Modifier.PRIVATE).build();
|
||||
builder.addField(observerField);
|
||||
|
||||
//添加url映射表
|
||||
FieldSpec mappingField = FieldSpec.builder(
|
||||
ParameterizedTypeName.get(ClassName.get(Map.class), ClassName.get(String.class),
|
||||
ParameterizedTypeName.get(ClassName.get(Set.class), ClassName.get(String.class))),
|
||||
"keyMapping").addModifiers(Modifier.PRIVATE).initializer("new $T()", HashMap.class).build();
|
||||
builder.addField(mappingField);
|
||||
|
||||
//添加注解方法
|
||||
for (TaskEnum te : entity.methods.keySet()) {
|
||||
Map<Class<? extends Annotation>, String> temp = entity.methods.get(te);
|
||||
if (temp != null) {
|
||||
for (Class<? extends Annotation> annotation : temp.keySet()) {
|
||||
MethodSpec method = createProxyMethod(te, annotation, temp.get(annotation));
|
||||
builder.addMethod(method);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
//增加构造函数
|
||||
CodeBlock.Builder cb = CodeBlock.builder();
|
||||
cb.add("Set<String> set = null;\n");
|
||||
for (String methodName : entity.keyMappings.keySet()) {
|
||||
//PrintLog.getInstance().info("methodName ====> " + methodName);
|
||||
Set<String> keys = entity.keyMappings.get(methodName);
|
||||
if (keys == null || keys.size() == 0) continue;
|
||||
StringBuilder sb = new StringBuilder();
|
||||
sb.append("set = new $T();\n");
|
||||
for (String key : keys) {
|
||||
if (key.isEmpty()) continue;
|
||||
sb.append("set.add(\"").append(key).append("\");\n");
|
||||
}
|
||||
|
||||
sb.append("keyMapping.put(\"").append(methodName).append("\", ").append("set);\n");
|
||||
cb.add(sb.toString(), ClassName.get(HashSet.class));
|
||||
}
|
||||
MethodSpec structure =
|
||||
MethodSpec.constructorBuilder().addModifiers(Modifier.PUBLIC).addCode(cb.build()).build();
|
||||
builder.addMethod(structure);
|
||||
|
||||
//添加设置代理的类
|
||||
ParameterSpec parameterSpec =
|
||||
ParameterSpec.builder(Object.class, "obj").addModifiers(Modifier.FINAL).build();
|
||||
MethodSpec listener = MethodSpec.methodBuilder(ProxyConstance.SET_LISTENER)
|
||||
.addModifiers(Modifier.PUBLIC)
|
||||
.returns(void.class)
|
||||
.addParameter(parameterSpec)
|
||||
.addAnnotation(Override.class)
|
||||
.addCode("this.obj = (" + entity.className + ")obj;\n")
|
||||
.build();
|
||||
builder.addJavadoc("该文件为Aria自动生成的代理文件，请不要修改该文件的任何代码！\n");
|
||||
|
||||
//创建父类参数
|
||||
ClassName superClass = ClassName.get("com.arialyy.aria.core.scheduler", "AbsSchedulerListener");
|
||||
//主任务泛型参数
|
||||
ClassName taskTypeVariable =
|
||||
ClassName.get(entity.mainTaskEnum.pkg, entity.mainTaskEnum.className);
|
||||
//子任务泛型参数
|
||||
ClassName subTaskTypeVariable =
|
||||
ClassName.get(entity.subTaskEnum.pkg, entity.subTaskEnum.className);
|
||||
|
||||
builder.superclass(
|
||||
ParameterizedTypeName.get(superClass, taskTypeVariable, subTaskTypeVariable));
|
||||
builder.addMethod(listener);
|
||||
return builder.build();
|
||||
}
|
||||
|
||||
private void createFile(JavaFile jf) throws IOException {
|
||||
if (ProxyConstance.DEBUG) {
|
||||
// 如果需要在控制台打印生成的文件，则去掉下面的注释
|
||||
jf.writeTo(System.out);
|
||||
} else {
|
||||
jf.writeTo(mFiler);
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,221 @@
|
||||
/*
|
||||
* Copyright (C) 2016 AriaLyy(https://github.com/AriaLyy/Aria)
|
||||
*
|
||||
* 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.arialyy.compiler;
|
||||
|
||||
import java.lang.annotation.Annotation;
|
||||
import java.util.Collections;
|
||||
import java.util.HashMap;
|
||||
import java.util.HashSet;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
import java.util.Set;
|
||||
import javax.annotation.processing.RoundEnvironment;
|
||||
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.PackageElement;
|
||||
import javax.lang.model.element.TypeElement;
|
||||
import javax.lang.model.element.VariableElement;
|
||||
import javax.lang.model.util.Elements;
|
||||
|
||||
/**
|
||||
* Created by Aria.Lao on 2017/9/6.
|
||||
* 构建代理文件的参数获取工具
|
||||
*/
|
||||
class ParamObtainUtil {
|
||||
private Map<String, ProxyClassParam> mMethodParams = new HashMap<>();
|
||||
private Map<String, Set<String>> mListenerClass = new HashMap<>();
|
||||
private Elements mElementUtil;
|
||||
|
||||
ParamObtainUtil(Elements elements) {
|
||||
mElementUtil = elements;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取需要创建的代理类参数
|
||||
*/
|
||||
Map<String, Set<String>> getListenerClass() {
|
||||
return mListenerClass;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取搜索到的代理方法参数
|
||||
*/
|
||||
Map<String, ProxyClassParam> getMethodParams() {
|
||||
return mMethodParams;
|
||||
}
|
||||
|
||||
/**
|
||||
* 查找并保存扫描到的方法
|
||||
*/
|
||||
void saveMethod(TaskEnum taskEnum, RoundEnvironment roundEnv,
|
||||
Class<? extends Annotation> annotationClazz, int annotationType) {
|
||||
for (Element element : roundEnv.getElementsAnnotatedWith(annotationClazz)) {
|
||||
ElementKind kind = element.getKind();
|
||||
if (kind == ElementKind.METHOD) {
|
||||
ExecutableElement method = (ExecutableElement) element;
|
||||
TypeElement classElement = (TypeElement) method.getEnclosingElement();
|
||||
PackageElement packageElement = mElementUtil.getPackageOf(classElement);
|
||||
checkDownloadMethod(taskEnum, method);
|
||||
String methodName = method.getSimpleName().toString();
|
||||
String className = method.getEnclosingElement().toString(); //全类名\
|
||||
String key = className + taskEnum.proxySuffix;
|
||||
ProxyClassParam proxyEntity = mMethodParams.get(key);
|
||||
|
||||
if (proxyEntity == null) {
|
||||
proxyEntity = new ProxyClassParam();
|
||||
proxyEntity.taskEnums = new HashSet<>();
|
||||
proxyEntity.packageName = packageElement.getQualifiedName().toString();
|
||||
proxyEntity.className = classElement.getSimpleName().toString();
|
||||
proxyEntity.proxyClassName = proxyEntity.className + taskEnum.proxySuffix;
|
||||
proxyEntity.mainTaskEnum = taskEnum;
|
||||
if (taskEnum == TaskEnum.DOWNLOAD_GROUP_SUB || taskEnum == TaskEnum.DOWNLOAD_GROUP) {
|
||||
proxyEntity.subTaskEnum = TaskEnum.DOWNLOAD;
|
||||
}
|
||||
mMethodParams.put(key, proxyEntity);
|
||||
}
|
||||
proxyEntity.taskEnums.add(taskEnum);
|
||||
if (proxyEntity.methods.get(taskEnum) == null) {
|
||||
proxyEntity.methods.put(taskEnum, new HashMap<Class<? extends Annotation>, String>());
|
||||
}
|
||||
proxyEntity.methods.get(taskEnum).put(annotationClazz, methodName);
|
||||
proxyEntity.keyMappings.put(methodName, getValues(taskEnum, method, annotationType));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取注解的内容
|
||||
*/
|
||||
private Set<String> getValues(TaskEnum taskEnum, ExecutableElement method, int annotationType) {
|
||||
String clsName = method.getEnclosingElement().toString();
|
||||
String[] keys = null;
|
||||
switch (taskEnum) {
|
||||
case DOWNLOAD:
|
||||
keys = ValuesUtil.getDownloadValues(method, annotationType);
|
||||
addListenerMapping(clsName, ProxyConstance.COUNT_DOWNLOAD);
|
||||
break;
|
||||
case UPLOAD:
|
||||
keys = ValuesUtil.getUploadValues(method, annotationType);
|
||||
addListenerMapping(clsName, ProxyConstance.COUNT_UPLOAD);
|
||||
break;
|
||||
case DOWNLOAD_GROUP:
|
||||
keys = ValuesUtil.getDownloadGroupValues(method, annotationType);
|
||||
addListenerMapping(clsName, ProxyConstance.COUNT_DOWNLOAD_GROUP);
|
||||
break;
|
||||
case DOWNLOAD_GROUP_SUB:
|
||||
keys = ValuesUtil.getDownloadGroupSubValues(method, annotationType);
|
||||
addListenerMapping(clsName, ProxyConstance.COUNT_DOWNLOAD_GROUP_SUB);
|
||||
break;
|
||||
}
|
||||
return keys == null ? null : convertSet(keys);
|
||||
}
|
||||
|
||||
/**
|
||||
* 添加方法映射
|
||||
*
|
||||
* @param clsName 注解事件的类
|
||||
* @param key {@link ProxyConstance#COUNT_DOWNLOAD}、{@link ProxyConstance#COUNT_UPLOAD}、{@link
|
||||
* ProxyConstance#COUNT_DOWNLOAD_GROUP}、{@link ProxyConstance#COUNT_DOWNLOAD_GROUP_SUB}
|
||||
*/
|
||||
void addListenerMapping(String clsName, String key) {
|
||||
Set<String> cls = mListenerClass.get(key);
|
||||
if (cls == null) {
|
||||
cls = new HashSet<>();
|
||||
mListenerClass.put(key, cls);
|
||||
}
|
||||
cls.add(clsName);
|
||||
}
|
||||
|
||||
/**
|
||||
* 检查和下载相关的方法，如果被注解的方法为private或参数不合法，则抛异常
|
||||
*/
|
||||
private void checkDownloadMethod(TaskEnum taskEnum, ExecutableElement method) {
|
||||
String methodName = method.getSimpleName().toString();
|
||||
String className = method.getEnclosingElement().toString();
|
||||
Set<Modifier> modifiers = method.getModifiers();
|
||||
if (modifiers.contains(Modifier.PRIVATE)) {
|
||||
throw new IllegalAccessError(className + "." + methodName + "不能为private方法");
|
||||
}
|
||||
List<VariableElement> params = (List<VariableElement>) method.getParameters();
|
||||
if (taskEnum == TaskEnum.DOWNLOAD_GROUP_SUB) {
|
||||
if (params.size() != 2) {
|
||||
throw new IllegalArgumentException(className
|
||||
+ "."
|
||||
+ methodName
|
||||
+ "参数错误, 参数只有两个，且第一个参数必须是"
|
||||
+ getCheckParams(taskEnum)
|
||||
+ "，第二个参数必须是"
|
||||
+ getCheckSubParams(taskEnum));
|
||||
}
|
||||
} else {
|
||||
if (params.size() != 1) {
|
||||
throw new IllegalArgumentException(
|
||||
className + "." + methodName + "参数错误, 参数只能有一个，且参数必须是" + getCheckParams(taskEnum));
|
||||
}
|
||||
}
|
||||
if (!params.get(0).asType().toString().equals(getCheckParams(taskEnum))) {
|
||||
throw new IllegalArgumentException(className
|
||||
+ "."
|
||||
+ methodName
|
||||
+ "参数【"
|
||||
+ params.get(0).getSimpleName()
|
||||
+ "】类型错误，参数必须是"
|
||||
+ getCheckParams(taskEnum));
|
||||
}
|
||||
if (taskEnum == TaskEnum.DOWNLOAD_GROUP_SUB) {
|
||||
if (!params.get(1).asType().toString().equals(getCheckSubParams(taskEnum))) {
|
||||
throw new IllegalArgumentException(className
|
||||
+ "."
|
||||
+ methodName
|
||||
+ "参数【"
|
||||
+ params.get(0).getSimpleName()
|
||||
+ "】类型错误，参数必须是"
|
||||
+ getCheckSubParams(taskEnum));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 字符串数组转set
|
||||
*
|
||||
* @param keys 注解中查到的key
|
||||
*/
|
||||
private Set<String> convertSet(final String[] keys) {
|
||||
if (keys == null || keys.length == 0) {
|
||||
return null;
|
||||
}
|
||||
if (keys[0].isEmpty()) return null;
|
||||
Set<String> set = new HashSet<>();
|
||||
Collections.addAll(set, keys);
|
||||
return set;
|
||||
}
|
||||
|
||||
private String getCheckParams(TaskEnum taskEnum) {
|
||||
return taskEnum.pkg + "." + taskEnum.className;
|
||||
}
|
||||
|
||||
/**
|
||||
* 检查任务组子任务第二个参数
|
||||
*/
|
||||
private String getCheckSubParams(TaskEnum taskEnum) {
|
||||
if (taskEnum == TaskEnum.DOWNLOAD_GROUP_SUB) {
|
||||
return TaskEnum.DOWNLOAD.pkg + "." + TaskEnum.DOWNLOAD.className;
|
||||
}
|
||||
return "";
|
||||
}
|
||||
}
|
||||
@@ -55,7 +55,7 @@ class PrintLog {
|
||||
mMessager.printMessage(Diagnostic.Kind.ERROR, String.format(msg, args));
|
||||
}
|
||||
|
||||
private void warning(String msg) {
|
||||
public void warning(String msg) {
|
||||
mMessager.printMessage(Diagnostic.Kind.WARNING, msg);
|
||||
}
|
||||
|
||||
|
||||
@@ -22,11 +22,30 @@ import java.util.Set;
|
||||
|
||||
/**
|
||||
* Created by Aria.Lao on 2017/6/7.
|
||||
* 创建代理方法的参数
|
||||
* 代理类参数
|
||||
*/
|
||||
class ProxyMethodParam {
|
||||
class ProxyClassParam {
|
||||
/**
|
||||
* 代理文件名
|
||||
*/
|
||||
String proxyClassName;
|
||||
/**
|
||||
* 被代理的类所在的包
|
||||
*/
|
||||
String packageName;
|
||||
/**
|
||||
* 被代理的类
|
||||
*/
|
||||
String className;
|
||||
/**
|
||||
* 主任务泛型参数
|
||||
*/
|
||||
TaskEnum mainTaskEnum;
|
||||
/**
|
||||
* 子任务泛型参数
|
||||
*/
|
||||
TaskEnum subTaskEnum = TaskEnum.NORMAL;
|
||||
|
||||
Set<TaskEnum> taskEnums;
|
||||
Map<String, Set<String>> keyMappings = new HashMap<>();
|
||||
Map<TaskEnum, Map<Class<? extends Annotation>, String>> methods = new HashMap<>();
|
||||
@@ -20,6 +20,7 @@ package com.arialyy.compiler;
|
||||
* 扫描器常量
|
||||
*/
|
||||
interface ProxyConstance {
|
||||
boolean DEBUG = false;
|
||||
/**
|
||||
* 设置观察者的方法
|
||||
*/
|
||||
@@ -41,6 +42,7 @@ interface ProxyConstance {
|
||||
|
||||
String COUNT_DOWNLOAD = "download";
|
||||
String COUNT_DOWNLOAD_GROUP = "downloadGroup";
|
||||
String COUNT_DOWNLOAD_GROUP_SUB = "downloadGroupSub";
|
||||
String COUNT_UPLOAD = "upload";
|
||||
|
||||
String COUNT_METHOD_DOWNLOAD = "getDownloadCounter";
|
||||
|
||||
@@ -22,9 +22,11 @@ package com.arialyy.compiler;
|
||||
enum TaskEnum {
|
||||
DOWNLOAD("com.arialyy.aria.core.download", "DownloadTask",
|
||||
"$$DownloadListenerProxy"), DOWNLOAD_GROUP("com.arialyy.aria.core.download",
|
||||
"DownloadGroupTask", "$$DownloadGroupListenerProxy"), UPLOAD("com.arialyy.aria.core.upload",
|
||||
"UploadTask", "$$UploadListenerProxy"), UPLOAD_GROUP("com.arialyy.aria.core.upload",
|
||||
"UploadGroupTask", "$$UploadGroupListenerProxy");
|
||||
"DownloadGroupTask", "$$DownloadGroupListenerProxy"), DOWNLOAD_GROUP_SUB(
|
||||
"com.arialyy.aria.core.download", "DownloadGroupTask",
|
||||
"$$DownloadGroupListenerProxy"), UPLOAD("com.arialyy.aria.core.upload", "UploadTask",
|
||||
"$$UploadListenerProxy"), UPLOAD_GROUP("com.arialyy.aria.core.upload", "UploadGroupTask",
|
||||
"$$UploadGroupListenerProxy"), NORMAL("com.arialyy.aria.core.inf", "AbsNormalTask", "");
|
||||
|
||||
String pkg, className, proxySuffix;
|
||||
|
||||
|
||||
175
AriaCompiler/src/main/java/com/arialyy/compiler/ValuesUtil.java
Normal file
175
AriaCompiler/src/main/java/com/arialyy/compiler/ValuesUtil.java
Normal file
@@ -0,0 +1,175 @@
|
||||
/*
|
||||
* Copyright (C) 2016 AriaLyy(https://github.com/AriaLyy/Aria)
|
||||
*
|
||||
* 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.arialyy.compiler;
|
||||
|
||||
import com.arialyy.annotations.Download;
|
||||
import com.arialyy.annotations.DownloadGroup;
|
||||
import com.arialyy.annotations.Upload;
|
||||
import javax.lang.model.element.ExecutableElement;
|
||||
|
||||
/**
|
||||
* Created by lyy on 2017/9/6.
|
||||
* 获取注解value工具
|
||||
*/
|
||||
final class ValuesUtil {
|
||||
/**
|
||||
* 获取下载任务组子任务的的注解数据
|
||||
*/
|
||||
static String[] getDownloadGroupSubValues(ExecutableElement method, int annotationType) {
|
||||
String[] values = null;
|
||||
switch (annotationType) {
|
||||
case ProxyConstance.TASK_PRE:
|
||||
values = method.getAnnotation(DownloadGroup.onSubTaskPre.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_START:
|
||||
values = method.getAnnotation(DownloadGroup.onSubTaskStart.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_RUNNING:
|
||||
values = method.getAnnotation(DownloadGroup.onSubTaskRunning.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_STOP:
|
||||
values = method.getAnnotation(DownloadGroup.onSubTaskStop.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_COMPLETE:
|
||||
values = method.getAnnotation(DownloadGroup.onSubTaskComplete.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_CANCEL:
|
||||
values = method.getAnnotation(DownloadGroup.onSubTaskCancel.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_FAIL:
|
||||
values = method.getAnnotation(DownloadGroup.onSubTaskFail.class).value();
|
||||
break;
|
||||
}
|
||||
return values;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取下载任务组的注解数据
|
||||
*/
|
||||
static String[] getDownloadGroupValues(ExecutableElement method, int annotationType) {
|
||||
String[] values = null;
|
||||
switch (annotationType) {
|
||||
case ProxyConstance.PRE:
|
||||
values = method.getAnnotation(DownloadGroup.onPre.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_PRE:
|
||||
values = method.getAnnotation(DownloadGroup.onTaskPre.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_RESUME:
|
||||
values = method.getAnnotation(DownloadGroup.onTaskResume.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_START:
|
||||
values = method.getAnnotation(DownloadGroup.onTaskStart.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_RUNNING:
|
||||
values = method.getAnnotation(DownloadGroup.onTaskRunning.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_STOP:
|
||||
values = method.getAnnotation(DownloadGroup.onTaskStop.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_COMPLETE:
|
||||
values = method.getAnnotation(DownloadGroup.onTaskComplete.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_CANCEL:
|
||||
values = method.getAnnotation(DownloadGroup.onTaskCancel.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_FAIL:
|
||||
values = method.getAnnotation(DownloadGroup.onTaskFail.class).value();
|
||||
break;
|
||||
}
|
||||
return values;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取上传的注解数据
|
||||
*/
|
||||
static String[] getUploadValues(ExecutableElement method, int annotationType) {
|
||||
String[] values = null;
|
||||
switch (annotationType) {
|
||||
case ProxyConstance.PRE:
|
||||
values = method.getAnnotation(Upload.onPre.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_PRE:
|
||||
//values = method.getAnnotation(Upload.onTaskPre.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_RESUME:
|
||||
values = method.getAnnotation(Upload.onTaskResume.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_START:
|
||||
values = method.getAnnotation(Upload.onTaskStart.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_RUNNING:
|
||||
values = method.getAnnotation(Upload.onTaskRunning.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_STOP:
|
||||
values = method.getAnnotation(Upload.onTaskStop.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_COMPLETE:
|
||||
values = method.getAnnotation(Upload.onTaskComplete.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_CANCEL:
|
||||
values = method.getAnnotation(Upload.onTaskCancel.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_FAIL:
|
||||
values = method.getAnnotation(Upload.onTaskFail.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_NO_SUPPORT_BREAKPOINT:
|
||||
//values = method.getAnnotation(Upload.onNoSupportBreakPoint.class).value();
|
||||
break;
|
||||
}
|
||||
return values;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取下载的注解数据
|
||||
*/
|
||||
static String[] getDownloadValues(ExecutableElement method, int annotationType) {
|
||||
String[] values = null;
|
||||
switch (annotationType) {
|
||||
case ProxyConstance.PRE:
|
||||
values = method.getAnnotation(Download.onPre.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_PRE:
|
||||
values = method.getAnnotation(Download.onTaskPre.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_RESUME:
|
||||
values = method.getAnnotation(Download.onTaskResume.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_START:
|
||||
values = method.getAnnotation(Download.onTaskStart.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_RUNNING:
|
||||
values = method.getAnnotation(Download.onTaskRunning.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_STOP:
|
||||
values = method.getAnnotation(Download.onTaskStop.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_COMPLETE:
|
||||
values = method.getAnnotation(Download.onTaskComplete.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_CANCEL:
|
||||
values = method.getAnnotation(Download.onTaskCancel.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_FAIL:
|
||||
values = method.getAnnotation(Download.onTaskFail.class).value();
|
||||
break;
|
||||
case ProxyConstance.TASK_NO_SUPPORT_BREAKPOINT:
|
||||
values = method.getAnnotation(Download.onNoSupportBreakPoint.class).value();
|
||||
break;
|
||||
}
|
||||
return values;
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user