'包结构调整'

src/main/java/net/tccn/servlet/BaseServlet.java

@@ -0,0 +1,87 @@
+package net.tccn.servlet;
+
+import com.arangodb.ArangoDBException;
+import com.google.gson.Gson;
+import net.tccn.base.JBean;
+import net.tccn.base.Kv;
+import net.tccn.base.arango.ArangoSource;
+import net.tccn.service.UserService;
+import net.tccn.user.User;
+import org.redkale.net.http.HttpRequest;
+import org.redkale.net.http.HttpResponse;
+import org.redkale.net.http.HttpScope;
+import org.redkale.net.http.HttpServlet;
+
+import javax.annotation.Resource;
+import java.io.IOException;
+import java.util.logging.Level;
+import java.util.logging.Logger;
+
+/**
+ * @author: liangxianyou at 2018/11/8 17:05.
+ */
+public class BaseServlet extends HttpServlet {
+
+    @Resource
+    private UserService userService;
+
+    public Logger logger = Logger.getLogger(this.getClass().getSimpleName());
+    protected static Gson gson = new Gson();
+
+    @Override
+    protected void preExecute(HttpRequest request, HttpResponse response) throws IOException {
+        String sessionid = request.getParameter("token");
+        if (sessionid == null) {
+            sessionid = request.getHeader("token");
+        }
+        if (sessionid == null) {
+            sessionid = request.getSessionid(true);
+        }
+
+        if (sessionid != null) {
+            User user = userService.current(sessionid);
+            request.setCurrentUser(user);
+        }
+
+        super.preExecute(request, response);
+    }
+
+    @Override
+    protected void authenticate(HttpRequest request, HttpResponse response) throws IOException {
+         //fixme: 权限拦截
+        if (request.currentUser() == null) {
+            if ("XMLHttpRequest".equals(request.getHeader("X-Requested-With"))){
+                response.finish(JBean.by(-2, "未登陆"));
+            }else {
+                response.finish(HttpScope.refer("/user/login.html"));
+            }
+            return;
+        }
+        super.authenticate(request, response);
+    }
+
+    @Override
+    public void execute(HttpRequest request, HttpResponse response) throws IOException {
+        try {
+            super.execute(request, response);
+        } catch (ArangoDBException e) {
+            logger.log(Level.INFO, "arangodb init!", e);
+            ArangoSource.init();
+        }
+    }
+
+    public Kv getParams(HttpRequest request, String... key) {
+        Kv kv = Kv.of();
+        for (String k : key) {
+            if (k.contains("=")) {   //如果没有值使用默认值
+                kv.put(k.split("=")[0], request.getParameter(k.split("=")[0], k.split("=")[1]));
+                continue;
+            } else if (k.contains("<")) {  //强制使用"<"右侧的值
+                kv.put(k.split("<")[0], k.split("<")[1]);
+                continue;
+            }
+            kv.put(k, request.getParameter(k));
+        }
+        return kv;
+    }
+}