d

jfwAptOrm/src/main/java/org/jfw/apt/orm/DataAccessObjectHandler.java

@@ -18,6 +18,7 @@ import org.jfw.apt.orm.daoHandler.DeleteBatchHandler;
import org.jfw.apt.orm.daoHandler.DeleteHandler;
import org.jfw.apt.orm.daoHandler.DeleteWithBatchHandler;
import org.jfw.apt.orm.daoHandler.DeleteWithHandler;
import org.jfw.apt.orm.daoHandler.InsertAndGenKeyHandler;
import org.jfw.apt.orm.daoHandler.InsertBatchHandler;
import org.jfw.apt.orm.daoHandler.InsertHandler;
import org.jfw.apt.orm.daoHandler.OracleLimitQueryHandler;
@@ -162,6 +163,7 @@ public class DataAccessObjectHandler extends CodeGenHandler {
		supportedClass.add(PostgreSQLLimitSelectHandler.class);
		supportedClass.add(OracleLimitQueryHandler.class);
		supportedClass.add(OracleLimitSelectHandler.class);
		supportedClass.add(InsertAndGenKeyHandler.class);
		
	}

jfwAptOrm/src/main/java/org/jfw/apt/orm/annotation/dao/method/operator/InsertAndGenKey.java

@@ -0,0 +1,13 @@
package org.jfw.apt.orm.annotation.dao.method.operator;
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.SOURCE)
public @interface InsertAndGenKey {
	String value();
	boolean isLong() default true;
}

jfwAptOrm/src/main/java/org/jfw/apt/orm/core/enums/DE.java

@@ -6,6 +6,8 @@ import org.jfw.apt.orm.core.defaultImpl.*;
public enum DE {
	invalid_de(AbstractColumnHandler.class, null, false,false,false, false, null, null),
	
	boolean_de(BooleanHandler.class, "CHAR(1)",false,true,true,true,null,null),
	Boolean_de(WBooleanHandler.class, "CHAR(1)", true,true,true,true,null,null),
	
@@ -86,8 +88,10 @@ public enum DE {
	DoubleChar(FixLenStringHandler.class,"CHAR(2)",true,true,true,true,null,null),
	
	md5_de(FixLenStringHandler.class,"CHAR(32)",false,true,true,true,null,null),
	Md5_de(FixLenStringHandler.class,"CHAR(32)",true,true,true,true,null,null)	
	Md5_de(FixLenStringHandler.class,"CHAR(32)",true,true,true,true,null,null)	,
	
	bigSerial(LongHandler.class,"bigserial",false,false,false,true,null,null),
	serial(IntHandler.class,"serial",false,false,false,true,null,null)
	;

jfwAptOrm/src/main/java/org/jfw/apt/orm/daoHandler/InsertAndGenKeyHandler.java

@@ -0,0 +1,171 @@
package org.jfw.apt.orm.daoHandler;
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.ListIterator;
import javax.lang.model.element.Element;
import org.jfw.apt.exception.AptException;
import org.jfw.apt.orm.annotation.dao.method.operator.InsertAndGenKey;
import org.jfw.apt.orm.core.ColumnHandlerFactory;
import org.jfw.apt.orm.core.ColumnWriterCache;
import org.jfw.apt.orm.core.model.CalcColumn;
import org.jfw.apt.orm.core.model.Column;
import org.jfw.apt.orm.core.model.DataEntry;
import org.jfw.apt.orm.core.model.DataEntryFactory;
import org.jfw.apt.orm.core.model.Table;
public class InsertAndGenKeyHandler extends BaseDaoHandler {
	protected List<ColumnWriterCache> sqlparams = new ArrayList<ColumnWriterCache>();
	protected Table fromTable;
	private boolean isLong = false;
	private String keyName;
	protected boolean doReplaceSource;
	@Override
	protected void prepare() throws AptException {
		this.sqlparams.clear();
	}
	@Override
	protected void writeMethodContent() throws AptException {
		this.prepareTable();
		this.prepareSqlParams();
		this.prepareSQL();
		if (this.doReplaceSource) {
			cw.l("try{");
		}
		cw.bL("java.sql.PreparedStatement ps = con.prepareStatement(sql");
		cw.el(",1);");
		cw.l("try{");
		this.handleSqlParameters();
		cw.l("ps.executeUpdate();");
		String rsname = cw.getMethodTempVarName();
		cw.bL("java.sql.ResultSet ").w(rsname).el(" =  ps.getGeneratedKeys();");
		cw.l("try{");
		cw.bL("if(").w(rsname).el(".next()){");
		cw.bL("return ").w(rsname).w(".get").w(this.isLong ? "Long" : "Int").w("(\"").w(this.keyName).el("\");");
		cw.l("}else{");
		cw.l("return -1;").l("}");
		cw.l("}finally{").writeClosing(rsname).l("}");
		cw.l("}finally{").writeClosing("ps").l("}");
		if (this.doReplaceSource) {
			cw.l("}finally{");
			this.replaceResource();
			cw.l("}");
		}
	}
	protected void handleSqlParameters() throws AptException {
		for (ListIterator<ColumnWriterCache> it = this.sqlparams.listIterator(); it.hasNext();) {
			it.next().writeValue();
		}
	}
	protected void prepareTable() throws AptException {
		String entryJavaName = null;
			if (this.me.getParams().size() != 2)
				throw new AptException(ref, "method paramters size must be 2 and the second paramter with @" + Table.class.getName());
			entryJavaName = this.me.getParams().get(1).getTypeName();
				this.fromTable = (Table) DataEntryFactory.get(entryJavaName, DataEntry.TABLE);
		if (null == this.fromTable)
			throw new AptException(ref, "class[" + entryJavaName + "] not with @" + org.jfw.apt.orm.annotation.entry.Table.class.getName());
	}
	@Override
	protected void checkJdbc() throws AptException {
		super.checkJdbc();
		InsertAndGenKey ing = this.ref.getAnnotation(InsertAndGenKey.class);
		this.isLong = ing.isLong();
		this.keyName = ing.value();
		String rt = this.me.getReturnType();
		if(isLong){
			if (!"long".equals(rt))
				throw new AptException(ref, "mehtod must return long");
		}else{
			if (!"int".equals(rt))
				throw new AptException(ref, "mehtod must return int");
	}}
	protected void replaceResource() {
		for (ColumnWriterCache cwc : this.sqlparams) {
			if (cwc.isReplaceResource())
				cwc.replaceResource();
		}
	}
	@Override
	public boolean match(Element ele) {
		return null != ele.getAnnotation(InsertAndGenKey.class);
	}
	protected void prepareSqlParams() {
		this.doReplaceSource = false;
		String pn = this.me.getParams().get(1).getName();
		LinkedList<Column> list = new LinkedList<Column>();
		for (ListIterator<CalcColumn> it = this.fromTable.getAllColumn().listIterator(); it.hasNext();) {
			Column col = (Column) it.next();
			if (col.isInsertable()) {
				if (null == col.getFixSqlValueWithInsert()) {
					list.add(col);
				} else {
					list.add(0, col);
				}
			}
		}
		cw.bL("String sql=\"INSERT INTO ").w(this.fromTable.getFromSentence()).w(" (");
		boolean first = true;
		for (ListIterator<Column> it = list.listIterator(); it.hasNext();) {
			if (first)
				first = false;
			else
				cw.w(",");
			Column col = it.next();
			cw.w(col.getSqlName());
		}
		cw.w(") VALUES (");
		first = true;
		for (ListIterator<Column> it = list.listIterator(); it.hasNext();) {
			if (first)
				first = false;
			else
				cw.w(",");
			Column col = it.next();
			if (null == col.getFixSqlValueWithInsert()) {
				cw.w("?");
				ColumnWriterCache cwc = ColumnWriterCache.build(pn + "." + col.getGetter() + "()", true, col.isNullable(), cw,
						ColumnHandlerFactory.get(col.getHandlerClass()));
				cwc.setDynamic(false);
				if (cwc.isReplaceResource())
					this.doReplaceSource = true;
				cwc.setDynamic(false);
				this.sqlparams.add(cwc);
			} else {
				cw.ws(col.getFixSqlValueWithInsert());
			}
		}
		cw.el(")\";");
		for (ColumnWriterCache cwc : this.sqlparams) {
			cwc.prepare();
		}
	}
	protected void prepareSQL() {
	}
}