完成Mybatis Plus基础模板

Author: makejavas

…resources/globalConfig/Default/import.vm …urces/globalConfig/Default/autoImport.vmsrc/main/resources/globalConfig/Default/import.vm renamed to src/main/resources/globalConfig/Default/autoImport.vm src/main/resources/globalConfig/Default/import.vm renamed to src/main/resources/globalConfig/Default/autoImport.vm

@@ -1,4 +1,4 @@
 ##自动导入包（仅导入实体属性需要的包，通常用于实体类）
-#foreach($item in $importList)
-import $!item;
+#foreach($import in $importList)
+import $!import;
 #end

src/main/resources/template/MybatisPlus/controller.vm

@@ -10,29 +10,45 @@ $!define
 ##包路径（宏定义）
 #setPackageSuffix("controller")
 
-import org.springframework.web.bind.annotation.*;
-import javax.annotation.Resource;
-import com.alibaba.fastjson.JSON;
+##定义服务名
+#set($serviceName = $!tool.append($!tool.firstLowerCase($!tableInfo.name), "Service"))
+
+##定义实体对象名
+#set($entityName = $!tool.firstLowerCase($!tableInfo.name))
+
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
+import com.baomidou.mybatisplus.extension.api.ApiController;
+import com.baomidou.mybatisplus.extension.api.ApiResult;
+import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import $!{tableInfo.savePackageName}.entity.$!tableInfo.name;
 import $!{tableInfo.savePackageName}.service.$!{tableInfo.name}Service;
+import org.springframework.web.bind.annotation.*;
+
+import javax.annotation.Resource;
+import java.io.Serializable;
 import java.util.List;
 
 ##表注释（宏定义）
 #tableComment("表控制层")
 @RestController
-@RequestMapping("$!tool.firstLowerCase($!{tableName})")
-public class $!{tableName} {
+@RequestMapping("$!tool.firstLowerCase($!tableInfo.name)")
+public class $!{tableName} extends ApiController {
+    /**
+     * 服务对象
+     */
     @Resource
-    private $!{tableInfo.name}Service $!tool.firstLowerCase($!tableInfo.name)Service;
+    private $!{tableInfo.name}Service $!{serviceName};
 
     /**
      * 分页查询所有数据
      *
+     * @param page 分页对象
+     * @param $!entityName 查询实体
      * @return 所有数据
      */
     @GetMapping
-    public String selectAll() {
-        return success(this.$!tool.firstLowerCase($!tableInfo.name)Service.selectAll());
+    public ApiResult selectAll(Page<UserInfo> page, $!tableInfo.name $!entityName) {
+        return success(this.$!{serviceName}.page(page, new QueryWrapper<>($!entityName)));
     }
 
     /**
@@ -42,30 +58,30 @@ public class $!{tableName} {
      * @return 单条数据
      */
     @GetMapping("{id}")
-    public String selectOne(@PathVariable Long id) {
-        return success(this.$!tool.firstLowerCase($!tableInfo.name)Service.queryById(id));
+    public ApiResult selectOne(@PathVariable Serializable id) {
+        return success(this.$!{serviceName}.getById(id));
     }
 
     /**
      * 新增数据
      *
-     * @param entity 实体对象
+     * @param $!entityName 实体对象
      * @return 新增结果
      */
     @PostMapping
-    public String insert(@RequestBody $!tableInfo.name entity) {
-        return success(this.$!tool.firstLowerCase($!tableInfo.name)Service.insert(entity));
+    public ApiResult insert(@RequestBody $!tableInfo.name $!entityName) {
+        return success(this.$!{serviceName}.save($!entityName));
     }
 
     /**
      * 修改数据
      *
-     * @param entity 实体对象
+     * @param $!entityName 实体对象
      * @return 修改结果
      */
     @PutMapping
-    public String update(@RequestBody $!tableInfo.name entity) {
-        return success(this.$!tool.firstLowerCase($!tableInfo.name)Service.update(entity));
+    public ApiResult update(@RequestBody $!tableInfo.name $!entityName) {
+        return success(this.$!{serviceName}.updateById($!entityName));
     }
 
     /**
@@ -75,17 +91,7 @@ public class $!{tableName} {
      * @return 删除结果
      */
     @DeleteMapping
-    public String delete(@RequestParam("idList") List<Long> idList) {
-        return success(this.$!tool.firstLowerCase($!tableInfo.name)Service.deleteByIdList(idList));
-    }
-
-    /**
-      * 成功返回方法
-      *
-      * @param obj 对象
-      * @return JSON字符串
-      */
-    public String success(Object obj) {
-        return JSON.toJSONString(obj);
+    public ApiResult delete(@RequestParam("idList") List<Long> idList) {
+        return success(this.$!{serviceName}.removeByIds(idList));
     }
 }

src/main/resources/template/MybatisPlus/dao.vm

@@ -10,8 +10,11 @@ $!define
 ##包路径（宏定义）
 #setPackageSuffix("dao")
 
+import com.baomidou.mybatisplus.core.mapper.BaseMapper;
+import $!{tableInfo.savePackageName}.entity.$!tableInfo.name;
+
 ##表注释（宏定义）
 #tableComment("表数据库访问层")
-public interface $!{tableName} {
+public interface $!{tableName} extends BaseMapper<$!tableInfo.name> {
 
 }

src/main/resources/template/MybatisPlus/entity.vm

@@ -8,11 +8,14 @@ $!define
 #setPackageSuffix("entity")
 
 ##导入包（全局变量）
-$!import
+$!autoImport
+import com.baomidou.mybatisplus.extension.activerecord.Model;
+import java.io.Serializable;
 
 ##表注释（宏定义）
 #tableComment("表实体类")
-public class $!{tableInfo.name} {
+@SuppressWarnings("serial")
+public class $!{tableInfo.name} extends Model<$!{tableInfo.name}> {
 #foreach($column in $tableInfo.fullColumn)
     #if(${column.comment})//${column.comment}#end
 
@@ -23,4 +26,16 @@ public class $!{tableInfo.name} {
 #getSetMethod($column)
 #end
 
+#foreach($column in $tableInfo.pkColumn)
+    /**
+     * 获取主键值
+     *
+     * @return 主键值
+     */
+    @Override
+    protected Serializable pkVal() {
+        return this.$!column.name;
+    }
+    #break
+#end
 }

src/main/resources/template/MybatisPlus/service.vm

@@ -10,8 +10,11 @@ $!define
 ##包路径（宏定义）
 #setPackageSuffix("service")
 
+import com.baomidou.mybatisplus.extension.service.IService;
+import $!{tableInfo.savePackageName}.entity.$!tableInfo.name;
+
 ##表注释（宏定义）
 #tableComment("表服务接口")
-public interface $!{tableName} {
+public interface $!{tableName} extends IService<$!tableInfo.name> {
 
 }

src/main/resources/template/MybatisPlus/serviceImpl.vm

@@ -10,8 +10,15 @@ $!define
 ##包路径（宏定义）
 #setPackageSuffix("service.impl")
 
+import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import $!{tableInfo.savePackageName}.dao.UserInfoDao;
+import $!{tableInfo.savePackageName}.entity.UserInfo;
+import $!{tableInfo.savePackageName}.service.UserInfoService;
+import org.springframework.stereotype.Service;
+
 ##表注释（宏定义）
 #tableComment("表服务实现类")
-public class $!{tableName} implements $!{tableInfo.name}Service {
+@Service("$!tool.firstLowerCase($tableInfo.name)Service")
+public class $!{tableName} extends ServiceImpl<$!{tableInfo.name}Dao, $!{tableInfo.name}> implements $!{tableInfo.name}Service {
 
 }