更新http服务器 剔除反射功能
This commit is contained in:
85
lib/base/webserver/impl/RouteGenerator.dart
Normal file
85
lib/base/webserver/impl/RouteGenerator.dart
Normal file
@@ -0,0 +1,85 @@
|
||||
import 'package:EasyDartModule/base/webserver/WebServer.dart';
|
||||
import 'package:build/build.dart';
|
||||
import 'package:analyzer/dart/element/element.dart';
|
||||
import 'package:source_gen/source_gen.dart';
|
||||
import 'package:path/path.dart';
|
||||
|
||||
Builder routeGenerator(BuilderOptions options) {
|
||||
return LibraryBuilder(
|
||||
RouteGenerator(),
|
||||
generatedExtension: '.route.dart',
|
||||
);
|
||||
}
|
||||
|
||||
class RouteGenerator extends GeneratorForAnnotation<RequestMapping> {
|
||||
call(String name) {}
|
||||
@override
|
||||
generateForAnnotatedElement(
|
||||
Element element, ConstantReader annotation, BuildStep buildStep) {
|
||||
// 读取注解字段
|
||||
String path = annotation.read('path').stringValue;
|
||||
String method = annotation.read('method').read("_name").stringValue;
|
||||
// String response = annotation.read("responseType").read("_name").stringValue;
|
||||
// print("url:$path $method");
|
||||
//0:路径 1:响应类型 2:调用函数 3:参数数量
|
||||
Map<HttpMethod, List> routes = {};
|
||||
if (method != HttpMethod.WS.name) {
|
||||
if (path.endsWith("/")) {
|
||||
path = path.substring(0, path.length - 1);
|
||||
}
|
||||
for (final method in element.children) {
|
||||
for (final metaData in method.metadata) {
|
||||
var anno = ConstantReader(metaData.computeConstantValue());
|
||||
if (!anno.instanceOf(typeChecker)) {
|
||||
continue;
|
||||
}
|
||||
|
||||
int pl = (method as MethodElement).parameters.length;
|
||||
// print(
|
||||
// "url:$path-${anno.read("path").stringValue} method:${anno.read("method").read("_name").stringValue} response:${anno.read("responseType").read("_name").stringValue}");
|
||||
//转换为枚举类型
|
||||
|
||||
var p = anno.read("path").stringValue;
|
||||
if (!p.startsWith("/")) {
|
||||
p = "$path/$p";
|
||||
} else {
|
||||
p = path + p;
|
||||
}
|
||||
//加入调用缓存
|
||||
routes[HttpMethod.valueOf(
|
||||
anno.read("method").read("_name").stringValue)] = [
|
||||
p,
|
||||
HttpResponseType.valueOf(
|
||||
anno.read("responseType").read("_name").stringValue),
|
||||
() {
|
||||
return method.name;
|
||||
},
|
||||
pl >= 2 ? 2 : 1
|
||||
];
|
||||
}
|
||||
// print("构建完毕");
|
||||
}
|
||||
} else {
|
||||
//单独处理ws
|
||||
routes[HttpMethod.valueOf(method)] = [path];
|
||||
}
|
||||
// 生成路由映射代码
|
||||
// var data = jsonEncode(routes);
|
||||
final routesMapString = routes.entries.map((entry) {
|
||||
return '${entry.key}: [[${entry.value.map((e) {
|
||||
if (e is String) {
|
||||
return '"$e"';
|
||||
} else if (e is Function) {
|
||||
return "_callHandler.${e()}";
|
||||
}
|
||||
return e;
|
||||
}).join(",")}]],';
|
||||
}).join('\n');
|
||||
// print(routes);
|
||||
return '''
|
||||
part of '${basename(buildStep.inputId.path)}';
|
||||
late var _callHandler;
|
||||
final Map<HttpMethod, List<List>> routes = {$routesMapString};
|
||||
''';
|
||||
}
|
||||
}
|
||||
@@ -1,5 +1,4 @@
|
||||
import 'dart:io';
|
||||
import 'dart:mirrors';
|
||||
|
||||
import 'package:EasyDartModule/EasyDartModule.dart';
|
||||
import 'package:EasyDartModule/base/logger/Logger.dart';
|
||||
@@ -50,7 +49,7 @@ class ShelfWebServer implements WebServer {
|
||||
} catch (e, s) {
|
||||
if (e is HijackException) {
|
||||
//不能处理该异常直接抛出
|
||||
throw e;
|
||||
rethrow;
|
||||
}
|
||||
stopwatch.stop();
|
||||
logger.error("| 服务器错误 | ${e.toString()} ${s.toString()}",
|
||||
@@ -135,12 +134,24 @@ class ShelfWebServer implements WebServer {
|
||||
}
|
||||
|
||||
@override
|
||||
void start(int port) async {
|
||||
void start(int port, {Function? interceptor}) async {
|
||||
//反射获取全部路由地址
|
||||
final handler = Pipeline()
|
||||
var pipeline = Pipeline()
|
||||
.addMiddleware(requestIdMiddleware())
|
||||
.addMiddleware(logRequests())
|
||||
.addHandler(_routerHandler);
|
||||
.addMiddleware(logRequests());
|
||||
//判断拦截器类型
|
||||
if (interceptor != null) {
|
||||
pipeline = pipeline.addMiddleware((Handler innerHandler) {
|
||||
return (Request request) async {
|
||||
var response = interceptor(request);
|
||||
if (response == null) {
|
||||
return innerHandler(request);
|
||||
}
|
||||
return response;
|
||||
};
|
||||
});
|
||||
}
|
||||
final handler = pipeline.addHandler(_routerHandler);
|
||||
_server = await serve(handler, InternetAddress.anyIPv4, port);
|
||||
print('Server listening on port ${_server?.port}');
|
||||
}
|
||||
@@ -152,88 +163,63 @@ class ShelfWebServer implements WebServer {
|
||||
|
||||
@override
|
||||
void addHandler(handler) {
|
||||
ClassMirror cm = reflectClass(handler.runtimeType);
|
||||
var im = reflect(handler);
|
||||
String path = "";
|
||||
for (var metadata in cm.metadata) {
|
||||
// 检查元数据是否为RequestMapping类型
|
||||
if (metadata.reflectee is RequestMapping) {
|
||||
// 获取实例
|
||||
RequestMapping annotation = metadata.reflectee;
|
||||
path = annotation.path;
|
||||
if (path.endsWith("/")) {
|
||||
path = path.substring(0, path.length - 1);
|
||||
try {
|
||||
handler.callHandler = handler;
|
||||
} catch (e) {
|
||||
print(
|
||||
"请在 ${handler.runtimeType} 中实现set方法 set callHandler(handler) => _callHandler = handler;");
|
||||
return;
|
||||
}
|
||||
try {
|
||||
handler.routeMap.forEach((k, v) {
|
||||
for (var p in v) {
|
||||
if (k == HttpMethod.WS) {
|
||||
addRouter(k, p[0], handler);
|
||||
continue;
|
||||
}
|
||||
addRouter(k, p[0], (req, [a, b, c, d, e, f]) async {
|
||||
//从request中获取token并解码传入
|
||||
String? token = req.headers["token"];
|
||||
dynamic payload;
|
||||
if (token != null) {
|
||||
try {
|
||||
payload = JWT.decode(token).payload;
|
||||
} catch (e) {
|
||||
//jwt解码失败
|
||||
print(e);
|
||||
}
|
||||
}
|
||||
var response;
|
||||
if (p[3] == 2) {
|
||||
response = await p[2](req, payload ?? <String, dynamic>{});
|
||||
} else {
|
||||
response = await p[2](req);
|
||||
}
|
||||
Map<String, String> headers = {...response.headers};
|
||||
headers['Access-Control-Allow-Origin'] = '*';
|
||||
headers['Access-Control-Allow-Headers'] =
|
||||
'Content-Type,X-Span-ID,X-Trace-ID,token';
|
||||
headers["Access-Control-Expose-Headers"] =
|
||||
"Content-Type,X-Span-ID,X-Trace-ID,token";
|
||||
headers['Access-Control-Allow-Methods'] =
|
||||
'GET, POST, PUT, DELETE, OPTIONS';
|
||||
headers["content-type"] = p[1].type;
|
||||
|
||||
response = response.change(headers: headers);
|
||||
|
||||
return response;
|
||||
});
|
||||
}
|
||||
if (annotation.method == HttpMethod.WS) {
|
||||
//wwebsocket处理句柄直接加入路由
|
||||
addRouter(annotation.method, path, handler);
|
||||
return;
|
||||
}
|
||||
break;
|
||||
});
|
||||
} catch (e) {
|
||||
if (e is NoSuchMethodError) {
|
||||
print(
|
||||
"请在 ${handler.runtimeType} 中实现get方法 Map<HttpMethod, List<List>> get routeMap => routes;");
|
||||
return;
|
||||
} else {
|
||||
rethrow;
|
||||
}
|
||||
}
|
||||
|
||||
im.type.declarations.forEach((k, v) {
|
||||
if (v is MethodMirror) {
|
||||
for (var m in v.metadata) {
|
||||
if (m.reflectee is RequestMapping) {
|
||||
var mp = m.reflectee.path as String;
|
||||
|
||||
if (mp.startsWith("/")) {
|
||||
mp = mp.substring(1, mp.length);
|
||||
}
|
||||
|
||||
String p = "$path/$mp";
|
||||
// print("method: ${m.reflectee.method} $p");
|
||||
//把地址加入路由
|
||||
|
||||
addRouter(m.reflectee.method, p, (req, [a, b, c, d, e, f]) async {
|
||||
//从request中获取token并解码传入
|
||||
String? token = req.headers["token"];
|
||||
dynamic payload;
|
||||
if (token != null) {
|
||||
try {
|
||||
payload = JWT.decode(token).payload;
|
||||
} catch (e) {
|
||||
//jwt解码失败
|
||||
print(e);
|
||||
}
|
||||
}
|
||||
List args;
|
||||
if (v.parameters.length == 2) {
|
||||
args = [req, payload ?? <String, dynamic>{}];
|
||||
} else {
|
||||
args = [req];
|
||||
}
|
||||
Response response =
|
||||
await im.invoke(v.simpleName, args).reflectee as Response;
|
||||
//判断contenttype是否定义
|
||||
// var t = response.headers["content-type"];
|
||||
// print(t);
|
||||
Map<String, String> headers = {...response.headers};
|
||||
// if (t == null) {
|
||||
//设置默认contenttype
|
||||
// response = response.change(headers: {
|
||||
// ...response.headers,
|
||||
// "content-type": m.reflectee.responseType.type
|
||||
// });
|
||||
// } else {
|
||||
headers['Access-Control-Allow-Origin'] = '*';
|
||||
headers['Access-Control-Allow-Headers'] =
|
||||
'Content-Type,X-Span-ID,X-Trace-ID';
|
||||
headers['Access-Control-Allow-Methods'] =
|
||||
'GET, POST, PUT, DELETE, OPTIONS';
|
||||
headers["content-type"] = m.reflectee.responseType.type;
|
||||
|
||||
// }
|
||||
response = response.change(headers: headers);
|
||||
|
||||
return response;
|
||||
});
|
||||
}
|
||||
}
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
void addRouter(HttpMethod method, String path, dynamic handler) {
|
||||
@@ -255,7 +241,7 @@ class ShelfWebServer implements WebServer {
|
||||
_router.options(path, (r) {
|
||||
return Response.ok('', headers: {
|
||||
'Access-Control-Allow-Origin': '*',
|
||||
'Access-Control-Allow-Headers': 'Content-Type,X-Span-ID,X-Trace-ID',
|
||||
'Access-Control-Allow-Headers': 'Content-Type,X-Span-ID,X-Trace-ID,token',
|
||||
'Access-Control-Allow-Methods': 'GET, POST, PUT, DELETE, OPTIONS',
|
||||
});
|
||||
});
|
||||
|
||||
@@ -6,7 +6,7 @@ class ShelfWebServer implements WebServer {
|
||||
final String tag = "webserver";
|
||||
ShelfWebServer(this.logger);
|
||||
@override
|
||||
void start(int port) async {}
|
||||
void start(int port, {Function? interceptor}) async {}
|
||||
|
||||
@override
|
||||
void stop() {}
|
||||
|
||||
Reference in New Issue
Block a user