package com.cun.plus;

import com.baomidou.mybatisplus.mapper.Wrapper;
import com.baomidou.mybatisplus.plugins.Page;
import com.cun.plus.entity.User;
import com.cun.plus.mapper.UserMapper;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.junit4.SpringRunner;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;

@RunWith(SpringRunner.class)
@SpringBootTest
public class PlusApplicationTests {

  @Autowired
  private UserMapper userMapper;

  /**
   * 1、增加 insert
   */
  @Test
  public void insertTest() {
    User user = new User();
    user.setUsername("綠茶");
    user.setPassword("lvcha");
    // 返回對數據庫影響操作數：1
    Integer insert1 = userMapper.insert(user); // 非空屬性才會出現在 sql 中
    // 可以立刻獲取插入新記錄的 id
    System.out.println(user.getId());
    // 同上
    Integer insert2 = userMapper.insertAllColumn(user); // 所有屬性都會出現在 sql 中
    // 同上
    System.out.println(user.getId());
  }

  /**
   * 2、修改 update
   */
  @Test
  public void updateTest() {
    User user = new User();
    user.setId(45);
    user.setUsername("cun");
    //user.setPassword("666");
    // 返回對數據庫影響操作數：1
    Integer integer1 = userMapper.updateById(user); // 屬性空則不修改
    System.out.println(integer1);
    // 同上
    Integer integer2 = userMapper.updateAllColumnById(user); // 屬性空則字段空
    System.out.println(integer2);
  }

  /**
   * 3、查詢 select
   */
  @Test
  public void selectTest() {
    // 根據id 查詢一條記錄
    User user = userMapper.selectById(46);
    System.out.println(user.getUsername());

    ArrayList<Integer> idList = new ArrayList<>();
    idList.add(61);
    idList.add(63);
    // 根據多個id 批量查詢
    List<User> users = userMapper.selectBatchIds(idList);
    System.out.println(users.get(0).getUsername() + users.get(1).getUsername());

    User user1 = new User();
    user1.setId(61);
    user1.setUsername("cun");
    // 根據多個條件返回一個對象，若有多個符合條件的記錄則將報錯
    User user2 = userMapper.selectOne(user1);
    System.out.println(user2.getPassword());

    // 根據多個條件返回對象組，注意Map 中的key 必須和數據庫表字段一直
    HashMap<String, Object> columnMap = new HashMap<>();
    columnMap.put("username", "cun");
    List<User> users1 = userMapper.selectByMap(columnMap);
    System.out.println(users1.size());


  }

  /**
   * 4、刪除 delete
   */
  @Test
  public void deleteTest() {
    // 根據一個id 刪除，返回對數據庫影響操作數：1
    Integer integer1 = userMapper.deleteById(65);// 根據id刪除一條記錄
    System.out.println(integer1);

    ArrayList<Integer> idList = new ArrayList<>(); // 根據id集合批量刪除
    idList.add(64);
    idList.add(66);
    // 根據多個id 批量刪除，返回對數據庫影響操作數：2
    Integer integer2 = userMapper.deleteBatchIds(idList);
    System.out.println(integer2);

    HashMap<String, Object> columnMap = new HashMap<>();
    columnMap.put("username", "cun");
    // 根據多個條件刪除，返回對數據庫影響操作數
    Integer integer3 = userMapper.deleteByMap(columnMap);
    System.out.println(integer3);
  }

  /**
   * 5、偽分頁（獲取全部數據再分頁）
   */
  @Test
  public void pageTest() {
    // 分頁查詢，注意如果設置了 PaginationInterceptor 分頁插件則會報錯，
    List<User> users2 = userMapper.selectPage(new Page<User>(1, 2), null); //當前頁、每頁大小
    System.out.println(users2.get(0).getUsername() + users2.get(1).getUsername());
    System.out.println(users2.size());
  }


  /**
   * 6、條件構造器
   */
  @Test
  public void wrapperTest(){
    List<User> users = userMapper.selectList(new EntityWrapper<User>()
        .eq("username", "linhongcun")
    );
    System.out.println(users.size());
  }

}

package com.cun.plus;

import com.baomidou.mybatisplus.enums.IdType;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.config.DataSourceConfig;
import com.baomidou.mybatisplus.generator.config.GlobalConfig;
import com.baomidou.mybatisplus.generator.config.PackageConfig;
import com.baomidou.mybatisplus.generator.config.StrategyConfig;
import com.baomidou.mybatisplus.generator.config.rules.DbType;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;

public class MpG {
  public static void main(String[] args) {
    //1. 全局配置
    GlobalConfig config = new GlobalConfig();
    config.setActiveRecord(false) // 是否支持AR模式
        .setAuthor("linhongcun") // 作者
        .setOutputDir("C:\\data\\mp") // 生成路徑
        .setFileOverride(true) // 文件覆蓋
        .setIdType(IdType.AUTO) // 主鍵策略
        .setServiceName("%sService") // 設置生成的service接口的名字的首字母是否為I
        // IUserService
        .setBaseResultMap(true)
        .setBaseColumnList(true);

    //2. 數據源配置
    DataSourceConfig dsConfig = new DataSourceConfig();
    dsConfig.setDbType(DbType.MYSQL) // 設置數據庫類型
        .setDriverName("com.mysql.jdbc.Driver")
        .setUrl("jdbc:mysql://120.79.197.130:3307/testspring?useSSL=true&verifyServerCertificate=false&allowMultiQueries=true&useUnicode=true&characterEncoding=UTF-8")
        .setUsername("root")
        .setPassword("123");

    //3. 策略配置
    StrategyConfig stConfig = new StrategyConfig();
    stConfig.setCapitalMode(true) //全局大寫命名
        .setDbColumnUnderline(true) // 指定表名 字段名是否使用下劃線
        .setNaming(NamingStrategy.underline_to_camel) // 數據庫表映射到實體的命名策略
        .setTablePrefix("tb_")
        .setInclude("tb_user"); // 生成的表

    //4. 包名策略配置
    PackageConfig pkConfig = new PackageConfig();
    pkConfig.setParent("com.cun.plus")
        .setMapper("mapper")
        .setService("service")
        .setController("controller")
        .setEntity("entity")
        .setXml("mapper");

    //5. 整合配置
    AutoGenerator ag = new AutoGenerator();
    ag.setGlobalConfig(config)
        .setDataSource(dsConfig)
        .setStrategy(stConfig)
        .setPackageInfo(pkConfig);

    //6. 執行
    ag.execute();
  }
}

// 選擇 freemarker 引擎，默認 Veloctiy
  // mpg.setTemplateEngine(new FreemarkerTemplateEngine());

 <!-- mybatis-plus-->
    <dependency>
      <groupId>com.baomidou</groupId>
      <artifactId>mybatis-plus-boot-starter</artifactId>
      <version>2.3</version>
    </dependency>
    <!-- 代碼生成器默認使用如下模版引擎 -->
    <dependency>
      <groupId>org.apache.velocity</groupId>
      <artifactId>velocity-engine-core</artifactId>
      <version>2.0</version>
    </dependency>