com.obatis.orm.sql.AbstractInsertMethod 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;

import com.obatis.convert.date.DateConvert;
import com.obatis.orm.annotation.NotColumn;
import com.obatis.orm.constant.CacheInfoConstant;
import com.obatis.orm.constant.SqlConstant;
import com.obatis.orm.model.CommonField;
import com.obatis.exception.HandleException;
import com.obatis.generator.NumberGenerator;
import com.obatis.tools.ValidateTool;
import org.apache.ibatis.jdbc.SQL;

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;

public abstract class AbstractInsertMethod {

	protected String handleInsertSql(Object object, Class clazz, String tableName) throws HandleException {

        SQL sql = new SQL();
        sql.INSERT_INTO(tableName);
        Map res = this.getInsertFields(object, clazz, tableName);
        if (res == null) {
            throw new HandleException("error：object is null");
        }

        sql.INTO_COLUMNS(res.get(SqlConstant.BEAN_FIELD));
        sql.INTO_VALUES(res.get(SqlConstant.BEAN_VALUE));
        return sql.toString();
    }
	
	private Map getInsertFields(Object object, Class clazz, String tableName) throws HandleException {
		List fields = new ArrayList<>();
		List values = new ArrayList<>();

		Map columnMap = CacheInfoConstant.COLUMN_CACHE.get(tableName);
		getInsertFieldVaule(clazz, columnMap, object, fields, values);

		if (fields.size() > 0) {
			Map res = new HashMap<>();
			res.put(SqlConstant.BEAN_FIELD, String.join(",", fields));
			res.put(SqlConstant.BEAN_VALUE, String.join(",", values));
			return res;
		} else {
			return null;
		}
	}
	
	private void getInsertFieldVaule(Class clazz, Map columnMap, Object obj, List fields, List values)
			throws HandleException {

		Field[] fieldArr = clazz.getDeclaredFields();
		for (Field field : fieldArr) {

			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);
				boolean addFlag = false;
				if (ValidateTool.isEmpty(value)) {
					if (CommonField.FIELD_ID.equals(columnName)) {
						field.set(obj, NumberGenerator.getNumber());
						addFlag = true;
					} else if (CommonField.FIELD_CREATE_TIME.equals(columnName)) {
						field.set(obj, DateConvert.getDateTime());
						addFlag = true;
					}
				} else {
					// 说明值不为空
					addFlag = true;
				}

				if (addFlag) {
					fields.add(columnName);
					values.add("#{request." + fieldName + "}");
				}
			} catch (Exception e) {
				e.printStackTrace();
				throw new HandleException("error: load class fields fail");
			}
		}

		Class supCls = clazz.getSuperclass();
		if (supCls != null) {
			getInsertFieldVaule(supCls, columnMap, obj, fields, values);
		}
	}
	
	protected abstract String handleBatchInsertSql(List list, Class cls, String tableName);
}