1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
/*
* Test FetchRefcursors setting
*/
#include "common.h"
static void print_all_results(HSTMT hstmt)
{
int i;
int rc = SQL_SUCCESS;
for (i = 1; rc == SQL_SUCCESS || rc == SQL_SUCCESS_WITH_INFO; i++)
{
printf("--%d ", i);
print_result(hstmt);
rc = SQLMoreResults(hstmt);
}
if (rc != SQL_NO_DATA)
CHECK_STMT_RESULT(rc, "SQLMoreResults failed", hstmt);
}
static void setup_procedure()
{
SQLRETURN rc;
HSTMT hstmt = SQL_NULL_HSTMT;
printf("Creating procedure 'refproc'\n");
test_connect();
rc = SQLAllocHandle(SQL_HANDLE_STMT, conn, &hstmt);
CHECK_CONN_RESULT(rc, "failed to allocate stmt handle", conn);
rc = SQLExecDirect(hstmt, "create or replace procedure refproc"
"(inout num_cursor integer, inout ref1 refcursor default null, inout ref2 refcursor default null) as "
"$procedure$ \n"
"DECLARE \n"
"BEGIN \n"
"IF num_cursor > 0 THEN \n"
" OPEN ref1 FOR SELECT id, t FROM testtab1 ORDER BY id ASC; \n"
"END IF; \n"
"IF num_cursor > 1 THEN \n"
" OPEN ref2 FOR SELECT t, id FROM testtab1 ORDER BY id DESC; \n"
"END IF; \n"
"IF num_cursor > 2 THEN \n"
" num_cursor := 2; \n"
"END IF; \n"
"END; \n"
"$procedure$ \n"
"LANGUAGE plpgsql\n"
, SQL_NTS);
CHECK_STMT_RESULT(rc, "create procedure refproc failed", hstmt);
rc = SQLFreeStmt(hstmt, SQL_CLOSE);
CHECK_STMT_RESULT(rc, "SQLFreeStmt failed", hstmt);
test_disconnect();
}
static void refcursor_test(char* connectparams, SQLUINTEGER autocommit, int numresults)
{
SQLRETURN rc;
HSTMT hstmt = SQL_NULL_HSTMT;
int num_cursor = numresults;
printf("\n-- TEST using %s, autocommit=%u, numresults=%d\n", connectparams, autocommit, numresults);
test_connect_ext(connectparams);
/* Conditionally start a transaction */
rc = SQLSetConnectAttr(conn, SQL_ATTR_AUTOCOMMIT, (SQLPOINTER) (SQLULEN) autocommit, SQL_IS_UINTEGER);
CHECK_STMT_RESULT(rc, "SQLSetConnectAttr failed", hstmt);
rc = SQLAllocHandle(SQL_HANDLE_STMT, conn, &hstmt);
CHECK_CONN_RESULT(rc, "failed to allocate stmt handle", conn);
rc = SQLPrepare(hstmt, "CALL refproc(?)", SQL_NTS);
CHECK_STMT_RESULT(rc, "SQLPrepare failed", hstmt);
rc = SQLBindParameter(hstmt, 1, SQL_PARAM_INPUT_OUTPUT,
SQL_C_LONG, /* value type */
SQL_INTEGER, /* param type */
0, /* column size */
0, /* dec digits */
&num_cursor, /* param value ptr */
0, /* buffer len */
NULL /* StrLen_or_IndPtr */);
CHECK_STMT_RESULT(rc, "SQLBindParameter failed", hstmt);
rc = SQLExecute(hstmt);
if (!SQL_SUCCEEDED(rc))
{
print_diag("SQLExecute failed", SQL_HANDLE_STMT, hstmt);
return;
}
printf("Output param num_cursor is %d\n", num_cursor);
print_all_results(hstmt);
rc = SQLFreeStmt(hstmt, SQL_CLOSE);
CHECK_STMT_RESULT(rc, "SQLFreeStmt failed", hstmt);
test_disconnect();
}
int main(int argc, char **argv)
{
setup_procedure();
/* Using a fetch cache size of 1 to test multiple fetches per cursor */
refcursor_test("Fetch=1;FetchRefcursors=0", SQL_AUTOCOMMIT_ON, 2);
refcursor_test("Fetch=1;FetchRefcursors=1", SQL_AUTOCOMMIT_ON, 2);
refcursor_test("Fetch=1;FetchRefcursors=1", SQL_AUTOCOMMIT_OFF, 0);
refcursor_test("Fetch=1;FetchRefcursors=1", SQL_AUTOCOMMIT_OFF, 1);
refcursor_test("Fetch=1;FetchRefcursors=1", SQL_AUTOCOMMIT_OFF, 2);
refcursor_test("Fetch=1;FetchRefcursors=1", SQL_AUTOCOMMIT_OFF, 3);
return 0;
}
|
