MyBatisGenerator工具生成的类与XML对Oracle数据库模糊查询的支持

MyBatis Generator 工具可以用来生成一些工具类和XML模板，用于为用户使用Java语言访问数据库提供便利。

我使用的数据库版本为Oracle11g，JDK版本为1.8。

建表语句如下：

CREATE TABLE PERSON_INFO
(
    ID NUMBER(12,0) PRIMARY KEY,NAME VARCHAR(50),COMPANY VARCHAR(200)
);

INSERT INTO PERSON_INFO (ID,NAME,COMPANY) VALUES (1,'Tsybius','Dog Head Inc');
INSERT INTO PERSON_INFO (ID,COMPANY) VALUES (2,'Galatea',COMPANY) VALUES (3,'Quintus',COMPANY) VALUES (4,'Gaius','State Grid Corporation of Mars');
INSERT INTO PERSON_INFO (ID,COMPANY) VALUES (5,'Titus','Tyrannosaurus Company Limited');
INSERT INTO PERSON_INFO (ID,COMPANY) VALUES (6,'Test1','Dog Head%');
INSERT INTO PERSON_INFO (ID,COMPANY) VALUES (7,'Test2','Dog Head%Test');
INSERT INTO PERSON_INFO (ID,COMPANY) VALUES (8,'Test3','Test%Dog Head%Test');

COMMIT;
/

如果我们要在这个表中查询所有COMPANY字段包含“Head%”的数据，可以使用下面的sql：

SELECT *
FROM   PERSON_INFO
WHERE  COMPANY LIKE '%Dog Head/%%' ESCAPE '/'
ORDER  BY ID ASC

查询结果如下：

我们使用 MyBatis Generator 工具生成四个文件：

1、PersonInfoMapper.java

2、PersonInfoMapper.xml

3、PersonInfo.java

4、PersonInfoExample.java

生成方法可参考下面两篇博客：

1、http://my.oschina.net/Tsybius2014/blog/676973

2、http://my.oschina.net/Tsybius2014/blog/678840

PersonInfoExample类中会生成andNameLike、andNameNotLike、andCompanyLike、andCompanyNotLike四个函数，但这四个函数并不能满足我们上面例子中的要求：一是传入的value必须左右手工添加百分号，二是这样传入的value不能支持查询“%”和“_”两个通配符（即oracle中的escape关键字）

为了支持在模糊查询时可以准确查询出通配符“_”和“%”，需要对PersonInfoExample.java和PersonInfoMapper.xml进行修改。

1、对PersonInfoExample.java的改动

修改的部分都用“【新加入的代码】”做了标记，改动的地方包括：

1）对所有Like和NotLike函数都做了改造

2）新加了一个布尔型的变量escapeValue，用于标记Like和NotLike需要转义

3）为Criterion类新加了一个构造函数，用于为Like和NotLike两类条件创建Criterion

package GenTarget.object_example_java;

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

public class PersonInfoExample {
    protected String orderByClause;

    protected boolean distinct;

    protected List<Criteria> oredCriteria;

    public PersonInfoExample() {
        oredCriteria = new ArrayList<Criteria>();
    }

    public void setOrderByClause(String orderByClause) {
        this.orderByClause = orderByClause;
    }

    public String getOrderByClause() {
        return orderByClause;
    }

    public void setDistinct(boolean distinct) {
        this.distinct = distinct;
    }

    public boolean isDistinct() {
        return distinct;
    }

    public List<Criteria> getOredCriteria() {
        return oredCriteria;
    }

    public void or(Criteria criteria) {
        oredCriteria.add(criteria);
    }

    public Criteria or() {
        Criteria criteria = createCriteriaInternal();
        oredCriteria.add(criteria);
        return criteria;
    }

    public Criteria createCriteria() {
        Criteria criteria = createCriteriaInternal();
        if (oredCriteria.size() == 0) {
            oredCriteria.add(criteria);
        }
        return criteria;
    }

    protected Criteria createCriteriaInternal() {
        Criteria criteria = new Criteria();
        return criteria;
    }

    public void clear() {
        oredCriteria.clear();
        orderByClause = null;
        distinct = false;
    }

    protected abstract static class GeneratedCriteria {
        protected List<Criterion> criteria;

        protected GeneratedCriteria() {
            super();
            criteria = new ArrayList<Criterion>();
        }

        public boolean isValid() {
            return criteria.size() > 0;
        }

        public List<Criterion> getAllCriteria() {
            return criteria;
        }

        public List<Criterion> getCriteria() {
            return criteria;
        }

        protected void addCriterion(String condition) {
            if (condition == null) {
                throw new RuntimeException("Value for condition cannot be null");
            }
            criteria.add(new Criterion(condition));
        }

        protected void addCriterion(String condition,Object value,String property) {
            if (value == null) {
                throw new RuntimeException("Value for " + property + " cannot be null");
            }
            criteria.add(new Criterion(condition,value));
        }

        protected void addCriterion(String condition,Object value1,Object value2,String property) {
            if (value1 == null || value2 == null) {
                throw new RuntimeException("Between values for " + property + " cannot be null");
            }
            criteria.add(new Criterion(condition,value1,value2));
        }

        /**
         * 【新加入的代码】生成带有转义字符的Criterion
         * @param condition
         * @param value1
         * @param value2
         * @param property
         */
        protected void addCriterionEscape(String condition,String property) {
            if (value1 == null || value2 == null) {
                throw new RuntimeException("Value or escape for " + property + " cannot be null");
            }
            criteria.add(new Criterion(condition,value2,null,null));
        }
        
        public Criteria andIdIsNull() {
            addCriterion("ID is null");
            return (Criteria) this;
        }

        public Criteria andIdIsNotNull() {
            addCriterion("ID is not null");
            return (Criteria) this;
        }

        public Criteria andIdEqualTo(Long value) {
            addCriterion("ID =",value,"id");
            return (Criteria) this;
        }

        public Criteria andIdNotEqualTo(Long value) {
            addCriterion("ID <>","id");
            return (Criteria) this;
        }

        public Criteria andIdGreaterThan(Long value) {
            addCriterion("ID >","id");
            return (Criteria) this;
        }

        public Criteria andIdGreaterThanOrEqualTo(Long value) {
            addCriterion("ID >=","id");
            return (Criteria) this;
        }

        public Criteria andIdLessThan(Long value) {
            addCriterion("ID <","id");
            return (Criteria) this;
        }

        public Criteria andIdLessThanOrEqualTo(Long value) {
            addCriterion("ID <=","id");
            return (Criteria) this;
        }

        public Criteria andIdIn(List<Long> values) {
            addCriterion("ID in",values,"id");
            return (Criteria) this;
        }

        public Criteria andIdNotIn(List<Long> values) {
            addCriterion("ID not in","id");
            return (Criteria) this;
        }

        public Criteria andIdBetween(Long value1,Long value2) {
            addCriterion("ID between","id");
            return (Criteria) this;
        }

        public Criteria andIdNotBetween(Long value1,Long value2) {
            addCriterion("ID not between","id");
            return (Criteria) this;
        }

        public Criteria andNameIsNull() {
            addCriterion("NAME is null");
            return (Criteria) this;
        }

        public Criteria andNameIsNotNull() {
            addCriterion("NAME is not null");
            return (Criteria) this;
        }

        public Criteria andNameEqualTo(String value) {
            addCriterion("NAME =","name");
            return (Criteria) this;
        }

        public Criteria andNameNotEqualTo(String value) {
            addCriterion("NAME <>","name");
            return (Criteria) this;
        }

        public Criteria andNameGreaterThan(String value) {
            addCriterion("NAME >","name");
            return (Criteria) this;
        }

        public Criteria andNameGreaterThanOrEqualTo(String value) {
            addCriterion("NAME >=","name");
            return (Criteria) this;
        }

        public Criteria andNameLessThan(String value) {
            addCriterion("NAME <","name");
            return (Criteria) this;
        }

        public Criteria andNameLessThanOrEqualTo(String value) {
            addCriterion("NAME <=","name");
            return (Criteria) this;
        }

        /**
         * 【新加入的代码】模糊查询
         * @param value
         * @return
         */
        public Criteria andNameLike(String value) {
            if(value != null){
                value = value.replaceAll("/","//").replaceAll("%","/%").replaceAll("_","/_");
                value = "%" + value + "%";
            }
            addCriterionEscape("NAME like","/","name");
            return (Criteria) this;
        }

        /**
         * 【新加入的代码】模糊查询
         * @param value
         * @return
         */
        public Criteria andNameNotLike(String value) {
            //模糊匹配的通配符
            if(value != null){
                value = value.replaceAll("/","/_");
                value = "%" + value + "%";
            }
            addCriterionEscape("NAME not like","name");
            return (Criteria) this;
        }

        public Criteria andNameIn(List<String> values) {
            addCriterion("NAME in","name");
            return (Criteria) this;
        }

        public Criteria andNameNotIn(List<String> values) {
            addCriterion("NAME not in","name");
            return (Criteria) this;
        }

        public Criteria andNameBetween(String value1,String value2) {
            addCriterion("NAME between","name");
            return (Criteria) this;
        }

        public Criteria andNameNotBetween(String value1,String value2) {
            addCriterion("NAME not between","name");
            return (Criteria) this;
        }

        public Criteria andCompanyIsNull() {
            addCriterion("COMPANY is null");
            return (Criteria) this;
        }

        public Criteria andCompanyIsNotNull() {
            addCriterion("COMPANY is not null");
            return (Criteria) this;
        }

        public Criteria andCompanyEqualTo(String value) {
            addCriterion("COMPANY =","company");
            return (Criteria) this;
        }

        public Criteria andCompanyNotEqualTo(String value) {
            addCriterion("COMPANY <>","company");
            return (Criteria) this;
        }

        public Criteria andCompanyGreaterThan(String value) {
            addCriterion("COMPANY >","company");
            return (Criteria) this;
        }

        public Criteria andCompanyGreaterThanOrEqualTo(String value) {
            addCriterion("COMPANY >=","company");
            return (Criteria) this;
        }

        public Criteria andCompanyLessThan(String value) {
            addCriterion("COMPANY <","company");
            return (Criteria) this;
        }

        public Criteria andCompanyLessThanOrEqualTo(String value) {
            addCriterion("COMPANY <=","company");
            return (Criteria) this;
        }

        /**
         * 【新加入的代码】模糊查询
         * @param value
         * @return
         */
        public Criteria andCompanyLike(String value) {
            //模糊匹配的通配符
            if(value != null){
                value = value.replaceAll("/","/_");
                value = "%" + value + "%";
            }
            addCriterionEscape("COMPANY like","company");
            return (Criteria) this;
        }

        /**
         * 【新加入的代码】模糊查询
         * @param value
         * @return
         */
        public Criteria andCompanyNotLike(String value) {
            if(value != null){
                value = value.replaceAll("/","/_");
                value = "%" + value + "%";
            }
            addCriterionEscape("COMPANY not like","company");
            return (Criteria) this;
        }

        public Criteria andCompanyIn(List<String> values) {
            addCriterion("COMPANY in","company");
            return (Criteria) this;
        }

        public Criteria andCompanyNotIn(List<String> values) {
            addCriterion("COMPANY not in","company");
            return (Criteria) this;
        }

        public Criteria andCompanyBetween(String value1,String value2) {
            addCriterion("COMPANY between","company");
            return (Criteria) this;
        }

        public Criteria andCompanyNotBetween(String value1,String value2) {
            addCriterion("COMPANY not between","company");
            return (Criteria) this;
        }
    }

    public static class Criteria extends GeneratedCriteria {

        protected Criteria() {
            super();
        }
    }

    public static class Criterion {
        private String condition;

        private Object value;

        private Object secondValue;

        private boolean noValue;

        private boolean singleValue;

        private boolean betweenValue;

        /** 【新加入的代码】为true时按转义字符处理添加escape关键字 */
        private boolean escapeValue;

        private boolean listValue;

        private String typeHandler;

        public String getCondition() {
            return condition;
        }

        public Object getValue() {
            return value;
        }

        public Object getSecondValue() {
            return secondValue;
        }

        public boolean isNoValue() {
            return noValue;
        }

        public boolean isSingleValue() {
            return singleValue;
        }

        public boolean isBetweenValue() {
            return betweenValue;
        }

        /**
         * 【新加入的代码】为true时按转义字符处理添加escape关键字
         * @return
         */
        public boolean isEscapeValue() {
            return escapeValue;
        }

        public boolean isListValue() {
            return listValue;
        }

        public String getTypeHandler() {
            return typeHandler;
        }

        protected Criterion(String condition) {
            super();
            this.condition = condition;
            this.typeHandler = null;
            this.noValue = true;
        }

        protected Criterion(String condition,String typeHandler) {
            super();
            this.condition = condition;
            this.value = value;
            this.typeHandler = typeHandler;
            if (value instanceof List<?>) {
                this.listValue = true;
            } else {
                this.singleValue = true;
            }
        }

        protected Criterion(String condition,Object value) {
            this(condition,null);
        }

        protected Criterion(String condition,Object secondValue,String typeHandler) {
            super();
            this.condition = condition;
            this.value = value;
            this.secondValue = secondValue;
            this.typeHandler = typeHandler;
            this.betweenValue = true;
        }

        protected Criterion(String condition,Object secondValue) {
            this(condition,secondValue,null);
        }
        
        /**
         * 【新加入的代码】生成带有转义字符的Criterion
         * @param condition
         * @param value
         * @param escapeValue
         * @param typeHandler
         * @param dummy
         */
        protected Criterion(String condition,Object escapeValue,String typeHandler,Object dummy) {
            super();
            this.condition = condition;
            this.value = value;
            this.secondValue = escapeValue;
            this.typeHandler = typeHandler;
            this.escapeValue = true;
        }
    }
}

2、对PersonInfoMapper.xml的改动

修改的部分都用“【新加入的代码】”做了标记，改动的地方包括：

1）sql 代码段：Example_Where_Clause

2）sql 代码段：Update_By_Example_Where_Clause

<?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" >
<mapper namespace="GenTarget.mapper_java.PersonInfoMapper" >
  <resultMap id="BaseResultMap" type="GenTarget.object_example_java.PersonInfo" >
    <id column="ID" property="id" jdbcType="DECIMAL" />
    <result column="NAME" property="name" jdbcType="VARCHAR" />
    <result column="COMPANY" property="company" jdbcType="VARCHAR" />
  </resultMap>
  <sql id="Example_Where_Clause" >
    <where >
      <foreach collection="oredCriteria" item="criteria" separator="or" >
        <if test="criteria.valid" >
          <trim prefix="(" suffix=")" prefixOverrides="and" >
            <foreach collection="criteria.criteria" item="criterion" >
              <choose >
                <when test="criterion.noValue" >
                  and ${criterion.condition}
                </when>
                <when test="criterion.singleValue" >
                  and ${criterion.condition} #{criterion.value}
                </when>
                <when test="criterion.betweenValue" >
                  and ${criterion.condition} #{criterion.value} and #{criterion.secondValue}
                </when>
                
                <!-- 【新加入的代码】添加转义字符 START -->
                <when test="criterion.escapeValue" >
                  and ${criterion.condition} #{criterion.value} escape #{criterion.secondValue}
                </when>
                <!-- 【新加入的代码】添加转义字符 END -->
                
                <when test="criterion.listValue" >
                  and ${criterion.condition}
                  <foreach collection="criterion.value" item="listItem" open="(" close=")" separator="," >
                    #{listItem}
                  </foreach>
                </when>
              </choose>
            </foreach>
          </trim>
        </if>
      </foreach>
    </where>
  </sql>
  <sql id="Update_By_Example_Where_Clause" >
    <where >
      <foreach collection="example.oredCriteria" item="criteria" separator="or" >
        <if test="criteria.valid" >
          <trim prefix="(" suffix=")" prefixOverrides="and" >
            <foreach collection="criteria.criteria" item="criterion" >
              <choose >
                <when test="criterion.noValue" >
                  and ${criterion.condition}
                </when>
                <when test="criterion.singleValue" >
                  and ${criterion.condition} #{criterion.value}
                </when>
                <when test="criterion.betweenValue" >
                  and ${criterion.condition} #{criterion.value} and #{criterion.secondValue}
                </when>
                
                <!-- 【新加入的代码】添加转义字符 START -->
                <when test="criterion.escapeValue" >
                  and ${criterion.condition} #{criterion.value} escape #{criterion.secondValue}
                </when>
                <!-- 【新加入的代码】添加转义字符 END -->
                
                <when test="criterion.listValue" >
                  and ${criterion.condition}
                  <foreach collection="criterion.value" item="listItem" open="(" close=")" separator="," >
                    #{listItem}
                  </foreach>
                </when>
              </choose>
            </foreach>
          </trim>
        </if>
      </foreach>
    </where>
  </sql>
  <sql id="Base_Column_List" >
    ID,COMPANY
  </sql>
  <select id="selectByExample" resultMap="BaseResultMap" parameterType="GenTarget.object_example_java.PersonInfoExample" >
    select
    <if test="distinct" >
      distinct
    </if>
    <include refid="Base_Column_List" />
    from PERSON_INFO
    <if test="_parameter != null" >
      <include refid="Example_Where_Clause" />
    </if>
    <if test="orderByClause != null" >
      order by ${orderByClause}
    </if>
  </select>
  <select id="selectByPrimaryKey" resultMap="BaseResultMap" parameterType="java.lang.Long" >
    select 
    <include refid="Base_Column_List" />
    from PERSON_INFO
    where ID = #{id,jdbcType=DECIMAL}
  </select>
  <delete id="deleteByPrimaryKey" parameterType="java.lang.Long" >
    delete from PERSON_INFO
    where ID = #{id,jdbcType=DECIMAL}
  </delete>
  <delete id="deleteByExample" parameterType="GenTarget.object_example_java.PersonInfoExample" >
    delete from PERSON_INFO
    <if test="_parameter != null" >
      <include refid="Example_Where_Clause" />
    </if>
  </delete>
  <insert id="insert" parameterType="GenTarget.object_example_java.PersonInfo" >
    insert into PERSON_INFO (ID,COMPANY
      )
    values (#{id,jdbcType=DECIMAL},#{name,jdbcType=VARCHAR},#{company,jdbcType=VARCHAR}
      )
  </insert>
  <insert id="insertSelective" parameterType="GenTarget.object_example_java.PersonInfo" >
    insert into PERSON_INFO
    <trim prefix="(" suffix=")" suffixOverrides="," >
      <if test="id != null" >
        ID,</if>
      <if test="name != null" >
        NAME,</if>
      <if test="company != null" >
        COMPANY,</if>
    </trim>
    <trim prefix="values (" suffix=")" suffixOverrides="," >
      <if test="id != null" >
        #{id,</if>
      <if test="name != null" >
        #{name,</if>
      <if test="company != null" >
        #{company,</if>
    </trim>
  </insert>
  <select id="countByExample" parameterType="GenTarget.object_example_java.PersonInfoExample" resultType="java.lang.Integer" >
    select count(*) from PERSON_INFO
    <if test="_parameter != null" >
      <include refid="Example_Where_Clause" />
    </if>
  </select>
  <update id="updateByExampleSelective" parameterType="map" >
    update PERSON_INFO
    <set >
      <if test="record.id != null" >
        ID = #{record.id,</if>
      <if test="record.name != null" >
        NAME = #{record.name,</if>
      <if test="record.company != null" >
        COMPANY = #{record.company,</if>
    </set>
    <if test="_parameter != null" >
      <include refid="Update_By_Example_Where_Clause" />
    </if>
  </update>
  <update id="updateByExample" parameterType="map" >
    update PERSON_INFO
    set ID = #{record.id,NAME = #{record.name,COMPANY = #{record.company,jdbcType=VARCHAR}
    <if test="_parameter != null" >
      <include refid="Update_By_Example_Where_Clause" />
    </if>
  </update>
  <update id="updateByPrimaryKeySelective" parameterType="GenTarget.object_example_java.PersonInfo" >
    update PERSON_INFO
    <set >
      <if test="name != null" >
        NAME = #{name,</if>
      <if test="company != null" >
        COMPANY = #{company,</if>
    </set>
    where ID = #{id,jdbcType=DECIMAL}
  </update>
  <update id="updateByPrimaryKey" parameterType="GenTarget.object_example_java.PersonInfo" >
    update PERSON_INFO
    set NAME = #{name,COMPANY = #{company,jdbcType=VARCHAR}
    where ID = #{id,jdbcType=DECIMAL}
  </update>
</mapper>

调用此功能的Java代码如下：

package GenMyBatisTest;
 
import java.io.InputStream;
import java.io.PrintWriter;
import java.io.StringWriter;
import java.util.ArrayList;
import java.util.List;

import org.apache.ibatis.io.Resources;
import org.apache.ibatis.session.sqlSession;
import org.apache.ibatis.session.sqlSessionFactory;
import org.apache.ibatis.session.sqlSessionFactoryBuilder;

import GenTarget.mapper_java.PersonInfoMapper;
import GenTarget.object_example_java.PersonInfo;
import GenTarget.object_example_java.PersonInfoExample;

/**
 * MyBatis使用测试
 * @author Tsybius2014
 * @date 2016年5月22日
 * @time 下午10:44:06
 * @remark
 */
public class MyBatisTest {
    public static void main(String[] args) {
        try {
            String resource = "mybatis-config.xml";
            InputStream inputStream = Resources.getResourceAsStream(resource);
            sqlSessionFactory sqlSessionFactory = new sqlSessionFactoryBuilder().build(inputStream);
            sqlSession session = sqlSessionFactory.openSession();
            try {
                PersonInfoMapper mapper = session.getMapper(PersonInfoMapper.class);
                //模糊查询
                PersonInfoExample example = new PersonInfoExample();
                PersonInfoExample.Criteria criteria = example.createCriteria();
                criteria.andCompanyLike("Dog Head%");
                example.setOrderByClause("id desc");
                List<PersonInfo> personInfoList = new ArrayList<PersonInfo>();
                personInfoList.addAll(mapper.selectByExample(example));
                if (personInfoList != null && personInfoList.size() > 0) {
                    for (PersonInfo personInfo : personInfoList) {
                        System.out.println(
                            "id:" + personInfo.getId() + ";" +
                            "name:" + personInfo.getName() + ";" +
                            "company:" + personInfo.getCompany());
                    }
                } else {
                    System.out.println("personInfoList is emply");
                }
                System.out.println("done.");
            } finally {
                session.close();
            }
        } catch (Exception ex) {
            StringWriter stringWriter = new StringWriter();
            PrintWriter printWriter = new PrintWriter(stringWriter);
            ex.printStackTrace(printWriter);
            System.out.println(stringWriter.toString());
        }
    }
}

Console中输出的结果如下：

id:8;name:Test3;company:Test%Dog Head%Test
id:7;name:Test2;company:Dog Head%Test
id:6;name:Test1;company:Dog Head%
done.

注：本文中介绍的只是实现此类功能的一种方法，仅供参考。

END

MyBatisGenerator工具生成的类与XML对Oracle数据库模糊查询的支持

猜你在找的XML相关文章