一步一步地配置Spring

亦凉 2022-07-24 08:17 420阅读 0赞

本文旨在从一个空工程一步一步地配置Spring,空工程见上一篇文章创建Maven父子工程。

##一、spring基本配置

###1. 添加spring依赖
父工程pom.xml添加spring依赖

  1. <dependencyManagement>
  2. <dependencies>
  3. <!-- 使用spring的BOM管理依赖 -->
  4. <dependency>
  5. <groupId>org.springframework</groupId>
  6. <artifactId>spring-framework-bom</artifactId>
  7. <version>4.2.6.RELEASE</version>
  8. <type>pom</type>
  9. <scope>import</scope>
  10. </dependency>
  11. </dependencies>
  12. </dependencyManagement>
  13. <dependencies>
  14. <!-- spring配置开始 -->
  15. <dependency>
  16. <groupId>org.springframework</groupId>
  17. <artifactId>spring-context</artifactId>
  18. </dependency>
  19. <dependency>
  20. <groupId>org.springframework</groupId>
  21. <artifactId>spring-web</artifactId>
  22. </dependency>
  23. <dependency>
  24. <groupId>org.springframework</groupId>
  25. <artifactId>spring-webmvc</artifactId>
  26. </dependency>
  27. <dependency>
  28. <groupId>org.springframework</groupId>
  29. <artifactId>spring-core</artifactId>
  30. <exclusions>
  31. <exclusion>
  32. <groupId>commons-logging</groupId>
  33. <artifactId>commons-logging</artifactId>
  34. </exclusion>
  35. </exclusions>
  36. </dependency>
  37. <!-- spring配置结束 -->
  38. </dependencies>

###2. web.xml添加spring配置
修改web.xml文件为如下:

  1. <?xml version="1.0" encoding="UTF-8"?>
  2. <web-app xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xmlns="http://java.sun.com/xml/ns/javaee" xmlns:web="http://java.sun.com/xml/ns/javaee" xsi:schemaLocation="http://java.sun.com/xml/ns/javaee http://java.sun.com/xml/ns/javaee/web-app_2_5.xsd" id="WebApp_ID" version="2.5">
  3. <context-param>
  4. <param-name>contextConfigLocation</param-name>
  5. <param-value>classpath*:META-INF/dispatcher.xml</param-value>
  6. </context-param>
  7. <servlet>
  8. <servlet-name>dispatcher</servlet-name>
  9. <servlet-class>org.springframework.web.servlet.DispatcherServlet</servlet-class>
  10. <init-param>
  11. <param-name>contextConfigLocation</param-name>
  12. <param-value></param-value>
  13. </init-param>
  14. <load-on-startup>1</load-on-startup>
  15. </servlet>
  16. <servlet-mapping>
  17. <servlet-name>dispatcher</servlet-name>
  18. <url-pattern>/</url-pattern>
  19. </servlet-mapping>
  20. <listener>
  21. <listener-class>org.springframework.web.context.ContextLoaderListener</listener-class>
  22. </listener>
  23. <filter>
  24. <filter-name>encodingFilter</filter-name>
  25. <filter-class>org.springframework.web.filter.CharacterEncodingFilter</filter-class>
  26. <init-param>
  27. <param-name>encoding</param-name>
  28. <param-value>utf8</param-value>
  29. </init-param>
  30. <init-param>
  31. <param-name>forceEncoding</param-name>
  32. <param-value>true</param-value>
  33. </init-param>
  34. </filter>
  35. <filter-mapping>
  36. <filter-name>encodingFilter</filter-name>
  37. <url-pattern>/*</url-pattern>
  38. </filter-mapping>
  39. <servlet-mapping>
  40. <servlet-name>default</servlet-name>
  41. <url-pattern>*.html</url-pattern>
  42. </servlet-mapping>
  43. <welcome-file-list>
  44. <welcome-file>index.html</welcome-file>
  45. </welcome-file-list>
  46. </web-app>

###3. 添加dispatcher.xml文件
在web工程的src/main/resources/META-INF目录下添加dispatcher.xml文件,并添加如下配置:

  1. <?xml version="1.0" encoding="UTF-8"?>
  2. <beans xmlns="http://www.springframework.org/schema/beans" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xmlns:p="http://www.springframework.org/schema/p" xmlns:context="http://www.springframework.org/schema/context" xmlns:mvc="http://www.springframework.org/schema/mvc" xsi:schemaLocation="http://www.springframework.org/schema/mvc http://www.springframework.org/schema/mvc/spring-mvc-3.1.xsd http://www.springframework.org/schema/beans http://www.springframework.org/schema/beans/spring-beans.xsd http://www.springframework.org/schema/context http://www.springframework.org/schema/context/spring-context.xsd">
  3. <!-- 加入下面这两个配置才能支持注解功能 -->
  4. <bean id="handlerMapping" class="org.springframework.web.servlet.mvc.annotation.DefaultAnnotationHandlerMapping" />
  5. <bean id="handlerAdapter" class="org.springframework.web.servlet.mvc.annotation.AnnotationMethodHandlerAdapter" />
  6. <!-- 本工程拦截器 -->
  7. <mvc:interceptors>
  8. <bean class="org.springframework.web.servlet.mvc.WebContentInterceptor">
  9. <property name="cacheSeconds" value="0" />
  10. <property name="useExpiresHeader" value="true" />
  11. <property name="useCacheControlHeader" value="true" />
  12. <property name="useCacheControlNoStore" value="true" />
  13. <property name="alwaysUseFullPath" value="true" />
  14. </bean>
  15. </mvc:interceptors>
  16. <!-- 扫描包,如果用扫描则不用再写 <context:annotation-config /> -->
  17. <context:component-scan base-package="com.alan.controller" />
  18. <context:component-scan base-package="com.alan.service.impl" />
  19. <context:component-scan base-package="com.alan.dao.impl" />
  20. </beans>

###4. 添加controller/service/dao等类
下面直接贴出各实现类。
(1)TestDaoImpl.java

  1. package com.alan.dao.impl;
  2. import org.springframework.stereotype.Repository;
  3. import com.alan.dao.TestDao;
  4. @Repository("testDao")
  5. public class TestDaoImpl implements TestDao {
  6. @Override
  7. public String test() {
  8. return "../index.html";
  9. }
  10. }

(2)TestServiceImpl.java

  1. package com.alan.service.impl;
  2. import org.springframework.beans.factory.annotation.Autowired;
  3. import org.springframework.stereotype.Service;
  4. import com.alan.dao.TestDao;
  5. import com.alan.service.TestService;
  6. @Service("testService")
  7. public class TestServiceImpl implements TestService {
  8. @Autowired
  9. private TestDao testDao;
  10. @Override
  11. public String test() {
  12. return testDao.test();
  13. }
  14. }

(3)TestController.java

  1. package com.alan.controller;
  2. import org.springframework.beans.factory.annotation.Autowired;
  3. import org.springframework.web.bind.annotation.RequestMapping;
  4. import org.springframework.web.bind.annotation.RestController;
  5. import com.alan.service.TestService;
  6. @RestController
  7. @RequestMapping("/alan")
  8. public class TestController {
  9. @Autowired
  10. private TestService testService;
  11. @RequestMapping("/test")
  12. public String test(){
  13. return testService.test();
  14. }
  15. }

###5. 访问
重新编译工程并启动tomcat,通过浏览器访问,如果出现以下内容则表示配置正确。
注意:若tomcat启动老是报org.springframework.web.context.ContextLoaderListener找不到,把Eclipse关了,并把工作空间中Server/.metadata/RemoteTempFiles等几个目录删除,重新打开Eclipse并配置Preferences且重新导入工程再试。
这里写图片描述

(防采集:本文唯一网址在CSDN上,博主会不定期更新加载其它配置,其它均为未经博主同意转载的,原文为http://blog.csdn.net/tangtong1/article/details/51442757)

##二、配置Json支持

###6. 添加Json依赖
修改父工程的pom.xml,添加以下内容:

  1. <!-- json配置开始 -->
  2. <dependency>
  3. <groupId>com.alibaba</groupId>
  4. <artifactId>fastjson</artifactId>
  5. <version>1.2.4</version>
  6. </dependency>
  7. <!-- json配置结束 -->

###7. 配置@ResponseBody返回Json
在web工程的dispatcher.xml文件中添加如下配置:

  1. <!-- 加入以下配置支持@ResponseBody返回Json格式 -->
  2. <mvc:annotation-driven>
  3. <mvc:message-converters>
  4. <bean class="com.alibaba.fastjson.support.spring.FastJsonHttpMessageConverter">
  5. <property name="supportedMediaTypes">
  6. <list>
  7. <value>application/json;charset=UTF-8</value>
  8. <value>text/html;charset=UTF-8</value>
  9. <value>text/plain;charset=UTF-8</value>
  10. </list>
  11. </property>
  12. <!-- 为null字段时是否显示 -->
  13. <!-- <property name="features"> <list> <value>WriteMapNullValue</value> <value>QuoteFieldNames</value> <value>WriteNullStringAsEmpty</value> </list> </property> -->
  14. </bean>
  15. </mvc:message-converters>
  16. </mvc:annotation-driven>

###8. 再次访问
重启tomcat,通过浏览器再次访问,如果返回以下内容表示配置@ResponseBody正确。
注:如果Controller类上注解为@RestController则相当于方法上注解了@ResponseBody。
这里写图片描述

###9. 添加返回Model方法
新建Model,并添加对应的dao/service/controller方法返回此Model。
(1)TestModel.java

  1. package com.alan.model;
  2. public class TestModel {
  3. private Integer id;
  4. private String name;
  5. public Integer getId() {
  6. return id;
  7. }
  8. public void setId(Integer id) {
  9. this.id = id;
  10. }
  11. public String getName() {
  12. return name;
  13. }
  14. public void setName(String name) {
  15. this.name = name;
  16. }
  17. }

(2)TestDaoImpl.java

  1. @Override
  2. public TestModel testModel() {
  3. TestModel testModel = new TestModel();
  4. testModel.setId(1);
  5. testModel.setName("tt");
  6. return testModel;
  7. }

(3)TestServiceImpl.java

  1. @Override
  2. public TestModel testModel() {
  3. return testDao.testModel();
  4. }

(4)TestController.java

  1. @RequestMapping("/test-model")
  2. public TestModel testModel(){
  3. return testService.testModel();
  4. }

###10. 访问
重启tomcat,通过浏览器访问,如果返回以下内容则表示json配置正确。
这里写图片描述

(防采集:本文唯一网址在CSDN上,博主会不定期更新加载其它配置,其它均为未经博主同意转载的,原文为http://blog.csdn.net/tangtong1/article/details/51442757)

##三、配置单元测试

###11. 添加单元测试依赖
修改父工程pom.xml,添加单元测试依赖。

  1. <!-- 单元测试开始 -->
  2. <dependency>
  3. <groupId>org.springframework</groupId>
  4. <artifactId>spring-test</artifactId>
  5. </dependency>
  6. <dependency>
  7. <groupId>junit</groupId>
  8. <artifactId>junit</artifactId>
  9. <version>4.11</version>
  10. <scope>test</scope>
  11. </dependency>
  12. <!-- 单元测试结束 -->

###12. 添加测试类
新建一个测试类。

  1. package com.alan.test;
  2. import org.junit.Test;
  3. import org.junit.runner.RunWith;
  4. import org.springframework.beans.factory.annotation.Autowired;
  5. import org.springframework.test.context.ContextConfiguration;
  6. import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
  7. import com.alan.service.TestService;
  8. @RunWith(SpringJUnit4ClassRunner.class)
  9. @ContextConfiguration(locations="classpath*:META-INF/dispatcher.xml")
  10. public class UnitTest {
  11. @Autowired
  12. private TestService testService;
  13. @Test
  14. public void test(){
  15. String result = testService.test();
  16. System.out.println(result);
  17. }
  18. }

###13. 运行测试类

Run As —> JUnit Test,运行结果如下表示配置单元测试正确。
这里写图片描述

(防采集:本文唯一网址在CSDN上,博主会不定期更新加载其它配置,其它均为未经博主同意转载的,原文为http://blog.csdn.net/tangtong1/article/details/51442757)

##四、配置日志管理

此处介绍两种日志管理工具,一个是log4j,一个是logback。

(一)log4j

###14. 添加log4j相关依赖
修改父工程的pom.xml,添加以下内容。

  1. <!-- 日志配置开始 -->
  2. <dependency>
  3. <groupId>org.slf4j</groupId>
  4. <artifactId>jcl-over-slf4j</artifactId>
  5. <version>1.5.8</version>
  6. </dependency>
  7. <dependency>
  8. <groupId>org.slf4j</groupId>
  9. <artifactId>slf4j-api</artifactId>
  10. <version>1.5.8</version>
  11. </dependency>
  12. <dependency>
  13. <groupId>org.slf4j</groupId>
  14. <artifactId>slf4j-log4j12</artifactId>
  15. <version>1.5.8</version>
  16. </dependency>
  17. <dependency>
  18. <groupId>log4j</groupId>
  19. <artifactId>log4j</artifactId>
  20. <version>1.2.14</version>
  21. </dependency>
  22. <!-- 日志配置结束 -->

###15. 添加log4j.properties
在web工程的src/main/resources目录下添加log4j.properties文件。

  1. log4j.rootCategory=DEBUG, stdout
  2. log4j.appender.stdout=org.apache.log4j.ConsoleAppender
  3. log4j.appender.stdout.layout=org.apache.log4j.PatternLayout
  4. log4j.appender.stdout.layout.ConversionPattern=%d{ABSOLUTE} %5p %t %c{2}:%L - %m%n
  5. log4j.category.org.springframework.beans.factory=DEBUG

修改log4j.rootCategory中的DEBUG为INFO/ERROR等即可输出相应级别的日志。

###16. 测试log4j
修改单元测试类为如下。

  1. package com.alan.test;
  2. import org.junit.Test;
  3. import org.junit.runner.RunWith;
  4. import org.slf4j.Logger;
  5. import org.slf4j.LoggerFactory;
  6. import org.springframework.beans.factory.annotation.Autowired;
  7. import org.springframework.test.context.ContextConfiguration;
  8. import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
  9. import com.alan.controller.TestController;
  10. import com.alan.service.TestService;
  11. @RunWith(SpringJUnit4ClassRunner.class)
  12. @ContextConfiguration(locations="classpath*:META-INF/dispatcher.xml")
  13. public class UnitTest {
  14. private Logger logger = LoggerFactory.getLogger(UnitTest.class);
  15. @Autowired
  16. private TestService testService;
  17. @Test
  18. public void test(){
  19. logger.debug("---------------------测试开始----------------------------");
  20. String result = testService.test();
  21. System.out.println(result);
  22. logger.debug("---------------------测试结束----------------------------");
  23. }
  24. }

运行测试类,如果输出结果为以下内容则表示log4j配置正确。
这里写图片描述

(一)logback

###17. 添加logback依赖
修改父工程的pom.xml,删除之前log4j的依赖,添加logback相关的依赖。

  1. <!-- 日志配置开始 -->
  2. <dependency>
  3. <groupId>org.slf4j</groupId>
  4. <artifactId>jcl-over-slf4j</artifactId>
  5. <version>1.7.12</version>
  6. </dependency>
  7. <!-- logback配置开始 -->
  8. <dependency>
  9. <groupId>ch.qos.logback</groupId>
  10. <artifactId>logback-classic</artifactId>
  11. <version>1.1.3</version>
  12. </dependency>
  13. <dependency>
  14. <groupId>org.logback-extensions</groupId>
  15. <artifactId>logback-ext-spring</artifactId>
  16. <version>0.1.2</version>
  17. </dependency>
  18. <!-- logback配置结束 -->
  19. <!-- 日志配置结束 -->

###18. 添加logback.xml文件
在web工程的src/main/resources目录下添加logback.xml文件,官网说会先检测classpath下的logback.groovy文件,如果没有再寻找logback-test.xml文件,最后才会找logback.xml文件,不过我们一般就直接使用logback.xml文件啦。

  1. <?xml version="1.0" encoding="UTF-8"?>
  2. <configuration>
  3. <property name="log.base" value="E:/workspace/self/log/alan-springmvc" />
  4. <appender name="stdout" class="ch.qos.logback.core.ConsoleAppender">
  5. <encoder>
  6. <pattern>%d{yyyy-MM-dd HH:mm:ss.SSS} [%thread] %-5level %logger{36}:%L -%msg%n</pattern>
  7. </encoder>
  8. </appender>
  9. <appender name="logfile" class="ch.qos.logback.core.rolling.RollingFileAppender">
  10. <Encoding>UTF-8</Encoding>
  11. <File>${log.base}/alan-springmvc.log</File>
  12. <encoder>
  13. <pattern>%d{yyyy-MM-dd HH:mm:ss.SSS} [%thread] %-5level %logger{36}:%L -%msg%n</pattern>
  14. </encoder>
  15. <rollingPolicy class="ch.qos.logback.core.rolling.TimeBasedRollingPolicy">
  16. <fileNamePattern>${log.base}/alan-springmvc-%d{yyyyMMdd}-%i.log</fileNamePattern>
  17. <MaxHistory>100</MaxHistory>
  18. <TimeBasedFileNamingAndTriggeringPolicy class="ch.qos.logback.core.rolling.SizeAndTimeBasedFNATP">
  19. <MaxFileSize>100MB</MaxFileSize>
  20. </TimeBasedFileNamingAndTriggeringPolicy>
  21. </rollingPolicy>
  22. </appender>
  23. <root level="debug">
  24. <appender-ref ref="stdout" />
  25. <appender-ref ref="logfile" />
  26. </root>
  27. </configuration>

修改root元素的level属性为debug/info/error等即可输出相应级别的日志。

###19. 添加logback监听
修改web工程的web.xml,添加以下内容。

  1. <!-- logback监听配置 -->
  2. <context-param>
  3. <param-name>logbackConfigLocation</param-name>
  4. <param-value>classpath:logback.xml</param-value>
  5. </context-param>
  6. <listener>
  7. <listener-class>ch.qos.logback.ext.spring.web.LogbackConfigListener</listener-class>
  8. </listener>

###20. 测试logback
运行之前的单元测试类,如果输出以下内容则表示logback配置正确。
注:从log4j转到logback之前类中的写法完全不用改变,这是slf4j的威力。
这里写图片描述

打开日志文件(logback.xml中配置的第二个appender),如果看到以下内容表示logback配置正确。
注:…/index.html这行是system.out.println输出的,所以不在日志文件中。
这里写图片描述

(防采集:本文唯一网址在CSDN上,博主会不定期更新加载其它配置,其它均为未经博主同意转载的,原文为http://blog.csdn.net/tangtong1/article/details/51442757)

##五、集成mybatis

本文采用bonecp管理数据库连接,数据库采用mysql。

###21. 添加bonecp/mybatis/mysql/jdbc依赖

其中,mysql的依赖版本最好跟数据库保持一致,否则可能会出现连不上的情况。

  1. <!-- 连接池配置开始 -->
  2. <dependency>
  3. <groupId>com.jolbox</groupId>
  4. <artifactId>bonecp</artifactId>
  5. <version>0.8.0.RELEASE</version>
  6. </dependency>
  7. <!-- 连接池配置结束 -->
  8. <!-- mybatis配置开始 -->
  9. <dependency>
  10. <groupId>org.mybatis</groupId>
  11. <artifactId>mybatis</artifactId>
  12. <version>3.4.0</version>
  13. </dependency>
  14. <dependency>
  15. <groupId>org.mybatis</groupId>
  16. <artifactId>mybatis-spring</artifactId>
  17. <version>1.3.0</version>
  18. <exclusions>
  19. <exclusion>
  20. <groupId>org.springframework</groupId>
  21. <artifactId>spring-jdbc</artifactId>
  22. </exclusion>
  23. <exclusion>
  24. <groupId>org.springframework</groupId>
  25. <artifactId>spring-test</artifactId>
  26. </exclusion>
  27. </exclusions>
  28. </dependency>
  29. <!-- mybatis配置结束 -->
  30. <!--mysql配置开始 -->
  31. <dependency>
  32. <groupId>mysql</groupId>
  33. <artifactId>mysql-connector-java</artifactId>
  34. <version>5.1.34</version>
  35. </dependency>
  36. <!--mysql配置结束 -->
  37. <!-- jdbc配置结束 -->
  38. <dependency>
  39. <groupId>org.springframework</groupId>
  40. <artifactId>spring-jdbc</artifactId>
  41. </dependency>
  42. <!-- jdbc配置结束 -->

###22. 添加spring-mybatis.xml
在web工程的src/main/resources/META-INF目录下增加配置文件spring-mybatis.xml。

  1. <?xml version="1.0" encoding="UTF-8"?>
  2. <beans xmlns="http://www.springframework.org/schema/beans" xmlns:context="http://www.springframework.org/schema/context" xmlns:aop="http://www.springframework.org/schema/aop" xmlns:tx="http://www.springframework.org/schema/tx" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://www.springframework.org/schema/beans http://www.springframework.org/schema/beans/spring-beans-3.0.xsd http://www.springframework.org/schema/aop http://www.springframework.org/schema/aop/spring-aop-3.0.xsd http://www.springframework.org/schema/tx http://www.springframework.org/schema/tx/spring-tx-3.0.xsd http://www.springframework.org/schema/context http://www.springframework.org/schema/context/spring-context-3.0.xsd">
  3. <!-- 配置DataSource数据源 -->
  4. <bean id="dataSource" class="com.jolbox.bonecp.BoneCPDataSource" destroy-method="close">
  5. <!-- 数据库驱动 -->
  6. <property name="driverClass" value="${jdbc_driverClassName}" />
  7. <!-- 相应驱动的jdbcUrl -->
  8. <property name="jdbcUrl" value="${jdbc_url}" />
  9. <!-- 数据库的用户名 -->
  10. <property name="username" value="${jdbc_username}" />
  11. <!-- 数据库的密码 -->
  12. <property name="password" value="${jdbc_password}" />
  13. <!-- /** Maximum age of an unused connection before it is closed off. */ -->
  14. <!-- 设置分区个数。这个参数默认为1,建议3-4(根据特定应用程序而定)。 为了减少锁竞争和改善性能,从当前线程分区(thread-affinity)中获取一个connection, 也就是这个样子:partitions[Thread.currentThread().getId() % partitionCount]。 当拥有充足的短期(short-lived)的线程时候,这个参数设置越大,性能越好。 当超过一定的阀值时,连接池的维护工作就可能对性能造成一定的负面影响(仅当分区上的connection使用耗尽时) -->
  15. <property name="partitionCount" value="${db.partitionCount}" />
  16. <!-- 设置每个分区含有connection最大个数。这个参数默认为2。如果小于2,BoneCP将设置为50。 比如:partitionCount设置为3,maxConnectionPerPartition设置为5,你就会拥有总共15个connection。 注意:BoneCP不会将这些connection一起创建出来,而是说在需要更多connection的时候从minConnectionsPerPartition参数开始逐步地增长connection数量。 -->
  17. <property name="maxConnectionsPerPartition" value="${db.maxConnectionsPerPartition}" />
  18. <!-- 设置每个分区含有connection最大小个数。这个参数默认为0。 -->
  19. <property name="minConnectionsPerPartition" value="${db.minConnectionsPerPartition}" />
  20. <!-- 设置分区中的connection增长数量。这个参数默认为1。 当每个分区中的connection大约快用完时,BoneCP动态批量创建connection, 这个属性控制一起创建多少个connection(不会大于maxConnectionsPerPartition)。 注意:这个配置属于每个分区的设置。 -->
  21. <property name="acquireIncrement" value="${db.acquireIncrement}" />
  22. <!-- 设置连接池阀值。这个参数默认为20。如果小于0或是大于100,BoneCP将设置为20。 连接池观察线程(PoolWatchThread)试图为每个分区维护一定数量的可用connection。 这个数量趋于maxConnectionPerPartition和minConnectionPerPartition之间。这个参数是以百分比的形式来计算的。 例如:设置为20,下面的条件如果成立:Free Connections / MaxConnections < poolAvailabilityThreshold;就会创建出新的connection。 换句话来说连接池为每个分区至少维持20%数量的可用connection。 设置为0时,每当需要connection的时候,连接池就要重新创建新connection,这个时候可能导致应用程序可能会为了获得新connection而小等一会。 -->
  23. <property name="poolAvailabilityThreshold" value="${db.poolAvailabilityThreshold}" />
  24. <!-- 设置获取connection超时的时间。这个参数默认为Long.MAX_VALUE;单位:毫秒。 在调用getConnection获取connection时,获取时间超过了这个参数,就视为超时并报异常。 -->
  25. <property name="connectionTimeoutInMs" value="${db.connectionTimeoutInMs}" />
  26. <!-- /** A connection older than maxConnectionAge will be destroyed and purged from the pool. */ -->
  27. <!-- 设置connection的存活时间。这个参数默认为0,单位:毫秒。设置为0该功能失效。 通过ConnectionMaxAgeThread观察每个分区中的connection,不管connection是否空闲, 如果这个connection距离创建的时间大于这个参数就会被清除。当前正在使用的connection不受影响,直到返回到连接池再做处理。 -->
  28. <!-- 48小时关闭一个链接 -->
  29. <property name="maxConnectionAgeInSeconds" value="${db.maxConnectionAgeInSeconds}" />
  30. <!-- /** SQL statement to use for keep-alive/test of connection. */ -->
  31. <property name="connectionTestStatement" value="${db.connectionTestStatement}" />
  32. <!-- 设置connection的空闲存活时间。这个参数默认为60,单位:分钟。设置为0该功能失效。 通过ConnectionTesterThread观察每个分区中的connection,如果这个connection距离最后使用的时间大于这个参数就会被清除。 注意:这个参数仅和idleConnectionTestPeriodInSeconds搭配使用,而且不要在这里设置任何挑衅的参数! -->
  33. <!-- 1小时回收空闲链接 -->
  34. <property name="idleMaxAgeInMinutes" value="${db.idleMaxAgeInMinutes}" />
  35. <!-- /** Connections older than this are sent a keep-alive statement. */ -->
  36. <!-- 设置测试connection的间隔时间。这个参数默认为240*60,单位:分钟。设置为0该功能失效。 通过ConnectionTesterThread观察每个分区中的connection, 如果这个connection距离最后使用的时间大于这个参数并且距离上一次测试的时间大于这个参数就会向数据库发送一条测试语句,如果执行失败则将这个connection清除。 注意:这个值仅和idleMaxAge搭配使用,而且不要在这里设置任何挑衅的参数! -->
  37. <!-- 4小时检测一次空闲链接 -->
  38. <property name="idleConnectionTestPeriodInMinutes" value="${db.idleConnectionTestPeriodInMinutes}" />
  39. <!-- /** After attempting to acquire a connection and failing, try to connect these many times before giving up. */ -->
  40. <!-- 设置重新获取连接的次数。这个参数默认为5。 获取某个connection失败之后会多次尝试重新连接,如果在这几次还是失败则放弃。 -->
  41. <property name="acquireRetryAttempts" value="${db.acquireRetryAttempts}" />
  42. <!-- 设置重新获取连接的次数间隔时间。这个参数默认为7000,单位:毫秒。如果小于等于0,BoneCP将设置为1000。 获取connection失败之后再次尝试获取connection的间隔时间。 -->
  43. <property name="acquireRetryDelayInMs" value="${db.acquireRetryDelayInMs}" />
  44. <!-- 设置连接池初始化功能。这个参数默认为false。 设置为true,连接池将会初始化为空,直到获取第一个connection。 -->
  45. <property name="lazyInit" value="${db.lazyInit}" />
  46. <!-- 设置是否关闭JMX功能。这个参数默认为false。 -->
  47. <property name="disableJMX" value="${db.disableJMX}" />
  48. <!-- 设置连接池名字。用于当作JMX和助手线程名字的后缀。 -->
  49. <property name="poolName" value="${db.poolName}" />
  50. <!-- /** Min no of prepared statements to cache. */ -->
  51. <!-- 设置statement缓存个数。这个参数默认为0。 -->
  52. <property name="statementsCacheSize" value="${db.statementsCacheSize}" />
  53. <!-- 设置是否开启记录SQL语句功能。这个参数默认是false。 将执行的SQL记录到日志里面(包括参数值)。 -->
  54. <property name="logStatementsEnabled" value="${db.logStatementsEnabled}" />
  55. <!-- 设置执行SQL的超时时间。这个参数默认为0;单位:毫秒。 当查询语句执行的时间超过这个参数,执行的情况就会被记录到日志中。 设置为0时,该功能失效。 -->
  56. <property name="queryExecuteTimeLimitInMs" value="${db.queryExecuteTimeLimit}" />
  57. </bean>
  58. <!-- 对数据源进行事务管理 -->
  59. <bean id="transactionManager" class="org.springframework.jdbc.datasource.DataSourceTransactionManager">
  60. <property name="dataSource" ref="dataSource" />
  61. </bean>
  62. <tx:annotation-driven transaction-manager="transactionManager" />
  63. <!-- 配置SqlSessionFactoryBean -->
  64. <bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
  65. <property name="dataSource" ref="dataSource" />
  66. <!-- mapper和resultmap配置路径 -->
  67. <property name="mapperLocations">
  68. <list>
  69. <value>classpath:mapper/*Mapper.xml</value>
  70. </list>
  71. </property>
  72. </bean>
  73. <!-- 通过扫描的模式 -->
  74. <bean class="org.mybatis.spring.mapper.MapperScannerConfigurer">
  75. <property name="basePackage" value="com.alan.mapper" />
  76. </bean>
  77. </beans>

###23. 修改dispatcher.xml
(1)增加对属性文件读取的支持;

  1. <!-- 本工程属性配置文件 -->
  2. <bean id="configProperties" class="org.springframework.beans.factory.config.PropertiesFactoryBean">
  3. <property name="locations">
  4. <list>
  5. <value>file:/etc/*.properties</value>
  6. <value>file:/E:/workspace/self/etc/*.properties</value>
  7. </list>
  8. </property>
  9. <property name="fileEncoding" value="utf-8" />
  10. </bean>
  11. <bean id="propertyConfigurer" class="org.springframework.beans.factory.config.PreferencesPlaceholderConfigurer">
  12. <property name="properties" ref="configProperties" />
  13. </bean>

(2)引入spring-mybatis.xml文件

  1. <import resource="classpath:META-INF/spring-mybatis.xml" />

###24. 添加jdbc.properties
在E:/workspace/self/etc目录下增加数据库连接配置文件jdbc.properties。

  1. jdbc_driverClassName=com.mysql.jdbc.Driver
  2. jdbc_url=jdbc:mysql://127.0.0.1:3306/fcuh_user?useUnicode=true&characterEncoding=UTF-8&characterSetResults=UTF-8&useOldAliasMetadataBehavior=true&noAccessToProcedureBodies=true
  3. jdbc_username=admin
  4. jdbc_password=1234
  5. #Number of partitions.
  6. db.partitionCount=3
  7. #Min number of connections per partition.
  8. db.minConnectionsPerPartition=2
  9. #Max number of connections per partition.
  10. db.maxConnectionsPerPartition=10
  11. #Number of new connections to create in 1 batch.
  12. db.acquireIncrement=2
  13. #Time to wait before a call to getConnection() times out and returns an error.
  14. db.connectionTimeoutInMs=30000
  15. #Number of release-connection helper threads to create per partition.
  16. #A connection older than maxConnectionAge will be destroyed and purged from the pool.
  17. db.maxConnectionAgeInSeconds=172800
  18. #SQL statement to use for keep-alive/test of connection.
  19. db.connectionTestStatement=select 1 from dual
  20. #Maximum age of an unused connection before it is closed off.
  21. db.idleMaxAgeInMinutes=60
  22. #Connections older than this are sent a keep-alive statement.
  23. db.idleConnectionTestPeriodInMinutes=60
  24. #After attempting to acquire a connection and failing, try to connect these many times before giving up.
  25. db.acquireRetryAttempts=5
  26. #After attempting to acquire a connection and failing, wait for this value before attempting to acquire a new connection again.
  27. db.acquireRetryDelayInMs=5000
  28. #If set to true, the connection pool will remain empty until the first connection is obtained.
  29. db.lazyInit=false
  30. #Set to true to disable JMX.
  31. db.disableJMX=false
  32. #Name of the pool for JMX and thread names.
  33. db.poolName=logistics
  34. #Min no of prepared statements to cache.
  35. db.statementsCacheSize=50
  36. #If set to true, log SQL statements being executed.
  37. db.logStatementsEnabled=false
  38. #Queries taking longer than this limit to execute are logged.
  39. db.queryExecuteTimeLimit=0
  40. #Create more connections when we hit x% of our possible number of connections.
  41. db.poolAvailabilityThreshold=60

###25. 修改dao/service/controller/mapper类
修改dao/service/controller类,在dao工程添加新的包com.alan.mapper,并添加新类TestMapper.java。
注:主要测试select/update/事务操作。
(1)TestMapper.java

  1. package com.alan.mapper;
  2. import java.util.List;
  3. public interface TestMapper {
  4. List<Object> testDB();
  5. int testUpdate();
  6. int testInsert();
  7. }

(2)TestDaoImpl.java

  1. package com.alan.dao.impl;
  2. import java.util.List;
  3. import org.springframework.beans.factory.annotation.Autowired;
  4. import org.springframework.stereotype.Repository;
  5. import com.alan.dao.TestDao;
  6. import com.alan.mapper.TestMapper;
  7. import com.alan.model.TestModel;
  8. @Repository("testDao")
  9. public class TestDaoImpl implements TestDao {
  10. @Autowired
  11. private TestMapper testMapper;
  12. @Override
  13. public String test() {
  14. return "../index.html";
  15. }
  16. @Override
  17. public TestModel testModel() {
  18. TestModel testModel = new TestModel();
  19. testModel.setId(1);
  20. testModel.setName("tt");
  21. return testModel;
  22. }
  23. @Override
  24. public List<Object> testDB() {
  25. return testMapper.testDB();
  26. }
  27. @Override
  28. public int testUpdate(){
  29. return testMapper.testUpdate();
  30. }
  31. @Override
  32. public int testInsert(){
  33. return testMapper.testInsert();
  34. }
  35. }

(3)TestServiceImpl.java
注:testTx()方法会抛出异常以便测试事务。

  1. package com.alan.service.impl;
  2. import java.util.List;
  3. import org.springframework.beans.factory.annotation.Autowired;
  4. import org.springframework.stereotype.Service;
  5. import org.springframework.transaction.annotation.Propagation;
  6. import org.springframework.transaction.annotation.Transactional;
  7. import com.alan.dao.TestDao;
  8. import com.alan.model.TestModel;
  9. import com.alan.service.TestService;
  10. @Service("testService")
  11. public class TestServiceImpl implements TestService {
  12. @Autowired
  13. private TestDao testDao;
  14. @Override
  15. public String test() {
  16. return testDao.test();
  17. }
  18. @Override
  19. public TestModel testModel() {
  20. return testDao.testModel();
  21. }
  22. @Override
  23. public List<Object> testDB() {
  24. return testDao.testDB();
  25. }
  26. @Override
  27. public int testUpdate(){
  28. return testDao.testUpdate();
  29. }
  30. @Override
  31. public int testInsert(){
  32. return testDao.testInsert();
  33. }
  34. @Override
  35. @Transactional(propagation=Propagation.REQUIRED,rollbackFor=Exception.class)
  36. public void testTx(Integer nullInt){
  37. testUpdate();
  38. nullInt.intValue();
  39. testInsert();
  40. }
  41. }

(4)TestController.java

  1. package com.alan.controller;
  2. import java.util.List;
  3. import org.slf4j.Logger;
  4. import org.slf4j.LoggerFactory;
  5. import org.springframework.beans.factory.annotation.Autowired;
  6. import org.springframework.web.bind.annotation.RequestMapping;
  7. import org.springframework.web.bind.annotation.RestController;
  8. import com.alan.model.TestModel;
  9. import com.alan.service.TestService;
  10. @RestController
  11. @RequestMapping("/alan")
  12. public class TestController {
  13. private Logger logger = LoggerFactory.getLogger(TestController.class);
  14. @Autowired
  15. private TestService testService;
  16. @RequestMapping("/test")
  17. public String test(){
  18. logger.debug("enter into test");
  19. return testService.test();
  20. }
  21. @RequestMapping("/test-model")
  22. public TestModel testModel(){
  23. return testService.testModel();
  24. }
  25. @RequestMapping("/test-db")
  26. public List<Object> testDB(){
  27. return testService.testDB();
  28. }
  29. @RequestMapping("/test-update")
  30. public int testUpdate(){
  31. return testService.testUpdate();
  32. }
  33. @RequestMapping("/test-tx")
  34. public String testTx(){
  35. testService.testTx(null);
  36. return "success";
  37. }
  38. }

###26. 添加TestMapper.xml
在dao工程的src/main/resources/mapper目录下,添加TestMapper.xml。

  1. <?xml version="1.0" encoding="UTF-8" ?>
  2. <!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd" >
  3. <mapper namespace="com.alan.mapper.TestMapper" >
  4. <select id="testDB" resultType="map">
  5. select user_id, mobile from basic_user_info limit 10
  6. </select>
  7. <update id="testUpdate">
  8. update basic_user_info
  9. set mobile = '12345678911'
  10. where user_id = 1
  11. </update>
  12. <insert id="testInsert">
  13. insert into basic_user_info(user_id,mobile,status)
  14. values(3,'123x456x789',1)
  15. </insert>
  16. </mapper>

###27. 测试
前提:数据库建好了表,并插入了相关数据,至少包含3列:user_id,mobile,status。
(1)测试select
这里写图片描述
(2)测试update
这里写图片描述
这里写图片描述
(3)测试事务
浏览器报错,后台日志回滚,数据库未插入id为3的数据,说明事务测试成功。
这里写图片描述
这里写图片描述

至此,表示mybatis配置正确,并支持事务。

(防采集:本文唯一网址在CSDN上,博主会不定期更新加载其它配置,其它均为未经博主同意转载的,原文为http://blog.csdn.net/tangtong1/article/details/51442757)

##六、编写通用类
我们知道项目中基本每个表都会有增删改查这些方法,要是针对每个表都要写这么一大堆重复的方法着实让人心烦,这章我们来写通用的类以提高开发的效率。

###28. 新建common工程

为方便我们直接建在父工程下了,按实际项目经验,common工程应该是一个独立的工程。
切换到父工程,执行以下命令。

  1. mvn archetype:generate -DarchetypeCatalog=internal -DgroupId=com.alan.springmvc.common -DartifactId=alan-springmvc-common -DarchetypeArtifactId=maven-archetype-quickstart -Dversion=1.0

同样,删除pom.xml文件中的groupId和versiong两行,删除自动生成的Junit依赖。

###29. 新建各通用类
以下类都放在common工程相应的包下。
(1)BaseMapper.java

  1. package com.alan.parent.mapper;
  2. import java.util.List;
  3. public interface BaseMapper<T> {
  4. T queryById(int id);
  5. List<T> queryAll();
  6. int add(T t);
  7. int update(T t);
  8. int delete(int id);
  9. }

(2)BaseSqlDao.java
注意:此接口是包访问权限。

  1. package com.alan.parent.dao;
  2. import java.util.List;
  3. import java.util.Map;
  4. interface BaseSqlDao {
  5. <V> V queryVoByParam(Map<String, Object> param);
  6. <V> List<V> queryListByParam(Map<String, Object> param);
  7. void multipleDML(Map<String, Object> param);
  8. }

(3)BaseSqlDaoImpl.java
注意:此类直接调用mybatis的API操作*Mapper.xml文件,跳过了*Mapper.java,具体使用方法后续介绍。其中,multipleDML方法暂不实现,有兴趣的可以参考另一篇博客spring+mybatis一个方法执行多条更新语句,实现批量DML。

  1. package com.alan.parent.dao;
  2. import java.util.List;
  3. import java.util.Map;
  4. import javax.annotation.Resource;
  5. import org.apache.ibatis.session.SqlSession;
  6. import org.apache.ibatis.session.SqlSessionFactory;
  7. class BaseSqlDaoImpl implements BaseSqlDao {
  8. @Resource
  9. private SqlSessionFactory sqlSessionFactory;
  10. private String baseMapperPath;
  11. public String getBaseMapperPath() {
  12. return baseMapperPath;
  13. }
  14. public void setBaseMapperPath(String baseMapperPath) {
  15. this.baseMapperPath = baseMapperPath;
  16. }
  17. @Override
  18. public <V> V queryVoByParam(Map<String, Object> param) {
  19. SqlSession session = sqlSessionFactory.openSession();
  20. try {
  21. return session.selectOne(baseMapperPath + "." + param.get("mapperName") + "." + param.get("selectId"),
  22. param);
  23. } finally {
  24. session.close();
  25. }
  26. }
  27. @Override
  28. public <V> List<V> queryListByParam(Map<String, Object> param) {
  29. SqlSession session = sqlSessionFactory.openSession();
  30. try {
  31. return session.selectList(baseMapperPath + "." + param.get("mapperName") + "." + param.get("selectId"),
  32. param);
  33. } finally {
  34. session.close();
  35. }
  36. }
  37. @Override
  38. public void multipleDML(Map<String, Object> param) {
  39. }
  40. }

(4)BaseDao.java

  1. package com.alan.parent.dao;
  2. import java.util.List;
  3. import java.util.Map;
  4. public interface BaseDao<T> {
  5. T queryById(int id);
  6. List<T> queryAll();
  7. int add(T t);
  8. int update(T t);
  9. int delete(int id);
  10. <V> V queryVoByParam(Map<String, Object> param);
  11. <V> List<V> queryListByParam(Map<String, Object> param);
  12. void multipleDML(Map<String, Object> param);
  13. }

(5)BaseDaoImpl.java
注意:此类中注入了BaseSqlDao实例。

  1. package com.alan.parent.dao;
  2. import java.util.List;
  3. import java.util.Map;
  4. import javax.annotation.PostConstruct;
  5. import javax.annotation.Resource;
  6. import com.alan.parent.mapper.BaseMapper;
  7. public abstract class BaseDaoImpl<T> implements BaseDao<T> {
  8. @Resource
  9. private BaseSqlDao baseSqlDao;
  10. private BaseMapper<T> mapper;
  11. @PostConstruct
  12. public void initMapper() {
  13. this.mapper = init();
  14. }
  15. public abstract BaseMapper<T> init();
  16. @Override
  17. public T queryById(int id) {
  18. return mapper.queryById(id);
  19. }
  20. @Override
  21. public List<T> queryAll() {
  22. return mapper.queryAll();
  23. }
  24. @Override
  25. public int add(T t) {
  26. return mapper.add(t);
  27. }
  28. @Override
  29. public int update(T t) {
  30. return mapper.update(t);
  31. }
  32. @Override
  33. public int delete(int id) {
  34. return mapper.delete(id);
  35. }
  36. @Override
  37. public <V> V queryVoByParam(Map<String, Object> param){
  38. return baseSqlDao.queryVoByParam(param);
  39. }
  40. @Override
  41. public <V> List<V> queryListByParam(Map<String, Object> param){
  42. return baseSqlDao.queryListByParam(param);
  43. }
  44. @Override
  45. public void multipleDML(Map<String, Object> param){
  46. baseSqlDao.multipleDML(param);
  47. }
  48. }

(6)BaseService.java

  1. package com.alan.parent.service;
  2. import java.util.List;
  3. import java.util.Map;
  4. public interface BaseService<T> {
  5. T queryById(int id);
  6. List<T> queryAll();
  7. int add(T t);
  8. int update(T t);
  9. int delete(int id);
  10. <V> V queryVoByParam(Map<String, Object> param);
  11. <V> List<V> queryListByParam(Map<String, Object> param);
  12. void multipleDML(Map<String, Object> param);
  13. }

(7)BaseServiceImpl.java

  1. package com.alan.parent.service;
  2. import java.util.List;
  3. import java.util.Map;
  4. import javax.annotation.PostConstruct;
  5. import com.alan.parent.dao.BaseDao;
  6. public abstract class BaseServiceImpl<T> implements BaseService<T> {
  7. private BaseDao<T> dao;
  8. @PostConstruct
  9. public void initDao(){
  10. this.dao = init();
  11. }
  12. public abstract BaseDao<T> init();
  13. @Override
  14. public T queryById(int id) {
  15. return dao.queryById(id);
  16. }
  17. @Override
  18. public List<T> queryAll() {
  19. return dao.queryAll();
  20. }
  21. @Override
  22. public int add(T t) {
  23. return dao.add(t);
  24. }
  25. @Override
  26. public int update(T t) {
  27. return dao.update(t);
  28. }
  29. @Override
  30. public int delete(int id) {
  31. return dao.delete(id);
  32. }
  33. @Override
  34. public <V> V queryVoByParam(Map<String, Object> param) {
  35. return dao.queryVoByParam(param);
  36. }
  37. @Override
  38. public <V> List<V> queryListByParam(Map<String, Object> param) {
  39. return dao.queryListByParam(param);
  40. }
  41. @Override
  42. public void multipleDML(Map<String, Object> param) {
  43. dao.multipleDML(param);
  44. }
  45. }

(8)BaseController.java
说明:

  • 前五个方法需要在对应的*Mapper.xml中定义相应的语句,比如queryById,需要定义一条select其id为queryById;
  • 之后的两个方法的注解会从url中获取selectId,这个即*Mapper.xml中select的id;
  • 后面三条不带@RequestMapping的用于自定义查询。

以上方法的使用会在后续介绍。

  1. package com.alan.parent.controller;
  2. import java.io.UnsupportedEncodingException;
  3. import java.lang.reflect.ParameterizedType;
  4. import java.lang.reflect.Type;
  5. import java.net.URLDecoder;
  6. import java.util.HashMap;
  7. import java.util.List;
  8. import java.util.Map;
  9. import java.util.Map.Entry;
  10. import javax.annotation.PostConstruct;
  11. import javax.servlet.http.HttpServletRequest;
  12. import org.slf4j.Logger;
  13. import org.slf4j.LoggerFactory;
  14. import org.springframework.web.bind.annotation.PathVariable;
  15. import org.springframework.web.bind.annotation.RequestMapping;
  16. import com.alan.parent.service.BaseService;
  17. public abstract class BaseController<T> {
  18. private Logger logger = LoggerFactory.getLogger(BaseController.class);
  19. private BaseService<T> service;
  20. @PostConstruct
  21. public void initService() {
  22. this.service = init();
  23. }
  24. public abstract BaseService<T> init();
  25. @RequestMapping("/queryById")
  26. public T queryById(int id) {
  27. return service.queryById(id);
  28. }
  29. @RequestMapping("/queryAll")
  30. public List<T> queryAll() {
  31. return service.queryAll();
  32. }
  33. @RequestMapping("/add")
  34. public int add(T t) {
  35. return service.add(t);
  36. }
  37. @RequestMapping("/update")
  38. public int update(T t) {
  39. return service.update(t);
  40. }
  41. @RequestMapping("/delete")
  42. public int delete(int id) {
  43. return service.delete(id);
  44. }
  45. /** * 从url上获取selectId并查询单条数据,其中selectId与*Mapper.xml中select的id对应 * @param request * @param selectId * @return */
  46. @RequestMapping("/queryVo/{selectId}")
  47. public Object queryVo(HttpServletRequest request, @PathVariable String selectId) {
  48. Map<String, Object> param = buildParam(request);
  49. param.put("selectId", selectId);
  50. param.put("mapperName", getMapperName());
  51. return queryVoByParam(param);
  52. }
  53. /** * 从url上获取selectId并查询多条数据,其中selectId与*Mapper.xml中select的id对应 * @param request * @param selectId * @return */
  54. @RequestMapping("/queryList/{selectId}")
  55. public List<Object> queryList(HttpServletRequest request, @PathVariable String selectId) {
  56. Map<String, Object> param = buildParam(request);
  57. param.put("selectId", selectId);
  58. param.put("mapperName", getMapperName());
  59. return queryListByParam(param);
  60. }
  61. public <V> V queryVoByParam(Map<String, Object> param) {
  62. return service.queryVoByParam(param);
  63. }
  64. public <V> List<V> queryListByParam(Map<String, Object> param) {
  65. return service.queryListByParam(param);
  66. }
  67. public void multipleDML(Map<String, Object> param) {
  68. service.multipleDML(param);
  69. }
  70. /** * 从request中构建出参数 * @param request * @return */
  71. private Map<String, Object> buildParam(HttpServletRequest request) {
  72. Map<String, Object> param = new HashMap<>();
  73. Map<String, String[]> paramMap = request.getParameterMap();
  74. for (Entry<String, String[]> e : paramMap.entrySet()) {
  75. if (e.getValue() == null || e.getValue().toString() == null || e.getValue().toString().length() == 0) {
  76. continue;
  77. }
  78. try {
  79. param.put(e.getKey(), URLDecoder.decode(e.getValue()[0], "utf-8"));
  80. } catch (UnsupportedEncodingException e1) {
  81. logger.error("decode error, the value is : " + e.getValue()[0]);
  82. }
  83. }
  84. return param;
  85. }
  86. /** * 根据泛型类型获取Mapper的名字 * @return */
  87. private String getMapperName() {
  88. Type genType = this.getClass().getGenericSuperclass();
  89. String paramType = ((ParameterizedType) genType).getActualTypeArguments()[0].getTypeName();
  90. String mapperName = paramType.substring(paramType.lastIndexOf('.') + 1) + "Mapper";
  91. return mapperName;
  92. }
  93. }

###30. 新建spring-common.xml配置文件

在common工程的src/main/resource/META-INFO目录下新建spring-common.xml文件,用于配置baseSqlDao。

  1. <?xml version="1.0" encoding="UTF-8"?>
  2. <beans xmlns="http://www.springframework.org/schema/beans" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xmlns:p="http://www.springframework.org/schema/p" xmlns:context="http://www.springframework.org/schema/context" xmlns:mvc="http://www.springframework.org/schema/mvc" xsi:schemaLocation="http://www.springframework.org/schema/mvc http://www.springframework.org/schema/mvc/spring-mvc-3.1.xsd http://www.springframework.org/schema/beans http://www.springframework.org/schema/beans/spring-beans.xsd http://www.springframework.org/schema/context http://www.springframework.org/schema/context/spring-context.xsd">
  3. <bean id="baseSqlDao" class="com.alan.parent.dao.BaseSqlDaoImpl">
  4. <property name="baseMapperPath" value="com.alan.mapper"></property>
  5. </bean>
  6. </beans>

###31. 引入spring-common.xml配置文件
在web工程的dispatcher.xml文件中引入上述配置文件。

  1. <import resource="classpath:META-INF/spring-common.xml" />

###32. 新建表
数据库中新建user_info及user_address表,并随意插入几条数据,用于测试上述通用类。

  1. create table user_info
  2. (
  3. id int,
  4. name varchar(50),
  5. mobile varchar(11)
  6. );
  7. create table user_address
  8. (
  9. id int primary key auto_increment,
  10. user_id int,
  11. address varchar(100)
  12. )

###33. 新建user_info表对应的各类

(1)实体类

  • UserInfo.java
    在dao工程下新建包com.alan.entity。

    package com.alan.entity;

    public class UserInfo {

    1. private Integer id;
    2. private String name;
    3. private String mobile;
    4. public Integer getId() {
    5. return id;
    6. }
    7. public void setId(Integer id) {
    8. this.id = id;
    9. }
    10. public String getName() {
    11. return name;
    12. }
    13. public void setName(String name) {
    14. this.name = name;
    15. }
    16. public String getMobile() {
    17. return mobile;
    18. }
    19. public void setMobile(String mobile) {
    20. this.mobile = mobile;
    21. }

    }

  • UserAddress.java

    package com.alan.entity;

    public class UserAddress {

    1. private Integer id;
    2. private Integer userId;
    3. private String address;
    4. public Integer getId() {
    5. return id;
    6. }
    7. public void setId(Integer id) {
    8. this.id = id;
    9. }
    10. public Integer getUserId() {
    11. return userId;
    12. }
    13. public void setUserId(Integer userId) {
    14. this.userId = userId;
    15. }
    16. public String getAddress() {
    17. return address;
    18. }
    19. public void setAddress(String address) {
    20. this.address = address;
    21. }

    }

  • UserAddressModel.java
    在dao工程下新建包com.alan.model,并新建UserAddressModel,其中包含address的列表。

    package com.alan.model;

    import java.util.List;

    import com.alan.entity.UserAddress;

    public class UserAddressModel {

    1. private Integer id;
    2. private String name;
    3. private List<UserAddress> addresses;
    4. public Integer getId() {
    5. return id;
    6. }
    7. public void setId(Integer id) {
    8. this.id = id;
    9. }
    10. public String getName() {
    11. return name;
    12. }
    13. public void setName(String name) {
    14. this.name = name;
    15. }
    16. public List<UserAddress> getAddresses() {
    17. return addresses;
    18. }
    19. public void setAddresses(List<UserAddress> addresses) {
    20. this.addresses = addresses;
    21. }

    }

(2)UserInfoMapper.java
直接继承BaseMapper接口。

  1. package com.alan.mapper;
  2. import com.alan.entity.UserInfo;
  3. import com.alan.parent.mapper.BaseMapper;
  4. public interface UserInfoMapper extends BaseMapper<UserInfo>{
  5. }

(3)UserInfoDao.java
直接继承BaseDao接口。

  1. package com.alan.dao;
  2. import com.alan.entity.UserInfo;
  3. import com.alan.parent.dao.BaseDao;
  4. public interface UserInfoDao extends BaseDao<UserInfo>{
  5. }

(4)UserInfoDaoImpl.java
继承BaseDaoImpl并实现UserInfoDao接口,注入userInfoMapper,并覆写init方法,传入userInfoMapper,这样就可以直接使用BaseDaoImpl中的方法了。

  1. package com.alan.dao.impl;
  2. import javax.annotation.Resource;
  3. import org.springframework.stereotype.Repository;
  4. import com.alan.dao.UserInfoDao;
  5. import com.alan.entity.UserInfo;
  6. import com.alan.mapper.UserInfoMapper;
  7. import com.alan.parent.dao.BaseDaoImpl;
  8. import com.alan.parent.mapper.BaseMapper;
  9. @Repository("userInfoDao")
  10. public class UserInfoDaoImpl extends BaseDaoImpl<UserInfo> implements UserInfoDao{
  11. @Resource
  12. private UserInfoMapper userInfoMapper;
  13. @Override
  14. public BaseMapper<UserInfo> init() {
  15. return userInfoMapper;
  16. }
  17. }

(5)UserInfoService.java

  1. package com.alan.service;
  2. import com.alan.entity.UserInfo;
  3. import com.alan.parent.service.BaseService;
  4. public interface UserInfoService extends BaseService<UserInfo> {
  5. }

(6)UserInfoServiceImpl.java

  1. package com.alan.service.impl;
  2. import javax.annotation.Resource;
  3. import org.springframework.stereotype.Service;
  4. import com.alan.dao.UserInfoDao;
  5. import com.alan.entity.UserInfo;
  6. import com.alan.parent.dao.BaseDao;
  7. import com.alan.parent.service.BaseServiceImpl;
  8. import com.alan.service.UserInfoService;
  9. @Service("userInfoService")
  10. public class UserInfoServiceImpl extends BaseServiceImpl<UserInfo> implements UserInfoService {
  11. @Resource
  12. private UserInfoDao userInfoDao;
  13. @Override
  14. public BaseDao<UserInfo> init() {
  15. return userInfoDao;
  16. }
  17. }

(7)UserInfoController.java
继承自BaseController,并初始化userInfoService,并新增加了三个方法queryUserInfoByMobile、queryUserInfoByName和queryRandomUser(假设手机号不重复,名字可重复),分别调用父类的queryVoByParam和queryListByParam方法,使用这两个方法,直接service/dao/mapper里的方法都不用写了,直接操作mapper.xml文件,非常之方便,当然,如果有点逻辑还是放在service里比较清晰一点。

  1. package com.alan.controller;
  2. import java.util.HashMap;
  3. import java.util.List;
  4. import java.util.Map;
  5. import javax.annotation.Resource;
  6. import org.springframework.web.bind.annotation.RequestMapping;
  7. import org.springframework.web.bind.annotation.RestController;
  8. import com.alan.entity.UserInfo;
  9. import com.alan.parent.controller.BaseController;
  10. import com.alan.parent.service.BaseService;
  11. import com.alan.service.UserInfoService;
  12. @RestController
  13. @RequestMapping("/userInfo")
  14. public class UserInfoController extends BaseController<UserInfo> {
  15. @Resource
  16. private UserInfoService userInfoService;
  17. @Override
  18. public BaseService<UserInfo> init() {
  19. return userInfoService;
  20. }
  21. @RequestMapping("/queryUserInfoByMobile")
  22. public UserInfo queryUserInfoByMobile(String mobile){
  23. Map<String, Object> param = new HashMap<>();
  24. param.put("mobile", mobile);
  25. param.put("mapperName","UserInfoMapper");
  26. param.put("selectId", "queryUserInfoByMobile");
  27. return queryVoByParam(param);
  28. }
  29. @RequestMapping("/queryUserInfoByName")
  30. public List<UserInfo> queryUserInfoByName(String name){
  31. Map<String, Object> param = new HashMap<>();
  32. param.put("name", name);
  33. param.put("mapperName","UserInfoMapper");
  34. param.put("selectId", "queryUserInfoByName");
  35. return queryListByParam(param);
  36. }
  37. @RequestMapping("/queryRandomUser")
  38. public Map<String, Object> queryRandomUser(){
  39. Map<String, Object> param = new HashMap<>();
  40. param.put("mapperName","UserInfoMapper");
  41. param.put("selectId", "queryRandomUser");
  42. return queryVoByParam(param);
  43. }
  44. }

###34. 新建UserInfoMapper.xml文件

在dao工程的src/main/resources/mapper目录中新建UserInfoMapper.xml文件。
说明:

  • 前面五个方法分别与BaseController中的方法对应(实际是BaseMapper);
  • 之后的三个方法分别与UserInfoController中自定义的方法对应;
  • 最后两个方法完全不在UserInfoController中,也不在其父类中,怎么调用呢?后续介绍-_-

    <?xml version=”1.0” encoding=”UTF-8” ?>
    <!DOCTYPE mapper PUBLIC “-//mybatis.org//DTD Mapper 3.0//EN” “http://mybatis.org/dtd/mybatis-3-mapper.dtd“ >

    1. <sql id="columns">
    2. id, name, mobile
    3. </sql>
    4. <sql id="columns-place-holder">
    5. #{id}, #{name}, #{mobile}
    6. </sql>
    7. <resultMap type="com.alan.entity.UserInfo" id="userInfoEntity">
    8. <id column="id" property="id" />
    9. <result column="name" property="name" />
    10. <result column="mobile" property="mobile" />
    11. </resultMap>
    12. <select id="queryById" resultMap="userInfoEntity ">
    13. select <include refid="columns"></include>
    14. from user_info
    15. where id=#{id}
    16. </select>
    17. <select id="queryAll" resultMap="userInfoEntity">
    18. select <include refid="columns"></include>
    19. from user_info
    20. </select>
    21. <insert id="add">
    22. insert into user_info (<include refid="columns"></include>)
    23. values(<include refid="columns-place-holder"></include>)
    24. </insert>
    25. <update id="update" parameterType="com.alan.entity.UserInfo">
    26. update user_info
    27. set id = #{id}
    28. <if test="name != null">
    29. ,name = #{name}
    30. </if>
    31. <if test="mobile != null">
    32. ,mobile = #{mobile}
    33. </if>
    34. where id = #{id}
    35. </update>
    36. <delete id="delete">
    37. delete from user_info where id = #{id}
    38. </delete>
  1. <select id="queryUserInfoByMobile" resultMap="userInfoEntity">
  2. select <include refid="columns"></include>
  3. from user_info
  4. where mobile = #{mobile}
  5. </select>
  6. <select id="queryUserInfoByName" resultMap="userInfoEntity">
  7. select <include refid="columns"></include>
  8. from user_info
  9. where name = #{name}
  10. </select>
  11. <select id="queryRandomUser" resultType="map">
  12. select <include refid="columns"></include>
  13. from user_info
  14. limit 1;
  15. </select>
  16. <select id="queryUserAddressByUserId" resultType="map">
  17. select ui.id, ui.name, ua.address
  18. from user_info ui, user_address ua
  19. where ui.id = ua.user_id
  20. and ui.id = #{userId}
  21. </select>
  22. <resultMap type="com.alan.entity.UserAddress" id="userAddressResult">
  23. <result column="address" property="address"/>
  24. </resultMap>
  25. <resultMap type="com.alan.model.UserAddressModel" id="userAddressModelResult">
  26. <id column="id" property="id"/>
  27. <result column="name" property="name"/>
  28. <collection property="addresses" resultMap="userAddressResult"></collection>
  29. </resultMap>
  30. <select id="queryUserAddressByUserIdReturnModel" resultMap="userAddressModelResult">
  31. select ui.id, ui.name, ua.address
  32. from user_info ui
  33. left join user_address ua on ui.id = ua.user_id
  34. where ui.id = #{userId}
  35. </select>
  36. </mapper>

###35. 测试
启动tomcat,浏览器中测试各方法的执行。
(1)queryById
这里写图片描述

(2)queryAll
这里写图片描述

(3)add
这里写图片描述
这里写图片描述

(4)update
这里写图片描述
这里写图片描述

(5)delete
这里写图片描述
这里写图片描述

(6)queryUserInfoByMobile
这里写图片描述

(7)queryUserInfoByName
这里写图片描述
这里写图片描述

(8)queryRandomUser
主要用于验证不返回model,直接返回map类型。
这里写图片描述

(9)queryUserAddressByUserId
通过/queryList/{selectId}访问,返回的是map对象。
这里写图片描述

(10)queryUserAddressByUserIdReturnModel
通过/queryList/{selectId}访问,返回的是具体的model,注意与上例比较。
这里写图片描述

(11)queryUserAddressByUserIdReturnModel
通过/queryVo/{selectId}访问,返回的是具体的model,注意与上例比较。
这里写图片描述

可见,以上各方法都可正确执行,表示我们的通用类编写完美。

###36. 通用类使用总结
(1)对于简单的增册改查,直接调用queryById、queryAll、add、update、delete方法即可;
(2)对于查询后就直接返回的,强烈推荐使用queryVo和queryList这两个方法,这种方式只需要在xml中定义好查询语句;
(3)如果还有业务逻辑需要处理,推荐调用父类的queryVoByParam和queryListByParam这两个方法,查询前后可处理业务逻辑,这种方式可以省去好几个方法的编写;
(4)此处只定义了查询的通用方法,像insert、update、delete也可以定义通用方法,与queryVoByParam和queryListByParam的定义类似,请读者自行参考mybatis源码定义。

###37.补充
补充insert、update、delete的通用方法,这里只写BaseSqlDaoImpl和BaseController两个类。
(1)BaseSqlDaoImpl.java

  1. @Override
  2. public int updateByParam(Map<String, Object> param){
  3. SqlSession session = sqlSessionFactory.openSession();
  4. try {
  5. return session.update(baseMapperPath + "." + param.get("mapperName") + "." + param.get("updateId"), param);
  6. } finally {
  7. session.close();
  8. }
  9. }
  10. @Override
  11. public int deleteByParam(Map<String, Object> param){
  12. SqlSession session = sqlSessionFactory.openSession();
  13. try {
  14. return session.delete(baseMapperPath + "." + param.get("mapperName") + "." + param.get("deleteId"), param);
  15. } finally {
  16. session.close();
  17. }
  18. }
  19. @Override
  20. public int insertByParam(Map<String, Object> param){
  21. SqlSession session = sqlSessionFactory.openSession();
  22. try {
  23. return session.insert(baseMapperPath + "." + param.get("mapperName") + "." + param.get("insertId"), param);
  24. } finally {
  25. session.close();
  26. }
  27. }

(2)BaseController.java

  1. /** * 从url上获取updateId并更新数据,其中updateId与*Mapper.xml中update的id对应 * @param request * @param updateId * @return */
  2. @RequestMapping("/updateCommon/{updateId}")
  3. public int updateCommon(HttpServletRequest request, @PathVariable String updateId) {
  4. Map<String, Object> param = buildParam(request);
  5. param.put("updateId", updateId);
  6. param.put("mapperName", getMapperName());
  7. return service.updateByParam(param);
  8. }
  9. /** * 从url上获取deleteId并更新数据,其中deleteId与*Mapper.xml中delete的id对应 * @param request * @param deleteId * @return */
  10. @RequestMapping("/deleteCommon/{deleteId}")
  11. public int deleteCommon(HttpServletRequest request, @PathVariable String deleteId) {
  12. Map<String, Object> param = buildParam(request);
  13. param.put("deleteId", deleteId);
  14. param.put("mapperName", getMapperName());
  15. return service.deleteByParam(param);
  16. }
  17. /** * 从url上获取insertId并更新数据,其中insertId与*Mapper.xml中insert的id对应 * @param request * @param insertId * @return */
  18. @RequestMapping("/insertCommon/{insertId}")
  19. public int insertCommon(HttpServletRequest request, @PathVariable String insertId) {
  20. Map<String, Object> param = buildParam(request);
  21. param.put("insertId", insertId);
  22. param.put("mapperName", getMapperName());
  23. return service.insertByParam(param);
  24. }

(防采集:本文唯一网址在CSDN上,博主会不定期更新加载其它配置,其它均为未经博主同意转载的,原文为http://blog.csdn.net/tangtong1/article/details/51442757)

##七、自动生成代码

从上述编写过程中可见,user_info这一个表,就要写实体类/mapper/dao/service/controller等这么多个类(包括mapper.xml),而且每个类中内容的形式基本保持一致,除了userInfo标识不同。那么,有没有方法不用写这些类呢, 答案是肯定的,下面我们就自己编写一个简单的代码生成工具。

###38. 生成代码思路
(1)拆分类并定义model
一个类可以拆分成以下几部分:全路径、访问修饰符、其它修饰符(static/final等)、父类、接口、注解、导入信息、字段、方法等,当然,还有内部类什么乱七八糟的这就不考虑了。如果我们把这些信息都定义好了,那么生成一个类还不是手到擒来。然后,方法、字段等又能分成一个一个的Model。

(2)重写上述Model的toString()方法
这样让Model类自己把生成的代码写好,方便后续生成。

(3)定义生成代码的类
比如,给个表名生成相应的dao/service等代码,给出表的列信息生成相应的entity/mapper等代码。

(4)定义数据库帮助类
因为要根据数据库中的表生成相应的entity/mapper等,所以需要从数据字典中获取这些信息,自然要连数据库。
以上两步结合就可以生成所有的代码了,但是还不够完美,要是能自己选择生成哪些表对应的代码,岂不更完美。

(5)编写界面类
没错,通过swing组件,编写一个界面,从数据库中查询出表信息,自己选择需要生成哪些表对应的代码,自己选择需要生成哪些层的代码。

(6)后悔药
如果生成错了咋办,还要一个一个文件把这些生成的文件删除掉,太麻烦了,“后悔药”即为生成代码后,可以快速把刚生成的删除。

###39. 成果展示

(1)操作界面
这里写图片描述

(2)生成的代码目录
这里写图片描述
这里写图片描述

(3)生成的代码
Mapper.xml
自动生成好相应的增删改查方法。

  1. <?xml version="1.0" encoding="UTF-8" ?>
  2. <!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd" >
  3. <mapper namespace="com.alan.mapper.UserTestXyzMapper">
  4. <sql id="columns">
  5. test_id, test_name, mobile, address
  6. </sql>
  7. <sql id="columns-place-holder">
  8. #{testId}, #{testName}, #{mobile}, #{address}
  9. </sql>
  10. <resultMap type="com.alan.entity.UserTestXyz" id="userTestXyzEntity">
  11. <id column="test_id" property="testId" />
  12. <result column="test_name" property="testName" />
  13. <result column="mobile" property="mobile" />
  14. <result column="address" property="address" />
  15. </resultMap>
  16. <select id="queryById" resultMap="userTestXyzEntity ">
  17. select <include refid="columns"></include>
  18. from user_test_xyz
  19. where test_id=#{testId}
  20. </select>
  21. <select id="queryAll" resultMap="userTestXyzEntity">
  22. select <include refid="columns"></include>
  23. from user_test_xyz
  24. </select>
  25. <insert id="add">
  26. insert into user_test_xyz (<include refid="columns"></include>)
  27. values(<include refid="columns-place-holder"></include>)
  28. </insert>
  29. <update id="update" parameterType="com.alan.entity.UserTestXyz">
  30. update user_test_xyz
  31. set test_id = #{testId}
  32. <if test="testName != null">
  33. ,test_name = #{testName}
  34. </if>
  35. <if test="mobile != null">
  36. ,mobile = #{mobile}
  37. </if>
  38. <if test="address != null">
  39. ,address = #{address}
  40. </if>
  41. where test_id = #{testId}
  42. </update>
  43. <delete id="delete">
  44. delete from user_test_xyz where test_id = #{testId}
  45. </delete>
  46. </mapper>

Controller.java
java文件有个弊端,import不会自动排好,Eclipse中ctrl+shift+o不知道怎么实现的。

  1. package com.alan.controller;
  2. import com.alan.parent.service.BaseService;
  3. import com.alan.entity.UserTestXyz;
  4. import javax.annotation.Resource;
  5. import org.springframework.web.bind.annotation.RequestMapping;
  6. import com.alan.service.UserTestXyzService;
  7. import org.springframework.web.bind.annotation.RestController;
  8. import com.alan.parent.controller.BaseController;
  9. @RestController
  10. @RequestMapping("/userTestXyz")
  11. public class UserTestXyzController extends BaseController<UserTestXyz> {
  12. @Resource
  13. private UserTestXyzService userTestXyzService;
  14. @Override
  15. public BaseService<UserTestXyz> init() {
  16. return userTestXyzService;
  17. }
  18. }

###40. 代码逻辑
注:此处是用纯java写的,后面博主采用了freemarker模板来写,事半功倍,有兴趣的同学可以去学习一下。
(1)Model类
在common工程下定义包com.alan.generate.model,并把所有生成代码相关model都置于其下。
(请注意model的set方法,及list的set改成了add)
ClassModel.java

  1. package com.alan.generate.model;
  2. import java.util.ArrayList;
  3. import java.util.HashSet;
  4. import java.util.List;
  5. import java.util.Set;
  6. import com.alan.generate.enums.ClassOrInterface;
  7. import com.alan.generate.enums.Modifier;
  8. import com.alan.generate.enums.Qualifier;
  9. import com.alan.generate.utils.Utils;
  10. public class ClassModel {
  11. private String name;
  12. private String simpleName;
  13. private Class<?> genericType;
  14. private Qualifier qualifier;
  15. private List<Modifier> modifiers;
  16. private ClassOrInterface classOrInterface;
  17. private List<AnnotationModel> annotations;
  18. private ClassTypeModel baseClass;
  19. private List<ClassTypeModel> interfaces;
  20. private List<FieldModel> fields;
  21. private List<MethodModel> methods;
  22. private Set<String> limitcs;
  23. @Override
  24. public String toString() {
  25. String result = "#{package}#{limitcs}#{annotations}#{qualifier}#{modifier}#{classType}#{genericType} #{simpleName}#{baseClass}#{interface} {\n\n#{fields}#{methods}}";
  26. if (Utils.isNotEmpty(name))
  27. result = result.replace("#{package}", "package " + name.substring(0, name.lastIndexOf(".")) + ";\n\n");
  28. if (Utils.isNotEmpty(limitcs)) {
  29. final StringBuilder sb = new StringBuilder();
  30. limitcs.forEach(l -> {
  31. if (!name.substring(0, name.lastIndexOf(".")).equals(l.substring(0, l.lastIndexOf("."))) && !l.startsWith("java.lang."))
  32. sb.append("import ").append(l).append(";\n");
  33. });
  34. result = result.replace("#{limitcs}", sb.toString() + "\n");
  35. }
  36. if (Utils.isNotEmpty(annotations)) {
  37. final StringBuilder sb = new StringBuilder();
  38. annotations.forEach(a -> sb.append(a.toString()).append("\n"));
  39. result = result.replace("#{annotations}", sb.toString());
  40. }
  41. if (Utils.isNotEmpty(qualifier))
  42. result = result.replace("#{qualifier}", qualifier.toString() + " ");
  43. if (Utils.isNotEmpty(modifiers)) {
  44. final StringBuilder sb = new StringBuilder();
  45. modifiers.forEach(m -> sb.append(m.toString()).append(" "));
  46. result = result.replace("#{modifier}", sb.toString());
  47. }
  48. result = result.replace("#{classType}", classOrInterface.toString());
  49. if (Utils.isNotEmpty(genericType))
  50. result = result.replace("#{genericType}", "<" + genericType.getSimpleName() + ">");
  51. result = result.replace("#{simpleName}", simpleName);
  52. if (Utils.isNotEmpty(baseClass))
  53. result = result.replace("#{baseClass}", " extends " + baseClass.toString());
  54. if (Utils.isNotEmpty(interfaces)) {
  55. final StringBuilder sb = new StringBuilder();
  56. interfaces.forEach(i -> sb.append(i.toString()).append(", "));
  57. result = result.replace("#{interface}", " implements " + sb.substring(0, sb.lastIndexOf(",")));
  58. }
  59. if (Utils.isNotEmpty(fields)) {
  60. StringBuilder sb = new StringBuilder();
  61. fields.forEach(f -> sb.append(f.toString()).append("\n"));
  62. result = result.replace("#{fields}", sb.toString() + "\n");
  63. }
  64. if (Utils.isNotEmpty(methods)) {
  65. StringBuilder sb = new StringBuilder();
  66. methods.forEach(m -> sb.append(m.toString()).append("\n\n"));
  67. result = result.replace("#{methods}", sb.toString());
  68. }
  69. return result.replaceAll("\\#\\{.+?\\}", "");
  70. }
  71. public Set<String> getLimitcs() {
  72. return limitcs;
  73. }
  74. public ClassModel addLimitc(String limitc) {
  75. if (limitcs == null) {
  76. limitcs = new HashSet<>();
  77. }
  78. this.limitcs.add(limitc);
  79. return this;
  80. }
  81. public Class<?> getGenericType() {
  82. return genericType;
  83. }
  84. public ClassModel setGenericType(Class<?> genericType) {
  85. this.genericType = genericType;
  86. return this;
  87. }
  88. public String getName() {
  89. return name;
  90. }
  91. public ClassModel setName(String name) {
  92. this.name = name;
  93. this.simpleName = name.substring(name.lastIndexOf(".") + 1);
  94. return this;
  95. }
  96. public String getSimpleName() {
  97. return simpleName;
  98. }
  99. public ClassModel setSimpleName(String simpleName) {
  100. this.simpleName = simpleName;
  101. return this;
  102. }
  103. public Qualifier getQualifier() {
  104. return qualifier;
  105. }
  106. public ClassModel setQualifier(Qualifier qualifier) {
  107. this.qualifier = qualifier;
  108. return this;
  109. }
  110. public List<Modifier> getModifiers() {
  111. return modifiers;
  112. }
  113. public ClassModel addModifier(Modifier modifier) {
  114. if (this.modifiers == null) {
  115. this.modifiers = new ArrayList<>();
  116. }
  117. this.modifiers.add(modifier);
  118. return this;
  119. }
  120. public ClassOrInterface getClassOrInterface() {
  121. return classOrInterface;
  122. }
  123. public ClassModel setClassOrInterface(ClassOrInterface classOrInterface) {
  124. this.classOrInterface = classOrInterface;
  125. return this;
  126. }
  127. public List<AnnotationModel> getAnnotations() {
  128. return annotations;
  129. }
  130. public ClassModel addAnnotation(AnnotationModel annotation) {
  131. if (this.annotations == null) {
  132. this.annotations = new ArrayList<>();
  133. }
  134. this.annotations.add(annotation);
  135. return this;
  136. }
  137. public ClassTypeModel getBaseClass() {
  138. return baseClass;
  139. }
  140. public ClassModel setBaseClass(ClassTypeModel baseClass) {
  141. this.baseClass = baseClass;
  142. return this;
  143. }
  144. public List<ClassTypeModel> getInterfaces() {
  145. return interfaces;
  146. }
  147. public ClassModel addInterface(ClassTypeModel interfacec) {
  148. if (this.interfaces == null) {
  149. this.interfaces = new ArrayList<>();
  150. }
  151. this.interfaces.add(interfacec);
  152. return this;
  153. }
  154. public List<FieldModel> getFields() {
  155. return fields;
  156. }
  157. public ClassModel addField(FieldModel field) {
  158. if (this.fields == null) {
  159. this.fields = new ArrayList<>();
  160. }
  161. this.fields.add(field);
  162. return this;
  163. }
  164. public List<MethodModel> getMethods() {
  165. return methods;
  166. }
  167. public ClassModel addMethod(MethodModel method) {
  168. if (this.methods == null) {
  169. this.methods = new ArrayList<>();
  170. }
  171. this.methods.add(method);
  172. return this;
  173. }
  174. }

ClassTypeModel.java

  1. package com.alan.generate.model;
  2. import com.alan.generate.utils.Utils;
  3. public class ClassTypeModel {
  4. private Class<?> classType;
  5. private String className;
  6. private Class<?> genericType;
  7. private String genericTypeName;
  8. private ClassModel classModel;
  9. public ClassTypeModel() {
  10. }
  11. public ClassTypeModel(ClassModel classModel) {
  12. this.classModel = classModel;
  13. }
  14. @Override
  15. public String toString() {
  16. return className + (Utils.isNotEmpty(genericTypeName) ? "<" + genericTypeName + ">" : "");
  17. }
  18. public String getClassName() {
  19. return className;
  20. }
  21. public ClassTypeModel setClassName(String className) {
  22. this.className = className;
  23. return this;
  24. }
  25. public String getGenericTypeName() {
  26. return genericTypeName;
  27. }
  28. public ClassTypeModel setGenericTypeName(String genericTypeName) {
  29. this.genericTypeName = genericTypeName;
  30. return this;
  31. }
  32. public ClassModel getClassModel() {
  33. return classModel;
  34. }
  35. public void setClassModel(ClassModel classModel) {
  36. this.classModel = classModel;
  37. }
  38. public Class<?> getClassType() {
  39. return classType;
  40. }
  41. public ClassTypeModel setClassType(Class<?> classType) {
  42. if (classModel != null) {
  43. classModel.addLimitc(classType.getName());
  44. }
  45. this.classType = classType;
  46. setClassName(classType.getSimpleName());
  47. return this;
  48. }
  49. public Class<?> getGenericType() {
  50. return genericType;
  51. }
  52. public ClassTypeModel setGenericType(Class<?> genericType) {
  53. this.genericType = genericType;
  54. this.genericTypeName = genericType.getSimpleName();
  55. return this;
  56. }
  57. }

AnnotationModel.java

  1. package com.alan.generate.model;
  2. public class AnnotationModel extends ClassTypeModel {
  3. private String value;
  4. public AnnotationModel() {
  5. }
  6. public AnnotationModel(ClassModel classModel) {
  7. setClassModel(classModel);
  8. }
  9. @Override
  10. public String toString() {
  11. return "@" + getClassType().getSimpleName() + (value == null ? "" : "(" + value + ")");
  12. }
  13. public Class<?> getAnnotationClass() {
  14. return getClassType();
  15. }
  16. public AnnotationModel setAnnotationClass(Class<?> annotationClass) {
  17. setClassType(annotationClass);
  18. return this;
  19. }
  20. public String getValue() {
  21. return value;
  22. }
  23. public AnnotationModel setValue(String value) {
  24. this.value = value;
  25. return this;
  26. }
  27. }

FieldModel.java

  1. package com.alan.generate.model;
  2. import java.util.ArrayList;
  3. import java.util.List;
  4. import com.alan.generate.enums.Modifier;
  5. import com.alan.generate.enums.Qualifier;
  6. import com.alan.generate.utils.Utils;
  7. public class FieldModel {
  8. private Qualifier qualifier;
  9. private List<Modifier> modifiers;
  10. private ClassTypeModel fieldType;
  11. private String fieldName;
  12. private String value;
  13. private List<AnnotationModel> annotations;
  14. private ClassModel classModel;
  15. public FieldModel() {
  16. }
  17. public FieldModel(ClassModel classModel) {
  18. this.classModel = classModel;
  19. }
  20. @Override
  21. public String toString() {
  22. String result = "#{annotations}\t#{qualifier}#{modifiers}#{fieldType} #{fieldName}#{value};";
  23. if (Utils.isNotEmpty(annotations)) {
  24. final StringBuilder sb = new StringBuilder();
  25. annotations.forEach(a -> sb.append("\t").append(a.toString()).append("\n"));
  26. result = result.replace("#{annotations}", sb.toString());
  27. }
  28. if (Utils.isNotEmpty(qualifier))
  29. result = result.replace("#{qualifier}", qualifier.toString() + " ");
  30. if (Utils.isNotEmpty(modifiers)) {
  31. final StringBuilder sb = new StringBuilder();
  32. modifiers.forEach(m -> sb.append(m.toString()).append(" "));
  33. result = result.replace("#{modifiers}", sb.toString());
  34. }
  35. result = result.replace("#{fieldType}", fieldType.toString());
  36. result = result.replace("#{fieldName}", fieldName);
  37. if (Utils.isNotEmpty(value))
  38. result = result.replace("#{value}", " = " + value);
  39. return result.replaceAll("\\#\\{.+?\\}", "");
  40. }
  41. public String valid() {
  42. return Utils.isNotEmpty(fieldType) && Utils.isNotEmpty(fieldName) ? "" : "not valid field";
  43. }
  44. public ClassModel getClassModel() {
  45. return classModel;
  46. }
  47. public FieldModel setClassModel(ClassModel classModel) {
  48. this.classModel = classModel;
  49. return this;
  50. }
  51. public String getValue() {
  52. return value;
  53. }
  54. public FieldModel setValue(String value) {
  55. this.value = value;
  56. return this;
  57. }
  58. public List<AnnotationModel> getAnnotations() {
  59. return annotations;
  60. }
  61. public FieldModel addAnnotation(AnnotationModel annotation) {
  62. if (this.annotations == null) {
  63. this.annotations = new ArrayList<>();
  64. }
  65. this.annotations.add(annotation);
  66. return this;
  67. }
  68. public Qualifier getQualifier() {
  69. return qualifier;
  70. }
  71. public FieldModel setQualifier(Qualifier qualifier) {
  72. this.qualifier = qualifier;
  73. return this;
  74. }
  75. public List<Modifier> getModifiers() {
  76. return modifiers;
  77. }
  78. public FieldModel addModifier(Modifier modifier) {
  79. if (this.modifiers == null) {
  80. this.modifiers = new ArrayList<>();
  81. }
  82. this.modifiers.add(modifier);
  83. return this;
  84. }
  85. public ClassTypeModel getFieldType() {
  86. return fieldType;
  87. }
  88. public FieldModel setFieldType(ClassTypeModel fieldType) {
  89. this.fieldType = fieldType;
  90. return this;
  91. }
  92. public String getFieldName() {
  93. return fieldName;
  94. }
  95. public FieldModel setFieldName(String fieldName) {
  96. this.fieldName = fieldName;
  97. return this;
  98. }
  99. public FieldModel generateSetMethod() {
  100. MethodModel methodModel = new MethodModel();
  101. methodModel.setQualifier(Qualifier.publicQ)
  102. .setMethodName("set" + this.fieldName.substring(0, 1).toUpperCase() + this.fieldName.substring(1))
  103. .addParamModel(new ParamModel().setParamType(this.fieldType).setParamName(this.fieldName))
  104. .addStatement("this." + this.fieldName + " = " + this.fieldName);
  105. classModel.addMethod(methodModel);
  106. return this;
  107. }
  108. public FieldModel generateGetMethod() {
  109. MethodModel methodModel = new MethodModel();
  110. methodModel.setQualifier(Qualifier.publicQ).setReturnType(this.fieldType)
  111. .setMethodName("get" + this.fieldName.substring(0, 1).toUpperCase() + this.fieldName.substring(1))
  112. .addStatement("return " + this.fieldName);
  113. classModel.addMethod(methodModel);
  114. return this;
  115. }
  116. }

MethodModel.java

  1. package com.alan.generate.model;
  2. import java.util.ArrayList;
  3. import java.util.List;
  4. import com.alan.generate.enums.Modifier;
  5. import com.alan.generate.enums.Qualifier;
  6. import com.alan.generate.utils.Utils;
  7. public class MethodModel {
  8. private Qualifier qualifier;
  9. private List<Modifier> modifiers;
  10. private ClassTypeModel returnType;
  11. private String methodName;
  12. private List<ParamModel> paramModels;
  13. private List<String> statements;
  14. private List<AnnotationModel> annotations;
  15. @Override
  16. public String toString() {
  17. String result = "#{annotations}\t#{qualifier}#{modifiers}#{returnType}#{methodName} (#{paramModels}) {\n#{statements}\t}";
  18. if (Utils.isNotEmpty(annotations)) {
  19. final StringBuilder sb = new StringBuilder();
  20. annotations.forEach(a -> sb.append("\t").append(a.toString()).append("\n"));
  21. result = result.replace("#{annotations}", sb.toString());
  22. }
  23. if (Utils.isNotEmpty(qualifier))
  24. result = result.replace("#{qualifier}", qualifier.toString() + " ");
  25. if (Utils.isNotEmpty(modifiers)) {
  26. final StringBuilder sb = new StringBuilder();
  27. modifiers.forEach(m -> sb.append(m.toString()).append(" "));
  28. result = result.replace("#{modifiers}", sb.toString());
  29. }
  30. if (Utils.isNotEmpty(returnType)) {
  31. result = result.replace("#{returnType}", returnType.toString() + " ");
  32. } else {
  33. result = result.replace("#{returnType}", "void ");
  34. }
  35. result = result.replace("#{methodName}", methodName);
  36. if (Utils.isNotEmpty(paramModels)) {
  37. final StringBuilder sb = new StringBuilder();
  38. paramModels.forEach(p -> sb.append(p.toString()).append(", "));
  39. result = result.replace("#{paramModels}", sb.substring(0, sb.lastIndexOf(",")));
  40. }
  41. if (Utils.isNotEmpty(statements)) {
  42. StringBuilder sb = new StringBuilder();
  43. statements.forEach(s -> {
  44. if (s.lastIndexOf(";") != s.length() - 1)
  45. s += ";";
  46. sb.append("\t\t").append(s).append("\n");
  47. });
  48. result = result.replace("#{statements}", sb.toString());
  49. }
  50. return result.replaceAll("\\#\\{.+?\\}", "");
  51. }
  52. public String valid() {
  53. return Utils.isNotEmpty(methodName) ? "" : "not valid method";
  54. }
  55. public List<AnnotationModel> getAnnotations() {
  56. return annotations;
  57. }
  58. public MethodModel addAnnotation(AnnotationModel annotation) {
  59. if (this.annotations == null) {
  60. this.annotations = new ArrayList<>();
  61. }
  62. this.annotations.add(annotation);
  63. return this;
  64. }
  65. public Qualifier getQualifier() {
  66. return qualifier;
  67. }
  68. public MethodModel setQualifier(Qualifier qualifier) {
  69. this.qualifier = qualifier;
  70. return this;
  71. }
  72. public List<Modifier> getModifiers() {
  73. return modifiers;
  74. }
  75. public MethodModel addModifier(Modifier modifier) {
  76. if (this.modifiers == null) {
  77. this.modifiers = new ArrayList<>();
  78. }
  79. this.modifiers.add(modifier);
  80. return this;
  81. }
  82. public ClassTypeModel getReturnType() {
  83. return returnType;
  84. }
  85. public MethodModel setReturnType(ClassTypeModel returnType) {
  86. this.returnType = returnType;
  87. return this;
  88. }
  89. public String getMethodName() {
  90. return methodName;
  91. }
  92. public MethodModel setMethodName(String methodName) {
  93. this.methodName = methodName;
  94. return this;
  95. }
  96. public List<ParamModel> getParamModels() {
  97. return paramModels;
  98. }
  99. public MethodModel addParamModel(ParamModel paramModel) {
  100. if (this.paramModels == null) {
  101. this.paramModels = new ArrayList<>();
  102. }
  103. this.paramModels.add(paramModel);
  104. return this;
  105. }
  106. public List<String> getStatements() {
  107. return statements;
  108. }
  109. public MethodModel addStatement(String statement) {
  110. if (this.statements == null) {
  111. this.statements = new ArrayList<>();
  112. }
  113. this.statements.add(statement);
  114. return this;
  115. }
  116. }

ParamModel.java

  1. package com.alan.generate.model;
  2. import java.util.ArrayList;
  3. import java.util.List;
  4. import com.alan.generate.utils.Utils;
  5. public class ParamModel {
  6. private ClassTypeModel paramType;
  7. private String paramName;
  8. private List<AnnotationModel> annotations;
  9. @Override
  10. public String toString() {
  11. String result = "#{annotations}#{paramType}#{paramTypeGenericType} #{paramName}";
  12. if (Utils.isNotEmpty(annotations)) {
  13. final StringBuilder sb = new StringBuilder();
  14. annotations.forEach(a -> sb.append(a.toString()).append(" "));
  15. result = result.replace("#{annotations}", sb.toString());
  16. }
  17. result = result.replace("#{paramType}", paramType.toString());
  18. result = result.replace("#{paramName}", paramName);
  19. return result.replaceAll("\\#\\{.+?\\}", "");
  20. }
  21. public String valid() {
  22. return Utils.isNotEmpty(paramType) && Utils.isNotEmpty(paramName) ? "" : "not valid param";
  23. }
  24. public List<AnnotationModel> getAnnotations() {
  25. return annotations;
  26. }
  27. public ParamModel addAnnotation(AnnotationModel annotation) {
  28. if (this.annotations == null) {
  29. this.annotations = new ArrayList<>();
  30. }
  31. this.annotations.add(annotation);
  32. return this;
  33. }
  34. public ClassTypeModel getParamType() {
  35. return paramType;
  36. }
  37. public ParamModel setParamType(ClassTypeModel paramType) {
  38. this.paramType = paramType;
  39. return this;
  40. }
  41. public String getParamName() {
  42. return paramName;
  43. }
  44. public ParamModel setParamName(String paramName) {
  45. this.paramName = paramName;
  46. return this;
  47. }
  48. }

数据库对应的model
TableModel.java

  1. package com.alan.generate.model;
  2. import java.util.ArrayList;
  3. import java.util.List;
  4. import com.alan.generate.utils.Utils;
  5. /** * 用于存储数据库中原生数据 * * @author alan * */
  6. public class TableModel {
  7. private String tableName;
  8. private ColumnModel primaryKey;
  9. // 不包含主键列
  10. private List<ColumnModel> columns;
  11. public TableModel() {
  12. }
  13. public TableModel(String tableName) {
  14. this.tableName = tableName;
  15. }
  16. public String getTableName() {
  17. return tableName;
  18. }
  19. public void setTableName(String tableName) {
  20. this.tableName = tableName;
  21. }
  22. public ColumnModel getPrimaryKey() {
  23. return primaryKey;
  24. }
  25. public void setPrimaryKey(ColumnModel primaryKey) {
  26. this.primaryKey = primaryKey;
  27. }
  28. public List<ColumnModel> getColumns() {
  29. return columns;
  30. }
  31. public void addColumn(ColumnModel column) {
  32. if (columns == null) {
  33. columns = new ArrayList<>();
  34. }
  35. this.columns.add(column);
  36. }
  37. @Override
  38. public String toString() {
  39. String result = "table name is: \n" + tableName + "\n\nprimary key is: \n" + primaryKey.getColumnName() + " "
  40. + primaryKey.getDataType() + "\n\n";
  41. if (Utils.isNotEmpty(columns)) {
  42. StringBuilder sb = new StringBuilder("columns are: \n");
  43. columns.forEach(c -> sb.append(c.getColumnName()).append(" ").append(c.getDataType()).append("\n"));
  44. result += sb.toString();
  45. }
  46. return result;
  47. }
  48. }

ColumnModel.java

  1. package com.alan.generate.model;
  2. public class ColumnModel {
  3. private String columnName;
  4. private String dataType;
  5. public String getColumnName() {
  6. return columnName;
  7. }
  8. public ColumnModel setColumnName(String columnName) {
  9. this.columnName = columnName;
  10. return this;
  11. }
  12. public String getDataType() {
  13. return dataType;
  14. }
  15. public ColumnModel setDataType(String dataType) {
  16. this.dataType = dataType;
  17. return this;
  18. }
  19. }

(2)枚举类
定义几个枚举类,在包com.alan.generate.enums下:
Qualifier.java

  1. package com.alan.generate.enums;
  2. public enum Qualifier {
  3. publicQ("public"), privateQ("private"), protectedQ("protected"), defaultQ("");
  4. private String code;
  5. private Qualifier(String code) {
  6. this.code = code;
  7. }
  8. public String getCode() {
  9. return code;
  10. }
  11. @Override
  12. public String toString() {
  13. return code;
  14. }
  15. public static void main(String[] args) {
  16. System.out.println(privateQ.toString());
  17. }
  18. }

Modifier.java
这里就定义了两个,还有几个不怎么用的,就不定义了。

  1. package com.alan.generate.enums;
  2. public enum Modifier {
  3. staticM("static"), finalM("final");
  4. private String code;
  5. private Modifier(String code) {
  6. this.code = code;
  7. }
  8. public String getCode() {
  9. return code;
  10. }
  11. @Override
  12. public String toString() {
  13. return code;
  14. }
  15. }

ClassOrInterface.java

  1. package com.alan.generate.enums;
  2. public enum ClassOrInterface {
  3. classC("class"), interfaceC("interface"), aInterfaceC("@interface");
  4. private String code;
  5. private ClassOrInterface(String code) {
  6. this.code = code;
  7. }
  8. public String getCode() {
  9. return code;
  10. }
  11. @Override
  12. public String toString() {
  13. return code;
  14. }
  15. }

(3)常量类
用于存放数据库连接及路径配置。
Constant.java

  1. package com.alan.generate.main;
  2. public class Constant {
  3. public static final String entityPkg = "com.alan.entity";
  4. public static final String mapperPkg = "com.alan.mapper";
  5. public static final String daoPkg = "com.alan.dao";
  6. public static final String servicePkg = "com.alan.service";
  7. public static final String controllerPkg = "com.alan.controller";
  8. public static final String daoProject = "alan-springmvc-dao";
  9. public static final String serviceProject = "alan-springmvc-service";
  10. public static final String webProject = "alan-springmvc-web";
  11. public static String driverClassName = "com.mysql.jdbc.Driver";
  12. public static String url = "jdbc:mysql://127.0.0.1:3306/alan_springmvc";
  13. public static String username = "root";
  14. public static String password = "1234";
  15. }

(4)帮助类
Utils.java
主要用于处理判空、数据库下划线转驼峰等。

  1. package com.alan.generate.utils;
  2. import java.util.ArrayList;
  3. import java.util.HashMap;
  4. import java.util.List;
  5. import java.util.Map;
  6. public class Utils {
  7. public static Map<String, String> dataTypeMapping;
  8. static {
  9. dataTypeMapping = new HashMap<String, String>();
  10. dataTypeMapping.put("varchar", "String");
  11. dataTypeMapping.put("bigint", "Long");
  12. dataTypeMapping.put("int", "Long");
  13. dataTypeMapping.put("datetime", "java.util.Date");
  14. dataTypeMapping.put("tinyint", "Integer");
  15. dataTypeMapping.put("decimal", "java.math.BigDecimal");
  16. dataTypeMapping.put("double", "Double");
  17. dataTypeMapping.put("char", "String");
  18. dataTypeMapping.put("text", "String");
  19. dataTypeMapping.put("float", "Double");
  20. dataTypeMapping.put("timestamp", "java.util.Date");
  21. }
  22. public static boolean isNotEmpty(Object object) {
  23. return !isEmpty(object);
  24. }
  25. public static boolean isEmpty(Object object) {
  26. if (object == null)
  27. return true;
  28. if (object instanceof String)
  29. return String.valueOf(object).length() == 0;
  30. if (object instanceof List<?>)
  31. return ((List<?>) object).size() == 0;
  32. return false;
  33. }
  34. public static String firstUpper(String string) {
  35. return string.substring(0, 1).toUpperCase() + string.substring(1);
  36. }
  37. public static String firstLower(String string) {
  38. return string.substring(0, 1).toLowerCase() + string.substring(1);
  39. }
  40. public static String fromUnderlineToCamel(String string) {
  41. char[] src = string.toLowerCase().toCharArray();
  42. char[] des = new char[src.length];
  43. boolean flag = false;
  44. int i = 0;
  45. for (char c : src) {
  46. if (c == '_') {
  47. flag = true;
  48. continue;
  49. }
  50. if (flag) {
  51. c = Character.toUpperCase(c);
  52. flag = false;
  53. }
  54. des[i++] = c;
  55. }
  56. return String.valueOf(des).trim();
  57. }
  58. public static String getTableName(String dbTableName) {
  59. return firstUpper(fromUnderlineToCamel(dbTableName));
  60. }
  61. public static String getColumnName(String dbColumnName) {
  62. return fromUnderlineToCamel(dbColumnName);
  63. }
  64. public static String join(List<String> list, String seperator) {
  65. String result = list.stream().reduce("", (sum, s) -> sum = sum + s + seperator);
  66. return result.substring(0, result.lastIndexOf(seperator));
  67. }
  68. public static void main(String[] args) {
  69. // String result = firstUpper("userInfo");
  70. // System.out.println(result);
  71. // System.out.println(fromUnderlineToCamel("user_info_table"));
  72. List<String> list = new ArrayList<>();
  73. list.add("a");
  74. list.add("b");
  75. list.add("b");
  76. list.add("b");
  77. list.add("b");
  78. System.out.println(join(list, ", "));
  79. }
  80. }

(5)生成代码帮助类
GenerateHelper.java

  1. package com.alan.generate.utils;
  2. import java.io.BufferedReader;
  3. import java.io.File;
  4. import java.io.FileOutputStream;
  5. import java.io.FileReader;
  6. import java.io.FileWriter;
  7. import java.io.IOException;
  8. import java.util.List;
  9. import javax.annotation.Resource;
  10. import org.springframework.stereotype.Repository;
  11. import org.springframework.stereotype.Service;
  12. import org.springframework.web.bind.annotation.RequestMapping;
  13. import org.springframework.web.bind.annotation.RestController;
  14. import com.alan.generate.enums.ClassOrInterface;
  15. import com.alan.generate.enums.Qualifier;
  16. import com.alan.generate.main.Constant;
  17. import com.alan.generate.model.AnnotationModel;
  18. import com.alan.generate.model.ClassModel;
  19. import com.alan.generate.model.ClassTypeModel;
  20. import com.alan.generate.model.ColumnModel;
  21. import com.alan.generate.model.FieldModel;
  22. import com.alan.generate.model.MethodModel;
  23. import com.alan.generate.model.TableModel;
  24. import com.alan.parent.controller.BaseController;
  25. import com.alan.parent.dao.BaseDao;
  26. import com.alan.parent.dao.BaseDaoImpl;
  27. import com.alan.parent.mapper.BaseMapper;
  28. import com.alan.parent.service.BaseService;
  29. import com.alan.parent.service.BaseServiceImpl;
  30. public class GenerateHelper {
  31. public static final String entityPkg = Constant.entityPkg;
  32. public static final String mapperPkg = Constant.mapperPkg;
  33. public static final String daoPkg = Constant.daoPkg;
  34. public static final String servicePkg = Constant.servicePkg;
  35. public static final String controllerPkg = Constant.controllerPkg;
  36. public static final String daoProject = Constant.daoProject;
  37. public static final String serviceProject = Constant.serviceProject;
  38. public static final String webProject = Constant.webProject;
  39. public static String daoPath;
  40. public static String servicePath;
  41. public static String webPath;
  42. public static File historyFile;
  43. public static boolean flag = false;
  44. static {
  45. File directory = new File("");// 参数为空
  46. File parentDir;
  47. try {
  48. parentDir = new File(new File(directory.getCanonicalPath()).getParent());
  49. File[] files = parentDir.listFiles();
  50. for (File file : files) {
  51. if (file.isDirectory()) {
  52. if (file.getName().equals(daoProject))
  53. daoPath = file.getAbsolutePath();
  54. if (file.getName().equals(serviceProject))
  55. servicePath = file.getAbsolutePath();
  56. if (file.getName().equals(webProject))
  57. webPath = file.getAbsolutePath();
  58. }
  59. }
  60. historyFile = new File(daoPath + "\\src\\main\\resources\\history.txt");
  61. } catch (IOException e) {
  62. e.printStackTrace();
  63. }
  64. }
  65. public static String getDaoInterfaceCode(String entityName) {
  66. ClassModel classModel = new ClassModel();
  67. classModel.setName(daoPkg + "." + entityName + "Dao").addLimitc(entityPkg + "." + entityName)
  68. .setQualifier(Qualifier.publicQ).setClassOrInterface(ClassOrInterface.interfaceC).setBaseClass(
  69. new ClassTypeModel(classModel).setClassType(BaseDao.class).setGenericTypeName(entityName));
  70. return classModel.toString();
  71. }
  72. public static String getDaoImplCode(String entityName) {
  73. ClassModel classModel = new ClassModel();
  74. classModel.setName(daoPkg + ".impl." + entityName + "DaoImpl")
  75. .addAnnotation(new AnnotationModel(classModel).setAnnotationClass(Repository.class)
  76. .setValue("\"" + Utils.firstLower(entityName) + "Dao\""))
  77. .setQualifier(Qualifier.publicQ).setClassOrInterface(ClassOrInterface.classC)
  78. .setBaseClass(
  79. new ClassTypeModel(classModel).setClassType(BaseDaoImpl.class).setGenericTypeName(entityName))
  80. .addInterface(new ClassTypeModel().setClassName(entityName + "Dao"))
  81. .addField(new FieldModel(classModel)
  82. .addAnnotation(new AnnotationModel(classModel).setAnnotationClass(Resource.class))
  83. .setQualifier(Qualifier.privateQ)
  84. .setFieldType(new ClassTypeModel().setClassName(entityName + "Mapper")).setFieldName(
  85. Utils.firstLower(entityName) + "Mapper"))
  86. .addMethod(new MethodModel()
  87. .addAnnotation(new AnnotationModel(classModel).setAnnotationClass(Override.class))
  88. .setQualifier(Qualifier.publicQ)
  89. .setReturnType(new ClassTypeModel(classModel).setClassType(BaseMapper.class)
  90. .setGenericTypeName(entityName))
  91. .setMethodName("init").addStatement("return " + Utils.firstLower(entityName) + "Mapper"))
  92. .addLimitc(entityPkg + "." + entityName).addLimitc(daoPkg + "." + entityName + "Dao")
  93. .addLimitc(mapperPkg + "." + entityName + "Mapper");
  94. return classModel.toString();
  95. }
  96. public static String getMapperCode(String entityName) {
  97. ClassModel classModel = new ClassModel();
  98. classModel.setName(mapperPkg + "." + entityName + "Mapper").addLimitc(entityPkg + "." + entityName)
  99. .setQualifier(Qualifier.publicQ).setClassOrInterface(ClassOrInterface.interfaceC).setBaseClass(
  100. new ClassTypeModel(classModel).setClassType(BaseMapper.class).setGenericTypeName(entityName));
  101. return classModel.toString();
  102. }
  103. public static String getServiceCode(String entityName) {
  104. ClassModel classModel = new ClassModel();
  105. classModel.setName(servicePkg + "." + entityName + "Service").addLimitc(entityPkg + "." + entityName)
  106. .setQualifier(Qualifier.publicQ).setClassOrInterface(ClassOrInterface.interfaceC).setBaseClass(
  107. new ClassTypeModel(classModel).setClassType(BaseService.class).setGenericTypeName(entityName));
  108. return classModel.toString();
  109. }
  110. public static String getServiceImplCode(String entityName) {
  111. ClassModel classModel = new ClassModel();
  112. classModel.setName(servicePkg + ".impl." + entityName + "ServiceImpl")
  113. .addAnnotation(new AnnotationModel(classModel).setAnnotationClass(Service.class)
  114. .setValue("\"" + Utils.firstLower(entityName) + "Service\""))
  115. .setQualifier(Qualifier.publicQ).setClassOrInterface(ClassOrInterface.classC)
  116. .setBaseClass(new ClassTypeModel(classModel).setClassType(BaseServiceImpl.class)
  117. .setGenericTypeName(entityName))
  118. .addInterface(new ClassTypeModel().setClassName(entityName + "Service"))
  119. .addField(new FieldModel(classModel)
  120. .addAnnotation(new AnnotationModel(classModel).setAnnotationClass(Resource.class)).setQualifier(
  121. Qualifier.privateQ)
  122. .setFieldType(new ClassTypeModel().setClassName(entityName + "Dao")).setFieldName(
  123. Utils.firstLower(entityName) + "Dao"))
  124. .addMethod(new MethodModel()
  125. .addAnnotation(new AnnotationModel(classModel).setAnnotationClass(Override.class))
  126. .setQualifier(Qualifier.publicQ)
  127. .setReturnType(new ClassTypeModel(classModel).setClassType(BaseDao.class)
  128. .setGenericTypeName(entityName))
  129. .setMethodName("init").addStatement("return " + Utils.firstLower(entityName) + "Dao"))
  130. .addLimitc(entityPkg + "." + entityName).addLimitc(servicePkg + "." + entityName + "Service")
  131. .addLimitc(daoPkg + "." + entityName + "Dao");
  132. return classModel.toString();
  133. }
  134. public static String getControllerCode(String entityName) {
  135. ClassModel classModel = new ClassModel();
  136. classModel.setName(controllerPkg + "." + entityName + "Controller")
  137. .addAnnotation(new AnnotationModel(classModel).setAnnotationClass(RestController.class))
  138. .addAnnotation(new AnnotationModel(classModel).setAnnotationClass(RequestMapping.class)
  139. .setValue("\"/" + Utils.firstLower(entityName) + "\""))
  140. .setQualifier(Qualifier.publicQ).setClassOrInterface(ClassOrInterface.classC)
  141. .setBaseClass(new ClassTypeModel(classModel).setClassType(BaseController.class)
  142. .setGenericTypeName(entityName))
  143. .addField(new FieldModel(classModel)
  144. .addAnnotation(new AnnotationModel(classModel).setAnnotationClass(Resource.class))
  145. .setQualifier(Qualifier.privateQ)
  146. .setFieldType(new ClassTypeModel().setClassName(entityName + "Service")).setFieldName(
  147. Utils.firstLower(entityName) + "Service"))
  148. .addMethod(new MethodModel()
  149. .addAnnotation(new AnnotationModel(classModel).setAnnotationClass(Override.class))
  150. .setQualifier(Qualifier.publicQ)
  151. .setReturnType(new ClassTypeModel(classModel).setClassType(BaseService.class)
  152. .setGenericTypeName(entityName))
  153. .setMethodName("init").addStatement("return " + Utils.firstLower(entityName) + "Service"))
  154. .addLimitc(entityPkg + "." + entityName).addLimitc(servicePkg + "." + entityName + "Service");
  155. return classModel.toString();
  156. }
  157. public static String getEntityCode(TableModel tableModel) {
  158. ClassModel classModel = new ClassModel();
  159. classModel.setName(entityPkg + "." + Utils.getTableName(tableModel.getTableName()))
  160. .setQualifier(Qualifier.publicQ).setClassOrInterface(ClassOrInterface.classC);
  161. classModel
  162. .addField(
  163. new FieldModel().setQualifier(Qualifier.privateQ)
  164. .setFieldType(new ClassTypeModel(classModel).setClassName(
  165. Utils.dataTypeMapping.get(tableModel.getPrimaryKey().getDataType())))
  166. .setFieldName(Utils.getColumnName(tableModel.getPrimaryKey().getColumnName()))
  167. .setClassModel(classModel).generateGetMethod().generateSetMethod());
  168. tableModel.getColumns()
  169. .forEach(
  170. c -> classModel.addField(new FieldModel().setQualifier(Qualifier.privateQ)
  171. .setFieldType(new ClassTypeModel(classModel)
  172. .setClassName(Utils.dataTypeMapping.get(c.getDataType())))
  173. .setFieldName(Utils.getColumnName(c.getColumnName())).setClassModel(classModel)
  174. .generateGetMethod().generateSetMethod()));
  175. return classModel.toString();
  176. }
  177. public static String getMapperXmlCode(TableModel tableModel) {
  178. String result = "<?xml version=\"1.0\" encoding=\"UTF-8\" ?>\n<!DOCTYPE mapper PUBLIC \"-//mybatis.org//DTD Mapper 3.0//EN\" \"http://mybatis.org/dtd/mybatis-3-mapper.dtd\" >\n<mapper namespace=\"#{mapperPkg}.#{entityName}Mapper\">\n\n\t<sql id=\"columns\">\n\t\t#{columns}\n\t</sql>\n\n\t<sql id=\"columns-place-holder\">\n\t\t#{columnsPlaceHolder}\n\t</sql>\n\n\t<resultMap type=\"#{entityPkg}.#{entityName}\" id=\"#{lowerEntityName}Entity\">\n\t\t#{primaryKeyColumnMap}\n#{otherColumnsMap}\t</resultMap>\n\t<select id=\"queryById\" resultMap=\"#{lowerEntityName}Entity \">\n\t\tselect <include refid=\"columns\"></include>\n\t\tfrom #{tableName}\n\t\twhere #{dbPrimaryKey}=#{javaPrimaryKey}\n\t</select>\n\n\t<select id=\"queryAll\" resultMap=\"#{lowerEntityName}Entity\">\n\t\tselect <include refid=\"columns\"></include>\n\t\tfrom #{tableName}\n\t</select>\n\n\t<insert id=\"add\">\n\t\tinsert into #{tableName} (<include refid=\"columns\"></include>)\n\t\tvalues(<include refid=\"columns-place-holder\"></include>)\n\t</insert>\n\n\t<update id=\"update\" parameterType=\"#{entityPkg}.#{entityName}\">\n\t\tupdate #{tableName}\n\t\tset #{dbPrimaryKey} = #{javaPrimaryKey}\n#{updateColumnMap}\n\t\twhere #{dbPrimaryKey} = #{javaPrimaryKey}\n\t</update>\n\n\t<delete id=\"delete\">\n\t\tdelete from #{tableName} where #{dbPrimaryKey} = #{javaPrimaryKey}\n\t</delete>\n\n</mapper>";
  179. result = result.replace("#{mapperPkg}", mapperPkg);
  180. result = result.replaceAll("\\#\\{entityName\\}", Utils.getTableName(tableModel.getTableName()));
  181. StringBuilder sbColumns = new StringBuilder();
  182. StringBuilder sbColumnsPlaceHolder = new StringBuilder();
  183. StringBuilder sbResultMap = new StringBuilder();
  184. StringBuilder sbUpdateMap = new StringBuilder();
  185. sbColumns.append(tableModel.getPrimaryKey().getColumnName()).append(", ");
  186. sbColumnsPlaceHolder.append("#{").append(Utils.getColumnName(tableModel.getPrimaryKey().getColumnName()))
  187. .append("}, ");
  188. tableModel.getColumns().forEach(c -> {
  189. sbColumns.append(c.getColumnName()).append(", ");
  190. sbColumnsPlaceHolder.append("#{").append(Utils.getColumnName(c.getColumnName())).append("}, ");
  191. sbResultMap.append("\t\t<result column=\"" + c.getColumnName() + "\" property=\""
  192. + Utils.getColumnName(c.getColumnName()) + "\" />\n");
  193. sbUpdateMap.append("\n\t\t<if test=\"" + Utils.getColumnName(c.getColumnName()) + " != null\">\n\t\t\t,"
  194. + c.getColumnName() + " = #{" + Utils.getColumnName(c.getColumnName()) + "}\n\t\t</if>");
  195. });
  196. result = result.replace("#{columns}", sbColumns.subSequence(0, sbColumns.lastIndexOf(",")));
  197. result = result.replace("#{columnsPlaceHolder}",
  198. sbColumnsPlaceHolder.subSequence(0, sbColumnsPlaceHolder.lastIndexOf(",")));
  199. result = result.replaceAll("\\#\\{entityPkg\\}", entityPkg);
  200. result = result.replaceAll("\\#\\{lowerEntityName\\}",
  201. Utils.firstLower(Utils.getTableName(tableModel.getTableName())));
  202. result = result.replace("#{primaryKeyColumnMap}", "<id column=\"" + tableModel.getPrimaryKey().getColumnName()
  203. + "\" property=\"" + Utils.getColumnName(tableModel.getPrimaryKey().getColumnName()) + "\" />");
  204. result = result.replace("#{otherColumnsMap}", sbResultMap.toString());
  205. result = result.replaceAll("\\#\\{dbPrimaryKey\\}", tableModel.getPrimaryKey().getColumnName());
  206. result = result.replaceAll("\\#\\{javaPrimaryKey\\}",
  207. "#{" + Utils.getColumnName(tableModel.getPrimaryKey().getColumnName()) + "}");
  208. result = result.replaceAll("\\#\\{tableName\\}", tableModel.getTableName());
  209. result = result.replace("#{updateColumnMap}", sbUpdateMap.toString());
  210. return result;
  211. }
  212. public static void generateDaoInterface(String tableName) {
  213. String code = getDaoInterfaceCode(Utils.getTableName(tableName));
  214. String path = daoPath + "\\src\\main\\java\\com\\alan\\dao\\" + Utils.getTableName(tableName) + "Dao.java";
  215. createFile(path, code);
  216. }
  217. public static void generateDaoImpl(String tableName) {
  218. String code = getDaoImplCode(Utils.getTableName(tableName));
  219. String path = daoPath + "\\src\\main\\java\\com\\alan\\dao\\impl\\" + Utils.getTableName(tableName)
  220. + "DaoImpl.java";
  221. createFile(path, code);
  222. }
  223. public static void generateMapper(String tableName) {
  224. String code = getMapperCode(Utils.getTableName(tableName));
  225. String path = daoPath + "\\src\\main\\java\\com\\alan\\mapper\\" + Utils.getTableName(tableName)
  226. + "Mapper.java";
  227. createFile(path, code);
  228. }
  229. public static void generateService(String tableName) {
  230. String code = getServiceCode(Utils.getTableName(tableName));
  231. String path = servicePath + "\\src\\main\\java\\com\\alan\\service\\" + Utils.getTableName(tableName)
  232. + "Service.java";
  233. createFile(path, code);
  234. }
  235. public static void generateServiceImpl(String tableName) {
  236. String code = getServiceImplCode(Utils.getTableName(tableName));
  237. String path = servicePath + "\\src\\main\\java\\com\\alan\\service\\impl\\" + Utils.getTableName(tableName)
  238. + "ServiceImpl.java";
  239. createFile(path, code);
  240. }
  241. public static void generateController(String tableName) {
  242. String code = getControllerCode(Utils.getTableName(tableName));
  243. String path = webPath + "\\src\\main\\java\\com\\alan\\controller\\" + Utils.getTableName(tableName)
  244. + "Controller.java";
  245. createFile(path, code);
  246. }
  247. public static void generateEntity(TableModel tableModel) {
  248. String code = getEntityCode(tableModel);
  249. String path = daoPath + "\\src\\main\\java\\com\\alan\\entity\\" + Utils.getTableName(tableModel.getTableName())
  250. + ".java";
  251. createFile(path, code);
  252. }
  253. public static void generateMapperXml(TableModel tableModel) {
  254. String code = getMapperXmlCode(tableModel);
  255. String path = daoPath + "\\src\\main\\resources\\mapper\\" + Utils.getTableName(tableModel.getTableName())
  256. + "Mapper.xml";
  257. createFile(path, code);
  258. }
  259. public static void createFile(String path, String code) {
  260. File file = new File(path);
  261. try {
  262. file.createNewFile();
  263. FileOutputStream output = new FileOutputStream(file);
  264. output.write(code.getBytes());
  265. output.close();
  266. writeHistory(path);
  267. } catch (IOException e) {
  268. e.printStackTrace();
  269. }
  270. }
  271. public static void writeHistory(String path) {
  272. try {
  273. if (!historyFile.exists()) {
  274. historyFile.createNewFile();
  275. }
  276. FileWriter writer = new FileWriter(historyFile, flag);
  277. writer.write(path + "\n");
  278. writer.close();
  279. flag = true;
  280. } catch (Exception e) {
  281. e.printStackTrace();
  282. }
  283. }
  284. public static void deleteFile(String path) {
  285. File file = new File(path);
  286. file.delete();
  287. }
  288. public static void deleteFilesJustGenerated() {
  289. try {
  290. if (historyFile.exists()) {
  291. BufferedReader reader = new BufferedReader(new FileReader(historyFile));
  292. String line = null;
  293. while ((line = reader.readLine()) != null) {
  294. deleteFile(line);
  295. }
  296. reader.close();
  297. historyFile.delete();
  298. }
  299. } catch (Exception e) {
  300. e.printStackTrace();
  301. }
  302. }
  303. public static void generate(List<String> layers, List<String> tables) {
  304. tables.forEach(t -> {
  305. layers.forEach(l -> {
  306. if (l.equals("Mapper")) {
  307. TableModel tableModel = JdbcHelper.getTableInfo(t);
  308. generateEntity(tableModel);
  309. generateMapperXml(tableModel);
  310. generateMapper(t);
  311. }
  312. if (l.equals("Dao")) {
  313. generateDaoInterface(t);
  314. generateDaoImpl(t);
  315. }
  316. if (l.equals("Service")) {
  317. generateService(t);
  318. generateServiceImpl(t);
  319. }
  320. if (l.equals("Controller")) {
  321. generateController(t);
  322. }
  323. });
  324. });
  325. };
  326. public static void main(String[] args) throws IOException {
  327. // testGenerateMapXml();
  328. // deleteFile("E:\\workspace\\my-springmvc\\alan-springmvc\\alan-springmvc-dao\\src\\main\\resources\\mapper\\UserAddressMapper.xml");
  329. deleteFilesJustGenerated();
  330. }
  331. public static void testGenerateMapXml() {
  332. TableModel tableModel = new TableModel();
  333. tableModel.setTableName("user_address_mobile");
  334. tableModel.setPrimaryKey(new ColumnModel().setColumnName("user_id").setDataType("bigint"));
  335. tableModel.addColumn(new ColumnModel().setColumnName("user_name").setDataType("varchar"));
  336. tableModel.addColumn(new ColumnModel().setColumnName("mobile").setDataType("varchar"));
  337. tableModel.addColumn(new ColumnModel().setColumnName("user_age").setDataType("int"));
  338. generateMapperXml(tableModel);
  339. generateEntity(tableModel);
  340. }
  341. }

(6)数据库连接帮助类
JdbcHelper.java

  1. package com.alan.generate.utils;
  2. import java.sql.Connection;
  3. import java.sql.DriverManager;
  4. import java.sql.PreparedStatement;
  5. import java.sql.ResultSet;
  6. import java.sql.SQLException;
  7. import java.util.ArrayList;
  8. import java.util.List;
  9. import com.alan.generate.main.Constant;
  10. import com.alan.generate.model.ColumnModel;
  11. import com.alan.generate.model.TableModel;
  12. public class JdbcHelper {
  13. private static String driverClassName = Constant.driverClassName;
  14. private static String url = Constant.url;
  15. private static String username = Constant.username;
  16. private static String password = Constant.password;
  17. static {
  18. try {
  19. Class.forName(driverClassName);
  20. } catch (ClassNotFoundException e) {
  21. e.printStackTrace();
  22. }
  23. }
  24. public static Connection getConnection() {
  25. try {
  26. return DriverManager.getConnection(url, username, password);
  27. } catch (SQLException e) {
  28. e.printStackTrace();
  29. }
  30. return null;
  31. }
  32. public static TableModel getTableInfo(String tableName) {
  33. if (Utils.isEmpty(tableName))
  34. return null;
  35. TableModel tableModel = new TableModel(tableName);
  36. String sql = "select column_name, data_type, column_key from information_schema.columns where table_name = ? order by ordinal_position";
  37. ResultSet rs = query(sql, tableName);
  38. try {
  39. while (rs.next()) {
  40. if (Utils.isNotEmpty(rs.getString(3)) && "PRI".equals(rs.getString(3))) {
  41. tableModel.setPrimaryKey(
  42. new ColumnModel().setColumnName(rs.getString(1)).setDataType(rs.getString(2)));
  43. } else {
  44. tableModel.addColumn(new ColumnModel().setColumnName(rs.getString(1)).setDataType(rs.getString(2)));
  45. }
  46. }
  47. } catch (SQLException e) {
  48. e.printStackTrace();
  49. }
  50. return tableModel;
  51. }
  52. public static List<String> getAllTables(String tableSchema) {
  53. List<String> list = new ArrayList<>();
  54. if (Utils.isEmpty(tableSchema))
  55. return list;
  56. String sql = "select table_name from information_schema.tables where table_schema = ? order by table_name";
  57. ResultSet rs = query(sql, tableSchema);
  58. try {
  59. while (rs.next()) {
  60. list.add(rs.getString(1));
  61. }
  62. } catch (SQLException e) {
  63. e.printStackTrace();
  64. }
  65. return list;
  66. }
  67. public static List<String> getTablesByFilter(String tableSchema, String filter) {
  68. if (filter == null)
  69. filter = "";
  70. List<String> list = new ArrayList<>();
  71. String sql = "select table_name from information_schema.tables where table_schema = ? and table_name like ? order by table_name";
  72. ResultSet rs = query(sql, tableSchema, "%" + filter + "%");
  73. try {
  74. while (rs.next()) {
  75. list.add(rs.getString(1));
  76. }
  77. } catch (SQLException e) {
  78. e.printStackTrace();
  79. }
  80. return list;
  81. }
  82. public static List<String> getAllSchema() {
  83. List<String> list = new ArrayList<>();
  84. String sql = "select schema_name from information_schema.schemata order by schema_name";
  85. ResultSet rs = query(sql);
  86. try {
  87. while (rs.next()) {
  88. list.add(rs.getString(1));
  89. }
  90. } catch (SQLException e) {
  91. e.printStackTrace();
  92. }
  93. return list;
  94. }
  95. private static ResultSet query(String sql, Object... objects) {
  96. try {
  97. PreparedStatement ps = getConnection().prepareStatement(sql);
  98. if (objects != null && objects.length != 0) {
  99. for (int i = 0; i < objects.length; i++) {
  100. ps.setObject(i + 1, objects[i]);
  101. }
  102. }
  103. return ps.executeQuery();
  104. } catch (SQLException e) {
  105. e.printStackTrace();
  106. }
  107. return null;
  108. }
  109. }

(7)生成代码的UI界面
GenerateUI.java

  1. package com.alan.generate.utils;
  2. import java.awt.Component;
  3. import java.awt.GridBagConstraints;
  4. import java.awt.GridBagLayout;
  5. import java.awt.event.ActionEvent;
  6. import java.awt.event.ActionListener;
  7. import java.util.ArrayList;
  8. import java.util.HashMap;
  9. import java.util.List;
  10. import java.util.Map;
  11. import javax.swing.Box;
  12. import javax.swing.BoxLayout;
  13. import javax.swing.JButton;
  14. import javax.swing.JCheckBox;
  15. import javax.swing.JComboBox;
  16. import javax.swing.JFrame;
  17. import javax.swing.JLabel;
  18. import javax.swing.JOptionPane;
  19. import javax.swing.JPanel;
  20. import javax.swing.JScrollPane;
  21. import javax.swing.JTable;
  22. import javax.swing.JTextField;
  23. import javax.swing.border.EmptyBorder;
  24. import javax.swing.event.DocumentEvent;
  25. import javax.swing.event.DocumentListener;
  26. public class GenerateUI extends JFrame {
  27. private static final long serialVersionUID = 1L;
  28. private JComboBox<String> schemaList;
  29. private JTextField filterText;
  30. private JPanel contentPanel;
  31. private JPanel topPanel;
  32. private JPanel middlePanel;
  33. private JPanel bottomPanel;
  34. private JPanel tablePanel;
  35. private JCheckBox mapperCheckBox;
  36. private JCheckBox daoCheckBox;
  37. private JCheckBox serviceCheckBox;
  38. private JCheckBox controllerCheckBox;
  39. private JCheckBox allCheckBox;
  40. private Map<String, Boolean> bottomCheckBoxSelected = new HashMap<>();
  41. private Map<String, Boolean> tableCheckBoxSelected = new HashMap<>();
  42. private TableCheckBoxListener tableCheckBoxListener = new TableCheckBoxListener();
  43. public void getUI() {
  44. createTopPanel();
  45. createMiddlePanel();
  46. createBottomPanel();
  47. this.setTitle("自动生成代码");
  48. this.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
  49. this.setBounds(400, 250, 500, 600);
  50. contentPanel = new JPanel();
  51. contentPanel.setBorder(new EmptyBorder(5, 5, 5, 5));
  52. this.setContentPane(contentPanel);
  53. contentPanel.setLayout(new GridBagLayout());
  54. GridBagConstraints c1 = new GridBagConstraints();
  55. c1.gridx = 0;
  56. c1.gridy = 0;
  57. c1.weightx = 1.0;
  58. c1.weighty = 0;
  59. c1.fill = GridBagConstraints.HORIZONTAL;
  60. contentPanel.add(topPanel, c1);
  61. GridBagConstraints c2 = new GridBagConstraints();
  62. c2.gridx = 0;
  63. c2.gridy = 1;
  64. c2.weightx = 1.0;
  65. c2.weighty = 1.0;
  66. c2.fill = GridBagConstraints.BOTH;
  67. // 加入 middlePanel
  68. contentPanel.add(middlePanel, c2);
  69. GridBagConstraints c3 = new GridBagConstraints();
  70. c3.gridx = 0;
  71. c3.gridy = 2;
  72. c3.weightx = 1.0;
  73. c3.weighty = 0;
  74. c3.fill = GridBagConstraints.HORIZONTAL;
  75. // 加入 bottomPanel
  76. contentPanel.add(bottomPanel, c3);
  77. this.setVisible(true);
  78. }
  79. public Component getSchemaLabel() {
  80. return new JLabel("schema: ");
  81. }
  82. public Component getSchemaList() {
  83. schemaList = new JComboBox<>();
  84. schemaList.setLightWeightPopupEnabled(false);
  85. schemaList.addItem("");
  86. JdbcHelper.getAllSchema().forEach(s -> schemaList.addItem(s));
  87. schemaList.addActionListener(new ActionListener() {
  88. @SuppressWarnings("unchecked")
  89. @Override
  90. public void actionPerformed(ActionEvent e) {
  91. String selectedItem = ((JComboBox<String>) (e.getSource())).getSelectedItem().toString().trim();
  92. updateTableList(selectedItem);
  93. if (selectedItem.length() == 0) {
  94. filterText.setEditable(false);
  95. } else {
  96. filterText.setEditable(true);
  97. }
  98. filterText.setText(null);
  99. }
  100. });
  101. return schemaList;
  102. }
  103. public Component getFilterLabel() {
  104. return new JLabel("filter: ");
  105. }
  106. public Component getFilterText() {
  107. filterText = new JTextField(20);
  108. filterText.setEditable(false);
  109. filterText.getDocument().addDocumentListener(new DocumentListener() {
  110. private void change() {
  111. updateTableListByFilter(schemaList.getSelectedItem().toString().trim(), filterText.getText().trim());
  112. }
  113. @Override
  114. public void removeUpdate(DocumentEvent e) {
  115. change();
  116. }
  117. @Override
  118. public void insertUpdate(DocumentEvent e) {
  119. change();
  120. }
  121. @Override
  122. public void changedUpdate(DocumentEvent e) {
  123. change();
  124. }
  125. });
  126. return filterText;
  127. }
  128. public void createTopPanel() {
  129. topPanel = new JPanel();
  130. topPanel.setLayout(new BoxLayout(topPanel, BoxLayout.Y_AXIS));
  131. JPanel linePanel = new JPanel();
  132. linePanel.setLayout(new BoxLayout(linePanel, BoxLayout.X_AXIS));
  133. linePanel.add(getSchemaLabel());
  134. linePanel.add(getSchemaList());
  135. linePanel.add(Box.createHorizontalGlue());
  136. linePanel.add(getFilterLabel());
  137. linePanel.add(getFilterText());
  138. topPanel.add(Box.createVerticalStrut(10));
  139. topPanel.add(linePanel);
  140. topPanel.add(Box.createVerticalStrut(10));
  141. }
  142. public void createMiddlePanel() {
  143. middlePanel = new JPanel();
  144. middlePanel.setLayout(new BoxLayout(middlePanel, BoxLayout.Y_AXIS));
  145. tablePanel = new JPanel();
  146. tablePanel.setLayout(new BoxLayout(tablePanel, BoxLayout.Y_AXIS));
  147. JScrollPane scrollPane = new JScrollPane(tablePanel);
  148. scrollPane.setVerticalScrollBarPolicy(JScrollPane.VERTICAL_SCROLLBAR_ALWAYS);
  149. middlePanel.add(scrollPane);
  150. }
  151. public void createBottomPanel() {
  152. bottomPanel = new JPanel();
  153. bottomPanel.setLayout(new BoxLayout(bottomPanel, BoxLayout.Y_AXIS));
  154. JPanel linePanel = new JPanel();
  155. linePanel.setLayout(new BoxLayout(linePanel, BoxLayout.X_AXIS));
  156. BottomCheckBoxListener cBoxListener = new BottomCheckBoxListener();
  157. mapperCheckBox = new JCheckBox("Mapper");
  158. daoCheckBox = new JCheckBox("Dao");
  159. serviceCheckBox = new JCheckBox("Service");
  160. controllerCheckBox = new JCheckBox("Controller");
  161. allCheckBox = new JCheckBox("All");
  162. mapperCheckBox.addActionListener(cBoxListener);
  163. daoCheckBox.addActionListener(cBoxListener);
  164. serviceCheckBox.addActionListener(cBoxListener);
  165. controllerCheckBox.addActionListener(cBoxListener);
  166. allCheckBox.addActionListener(cBoxListener);
  167. linePanel.add(mapperCheckBox);
  168. linePanel.add(Box.createHorizontalStrut(20));
  169. linePanel.add(daoCheckBox);
  170. linePanel.add(Box.createHorizontalStrut(20));
  171. linePanel.add(serviceCheckBox);
  172. linePanel.add(Box.createHorizontalStrut(20));
  173. linePanel.add(controllerCheckBox);
  174. linePanel.add(Box.createHorizontalStrut(20));
  175. linePanel.add(allCheckBox);
  176. linePanel.add(Box.createHorizontalStrut(140));
  177. JPanel linePanel2 = new JPanel();
  178. linePanel2.setLayout(new BoxLayout(linePanel2, BoxLayout.X_AXIS));
  179. linePanel2.add(new JLabel("注:生成Mapper时会重新生成Entity、Mapper.java及Mapper.xml文件!"));
  180. JButton button = new JButton("生成");
  181. linePanel2.add(button);
  182. button.addActionListener(new ActionListener() {
  183. @Override
  184. public void actionPerformed(ActionEvent e) {
  185. List<String> layers = new ArrayList<>();
  186. List<String> tables = new ArrayList<>();
  187. bottomCheckBoxSelected.forEach((id, val) -> {
  188. if (val)
  189. layers.add(id);
  190. });
  191. if (layers.size() == 0) {
  192. JOptionPane.showMessageDialog(null, "您还没有选择生成哪些层!");
  193. return;
  194. }
  195. tableCheckBoxSelected.forEach((id, val) -> {
  196. if (val)
  197. tables.add(id);
  198. });
  199. if (tables.size() == 0) {
  200. JOptionPane.showMessageDialog(null, "您还没有选择表!");
  201. return;
  202. }
  203. int response = JOptionPane.showConfirmDialog(null,
  204. "您选择的信息如下:\n层:" + Utils.join(layers, ", ") + "\n表:" + Utils.join(tables, ", "), "确定生成代码?",
  205. JOptionPane.YES_NO_OPTION);
  206. if (response == 0) {
  207. GenerateHelper.generate(layers, tables);
  208. JOptionPane.showMessageDialog(null,
  209. "恭喜您,生成成功,详细信息见:" + GenerateHelper.daoPath + "\\src\\main\\resources\\history.txt");
  210. }
  211. }
  212. });
  213. JPanel linePanel3 = new JPanel();
  214. linePanel3.setLayout(new BoxLayout(linePanel3, BoxLayout.X_AXIS));
  215. linePanel3.add(new JLabel("其它功能:"));
  216. JButton button1 = new JButton("删除刚生成的文件");
  217. linePanel3.add(button1);
  218. linePanel3.add(Box.createHorizontalStrut(500));
  219. button1.addActionListener(new ActionListener() {
  220. @Override
  221. public void actionPerformed(ActionEvent e) {
  222. int response = JOptionPane.showConfirmDialog(null, "您确定要删除刚才生成的代码?删除后将无法恢复!!", "确定删除?",
  223. JOptionPane.YES_NO_OPTION);
  224. if (response == 0) {
  225. GenerateHelper.deleteFilesJustGenerated();
  226. JOptionPane.showMessageDialog(null, "删除完毕!");
  227. }
  228. }
  229. });
  230. JPanel verticalPanel = new JPanel();
  231. verticalPanel.setLayout(new BoxLayout(verticalPanel, BoxLayout.Y_AXIS));
  232. verticalPanel.add(Box.createVerticalStrut(10));
  233. verticalPanel.add(linePanel);
  234. verticalPanel.add(Box.createVerticalStrut(10));
  235. verticalPanel.add(linePanel2);
  236. verticalPanel.add(Box.createVerticalStrut(20));
  237. verticalPanel.add(linePanel3);
  238. verticalPanel.add(Box.createVerticalStrut(20));
  239. bottomPanel.add(verticalPanel);
  240. }
  241. public void updateTableList(String tableSchema) {
  242. tablePanel.removeAll();
  243. tableCheckBoxSelected.clear();
  244. JdbcHelper.getAllTables(tableSchema).forEach(s -> {
  245. JCheckBox checkBox = new JCheckBox(s);
  246. checkBox.addActionListener(tableCheckBoxListener);
  247. tablePanel.add(checkBox);
  248. });
  249. contentPanel.updateUI();
  250. contentPanel.repaint();
  251. }
  252. public void updateTableListByFilter(String tableSchema, String filter) {
  253. tablePanel.removeAll();
  254. JdbcHelper.getTablesByFilter(tableSchema, filter).forEach(s -> {
  255. Boolean selected = tableCheckBoxSelected.get(s);
  256. if (selected == null)
  257. selected = false;
  258. JCheckBox checkBox = new JCheckBox(s, selected);
  259. checkBox.addActionListener(tableCheckBoxListener);
  260. tablePanel.add(checkBox);
  261. });
  262. contentPanel.updateUI();
  263. contentPanel.repaint();
  264. }
  265. public Component getTableView() {
  266. JTable jTable = new JTable();
  267. return jTable;
  268. }
  269. public static void main(String[] args) {
  270. new GenerateUI().getUI();
  271. }
  272. class BottomCheckBoxListener implements ActionListener {
  273. @Override
  274. public void actionPerformed(ActionEvent e) {
  275. JCheckBox checkBox = (JCheckBox) e.getSource();
  276. if ("All".equals(checkBox.getText())) {
  277. if (checkBox.isSelected()) {
  278. mapperCheckBox.setSelected(true);
  279. daoCheckBox.setSelected(true);
  280. serviceCheckBox.setSelected(true);
  281. controllerCheckBox.setSelected(true);
  282. bottomCheckBoxSelected.put(mapperCheckBox.getText(), true);
  283. bottomCheckBoxSelected.put(daoCheckBox.getText(), true);
  284. bottomCheckBoxSelected.put(serviceCheckBox.getText(), true);
  285. bottomCheckBoxSelected.put(controllerCheckBox.getText(), true);
  286. } else {
  287. mapperCheckBox.setSelected(false);
  288. daoCheckBox.setSelected(false);
  289. serviceCheckBox.setSelected(false);
  290. controllerCheckBox.setSelected(false);
  291. bottomCheckBoxSelected.put(mapperCheckBox.getText(), false);
  292. bottomCheckBoxSelected.put(daoCheckBox.getText(), false);
  293. bottomCheckBoxSelected.put(serviceCheckBox.getText(), false);
  294. bottomCheckBoxSelected.put(controllerCheckBox.getText(), false);
  295. }
  296. return;
  297. }
  298. bottomCheckBoxSelected.put(checkBox.getText(), checkBox.isSelected());
  299. }
  300. }
  301. class TableCheckBoxListener implements ActionListener {
  302. @Override
  303. public void actionPerformed(ActionEvent e) {
  304. JCheckBox checkBox = (JCheckBox) e.getSource();
  305. tableCheckBoxSelected.put(checkBox.getText(), checkBox.isSelected());
  306. }
  307. }
  308. }

###41. 使用说明

(1)代码可以放在dao/service/controller随便哪个工程下,或放在另外一个与上述工程同级的工程下,我这里是放在spring-common工程下
(2)修改配置文件com.alan.generate.main.Constant.java
(3)运行com.alan.generate.main.Main.java
(4)点击schema后面的下拉列表选择相应的库
(5)可以通过filter筛选
(6)选择生成的层,可通过All选择所有层
(7)点击生成
(8)查看结果
(9)如果生成的类不想要了,可以点击最下方的按钮删除刚才生成的文件
(10)使用过程中有任何问题,请在下方评论,谢谢~~
这里写图片描述

##八、自动生成接口文档

###42. 思路
(1)首先要定义一个页面显示我们的接口文档,一般结构是左边的接口列表,右边是具体的接口信息,点击左右的接口,右边跟着变,类似下面这种结构:
这里写图片描述
(2)上述页面调用接口获取数据,假如叫/getApi;
(3)每个接口初始化的时候把自己的信息加到一个全局变量中保存起来,主要包括接口url、接口描述、接口参数、接口返回值等;
(4)调用/getApi时把上述全局变量中的数据返回;
(5)关键在于,怎么初始化的时候获取接口的信息?这个问题想到了就很简单,在父Controller的默认构造方法中通过反射获取子Controller的方法,然后解析出相应的信息即可,因为子类在初始化的时候会先调用父类的默认构造方法。

###43. 限制与要求
(1)通过Java的反射无法获取参数的名称,所以我们的所有参数都需要封装在Model中;(通过javassist可以搞,不过我们这里直接这样定了,这样更明确)
(2)为了便于前端解析,所有的接口返回统一的Model,假如叫RestResult,包括code/message/result三个字段,其中result是Object类型,放入具体的返回值;
(3)有些参数是必填参数,添加@NotNull注解;
(4)需要显示接口描述、参数描述、返回值描述,添加@Description注解;
(5)有些接口不需要生成接口文档,添加@SuppressApi注解;
(6)具体的返回值放在了RestResult的result中,无法获取具体类型,添加@RestReturn指定具体的返回类型;

###43. 代码实现
注:此处是用纯java写的,后面博主采用了freemarker模板来写,事半功倍,有兴趣的同学可以去学习一下。
生成接口文档相关代码放在alan-springmvc-common工程的com.alan.api包下。
(1)定义几个注解类型
新建包com.alan.api.annotation。
RestReturn.java

  1. package com.alan.api.annotation;
  2. import java.lang.annotation.Documented;
  3. import java.lang.annotation.ElementType;
  4. import java.lang.annotation.Retention;
  5. import java.lang.annotation.RetentionPolicy;
  6. import java.lang.annotation.Target;
  7. /** * 用于生成接口文档,标记返回类型 * @author Alan * */
  8. @Target(ElementType.METHOD)
  9. @Retention(RetentionPolicy.RUNTIME)
  10. @Documented
  11. public @interface RestReturn {
  12. Class<?> value();
  13. boolean collection() default false;
  14. }

NotNull.java

  1. package com.alan.api.annotation;
  2. import java.lang.annotation.Documented;
  3. import java.lang.annotation.ElementType;
  4. import java.lang.annotation.Retention;
  5. import java.lang.annotation.RetentionPolicy;
  6. import java.lang.annotation.Target;
  7. @Target({ ElementType.FIELD, ElementType.PARAMETER })
  8. @Retention(RetentionPolicy.RUNTIME)
  9. @Documented
  10. public @interface NotNull {
  11. }

Description.java

  1. package com.alan.api.annotation;
  2. import java.lang.annotation.Documented;
  3. import java.lang.annotation.ElementType;
  4. import java.lang.annotation.Retention;
  5. import java.lang.annotation.RetentionPolicy;
  6. import java.lang.annotation.Target;
  7. @Target({ ElementType.FIELD, ElementType.METHOD, ElementType.PARAMETER})
  8. @Retention(RetentionPolicy.RUNTIME)
  9. @Documented
  10. public @interface Description {
  11. String value() default "";
  12. }

Suppress.java

  1. package com.alan.api.annotation;
  2. import java.lang.annotation.Documented;
  3. import java.lang.annotation.ElementType;
  4. import java.lang.annotation.Retention;
  5. import java.lang.annotation.RetentionPolicy;
  6. import java.lang.annotation.Target;
  7. @Target({ ElementType.FIELD,ElementType.METHOD})
  8. @Retention(RetentionPolicy.RUNTIME)
  9. @Documented
  10. public @interface SuppressApi {
  11. }

(2)定义几个Model
新建包com.alan.api.rest。
RestParam.java

  1. package com.alan.api.rest;
  2. /** * 记录接口的参数信息 * @author Alan * */
  3. public class RestParam {
  4. private String type;
  5. private String name;
  6. private String description;
  7. private Boolean required = false;
  8. public String getType() {
  9. return type;
  10. }
  11. public void setType(String type) {
  12. this.type = type;
  13. }
  14. public String getName() {
  15. return name;
  16. }
  17. public void setName(String name) {
  18. this.name = name;
  19. }
  20. public String getDescription() {
  21. return description;
  22. }
  23. public void setDescription(String description) {
  24. this.description = description;
  25. }
  26. public Boolean getRequired() {
  27. return required;
  28. }
  29. public void setRequired(Boolean required) {
  30. this.required = required;
  31. }
  32. }

RestReturnParam.java

  1. package com.alan.api.rest;
  2. /** * 接口的返回参数信息 * @author Alan * */
  3. public class RestReturnParam {
  4. private String type;
  5. private String name;
  6. private String description;
  7. public String getType() {
  8. return type;
  9. }
  10. public void setType(String type) {
  11. this.type = type;
  12. }
  13. public String getName() {
  14. return name;
  15. }
  16. public void setName(String name) {
  17. this.name = name;
  18. }
  19. public String getDescription() {
  20. return description;
  21. }
  22. public void setDescription(String description) {
  23. this.description = description;
  24. }
  25. }

RestMethod.java

  1. package com.alan.api.rest;
  2. import java.util.ArrayList;
  3. import java.util.List;
  4. /** * 用于记录每个接口的信息 * @author Alan * */
  5. public class RestMethod {
  6. private String url;
  7. private String description;
  8. private List<RestParam> params = new ArrayList<>();
  9. private List<RestReturnParam> returnParams = new ArrayList<>();
  10. public String getUrl() {
  11. return url;
  12. }
  13. public void setUrl(String url) {
  14. this.url = url;
  15. }
  16. public String getDescription() {
  17. return description;
  18. }
  19. public void setDescription(String description) {
  20. this.description = description;
  21. }
  22. public List<RestParam> getParams() {
  23. return params;
  24. }
  25. public void addParam(RestParam param) {
  26. this.params.add(param);
  27. }
  28. public List<RestReturnParam> getReturnParams() {
  29. return returnParams;
  30. }
  31. public void addReturnParam(RestReturnParam returnParam) {
  32. this.returnParams.add(returnParam);
  33. }
  34. }

RestResult.java

  1. package com.alan.api.rest;
  2. /** * 统一的返回值 * @author Alan * */
  3. public class RestResult {
  4. private Integer code;
  5. private String message;
  6. private Object result;
  7. public RestResult() {
  8. this.code = 0;
  9. this.message = "success";
  10. }
  11. public RestResult(Object result) {
  12. this();
  13. this.result = result;
  14. }
  15. public Integer getCode() {
  16. return code;
  17. }
  18. public void setCode(Integer code) {
  19. this.code = code;
  20. }
  21. public String getMessage() {
  22. return message;
  23. }
  24. public void setMessage(String message) {
  25. this.message = message;
  26. }
  27. public Object getResult() {
  28. return result;
  29. }
  30. public void setResult(Object result) {
  31. this.result = result;
  32. }
  33. }

(3)定义生成接口文档的父Controller
新建包com.alan.api.controller。
BaseApiController.java

  1. package com.alan.api.controller;
  2. import java.lang.annotation.Annotation;
  3. import java.lang.reflect.Field;
  4. import java.lang.reflect.Method;
  5. import java.lang.reflect.ParameterizedType;
  6. import java.lang.reflect.Type;
  7. import java.sql.Timestamp;
  8. import java.util.ArrayList;
  9. import java.util.Date;
  10. import java.util.List;
  11. import javax.servlet.http.HttpServletRequest;
  12. import javax.servlet.http.HttpServletResponse;
  13. import org.springframework.ui.Model;
  14. import org.springframework.web.bind.annotation.RequestMapping;
  15. import org.springframework.web.servlet.ModelAndView;
  16. import com.alan.api.annotation.Description;
  17. import com.alan.api.annotation.NotNull;
  18. import com.alan.api.annotation.RestReturn;
  19. import com.alan.api.annotation.SuppressApi;
  20. import com.alan.api.rest.RestMethod;
  21. import com.alan.api.rest.RestParam;
  22. import com.alan.api.rest.RestReturnParam;
  23. import com.alan.generate.utils.Utils;
  24. /** * 用于生成接口文档的父Controller * @author Alan * */
  25. public class BaseApiController {
  26. private static final int LIST_REPEAT = 4;
  27. private static final List<RestMethod> REST_LIST = new ArrayList<>();
  28. public BaseApiController() {
  29. REST_LIST.addAll(findRestMethod(this.getClass()));
  30. }
  31. public static List<RestMethod> getRestList() {
  32. return REST_LIST;
  33. }
  34. private List<RestMethod> findRestMethod(Class<? extends BaseApiController> clazz) {
  35. List<RestMethod> restList = new ArrayList<>();
  36. String root = "/";
  37. RequestMapping classRequestMapping = clazz.getAnnotation(RequestMapping.class);
  38. if (classRequestMapping != null && classRequestMapping.value() != null) {
  39. root = classRequestMapping.value()[0];
  40. }
  41. Method[] methods = clazz.getMethods();
  42. for (Method method : methods) {
  43. RequestMapping requestMapping = method.getAnnotation(RequestMapping.class);
  44. SuppressApi suppressApi = method.getAnnotation(SuppressApi.class);
  45. if (requestMapping != null && suppressApi == null) {
  46. RestMethod restMethod = new RestMethod();
  47. if (requestMapping.value() != null) {
  48. restMethod.setUrl((root + requestMapping.value()[0]).replaceAll("//", "/"));
  49. }
  50. Description description = method.getAnnotation(Description.class);
  51. if (description != null) {
  52. restMethod.setDescription(description.value());
  53. }
  54. setParams(restMethod, method);
  55. setReturnParams(restMethod, method);
  56. restList.add(restMethod);
  57. }
  58. }
  59. return restList;
  60. }
  61. private boolean isTypeSkip(Class<?> clazz) {
  62. return clazz == HttpServletRequest.class || clazz == HttpServletResponse.class || clazz == Model.class
  63. || clazz == ModelAndView.class;
  64. }
  65. private void setParams(RestMethod restMethod, Method method) {
  66. Class<?>[] paramTypes = method.getParameterTypes();
  67. // Parameter[] params = method.getParameters();
  68. // Annotation[][] annotations = method.getParameterAnnotations();
  69. for (int i = 0; i < paramTypes.length; i++) {
  70. Class<?> paramType = paramTypes[i];
  71. if(isTypeSkip(paramType)){
  72. continue;
  73. }
  74. if (isBasicType(paramType)) {
  75. throw new RuntimeException("You cannot set basic type as controller method parameter!");
  76. // setBasicTypeParam(restMethod, paramType.getSimpleName(),
  77. // params[i].getName(), annotations[i]);
  78. } else {
  79. setObjectTypeParam(restMethod, paramType, null);
  80. }
  81. }
  82. }
  83. private void setBasicTypeParam(RestMethod restMethod, String paramTypeName, String paramName,
  84. Annotation[] annotations) {
  85. RestParam restParam = new RestParam();
  86. restParam.setName(paramName);
  87. restParam.setType(paramTypeName);
  88. if (annotations != null && annotations.length > 0) {
  89. for (Annotation annotation : annotations) {
  90. if (annotation instanceof NotNull) {
  91. restParam.setRequired(true);
  92. continue;
  93. }
  94. if (annotation instanceof Description) {
  95. restParam.setDescription(((Description) annotation).value());
  96. continue;
  97. }
  98. }
  99. }
  100. restMethod.addParam(restParam);
  101. }
  102. private void setObjectTypeParam(RestMethod restMethod, Class<?> paramType, String prefix) {
  103. Field[] fields = paramType.getDeclaredFields();
  104. for (Field field : fields) {
  105. // 字段的修饰符,default: 0, public: 1, private: 2, static: 8, final: 16
  106. // 这样可以排除掉static或final的字段
  107. if ((field.getModifiers() & 8) == 0) {
  108. if (isBasicType(field.getType())) {
  109. setBasicTypeParam(restMethod, field.getType().getSimpleName(),
  110. Utils.isEmpty(prefix) ? field.getName() : prefix + "." + field.getName(),
  111. field.getAnnotations());
  112. } else if (List.class.isAssignableFrom(field.getType())) {
  113. Class<?> listActualType = getCollectionFieldActualType(field);
  114. String name = Utils.isEmpty(prefix) ? field.getName() : prefix + "." + field.getName();
  115. if (isBasicType(listActualType)) {
  116. for (int i = 0; i < LIST_REPEAT; i++) {
  117. setBasicTypeParam(restMethod, listActualType.getSimpleName(), name + "[" + i + "]",
  118. field.getAnnotations());
  119. }
  120. } else {
  121. for (int i = 0; i < LIST_REPEAT; i++) {
  122. setObjectTypeParam(restMethod, listActualType, name + "[" + i + "]");
  123. }
  124. }
  125. } else {
  126. setObjectTypeParam(restMethod, field.getType(),
  127. Utils.isEmpty(prefix) ? field.getName() : prefix + "." + field.getName());
  128. }
  129. }
  130. }
  131. }
  132. private void setReturnParams(RestMethod restMethod, Method method) {
  133. RestReturn restReturn = method.getAnnotation(RestReturn.class);
  134. boolean isCollection = restReturn.collection();
  135. Class<?> returnType = restReturn.value();
  136. if (isBasicType(returnType)) {
  137. throw new RuntimeException("You cannot return basic type!");
  138. } else {
  139. if (isCollection) {
  140. setCollectionReturnParam(restMethod, returnType, " ");
  141. } else {
  142. setObjectReturnParam(restMethod, returnType, " ");
  143. }
  144. }
  145. }
  146. private void setBasicReturnParam(RestMethod restMethod, String returnParamTypeName, String returnParamName,
  147. Annotation[] annotations) {
  148. RestReturnParam returnParam = new RestReturnParam();
  149. returnParam.setName(returnParamName);
  150. returnParam.setType(returnParamTypeName);
  151. if (annotations != null && annotations.length > 0) {
  152. for (Annotation annotation : annotations) {
  153. if (annotation instanceof Description) {
  154. returnParam.setDescription(((Description) annotation).value());
  155. break;
  156. }
  157. }
  158. }
  159. restMethod.addReturnParam(returnParam);
  160. }
  161. private void setObjectReturnParam(RestMethod restMethod, Class<?> returnType, String prefix) {
  162. Field[] fields = returnType.getDeclaredFields();
  163. for (Field field : fields) {
  164. if ((field.getModifiers() & 8) == 0) {
  165. if (isBasicType(field.getType())) {
  166. setBasicReturnParam(restMethod, field.getType().getSimpleName(),
  167. Utils.isEmpty(prefix) ? field.getName() : prefix + field.getName(), field.getAnnotations());
  168. } else if (List.class.isAssignableFrom(field.getType())) {
  169. setBasicReturnParam(restMethod, field.getType().getSimpleName(),
  170. Utils.isEmpty(prefix) ? field.getName() : prefix + field.getName(), field.getAnnotations());
  171. Class<?> listActualType = getCollectionFieldActualType(field);
  172. prefix = " " + prefix;
  173. addStartCollectionReturnParam(restMethod, prefix);
  174. if (!isBasicType(listActualType)) {
  175. setObjectReturnParam(restMethod, listActualType, prefix);
  176. }
  177. addEndCollectionReturnParam(restMethod, prefix);
  178. }
  179. }
  180. }
  181. }
  182. private void setCollectionReturnParam(RestMethod restMethod, Class<?> returnType, String prefix) {
  183. addStartCollectionReturnParam(restMethod, prefix);
  184. setObjectReturnParam(restMethod, returnType, prefix);
  185. addEndCollectionReturnParam(restMethod, prefix);
  186. }
  187. private void addStartCollectionReturnParam(RestMethod restMethod, String prefix) {
  188. setBasicReturnParam(restMethod, null, Utils.isEmpty(prefix) ? "[" : prefix + "[", null);
  189. }
  190. private void addEndCollectionReturnParam(RestMethod restMethod, String prefix) {
  191. setBasicReturnParam(restMethod, null, Utils.isEmpty(prefix) ? "]" : prefix + "]", null);
  192. }
  193. private Class<?> getCollectionFieldActualType(Field field) {
  194. Type genType = field.getGenericType();
  195. if (genType instanceof ParameterizedType) {
  196. ((ParameterizedType) genType).getActualTypeArguments();
  197. Type[] parameterizedTypes = ((ParameterizedType) genType).getActualTypeArguments();
  198. if (parameterizedTypes.length > 0) {
  199. return ((Class<?>) parameterizedTypes[0]);
  200. } else {
  201. throw new RuntimeException("All List type must set its generic type!");
  202. }
  203. } else {
  204. throw new RuntimeException("All List type must set its generic type!");
  205. }
  206. }
  207. private boolean isBasicType(Class<?> paramType) {
  208. return paramType.isPrimitive() || paramType == String.class || Number.class.isAssignableFrom(paramType)
  209. || Date.class.isAssignableFrom(paramType) || Timestamp.class.isAssignableFrom(paramType)
  210. || Enum.class.isAssignableFrom(paramType);
  211. }
  212. }

(4)新建获取接口文档的Controller,其继承自BaseApiController,并写几个测试的接口
位于alan-springmvc-web工程的com.alan.controller包下。
ApiController.java

  1. package com.alan.controller;
  2. import java.util.ArrayList;
  3. import java.util.List;
  4. import javax.servlet.http.HttpServletRequest;
  5. import org.springframework.stereotype.Controller;
  6. import org.springframework.web.bind.annotation.RequestMapping;
  7. import org.springframework.web.bind.annotation.ResponseBody;
  8. import com.alan.api.annotation.Description;
  9. import com.alan.api.annotation.RestReturn;
  10. import com.alan.api.annotation.SuppressApi;
  11. import com.alan.api.controller.BaseApiController;
  12. import com.alan.api.rest.RestResult;
  13. import com.alan.entity.UserInfo;
  14. import com.alan.entity.UserTestXyz;
  15. @Controller
  16. public class ApiController extends BaseApiController {
  17. @SuppressApi
  18. @RequestMapping("/api")
  19. public String api() {
  20. return "api-index.html";
  21. }
  22. @SuppressApi
  23. @RequestMapping("/apiList")
  24. @ResponseBody
  25. public RestResult apiList(HttpServletRequest request) {
  26. String contextPath = request.getContextPath();
  27. RestResult restResult = new RestResult(BaseApiController.getRestList());
  28. restResult.setMessage(contextPath);
  29. return restResult;
  30. }
  31. @RequestMapping("/test")
  32. @ResponseBody
  33. @Description("测试")
  34. @RestReturn(UserInfo.class)
  35. public RestResult test(UserInfo userInfo){
  36. return new RestResult(userInfo);
  37. }
  38. @RequestMapping("/test2")
  39. @ResponseBody
  40. @Description("测试2")
  41. @RestReturn(value=UserTestXyz.class,collection=true)
  42. public RestResult test2(UserTestXyz userTestXyz){
  43. List<UserTestXyz> list = new ArrayList<>();
  44. list.add(userTestXyz);
  45. return new RestResult(list);
  46. }
  47. }

(5)编写页面及javascript
api-index.html

  1. <html>
  2. <head>
  3. <meta http-equiv="Content-Type" content="text/html; charset=UTF-8">
  4. <title>接口文档</title>
  5. <style type="text/css"> #left { border: 1px solid red; width: 25%; float: left; } #right { border: 1px solid green; width: 74%; float: right; padding-top: 50px; } #result { border: 1px solid blue; width: 500px; margin-left: 50px; text-align: left; padding: 10px 10px 40px 10px; } table { border: solid #add9c0; border-width: 1px 0px 0px 1px; width: 500px; margin-left: 50px; } td, th { border: solid #add9c0; border-width: 0px 1px 1px 0px; padding: 10px 0px; } </style>
  6. <body>
  7. <div id="left"></div>
  8. <div id="right">
  9. <form action="/api" onsubmit="return false;" id="form">
  10. <table id="param-list">
  11. </table>
  12. <br />
  13. <input type="submit" value="提交" id="submit" style="margin-left: 350px;">
  14. <br />
  15. <br />
  16. <table id="return-list">
  17. </table>
  18. </form>
  19. <div id="result"></div>
  20. </div>
  21. <script type="text/javascript" src="jquery.js"></script>
  22. <script type="text/javascript" src="api-index.js"></script>
  23. </body>
  24. </html>

api-index.js

  1. var apiIndex = {
  2. apiList: null,
  3. nowUrl: null,
  4. contextPath: null,
  5. getApiList: function(){
  6. var apiListUrl = "/alan-springmvc-web/apiList";
  7. $.post(apiListUrl, null, function(data){
  8. if(data && data.result && data.result.length){
  9. apiIndex.apiList = data.result;
  10. apiIndex.contextPath = data.message;
  11. var html = "";
  12. for(var i=0; i<apiIndex.apiList.length; i++){
  13. html += "<p class='item' data-index="+i+"><a href='#'>"+apiIndex.apiList[i].url+"</a><span> "+apiIndex.apiList[i].description+"</span></p>";
  14. }
  15. $("#left").html(html);
  16. $(".item").click(function(){
  17. apiIndex.getApiDetail($(this).attr("data-index"));
  18. });
  19. apiIndex.getApiDetail(0);
  20. }
  21. },"json");
  22. },
  23. getApiDetail: function(index){
  24. if(this.apiList && this.apiList.length){
  25. $("#result").text("");
  26. $("#param-list").html('<tr>\ <td colspan="4" id="api-name"></td>\ </tr>\ <tr>\ <td colspan="4"><b>参数列表</b></td>\ </tr>\ <tr>\ <th>名称</th>\ <th>值</th>\ <th>类型</th>\ <th>描述</th>\ </tr>');
  27. $("#return-list").html('<tr>\ <td colspan="3"><b>返回值列表</b></td>\ </tr>\ <tr>\ <th>名称</th>\ <th>类型</th>\ <th>描述</th>\ </tr>\ <tr>\ <td>code</td>\ <td>Integer</td>\ <td>返回代码,默认为0</td>\ </tr>\ <tr>\ <td>message</td>\ <td>String</td>\ <td>返回消息,默认为success</td>\ </tr>\ <tr>\ <td>result</td>\ <td></td>\ <td>返回结果的实际值</td>\ </tr>');
  28. var html = "";
  29. var paramTemplate = "<tr>\ <td>#{name}</td>\ <td><input type='text' name='#{name}' /></td>\ <td>#{type}</td>\ <td>#{description}</td>\ </tr>";
  30. var paramRequiredTemplate = "<tr>\ <td><span style='color:red;'>#{name}</span></td>\ <td><input type='text' name='#{name}' /></td>\ <td>#{type}</td>\ <td>#{description}</td>\ </tr>";
  31. var paramList = this.apiList[index].params;
  32. this.nowUrl = this.apiList[index].url;
  33. $("#api-name").text(this.nowUrl);
  34. for(var i=0; i<paramList.length; i++){
  35. var param = paramList[i];
  36. var content = null;
  37. if(param.required){
  38. content = paramRequiredTemplate;
  39. }else{
  40. content = paramTemplate;
  41. }
  42. var content = content.replace(/#{name}/g,param.name?param.name:"");
  43. var content = content.replace(/#{type}/,param.type?param.type:"");
  44. var content = content.replace(/#{description}/,param.description?param.description:"");
  45. html += content;
  46. }
  47. $("#param-list").append(html);
  48. var returnTemplate = "<tr>\ <td>#{name}</td>\ <td>#{type}</td>\ <td>#{description}</td>\ </tr>";
  49. html = "";
  50. var returnList = this.apiList[index].returnParams;
  51. for(var i=0; i<returnList.length; i++){
  52. var returnParam = returnList[i];
  53. var content = returnTemplate.replace(/#{name}/,returnParam.name?returnParam.name:"");
  54. var content = content.replace(/#{type}/,returnParam.type?returnParam.type:"");
  55. var content = content.replace(/#{description}/,returnParam.description?returnParam.description:"");
  56. html += content;
  57. }
  58. $("#return-list").append(html);
  59. }
  60. }
  61. }
  62. $(function(){
  63. apiIndex.getApiList();
  64. $("#submit").click(function(){
  65. var param = { };
  66. var inputs = $("#form input");
  67. for(var i=0; i<inputs.length; i++){
  68. if(inputs[i].type=="text"){
  69. param[inputs[i].name] = inputs[i].value;
  70. }
  71. }
  72. $.post(apiIndex.contextPath+apiIndex.nowUrl,param,function(data){
  73. $("#result").text(data);
  74. },'text');
  75. });
  76. });

###44. 效果展示
(1)启动tomcat,访问http://localhost:8080/alan-springmvc-web/api
(2)输入一些参数,点提交
(3)查看效果
这里写图片描述

###45. 总结
(1)接口的参数无法支持基本类型(包括String),这是因为Java本身反射无法取得参数名称,可通过javassit解决,有兴趣的同学自己解决下;
(2)统一返回类型为RestResult;
(3)页面上红色的参数表示必填参数,目前未实现必填参数不填报错,可通过拦截器统一处理;
(4)我们自己的写的参数Model和返回值Model可直接拿给Android使用,重新修改Model直接重新给他们一份即可,减少了双方因接口沟通的时间,这也是接口参数不支持基本类型的原因之一,IOS的Model也可以根据脚本自动生成一份对应的Model;
(5)可直接在页面上填写参数进行调试,非常方便;
(6)可以让BaseController继承自BaseApiController,使原有接口支持生成接口文档,不过一些通用方法可能要修改。
(7)局限性,要写好多Model,每个接口都要定义一个方法,有点麻烦,不过结构更清晰;

##九、配置redis

##十、配置dubbo

##十一、配置RabbitMQ

##十二、配置多数据库


欢迎关注我的公众号“彤哥读源码”,查看更多“源码&架构&算法”系列文章, 与彤哥一起畅游源码的海洋。

qrcode

发表评论

表情:
评论列表 (有 0 条评论,420人围观)

还没有评论,来说两句吧...

相关阅读

    相关 配置Spring

    本文旨在从一个空工程一步一步地配置Spring,空工程见上一篇文章[创建Maven父子工程][Maven]。 \\一、spring基本配置 \\\1. 添加spring依赖