diff --git a/src/org/redkale/net/http/Rest.java b/src/org/redkale/net/http/Rest.java index 639e74e12..cb08f33f6 100644 --- a/src/org/redkale/net/http/Rest.java +++ b/src/org/redkale/net/http/Rest.java @@ -148,9 +148,16 @@ public final class Rest { final String resourceDescriptor = sb1.toString(); final String resourceGenericDescriptor = sb1.length() == sb2.length() ? null : sb2.toString(); - //---------------------------------------------------------------------------------------- + //---------------------------------------------------------------------------------------- + final Map> asmParamMap = MethodParamClassVisitor.getMethodParamNames(webSocketType); + final Set messageNames = new HashSet<>(); for (Method method : webSocketType.getMethods()) { - + RestOnMessage rom = method.getAnnotation(RestOnMessage.class); + if (rom == null) continue; + String name = rom.name(); + if (name.isEmpty()) throw new RuntimeException(method + " RestOnMessage.name is empty createRestWebSocketServlet"); + if (messageNames.contains(name)) throw new RuntimeException(method + " repeat RestOnMessage.name(" + name + ") createRestWebSocketServlet"); + messageNames.add(name); } //---------------------------------------------------------------------------------------- @@ -292,7 +299,7 @@ public final class Rest { } public static void main(String[] args) throws Throwable { - System.out.println(createRestWebSocketServlet((Class)Class.forName("org.redkale.test.ws.ChatWebSocket"))); + System.out.println(createRestWebSocketServlet((Class) Class.forName("org.redkale.test.ws.ChatWebSocket"))); } static T createRestServlet(final Class userType0, final Class baseServletType, final Class serviceType) { diff --git a/src/org/redkale/net/http/RestOnMessage.java b/src/org/redkale/net/http/RestOnMessage.java index 1da2a919a..099b597d5 100644 --- a/src/org/redkale/net/http/RestOnMessage.java +++ b/src/org/redkale/net/http/RestOnMessage.java @@ -28,7 +28,7 @@ public @interface RestOnMessage { * * @return String */ - String name() default ""; + String name(); /** * 备注描述