cn.coder.struts.wrapper.EntityWrapper Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of struts Show documentation
A simple, light Java WEB + ORM framework.
There is a newer version: 1.0.0
package cn.coder.struts.wrapper;

import java.lang.reflect.Field;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Set;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import cn.coder.struts.annotation.Column;
import cn.coder.struts.annotation.Id;
import cn.coder.struts.annotation.Table;
import cn.coder.struts.util.FieldUtils;

public class EntityWrapper {
	private static final Logger logger = LoggerFactory.getLogger(EntityWrapper.class);
	private String tableName;
	private Field generateKeyField;
	private ArrayList primaryKeys;
	private HashMap columns;
	private Object obj;
	private SQLType type;
	private Object[] data;
	private String sql;

	public EntityWrapper(Class clazz) {
		findTable(clazz);
		findColumns(clazz);
	}

	private void findTable(Class clazz) {
		Table table = clazz.getAnnotation(Table.class);
		this.tableName = table.value();
	}

	public EntityWrapper setObject(Object data, SQLType type) {
		this.obj = data;
		this.type = type;
		fillColumns();
		switch (type) {
		case INSERT:
			buildInsert();
			break;
		case UPDATE:
			buildUpdate();
			break;
		default:
			break;
		}
		return this;
	}

	public String getSql() {
		return this.sql;
	}

	public Object[] getData() {
		return this.data;
	}

	public boolean returnGeneratedKey() {
		return type == SQLType.INSERT && generateKeyField != null;
	}

	public void setGeneratedKey(int value) {
		try {
			FieldUtils.setValue(this.generateKeyField, this.obj, value);
		} catch (SQLException e) {
			e.printStackTrace();
		}
	}

	public void clear() {
		Set keys = columns.keySet();
		for (String key : keys) {
			columns.put(key, null);
		}
		this.obj = null;
		this.type = null;
		this.sql = null;
		this.data = null;
	}

	private void findColumns(Class clazz) {
		Column col;
		this.columns = new HashMap<>();
		this.primaryKeys = new ArrayList<>();
		Set fields = FieldUtils.getDeclaredFields(clazz);
		for (Field field : fields) {
			col = field.getAnnotation(Column.class);
			if (col != null) {
				columns.put(col.value(), null);
				Id id = field.getAnnotation(Id.class);
				if (id != null) {
					primaryKeys.add(col.value());
					if (id.value()) {
						this.generateKeyField = field;
					}
				}
			}
		}
	}

	private void fillColumns() {
		Column col;
		Set fields = FieldUtils.getDeclaredFields(obj.getClass());
		for (Field field : fields) {
			col = field.getAnnotation(Column.class);
			if (col != null && columns.containsKey(col.value())) {
				try {
					if (!field.isAccessible())
						field.setAccessible(true);
					columns.put(col.value(), field.get(obj));
				} catch (IllegalArgumentException | IllegalAccessException e) {
					e.printStackTrace();
				}
			}
		}
	}

	private void buildInsert() {
		try {
			long start = System.nanoTime();
			List