com.obatis.orm.sql.mysql.HandleInsertBatchMethod Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of obatis-orm Show documentation
obatis-orm, apply database project
The newest version!
package com.obatis.orm.sql.mysql;

import com.obatis.convert.date.DateConvert;
import com.obatis.orm.model.CommonField;
import com.obatis.orm.model.CommonModel;
import com.obatis.orm.annotation.NotColumn;
import com.obatis.orm.constant.CacheInfoConstant;
import com.obatis.orm.constant.SqlConstant;
import com.obatis.exception.HandleException;
import com.obatis.orm.sql.AbstractInsertMethod;
import com.obatis.generator.NumberGenerator;
import com.obatis.tools.ValidateTool;

import java.lang.reflect.Field;
import java.lang.reflect.Modifier;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * MySQL 批量添加方法实现
 * @author HuangLongPu
 */
public class HandleInsertBatchMethod extends AbstractInsertMethod {

	@Override
	protected String handleBatchInsertSql(List list, Class cls, String tableName) {
		StringBuffer sql = new StringBuffer("insert into " + tableName + "(");
        Map res = this.getBatchInsertFields(list, cls, tableName);
        if (res == null) {
            throw new HandleException("error：object is empty");
        }

        sql.append(res.get(SqlConstant.BEAN_FIELD) + ")");
        return sql.toString() + " values " + res.get(SqlConstant.BEAN_VALUE);
	}

	protected Map getBatchInsertFields(List list, Class cls, String tableName) {

		List fieldArr = new ArrayList<>();
		List valueArr = new ArrayList<>();

		if (list.isEmpty() || list.size() == 0) {
			throw new HandleException("error: batch insert list is empty");
		}

		Map columnMap = CacheInfoConstant.COLUMN_CACHE.get(tableName);
		for (int i = 0, j = list.size(); i < j; i++) {
			Object obj = list.get(i);
			if (!(obj instanceof CommonModel)) {
				throw new HandleException("error: entity is not instanceof CommonModel");
			}
			List colValueArr = new ArrayList<>();
			getBatchInsertValidColumnFields(obj, cls, columnMap, i, fieldArr, colValueArr);
			valueArr.add("(" + String.join(",", colValueArr) + ")");
		}

		if (fieldArr.size() > 0) {
			Map res = new HashMap<>();
			res.put(SqlConstant.BEAN_FIELD, String.join(",", fieldArr));
			res.put(SqlConstant.BEAN_VALUE, String.join(",", valueArr));
			return res;
		} else {
			return null;
		}
	}

	private void getBatchInsertValidColumnFields(Object obj, Class cls, Map columnMap, int index, List fieldArr,
												 List colValueArr) {

		Field[] fields = cls.getDeclaredFields();
		for (Field field : fields) {

			boolean isStatic = Modifier.isStatic(field.getModifiers());
			if (isStatic) {
				continue;
			}
			NotColumn ts = field.getAnnotation(NotColumn.class);
			if (ts != null) {
				continue;
			}

			String fieldName = field.getName();
			if (!columnMap.containsKey(fieldName)) {
				continue;
			}
			String columnName = columnMap.get(fieldName);

			try {
				field.setAccessible(true);
				Object value = field.get(obj);
				if (ValidateTool.isEmpty(value)) {
					if (CommonField.FIELD_ID.equals(columnName)) {
						field.set(obj, NumberGenerator.getNumber());
					} else if (CommonField.FIELD_CREATE_TIME.equals(columnName)) {
						field.set(obj, DateConvert.getDateTime());
					}
				}

				if (index == 0) {
					fieldArr.add(columnName);
				}
				colValueArr.add("#{request[" + index + "]." + fieldName + "}");
			} catch (Exception e) {
				e.printStackTrace();
				throw new HandleException("error: load class fields fail");
			}
		}

		Class supCls = cls.getSuperclass();
		if (supCls != null) {
			getBatchInsertValidColumnFields(obj, supCls, columnMap, index, fieldArr, colValueArr);
		}
	}
}