done_handling.c

在Linux/Unix下面访问WINDOWS SQLSERVER 的ODBC驱动程序

#include "common.h"

static char software_version[] = "$Id: done_handling.c,v 1.7 2007/12/28 23:00:42 jklowden Exp $";
static void *no_unused_var_warn[] = { software_version, no_unused_var_warn };

/*
 * This test try do discovery how dblib process token looking for state
 * at every iteration. It issue a query to server and check
 * - row count (valid number means DONE processed)
 * - if possible to send another query (means state IDLE)
 * - if error readed (means ERROR token readed)
 * - if status present (PARAMRESULT token readed)
 * - if parameter present (PARAM token readed)
 * It try these query types:
 * - normal row
 * - normal row with no count
 * - normal row without rows
 * - error query
 * - store procedure call with output parameters
 */

/* Forward declarations of the error handler and message handler. */
static int err_handler(DBPROCESS * dbproc, int severity, int dberr, int oserr, char *dberrstr, char *oserrstr);
static int msg_handler(DBPROCESS * dbproc, DBINT msgno, int msgstate, int severity, char *msgtext, char *srvname, char *procname,
		       int line);

static DBPROCESS *dbproc;
static int silent = 0;
static int check_idle = 0;

static void
query(const char *query)
{
	printf("query: %s\n", query);
	dbcmd(dbproc, (char *) query);
	dbsqlexec(dbproc);
	while (dbresults(dbproc) == SUCCEED) {
		/* nop */
	}
}

static void
check_state(void)
{
	printf("State ");
	if (dbnumcols(dbproc) > 0)
		printf("COLS(%d) ", dbnumcols(dbproc));
	/* row count */
	if (dbcount(dbproc) >= 0)
		printf("ROWS(%d) ", (int) dbcount(dbproc));
	silent = 1;
	if (dbdata(dbproc, 1))
		printf("DATA ");
	silent = 0;
	/* if status present */
	if (dbretstatus(dbproc) == TRUE)
		printf("STATUS %d ", (int) dbretstatus(dbproc));
	/* if parameter present */
	if (dbnumrets(dbproc) > 0)
		printf("PARAMS ");
	/*
	 * if possible to send another query
	 * NOTE this must be the last
	 */
	if (check_idle) {
		silent = 1;
		dbcmd(dbproc, "declare @i int ");
		if (FAIL != dbsqlexec(dbproc))
			printf("IDLE ");
		silent = 0;
	}
	printf("\n");
}

static void
do_test(const char *query)
{
	int ret;

	printf("test query %s\n", query);
	dbcmd(dbproc, (char *) query);

	printf("sqlexec ");
	dbsqlexec(dbproc);

	check_state();

	printf("nextrow ");
	dbnextrow(dbproc);

	check_state();

	printf("nextrow ");
	dbnextrow(dbproc);

	check_state();

	printf("results ");
	dbresults(dbproc);

	check_state();

	printf("nextrow ");
	dbnextrow(dbproc);

	check_state();

	printf("nextrow ");
	dbnextrow(dbproc);

	check_state();

	check_idle = 0;
	for (;;) {
		printf("results \n");
		ret = dbresults(dbproc);
		check_state();
		if (ret != SUCCEED)
			break;

		do {
			printf("nextrow ");
			ret = dbnextrow(dbproc);
			check_state();
		} while (ret == SUCCEED);
	}
	check_state();
}

int
main(int argc, char *argv[])
{
	LOGINREC *login;	/* Our login information. */

	setbuf(stdout, NULL);
	read_login_info(argc, argv);

	if (dbinit() == FAIL)
		exit(1);

	dberrhandle(err_handler);
	dbmsghandle(msg_handler);

	login = dblogin();
	DBSETLUSER(login, USER);
	DBSETLPWD(login, PASSWORD);
	DBSETLAPP(login, __FILE__);

	dbproc = dbopen(login, SERVER);
	dbloginfree(login);
	if (!dbproc)
		exit(1);
	if (strlen(DATABASE))
		dbuse(dbproc, DATABASE);

	query("create table #dummy (s char(10))");
	query("insert into #dummy values('xxx')");
	query("if object_id('done_test') is not NULL drop proc done_test");
	query("create proc done_test @a varchar(10) output as select * from #dummy");
	query("if object_id('done_test2') is not NULL drop proc done_test2");
	query("create proc done_test2 as select * from #dummy where s = 'aaa' select * from #dummy");

	check_idle = 1;

	/* normal row */
	do_test("select * from #dummy");
	/* normal row with no count */
	query("set nocount on");
	do_test("select * from #dummy");
	query("set nocount off");
	/* normal row without rows */
	do_test("select * from #dummy where 0=1");
	/* error query */
	do_test("select dklghdlgkh from #dummy");
	/* store procedure call with output parameters */
	do_test("declare @s varchar(10) exec done_test @s output");

	do_test("declare @i int");

	do_test("exec done_test2");
	do_test("declare @i int");

	query("drop proc done_test");
	query("drop proc done_test2");

	dbexit();
	return 0;
}

static int
err_handler(DBPROCESS * dbproc, int severity, int dberr, int oserr, char *dberrstr, char *oserrstr)
{
	if (silent)
		return INT_CANCEL;

	fflush(stdout);
	fprintf(stderr, "DB-Library error (severity %d):\n\t%s\n", severity, dberrstr);

	if (oserr != DBNOERR)
		fprintf(stderr, "Operating-system error:\n\t%s\n", oserrstr);
	fflush(stderr);

	return INT_CANCEL;
}

static int
msg_handler(DBPROCESS * dbproc, DBINT msgno, int msgstate, int severity, char *msgtext, char *srvname, char *procname, int line)
{
	if (silent)
		return 0;

	fflush(stdout);
	fprintf(stderr, "Msg %d, Level %d, State %d\n", (int) msgno, severity, msgstate);

	if (strlen(srvname) > 0)
		fprintf(stderr, "Server '%s', ", srvname);
	if (procname && strlen(procname) > 0) {
		fprintf(stderr, "Procedure '%s', ", procname);
		if (line > 0)
			fprintf(stderr, "Line %d", line);
	}

	fprintf(stderr, "\n\t%s\n", msgtext);
	fflush(stderr);

	return 0;
}