com.scudata.server.unit.JdbcTask Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of esproc Show documentation
SPL(Structured Process Language) A programming language specially for structured data computing.
There is a newer version: 20240823
package com.scudata.server.unit;

import java.io.Serializable;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.Map;

import com.esproc.jdbc.JDBCUtil;
import com.scudata.cellset.datamodel.PgmCellSet;
import com.scudata.common.StringUtils;
import com.scudata.dm.Context;
import com.scudata.dm.RetryException;
import com.scudata.dm.Sequence;
import com.scudata.dm.cursor.ICursor;
import com.scudata.parallel.Request;

/**
 * JDBCִ??????
 * 
 * @author Joancy
 *
 */
public class JdbcTask {
	String cmd;
	ArrayList args;
	Context context;
	Map envParams;
	Thread execThread = null;

	/**
	 * ????һ??JDBC????
	 * @param cmd ????ִ?е?????
	 * @param args ?????б?
	 * @param ctx ??????????
	 * @param envParams ????????
	 */
	public JdbcTask(String cmd, ArrayList args, Context ctx,
			Map envParams) {
		this.cmd = cmd;
		this.args = (ArrayList) args;
		this.context = ctx;
		this.envParams = envParams;
	}

	/**
	 * ִ?е?ǰ????
	 * @return ??????
	 * @throws Exception
	 */
	public Sequence execute() throws Exception {
		Object result = executeJDBC();
		if (result == null)
			return null;
		Sequence seq = new Sequence();
		if (result instanceof PgmCellSet) {
			PgmCellSet cs = (PgmCellSet) result;
			while (cs.hasNextResult()) {
				seq.add(checkResult(cs.nextResult()));
			}
		} else {
			seq.add(checkResult(result));
		}
		return seq;
	}

	public static Object checkResult(Object r) throws Exception {
		if(r==null)
			return null;
		if (r instanceof ICursor) {// ?α겻??ʵ?????л????????Զ???α?
			return r;
		}
		if (!(r instanceof Serializable)) {
			throw new Exception("Return result " + r.getClass().getName()
					+ " is not supportted.");
		}
		return r;
	}

	/**
	 * ȡ????ǰ????
	 * @return ?ɹ?ȡ??????true
	 * @throws Exception
	 */
	public boolean cancel() throws Exception {
		if (execThread != null) {
			try {
				execThread.stop();
			} catch (Throwable t1) {
			}
			try {
				execThread.destroy();
			} catch (Throwable t1) {
			}
			execThread = null;
		}
		return true;
	}

	/**
	 * ??ȡִ?????????dfx????
	 * @return ????
	 */
	public String getCmd() {
		return cmd;
	}

	/**
	 * ??ȡ?????б?
	 * @return dfx????
	 */
	public ArrayList getArgs() {
		return args;
	}

	private Exception ex;
	private boolean isCanceled = false;
	private Object result;

	private Object executeJDBC() throws Exception {
		try {
			ex = null;
			result = null;
			isCanceled = false;
			execThread = new Thread() {
				public void run() {
					try {
						Object gateway = envParams
								.get(Request.PREPARE_ENV_GATEWAY);
						boolean execGateway = false;
						if (StringUtils.isValidString(gateway)) {
							try {
								result = JDBCUtil.executeGateway(cmd, args,
										context, (String) gateway);
								execGateway = true;
							} catch (RetryException re) {
								// ??end??ʽ????ʱ???????????ط?ʽִ??
							}
						}
						if (!execGateway)
							result = JDBCUtil.execute(cmd, args, context);
					} catch (ThreadDeath td) {
						isCanceled = true;
					} catch (SQLException e) {
						ex = e;
					} catch (Exception e) {
						Throwable t = e;
						while (t != null) {
							if (t instanceof ThreadDeath) {
								ex = new InterruptedException();
								break;
							}
							t = t.getCause();
						}
						if (ex == null)
							ex = new SQLException(e.getMessage(), e);
					}
				}
			};
			execThread.start();
			execThread.join();
			if (ex != null) {
				throw ex;
			}
		} catch (ThreadDeath td) {
			isCanceled = true;
		}
		if (isCanceled)
			throw new InterruptedException();
		return result;
	}
}