cn.tom.db.jdbc.ProcedureCallback Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of cocook-mvc Show documentation
cocook mvc kits
There is a newer version: 1.0.3
package cn.tom.db.jdbc;

import java.sql.CallableStatement;
import java.sql.Connection;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.LinkedHashMap;
import java.util.Map;

import cn.tom.db.jdbc.simple.DBUtil;
import cn.tom.db.jdbc.simple.ProceUtils;


public abstract class ProcedureCallback {

	String str_sql = null;

	Map paramsMap = new LinkedHashMap();

	public ProcedureCallback() {
	}

	protected void registerInParameter() {
		paramsMap.put(paramsMap.size() + 1, null);
	}

	protected void registerOutParameter(String type, String name) {
		paramsMap.put(paramsMap.size() + 1, new Object[] { name, type });
	}

	public ProceRecord doInCallback(Connection conn, String sql, Object... objects) throws SQLException {

		this.str_sql = parse(sql); // parse(sql) 方法需要被重写

		CallableStatement cs = null;

		cs = conn.prepareCall(this.str_sql);

		try {

			for (int i = 0; i < paramsMap.size(); i++) {
				Integer key = i + 1;
				Object[] p = paramsMap.get(key);
				if (p == null) {
					if (objects == null || i >= objects.length) {
						throw new IllegalArgumentException("not enought parameter");
					}
					ProceUtils.bindStatementInput(cs, key, objects[i]); // 设置输入参数
				} else {
					String type = (String) p[1];
					ProceUtils.bindStatementOutput(cs, key, type); // 设置输出参数u
				}
			}

			return getResult(cs, cs.execute());
		} finally {
			DBUtil.close(conn);
		}
	}

	public String parse(String sql) { // 解析正确的存储过程需要重写
		return sql;
	}

	protected ProceRecord getResult(CallableStatement cstmt, boolean executeResult) throws SQLException {

		Map retmap = new LinkedHashMap();

		int index = 0;

		if (executeResult) { // 有结果集返回
			ResultSet rs = null;
			try {
				rs = cstmt.getResultSet();
				retmap.put("DATASET_" + index++, ProceUtils.convertResultSetToMapList(rs, false));

				while (cstmt.getMoreResults()) {// 还有结果集
					rs = cstmt.getResultSet();
					retmap.put("DATASET_" + index++, ProceUtils.convertResultSetToMapList(rs, false));
				}
			} finally {
				DBUtil.close(rs);
			}
		}

		for (int i = 0; i < paramsMap.size(); i++) {
			Integer key = i + 1;
			Object[] p = paramsMap.get(key);
			if (p != null) {
				String name = (String) p[0]; // 输出参数名称
				String type = (String) p[1]; // 输出参数类型
				retmap.put(name.toUpperCase(), ProceUtils.convertType(cstmt, key, type)); // 输出参数如果是Oracle
																							// cursor直接转成ResultSet
			}
		}

		return new ProceRecord(retmap);
	}
}