wildfire chat server source initial submit

2019-03-16 17:04:23 +08:00
parent e59217bd10
commit cab3ca0916
1466 changed files with 528862 additions and 15919 deletions
@@ -0,0 +1,105 @@
+/*
+ * This file is part of the Wildfire Chat package.
+ * (c) Heavyrain2012 <[email protected]>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+package com.xiaoleilu.loServer.action;
+
+import com.google.gson.Gson;
+import com.xiaoleilu.loServer.RestResult;
+import com.xiaoleilu.loServer.action.admin.GetIMTokenAction;
+import com.xiaoleilu.loServer.annotation.RequireAuthentication;
+import com.xiaoleilu.loServer.handler.Request;
+import com.xiaoleilu.loServer.handler.Response;
+import io.moquette.spi.IMessagesStore;
+import io.moquette.spi.ISessionsStore;
+import io.moquette.spi.impl.Utils;
+import io.netty.channel.ChannelHandlerContext;
+import io.netty.handler.codec.http.FullHttpRequest;
+import io.netty.handler.codec.http.HttpRequest;
+import io.netty.handler.codec.http.HttpResponseStatus;
+import win.liyufan.im.DBUtil;
+import win.liyufan.im.ErrorCode;
+
+import static win.liyufan.im.ErrorCode.ERROR_CODE_SUCCESS;
+
+/**
+ * 请求处理接口<br>
+ * 当用户请求某个Path，则调用相应Action的doAction方法
+ * @author Looly
+ *
+ */
+
+abstract public class Action {
+    public static IMessagesStore messagesStore = null;
+    public static ISessionsStore sessionsStore = null;
+
+    public ChannelHandlerContext ctx;
+
+    protected boolean isSync = true;
+
+    public ErrorCode preAction(Request request, Response response) {
+        if (getClass().getAnnotation(RequireAuthentication.class) != null) {
+            //do authentication
+        }
+
+        return ERROR_CODE_SUCCESS;
+    }
+	public boolean doAction(Request request, Response response) {
+        ErrorCode errorCode = preAction(request, response);
+        if (errorCode == ErrorCode.ERROR_CODE_SUCCESS) {
+            //事务逻辑有缺陷，先注释掉
+//            if (isTransactionAction() && !(this instanceof IMAction)) {
+//                DBUtil.beginTransaction();
+//                try {
+//                    action(request, response);
+//                    DBUtil.commit();
+//                } catch (Exception e) {
+//                    e.printStackTrace();
+//                    DBUtil.roolback();
+//                    throw e;
+//                }
+//            } else {
+                action(request, response);
+//            }
+            afterAction(request, response);
+        } else {
+            response.setStatus(HttpResponseStatus.OK);
+            if (errorCode == null) {
+                errorCode = ErrorCode.ERROR_CODE_SUCCESS;
+            }
+
+            RestResult result = RestResult.resultOf(errorCode, errorCode.getMsg(), RestResult.resultOf(errorCode));
+            response.setContent(new Gson().toJson(result));
+            response.send();
+        }
+
+        if(this instanceof IMAction || this instanceof RouteAction || this instanceof GetIMTokenAction) {
+            isSync = false;
+        }
+        return isSync;
+    }
+    public boolean isTransactionAction() {
+        return false;
+    }
+    abstract public void action(Request request, Response response);
+    public void afterAction(Request request, Response response) {
+
+    }
+
+
+    protected <T> T getRequestBody(HttpRequest request, Class<T> cls) {
+        if (request instanceof FullHttpRequest) {
+            FullHttpRequest fullHttpRequest = (FullHttpRequest) request;
+            byte[] bytes = Utils.readBytesAndRewind(fullHttpRequest.content());
+            String content = new String(bytes);
+            Gson gson = new Gson();
+            T t = gson.fromJson(content, cls);
+            return t;
+        }
+        return null;
+    }
+}