#393

Committed 19 May 2024 12:06PM CUT coverage: 72.18% (-17.3%) from 89.44%

Build # #393

Build Type

push

Committed by

Commit Message

add ci

Run Details

1977 of 2739 relevant lines covered (72.18%)

0.72 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

0.0

/src/main/java/com/github/wz2cool/dynamic/mybatis/mapper/provider/InsertIgnoreProvider.java

package com.github.wz2cool.dynamic.mybatis.mapper.provider;

import com.github.wz2cool.dynamic.mybatis.mapper.helper.EnhancedSqlHelper;
import org.apache.ibatis.mapping.MappedStatement;
import tk.mybatis.mapper.MapperException;
import tk.mybatis.mapper.entity.EntityColumn;
import tk.mybatis.mapper.mapperhelper.EntityHelper;
import tk.mybatis.mapper.mapperhelper.MapperHelper;
import tk.mybatis.mapper.mapperhelper.SelectKeyHelper;
import tk.mybatis.mapper.mapperhelper.SqlHelper;
import tk.mybatis.mapper.provider.base.BaseInsertProvider;

import java.util.Set;

/**
 * @author frank
 */
public class InsertIgnoreProvider extends BaseInsertProvider {

    public InsertIgnoreProvider(Class<?> mapperClass, MapperHelper mapperHelper) {
        super(mapperClass, mapperHelper);
    }

    public String insertIgnore(MappedStatement ms) {
        Class<?> entityClass = getEntityClass(ms);
        StringBuilder sql = new StringBuilder();
        //获取全部列
        Set<EntityColumn> columnList = EntityHelper.getColumns(entityClass);
        EntityColumn logicDeleteColumn = SqlHelper.getLogicDeleteColumn(entityClass);
        processKey(sql, entityClass, ms, columnList);
        sql.append(EnhancedSqlHelper.insertIgnoreIntoTable(entityClass, tableName(entityClass)));
        sql.append(SqlHelper.insertColumns(entityClass, false, false, false));
        sql.append("<trim prefix=\"VALUES(\" suffix=\")\" suffixOverrides=\",\">");
        for (EntityColumn column : columnList) {
            if (!column.isInsertable()) {
                continue;
            }
            if (logicDeleteColumn != null && logicDeleteColumn == column) {
                sql.append(SqlHelper.getLogicDeletedValue(column, false)).append(",");
                continue;
            }
            //优先使用传入的属性值,当原属性property!=null时，用原属性
            //自增的情况下,如果默认有值,就会备份到property_cache中,所以这里需要先判断备份的值是否存在
            if (column.isIdentity()) {
                sql.append(SqlHelper.getIfCacheNotNull(column, column.getColumnHolder(null, "_cache", ",")));
            } else {
                //其他情况值仍然存在原property中
                sql.append(SqlHelper.getIfNotNull(column, column.getColumnHolder(null, null, ","), isNotEmpty()));
            }
            //当属性为null时，如果存在主键策略，会自动获取值，如果不存在，则使用null
            if (column.isIdentity()) {
                sql.append(SqlHelper.getIfCacheIsNull(column, column.getColumnHolder() + ","));
            } else {
                //当null的时候，如果不指定jdbcType，oracle可能会报异常，指定VARCHAR不影响其他
                sql.append(SqlHelper.getIfIsNull(column, column.getColumnHolder(null, null, ","), isNotEmpty()));
            }
        }
        sql.append("</trim>");
        return sql.toString();
    }

    public String insertIgnoreSelective(MappedStatement ms) {
        Class<?> entityClass = getEntityClass(ms);
        StringBuilder sql = new StringBuilder();
        //获取全部列
        Set<EntityColumn> columnList = EntityHelper.getColumns(entityClass);
        EntityColumn logicDeleteColumn = SqlHelper.getLogicDeleteColumn(entityClass);
        processKey(sql, entityClass, ms, columnList);
        sql.append(EnhancedSqlHelper.insertIgnoreIntoTable(entityClass, tableName(entityClass)));
        sql.append("<trim prefix=\"(\" suffix=\")\" suffixOverrides=\",\">");
        for (EntityColumn column : columnList) {
            if (!column.isInsertable()) {
                continue;
            }
            if (column.isIdentity()) {
                sql.append(column.getColumn()).append(",");
            } else {
                if (logicDeleteColumn != null && logicDeleteColumn == column) {
                    sql.append(column.getColumn()).append(",");
                    continue;
                }
                sql.append(SqlHelper.getIfNotNull(column, column.getColumn() + ",", isNotEmpty()));
            }
        }
        sql.append("</trim>");

        sql.append("<trim prefix=\"VALUES(\" suffix=\")\" suffixOverrides=\",\">");
        for (EntityColumn column : columnList) {
            if (!column.isInsertable()) {
                continue;
            }
            if (logicDeleteColumn != null && logicDeleteColumn == column) {
                sql.append(SqlHelper.getLogicDeletedValue(column, false)).append(",");
                continue;
            }
            //优先使用传入的属性值,当原属性property!=null时，用原属性
            //自增的情况下,如果默认有值,就会备份到property_cache中,所以这里需要先判断备份的值是否存在
            if (column.isIdentity()) {
                sql.append(SqlHelper.getIfCacheNotNull(column, column.getColumnHolder(null, "_cache", ",")));
            } else {
                //其他情况值仍然存在原property中
                sql.append(SqlHelper.getIfNotNull(column, column.getColumnHolder(null, null, ","), isNotEmpty()));
            }
            //当属性为null时，如果存在主键策略，会自动获取值，如果不存在，则使用null
            //序列的情况
            if (column.isIdentity()) {
                sql.append(SqlHelper.getIfCacheIsNull(column, column.getColumnHolder() + ","));
            }
        }
        sql.append("</trim>");
        return sql.toString();
    }

    private void processKey(StringBuilder sql, Class<?> entityClass, MappedStatement ms, Set<EntityColumn> columnList) {
        //Identity列只能有一个
        Boolean hasIdentityKey = false;
        //先处理cache或bind节点
        for (EntityColumn column : columnList) {
            if (column.isIdentity()) {
                //这种情况下,如果原先的字段有值,需要先缓存起来,否则就一定会使用自动增长
                //这是一个bind节点
                sql.append(SqlHelper.getBindCache(column));
                //如果是Identity列，就需要插入selectKey
                //如果已经存在Identity列，抛出异常
                if (hasIdentityKey) {
                    //jdbc类型只需要添加一次
                    if (column.getGenerator() != null && "JDBC".equals(column.getGenerator())) {
                        continue;
                    }
                    throw new MapperException(ms.getId() + "对应的实体类" + entityClass.getName() + "中包含多个MySql的自动增长列,最多只能有一个!");
                }
                //插入selectKey
                SelectKeyHelper.newSelectKeyMappedStatement(ms, column, entityClass, isBEFORE(), getIDENTITY(column));
                hasIdentityKey = true;
            } else if (column.getGenIdClass() != null) {
                sql.append("<bind name=\"").append(column.getColumn()).append("GenIdBind\" value=\"@tk.mybatis.mapper.genid.GenIdUtil@genId(");
                sql.append("_parameter").append(", '").append(column.getProperty()).append("'");
                sql.append(", @").append(column.getGenIdClass().getName()).append("@class");
                sql.append(", '").append(tableName(entityClass)).append("'");
                sql.append(", '").append(column.getColumn()).append("')");
                sql.append("\"/>");
            }

        }
    }
}

1	package com.github.wz2cool.dynamic.mybatis.mapper.provider;
2
3	import com.github.wz2cool.dynamic.mybatis.mapper.helper.EnhancedSqlHelper;
4	import org.apache.ibatis.mapping.MappedStatement;
5	import tk.mybatis.mapper.MapperException;
6	import tk.mybatis.mapper.entity.EntityColumn;
7	import tk.mybatis.mapper.mapperhelper.EntityHelper;
8	import tk.mybatis.mapper.mapperhelper.MapperHelper;
9	import tk.mybatis.mapper.mapperhelper.SelectKeyHelper;
10	import tk.mybatis.mapper.mapperhelper.SqlHelper;
11	import tk.mybatis.mapper.provider.base.BaseInsertProvider;
12
13	import java.util.Set;
14
15	/**
16	* @author frank
17	*/
18	public class InsertIgnoreProvider extends BaseInsertProvider {
19
20	public InsertIgnoreProvider(Class<?> mapperClass, MapperHelper mapperHelper) {
21	super(mapperClass, mapperHelper);	×
22	}	×
23
24	public String insertIgnore(MappedStatement ms) {
25	Class<?> entityClass = getEntityClass(ms);	×
26	StringBuilder sql = new StringBuilder();	×
27	//获取全部列
28	Set<EntityColumn> columnList = EntityHelper.getColumns(entityClass);	×
29	EntityColumn logicDeleteColumn = SqlHelper.getLogicDeleteColumn(entityClass);	×
30	processKey(sql, entityClass, ms, columnList);	×
31	sql.append(EnhancedSqlHelper.insertIgnoreIntoTable(entityClass, tableName(entityClass)));	×
32	sql.append(SqlHelper.insertColumns(entityClass, false, false, false));	×
33	sql.append("<trim prefix=\"VALUES(\" suffix=\")\" suffixOverrides=\",\">");	×
34	for (EntityColumn column : columnList) {	×
35	if (!column.isInsertable()) {	×
36	continue;	×
37	}
38	if (logicDeleteColumn != null && logicDeleteColumn == column) {	×
39	sql.append(SqlHelper.getLogicDeletedValue(column, false)).append(",");	×
40	continue;	×
41	}
42	//优先使用传入的属性值,当原属性property!=null时，用原属性
43	//自增的情况下,如果默认有值,就会备份到property_cache中,所以这里需要先判断备份的值是否存在
44	if (column.isIdentity()) {	×
45	sql.append(SqlHelper.getIfCacheNotNull(column, column.getColumnHolder(null, "_cache", ",")));	×
46	} else {
47	//其他情况值仍然存在原property中
48	sql.append(SqlHelper.getIfNotNull(column, column.getColumnHolder(null, null, ","), isNotEmpty()));	×
49	}
50	//当属性为null时，如果存在主键策略，会自动获取值，如果不存在，则使用null
51	if (column.isIdentity()) {	×
52	sql.append(SqlHelper.getIfCacheIsNull(column, column.getColumnHolder() + ","));	×
53	} else {
54	//当null的时候，如果不指定jdbcType，oracle可能会报异常，指定VARCHAR不影响其他
55	sql.append(SqlHelper.getIfIsNull(column, column.getColumnHolder(null, null, ","), isNotEmpty()));	×
56	}
57	}	×
58	sql.append("</trim>");	×
59	return sql.toString();	×
60	}
61
62	public String insertIgnoreSelective(MappedStatement ms) {
63	Class<?> entityClass = getEntityClass(ms);	×
64	StringBuilder sql = new StringBuilder();	×
65	//获取全部列
66	Set<EntityColumn> columnList = EntityHelper.getColumns(entityClass);	×
67	EntityColumn logicDeleteColumn = SqlHelper.getLogicDeleteColumn(entityClass);	×
68	processKey(sql, entityClass, ms, columnList);	×
69	sql.append(EnhancedSqlHelper.insertIgnoreIntoTable(entityClass, tableName(entityClass)));	×
70	sql.append("<trim prefix=\"(\" suffix=\")\" suffixOverrides=\",\">");	×
71	for (EntityColumn column : columnList) {	×
72	if (!column.isInsertable()) {	×
73	continue;	×
74	}
75	if (column.isIdentity()) {	×
76	sql.append(column.getColumn()).append(",");	×
77	} else {
78	if (logicDeleteColumn != null && logicDeleteColumn == column) {	×
79	sql.append(column.getColumn()).append(",");	×
80	continue;	×
81	}
82	sql.append(SqlHelper.getIfNotNull(column, column.getColumn() + ",", isNotEmpty()));	×
83	}
84	}	×
85	sql.append("</trim>");	×
86
87	sql.append("<trim prefix=\"VALUES(\" suffix=\")\" suffixOverrides=\",\">");	×
88	for (EntityColumn column : columnList) {	×
89	if (!column.isInsertable()) {	×
90	continue;	×
91	}
92	if (logicDeleteColumn != null && logicDeleteColumn == column) {	×
93	sql.append(SqlHelper.getLogicDeletedValue(column, false)).append(",");	×
94	continue;	×
95	}
96	//优先使用传入的属性值,当原属性property!=null时，用原属性
97	//自增的情况下,如果默认有值,就会备份到property_cache中,所以这里需要先判断备份的值是否存在
98	if (column.isIdentity()) {	×
99	sql.append(SqlHelper.getIfCacheNotNull(column, column.getColumnHolder(null, "_cache", ",")));	×
100	} else {
101	//其他情况值仍然存在原property中
102	sql.append(SqlHelper.getIfNotNull(column, column.getColumnHolder(null, null, ","), isNotEmpty()));	×
103	}
104	//当属性为null时，如果存在主键策略，会自动获取值，如果不存在，则使用null
105	//序列的情况
106	if (column.isIdentity()) {	×
107	sql.append(SqlHelper.getIfCacheIsNull(column, column.getColumnHolder() + ","));	×
108	}
109	}	×
110	sql.append("</trim>");	×
111	return sql.toString();	×
112	}
113
114	private void processKey(StringBuilder sql, Class<?> entityClass, MappedStatement ms, Set<EntityColumn> columnList) {
115	//Identity列只能有一个
116	Boolean hasIdentityKey = false;	×
117	//先处理cache或bind节点
118	for (EntityColumn column : columnList) {	×
119	if (column.isIdentity()) {	×
120	//这种情况下,如果原先的字段有值,需要先缓存起来,否则就一定会使用自动增长
121	//这是一个bind节点
122	sql.append(SqlHelper.getBindCache(column));	×
123	//如果是Identity列，就需要插入selectKey
124	//如果已经存在Identity列，抛出异常
125	if (hasIdentityKey) {	×
126	//jdbc类型只需要添加一次
127	if (column.getGenerator() != null && "JDBC".equals(column.getGenerator())) {	×
128	continue;	×
129	}
130	throw new MapperException(ms.getId() + "对应的实体类" + entityClass.getName() + "中包含多个MySql的自动增长列,最多只能有一个!");	×
131	}
132	//插入selectKey
133	SelectKeyHelper.newSelectKeyMappedStatement(ms, column, entityClass, isBEFORE(), getIDENTITY(column));	×
134	hasIdentityKey = true;	×
135	} else if (column.getGenIdClass() != null) {	×
136	sql.append("<bind name=\"").append(column.getColumn()).append("GenIdBind\" value=\"@tk.mybatis.mapper.genid.GenIdUtil@genId(");	×
137	sql.append("_parameter").append(", '").append(column.getProperty()).append("'");	×
138	sql.append(", @").append(column.getGenIdClass().getName()).append("@class");	×
139	sql.append(", '").append(tableName(entityClass)).append("'");	×
140	sql.append(", '").append(column.getColumn()).append("')");	×
141	sql.append("\"/>");	×
142	}
143
144	}	×
145	}	×
146	}

wz2cool / mybatis-dynamic-query / #393

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous