【MyBaits】Maven项目整合Mybatis写数据到MySQL

介绍

Maven项目整合MyBatis

项目结构

项目文件

POM

需要导入MySQL，MyBatis，Lombok的依赖

<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0"
         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
    <modelVersion>4.0.0</modelVersion>

    <groupId>org.example</groupId>
    <artifactId>mavenmybatis</artifactId>
    <version>1.0-SNAPSHOT</version>

    <properties>
        <maven.compiler.source>8</maven.compiler.source>
        <maven.compiler.target>8</maven.compiler.target>
    </properties>

    <dependencies>
        <dependency>
            <groupId>org.mybatis</groupId>
            <artifactId>mybatis</artifactId>
            <version>3.5.2</version>
        </dependency>

        <dependency>
            <groupId>org.projectlombok</groupId>
            <artifactId>lombok</artifactId>
            <version>1.18.16</version>
        </dependency>

        <dependency>
            <groupId>mysql</groupId>
            <artifactId>mysql-connector-java</artifactId>
            <version>5.1.49</version>
        </dependency>
    </dependencies>

</project>

mybatis.xml

<?xml version="1.0" encoding="UTF-8" ?>
<!DOCTYPE configuration
        PUBLIC "-//mybatis.org//DTD Config 3.0//EN"
        "http://mybatis.org/dtd/mybatis-3-config.dtd">
<configuration>
    <environments default="mysql">
        <environment id="mysql">
            <!--配置事务管理器 -->
            <transactionManager type="JDBC"></transactionManager>
            <!-- 配置数据源-->
            <dataSource type="POOLED">
                <property name="driver" value="com.mysql.jdbc.Driver"/>
                <property name="url" value="jdbc:mysql://localhost:3306/wordcount?useSSL=false&amp;useUnicode=true&amp;characterEncoding=utf8&amp;
                          serverTimezone=Asia/Shanghai"/>
                <property name="username" value="root"/>
                <property name="password" value="root"/>
            </dataSource>
        </environment>
    </environments>

    <!--指明映射文件的位置。-->
    <mappers>
        <mapper resource="mapper/WordCountMapper.xml"></mapper>
    </mappers>

</configuration>

SqlSessionConfig

获取MyBatis.xml的配置信息，生成SqlSessionFactory对象

package config;

import org.apache.ibatis.io.Resources;
import org.apache.ibatis.session.SqlSessionFactory;
import org.apache.ibatis.session.SqlSessionFactoryBuilder;

import java.io.IOException;
import java.io.Reader;

public class SqlSessionConfig {
    public static SqlSessionFactory GetConn() {
        Reader reader = null;
        try {
            reader = Resources.getResourceAsReader("mybatis.xml");
            return new SqlSessionFactoryBuilder().build(reader);
        } catch (IOException e) {
            e.printStackTrace();
        }
        return null;
    }
}

WordCountMapper

MyBatis的Mapper接口文件，定义与数据库交互的行为规范

package mapper;

import entity.WordCount;
import org.apache.ibatis.annotations.Param;

import java.util.List;

public interface WordCountMapper {

    List<WordCount> selectAll();

    void saveOrUpdateBatch(@Param("wordCountList") List<WordCount> wordCountList);

}

WordCountMapper.xml

对WordCountMapper接口的具体实现

<?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">
<!--namespace：命名空间。目前取值没有要求，建议取当前文件的完整路径名，省略xml-->
<mapper namespace="mapper.WordCountMapper">

    <select id="selectAll" resultType="entity.WordCount">
        select * from wordcounts
    </select>

    <insert id="saveOrUpdateBatch" parameterType="list">
        insert into wordcounts (`word`, `count`) values
        <foreach collection="wordCountList" item="item" index="index" separator=",">
            (#{item.word},#{item.count})
        </foreach>
        ON DUPLICATE KEY UPDATE
        `word` = values(`word`),
        `count` = values(`count`)
    </insert>

</mapper>

WordCountDao

通过SqlSessionConfig配置获取SqlSessionFactory对象，以及WordCountMapper接口，生成与数据库交互的SqlSession，并通过main函数测试

package dao;

import config.SqlSessionConfig;
import entity.WordCount;
import mapper.WordCountMapper;
import org.apache.ibatis.session.SqlSession;
import org.apache.ibatis.session.SqlSessionFactory;

import java.util.*;

public class WordCountDao {

    /**
     * 获取所有信息
     * @return
     */
    public static List<WordCount> selectAll(){
        SqlSessionFactory factory = SqlSessionConfig.GetConn();
        SqlSession session = factory.openSession();
        WordCountMapper db = session.getMapper(WordCountMapper.class);
        List<WordCount> list = db.selectAll();
        session.commit();
        session.close();
        return list;
    }

    /**
     * 批量跟新或插入
     * @param wordCountList
     * @return
     */
    public static boolean saveOrUpdateBatch(List<WordCount> wordCountList) {
        SqlSessionFactory factory = SqlSessionConfig.GetConn();
        SqlSession session = factory.openSession();
        WordCountMapper db = session.getMapper(WordCountMapper.class);
        db.saveOrUpdateBatch(wordCountList);
        session.commit();
        session.close();
        return true;
    }

    public static final String[] words = {"Java", "MySQL", "Tomcat", "Redis", "Kafka", "Flink",
                                            "JUC", "Storm", "Zookeeper", "Maven", "MyBatis", "Linux"};

    public static void main(String[] args) {

        Random random = new Random();
        //计数
        Map<String, Integer> counters = new HashMap<>();
        for (int i = 0; i < 100000; i++) {
            String word = words[random.nextInt(words.length)];
            if(!counters.containsKey(word)){
                counters.put(word, 1);
            }else{
                Integer c = counters.get(word) + 1;
                counters.put(word, c);
            }
        }

        List<WordCount> wordCounts = new ArrayList<>();
        for (Map.Entry<String, Integer> entry : counters.entrySet()) {
            WordCount wordCount = new WordCount();
            wordCount.setCount(entry.getValue());
            wordCount.setWord(entry.getKey());
            wordCounts.add(wordCount);
        }

        System.out.println(saveOrUpdateBatch(wordCounts));
        System.err.println("\n=================\n");
        System.out.println(selectAll());
    }

}

效果

数据正常写入数据库