Updating customer injection for per check.

Author: liuhongyuand

EasyRest/src/main/java/tech/dbgsoftware/easyrest/actors/request/ControllerInvokeActor.java

@@ -7,6 +7,7 @@
 import tech.dbgsoftware.easyrest.actors.Signal;
 import tech.dbgsoftware.easyrest.actors.remote.model.RemoteInvokeObject;
 import tech.dbgsoftware.easyrest.actors.response.ResponseProcessActor;
+import tech.dbgsoftware.easyrest.aop.customer.CustomInjection;
 import tech.dbgsoftware.easyrest.ioc.utils.BeanOperationUtils;
 import tech.dbgsoftware.easyrest.model.HttpEntity;
 import tech.dbgsoftware.easyrest.model.ResponseEntity;
@@ -23,22 +24,22 @@ public Receive createReceive() {
                 Method method = httpEntity.getMethod();
                 Class<?> controller = httpEntity.getController();
                 if (method.getReturnType().getName().equalsIgnoreCase(Void.class.getSimpleName())) {
-                    invokeMethod(method, controller, httpEntity.getArgs(), null);
+                    invokeMethod(method, controller, httpEntity.getArgs(), null, httpEntity);
                     httpEntity.setResponseEntity(ResponseEntity.buildOkResponse());
                 } else if (method.getReturnType().equals(ResponseEntity.class)) {
-                    httpEntity.setResponseEntity((ResponseEntity) invokeMethod(method, controller, httpEntity.getArgs(), null));
+                    httpEntity.setResponseEntity((ResponseEntity) invokeMethod(method, controller, httpEntity.getArgs(), null, httpEntity));
                 } else {
-                    httpEntity.setResponseEntity(ResponseEntity.buildBaseResponse(invokeMethod(method, controller, httpEntity.getArgs(), null)));
+                    httpEntity.setResponseEntity(ResponseEntity.buildBaseResponse(invokeMethod(method, controller, httpEntity.getArgs(), null, httpEntity)));
                 }
                 ActorFactory.createActor(ResponseProcessActor.class).tell(httpEntity, ActorRef.noSender());
             } catch (Exception e) {
                 LogUtils.error(e.getMessage(), e);
-                httpEntity.addError(e);
+                httpEntity.addError(e.getCause());
                 ActorFactory.createActor(ExceptionHandleActor.class).tell(httpEntity, ActorRef.noSender());
             }
         })).match(RemoteInvokeObject.class, (remoteInvokeObject -> {
             try {
-                Object result = invokeMethod(remoteInvokeObject.getMethod(), remoteInvokeObject.getImplClass(), remoteInvokeObject.getArgs(), remoteInvokeObject.getInvokeBeanName());
+                Object result = invokeMethod(remoteInvokeObject.getMethod(), remoteInvokeObject.getImplClass(), remoteInvokeObject.getArgs(), remoteInvokeObject.getInvokeBeanName(), null);
                 if (remoteInvokeObject.getMethod().getReturnType().getName().equalsIgnoreCase(Void.class.getSimpleName())) {
                     remoteInvokeObject.setResult(ResponseEntity.buildOkResponse());
                 } else {
@@ -52,20 +53,30 @@ public Receive createReceive() {
         })).build();
     }
 
-    private Object invokeMethod(Method method, Class<?> controller, Object[] args, String invokeBeanName) throws Exception {
+    private Object invokeMethod(Method method, Class<?> controller, Object[] args, String invokeBeanName, HttpEntity httpEntity) throws Exception {
         if (args.length > 0) {
             if (invokeBeanName == null || invokeBeanName.equalsIgnoreCase("null")) {
+                invokePerCheck(BeanOperationUtils.getBean(controller), httpEntity);
                 return method.invoke(BeanOperationUtils.getBean(controller), args);
             } else {
+                invokePerCheck(BeanOperationUtils.getBean(invokeBeanName, controller), httpEntity);
                 return method.invoke(BeanOperationUtils.getBean(invokeBeanName, controller), args);
             }
         } else {
             if (invokeBeanName == null || invokeBeanName.equalsIgnoreCase("null")) {
+                invokePerCheck(BeanOperationUtils.getBean(controller), httpEntity);
                 return method.invoke(BeanOperationUtils.getBean(controller));
             } else {
+                invokePerCheck(BeanOperationUtils.getBean(invokeBeanName, controller), httpEntity);
                 return method.invoke(BeanOperationUtils.getBean(invokeBeanName, controller));
             }
         }
     }
 
+    private void invokePerCheck(Object classz, HttpEntity httpEntity) throws Exception {
+        if (httpEntity != null && classz instanceof CustomInjection) {
+            CustomInjection.class.getMethods()[0].invoke(classz, httpEntity);
+        }
+    }
+
 }

EasyRest/src/main/java/tech/dbgsoftware/easyrest/aop/customer/CustomInjection.java

@@ -0,0 +1,9 @@
+package tech.dbgsoftware.easyrest.aop.customer;
+
+import tech.dbgsoftware.easyrest.model.HttpEntity;
+
+public interface CustomInjection {
+
+    HttpEntity preCheck(HttpEntity httpEntity);
+
+}

EasyRest/src/main/java/tech/dbgsoftware/easyrest/model/HttpEntity.java

@@ -74,6 +74,10 @@ public void addError(Exception e){
         errorMap.put(e.getMessage(), e.getClass().getSimpleName());
     }
 
+    public void addError(Throwable e){
+        errorMap.put(e.getMessage(), e.getClass().getSimpleName());
+    }
+
     public RestObject getRestObject() {
         return restObject;
     }

Example/Example-Quick-Start/src/main/java/com/example/rest/TestRestService.java

@@ -4,9 +4,12 @@
 import tech.dbgsoftware.easyrest.annotations.method.Get;
 import tech.dbgsoftware.easyrest.annotations.method.Post;
 import tech.dbgsoftware.easyrest.annotations.parameter.AllDefined;
+import tech.dbgsoftware.easyrest.aop.customer.CustomInjection;
+import tech.dbgsoftware.easyrest.exception.PermissionException;
+import tech.dbgsoftware.easyrest.model.HttpEntity;
 
 @BindURL("/test")
-public interface TestRestService {
+public interface TestRestService extends CustomInjection {
 
     @Get
     String ping();
@@ -15,4 +18,8 @@ public interface TestRestService {
     @AllDefined
     String test(String a, String b);
 
+    @Override
+    default HttpEntity preCheck(HttpEntity httpEntity) {
+        throw new PermissionException("No permission for this rest api.");
+    }
 }