Merge branch 'origin/main' into main

vison.cao · vison.cao · commit a209bafcf25a · 2021-01-06T16:06:48.000+08:00
diff --git a/pom.xml b/pom.xml
@@ -66,6 +66,27 @@
             <artifactId>pebble</artifactId>
             <version>3.1.4</version>
         </dependency>
+        <!-- https://mvnrepository.com/artifact/javax.persistence/javax.persistence-api -->
+        <dependency>
+            <groupId>javax.persistence</groupId>
+            <artifactId>javax.persistence-api</artifactId>
+            <version>2.2</version>
+            <scope>compile</scope>
+        </dependency>
+        <dependency>
+            <groupId>mysql</groupId>
+            <artifactId>mysql-connector-java</artifactId>
+            <version>5.1.26</version>
+        </dependency>
+        <!-- https://mvnrepository.com/artifact/org.mybatis/mybatis -->
+        <dependency>
+            <groupId>org.mybatis</groupId>
+            <artifactId>mybatis</artifactId>
+            <version>3.5.6</version>
+        </dependency>
+
+
+
     </dependencies>
 
     <build>
diff --git a/src/main/java/com/vison/webmvc/config/App.java b/src/main/java/com/vison/webmvc/config/App.java
@@ -1,5 +1,7 @@
 package com.vison.webmvc.config;
 
+import org.reflections.Reflections;
+
 /**
  *
  * @author vison.cao <visonforcoding@gmail.com>
@@ -9,4 +11,5 @@ public class App {
     public static final String SESSION_USER = "user"; // 用户对象
     public static String _uniq_req_no = null;
     public static String OrderNoIncKey = ":order:no";
+    public static Reflections f;
 }
diff --git a/src/main/java/com/vison/webmvc/config/DatabaseConfig.java b/src/main/java/com/vison/webmvc/config/DatabaseConfig.java
@@ -0,0 +1,30 @@
+package com.vison.webmvc.config;
+
+import java.util.Properties;
+import javax.sql.DataSource;
+import org.apache.ibatis.datasource.unpooled.UnpooledDataSourceFactory;
+
+/**
+ *
+ * @author vison.cao <visonforcoding@gmail.com>
+ */
+public class DatabaseConfig {
+
+    static String driver = "com.mysql.jdbc.Driver";
+    static String url = "jdbc:mysql://localhost:3306/db_itdoc?useSSL=false";
+    static String username = "root";
+    static String password = "12345678";
+
+    public static DataSource getDataSource() {
+        Properties properties = new Properties();
+        properties.setProperty("driver", driver);
+        properties.setProperty("url", url);
+        properties.setProperty("username", username);
+        properties.setProperty("password", password);
+        UnpooledDataSourceFactory unpooledDataSourceFactory = new UnpooledDataSourceFactory();
+        unpooledDataSourceFactory.setProperties(properties);
+        DataSource dataSource = unpooledDataSourceFactory.getDataSource();
+        return dataSource;
+    }
+
+}
diff --git a/src/main/java/com/vison/webmvc/controller/UserController.java b/src/main/java/com/vison/webmvc/controller/UserController.java
@@ -4,8 +4,14 @@
 import com.vison.webmvc.framework.GetMapping;
 import com.vison.webmvc.Response;
 import com.vison.webmvc.config.Log;
+import com.vison.webmvc.dao.UserMapper;
+import com.vison.webmvc.framework.MybatisLoader;
+import com.vison.webmvc.framework.PostMapping;
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
+import org.apache.ibatis.session.SqlSession;
+//import org.hibernate.Session;
+//import org.hibernate.Transaction;
 
 /**
  *
@@ -23,12 +29,26 @@ public String profile(HttpServletRequest request, HttpServletResponse response)
     }
 
     @GetMapping(path = "/user")
-    public Response user() {
-        User user = new User();
-        user.setEmail("visonforcoding@gmail.com");
-        user.setName("曹麦穗");
-        Log.debug("request user", user);
+    public Response user(HttpServletRequest request, int id) {
+        User user = null;
+        SqlSession session = MybatisLoader.getSqlSessionFactory().openSession();
+        UserMapper mapper = session.getMapper(UserMapper.class);
+        user = mapper.selectUser(id);
         return new Response(0, "获取成功", user);
     }
 
+    @PostMapping(path = "/user/add")
+    public Response add(User user) {
+        Log.info("request user", user);
+        try {
+            SqlSession session = MybatisLoader.getSqlSessionFactory().openSession();
+            UserMapper mapper = session.getMapper(UserMapper.class);
+            int id = mapper.insertUser(user);
+            session.commit();
+            Log.debug("返回", id);
+        } catch (Exception e) {
+            Log.error("保存失败", e);
+        }
+        return new Response(0, "保存", user);
+    }
 }
diff --git a/src/main/java/com/vison/webmvc/dao/UserMapper.java b/src/main/java/com/vison/webmvc/dao/UserMapper.java
@@ -0,0 +1,19 @@
+package com.vison.webmvc.dao;
+
+import com.vison.webmvc.entity.User;
+import org.apache.ibatis.annotations.Insert;
+import org.apache.ibatis.annotations.Select;
+
+/**
+ *
+ * @author vison.cao <visonforcoding@gmail.com>
+ */
+public interface UserMapper {
+
+    @Select("SELECT * FROM user WHERE id = #{id}")
+    User selectUser(int id);
+
+    @Insert("INSERT INTO user(name,email) VALUES(#{name}, #{email})")
+    int insertUser(User user);
+
+}
diff --git a/src/main/java/com/vison/webmvc/framework/ContextServletListener.java b/src/main/java/com/vison/webmvc/framework/ContextServletListener.java
@@ -0,0 +1,37 @@
+package com.vison.webmvc.framework;
+
+import com.vison.webmvc.config.App;
+import com.vison.webmvc.config.Log;
+import javax.servlet.ServletContextEvent;
+import javax.servlet.ServletContextListener;
+import javax.servlet.annotation.WebListener;
+import org.reflections.Reflections;
+import org.reflections.scanners.MethodAnnotationsScanner;
+import org.reflections.util.ClasspathHelper;
+import org.reflections.util.ConfigurationBuilder;
+import org.reflections.util.FilterBuilder;
+
+/**
+ * Web application lifecycle listener.
+ *
+ * @author vison.cao <visonforcoding@gmail.com>
+ */
+@WebListener
+public class ContextServletListener implements ServletContextListener {
+
+    @Override
+    public void contextInitialized(ServletContextEvent sce) {
+        Log.info("应用程序启动...");
+        String packageName = "com.vison.webmvc.controller";
+        ConfigurationBuilder config = new ConfigurationBuilder();
+        config.filterInputsBy(new FilterBuilder().includePackage(packageName));
+        config.addUrls(ClasspathHelper.forPackage(packageName));
+        config.setScanners(new MethodAnnotationsScanner());
+        App.f = new Reflections(config);
+        MybatisLoader.getSqlSessionFactory();
+    }
+
+    @Override
+    public void contextDestroyed(ServletContextEvent sce) {
+    }
+}
diff --git a/src/main/java/com/vison/webmvc/framework/DispatchServlet.java b/src/main/java/com/vison/webmvc/framework/DispatchServlet.java
@@ -1,12 +1,18 @@
 package com.vison.webmvc.framework;
 
 import com.google.gson.Gson;
+import com.vison.webmvc.config.App;
 import com.vison.webmvc.config.Log;
 import com.vison.webmvc.framework.exception.NullRouteException;
+import java.io.BufferedReader;
 import java.io.IOException;
+import java.lang.annotation.Annotation;
+import java.lang.reflect.InvocationTargetException;
 import java.lang.reflect.Method;
 import java.lang.reflect.Parameter;
 import java.util.Set;
+import java.util.logging.Level;
+import java.util.logging.Logger;
 import javax.servlet.ServletException;
 import javax.servlet.annotation.WebServlet;
 import javax.servlet.http.HttpServlet;
@@ -26,34 +32,66 @@
 @WebServlet(name = "DispatchServlet", urlPatterns = {"/"})
 public class DispatchServlet extends HttpServlet {
 
+    private Reflections f;
+
     @Override
     public void init() throws ServletException {
         ViewEngine.load(this.getServletContext());
+        f = App.f;
+
     }
 
     @Override
     protected void doGet(HttpServletRequest req, HttpServletResponse resp)
             throws ServletException, IOException {
+        try {
+            process(req, resp, RequestMethod.GET);
+        } catch (NullRouteException ex) {
+            Logger.getLogger(DispatchServlet.class.getName()).log(Level.SEVERE, null, ex);
+        }
+    }
+
+    @Override
+    protected void doPost(HttpServletRequest req, HttpServletResponse resp)
+            throws ServletException, IOException {
+        try {
+            process(req, resp, RequestMethod.POST);
+        } catch (NullRouteException ex) {
+            Logger.getLogger(DispatchServlet.class.getName()).log(Level.SEVERE, null, ex);
+        }
+
+    }
+
+    private void process(HttpServletRequest req, HttpServletResponse resp, RequestMethod requestMethod)
+            throws NullRouteException, IOException {
         resp.setContentType("text/html");
         resp.setCharacterEncoding("UTF-8");
         String path = req.getRequestURI().substring(req.getContextPath().length());
-        Object res;
+        Object res = null;
         try {
-            res = dispatch(path, req, resp);
+            Method invokeMethod = this.getMaps(path, requestMethod);
+            switch (requestMethod) {
+                case GET:
+                    res = getDispatch(req, resp, invokeMethod);
+                    break;
+                case POST:
+                    res = postDispatch(req, resp, invokeMethod);
+                    break;
+            }
         } catch (NullRouteException ex) {
             res = "404 Not Found";
+        } catch (Exception e) {
+            Log.error("捕获错误", e);
         }
         String responseBody = this.handInvokeRes(res);
-        Log.info("返回信息", responseBody);
         resp.getWriter().write(responseBody);
         resp.getWriter().flush();
     }
 
-    private Object dispatch(String path, HttpServletRequest request, HttpServletResponse response)
-            throws NullRouteException {
-        Method method = this.getMaps(path);
+    private Object getDispatch(HttpServletRequest request, HttpServletResponse response, Method invokeMethod) {
+
         Object res = null;
-        Parameter[] parameters = method.getParameters();
+        Parameter[] parameters = invokeMethod.getParameters();
         Object[] arguments = new Object[parameters.length];
         for (int i = 0; i < parameters.length; i++) {
             Parameter parameter = parameters[i];
@@ -80,12 +118,48 @@ private Object dispatch(String path, HttpServletRequest request, HttpServletResp
         }
         Object obj;
         try {
-            obj = method.getDeclaringClass().getDeclaredConstructor().newInstance();
-            res = method.invoke(obj, arguments);
+            obj = invokeMethod.getDeclaringClass().getDeclaredConstructor().newInstance();
+            res = invokeMethod.invoke(obj, arguments);
+        } catch (Exception e) {
+            InvocationTargetException targetEx = (InvocationTargetException) e;
+            Throwable trowEx = targetEx.getTargetException();
+            Log.error("方法invoke失败", trowEx);
+        }
+        return res;
+    }
+
+    private Object postDispatch(HttpServletRequest request, HttpServletResponse response, Method invokeMethod)
+            throws IOException {
+        Object res = null;
+        Parameter[] parameters = invokeMethod.getParameters();
+        Object[] arguments = new Object[parameters.length];
+        for (int i = 0; i < parameters.length; i++) {
+            Parameter parameter = parameters[i];
+            Log.info("参数类型", parameter.getType());
+            Class<?> parameterClass = parameter.getType();
+            if (parameterClass == HttpServletRequest.class) {
+                arguments[i] = request;
+            } else if (parameterClass == HttpServletResponse.class) {
+                arguments[i] = response;
+            } else if (parameterClass == HttpSession.class) {
+                arguments[i] = request.getSession();
+            } else {
+                // 读取JSON并解析为JavaBean:
+                request.setCharacterEncoding("UTF-8");
+                BufferedReader reader = request.getReader();
+                Gson gson = new Gson();
+                arguments[i] = gson.fromJson(reader, parameterClass);
+            }
+        }
+        Object obj;
+        try {
+            obj = invokeMethod.getDeclaringClass().getDeclaredConstructor().newInstance();
+            res = invokeMethod.invoke(obj, arguments);
         } catch (Exception e) {
             Log.error("方法invoke失败", e);
         }
         return res;
+
     }
 
     private String getOrDefault(HttpServletRequest request, String name, String defaultValue) {
@@ -108,22 +182,27 @@ private String handInvokeRes(Object obj) {
         return jsonRes;
     }
 
-    private Method getMaps(String path) throws NullRouteException {
-
-        String packageName = "com.vison.webmvc.controller";
-        ConfigurationBuilder config = new ConfigurationBuilder();
-        config.filterInputsBy(new FilterBuilder().includePackage(packageName));
-        config.addUrls(ClasspathHelper.forPackage(packageName));
-        config.setScanners(new MethodAnnotationsScanner());
-        Reflections f = new Reflections(config);
-        Set<Method> resources = f.getMethodsAnnotatedWith(GetMapping.class);
+    private Method getMaps(String path, RequestMethod requestMethod) throws NullRouteException {
 
-        for (Method method : resources) {
-            GetMapping annotation = method.getAnnotation(GetMapping.class);
-            if (path.equals(annotation.path())) {
-                return method;
+        if (requestMethod == RequestMethod.GET) {
+            Set<Method> resources = f.getMethodsAnnotatedWith(GetMapping.class);
+            for (Method method : resources) {
+                GetMapping annotation = method.getAnnotation(GetMapping.class);
+                if (annotation.path().equals(path)) {
+                    return method;
+                }
             }
         }
+        if (requestMethod == RequestMethod.POST) {
+            Set<Method> resources = f.getMethodsAnnotatedWith(PostMapping.class);
+            for (Method method : resources) {
+                PostMapping annotation = method.getAnnotation(PostMapping.class);
+                if (annotation.path().equals(path)) {
+                    return method;
+                }
+            }
+        }
+
         throw new NullRouteException();
     }
 }
diff --git a/src/main/java/com/vison/webmvc/framework/MybatisLoader.java b/src/main/java/com/vison/webmvc/framework/MybatisLoader.java
@@ -0,0 +1,36 @@
+package com.vison.webmvc.framework;
+
+import com.vison.webmvc.config.DatabaseConfig;
+import com.vison.webmvc.dao.UserMapper;
+import javax.sql.DataSource;
+import org.apache.ibatis.mapping.Environment;
+import org.apache.ibatis.session.SqlSessionFactory;
+import org.apache.ibatis.transaction.TransactionFactory;
+import org.apache.ibatis.transaction.jdbc.JdbcTransactionFactory;
+import org.apache.ibatis.session.Configuration;
+import org.apache.ibatis.session.SqlSessionFactoryBuilder;
+
+/**
+ *
+ * @author vison.cao <visonforcoding@gmail.com>
+ */
+public class MybatisLoader {
+
+    static SqlSessionFactory sqlSessionFactory = null;
+
+    public static SqlSessionFactory getSqlSessionFactory() {
+        if (sqlSessionFactory == null) {
+            DataSource dataSource = DatabaseConfig.getDataSource();
+            TransactionFactory transactionFactory
+                    = new JdbcTransactionFactory();
+            Environment environment
+                    = new Environment("development", transactionFactory, dataSource);
+            Configuration configuration = new Configuration(environment);
+            configuration.addMapper(UserMapper.class);
+            sqlSessionFactory
+                    = new SqlSessionFactoryBuilder().build(configuration);
+        }
+        return sqlSessionFactory;
+    }
+
+}
diff --git a/src/main/java/com/vison/webmvc/framework/PostMapping.java b/src/main/java/com/vison/webmvc/framework/PostMapping.java
diff --git a/src/main/java/com/vison/webmvc/framework/RequestMethod.java b/src/main/java/com/vison/webmvc/framework/RequestMethod.java
diff --git a/src/main/webapp/WEB-INF/web.xml b/src/main/webapp/WEB-INF/web.xml

Original file line number	Diff line number	Diff line change
`@@ -1,5 +1,7 @@`
`1`	`1`	`package com.vison.webmvc.config;`
`2`	`2`
	`3`	`+import org.reflections.Reflections;`
	`4`	`+`
`3`	`5`	`/**`
`4`	`6`	`*`
`5`	`7`	`* @author vison.cao <visonforcoding@gmail.com>`
`@@ -9,4 +11,5 @@ public class App {`
`9`	`11`	`public static final String SESSION_USER = "user"; // 用户对象`
`10`	`12`	`public static String _uniq_req_no = null;`
`11`	`13`	`public static String OrderNoIncKey = ":order:no";`
	`14`	`+ public static Reflections f;`
`12`	`15`	`}`