Line data Source code
1 : #include <config.h>
2 :
3 : #if HAVE_STRING_H
4 : #include <string.h>
5 : #endif /* HAVE_STRING_H */
6 :
7 : #include <stdio.h>
8 : #include <ctpublic.h>
9 : #include "common.h"
10 :
11 : static CS_RETCODE ex_servermsg_cb(CS_CONTEXT * context, CS_CONNECTION * connection, CS_SERVERMSG * errmsg);
12 : static int compute_supported = 1;
13 :
14 : /* Testing: Retrieve compute results */
15 : int
16 8 : main(int argc, char *argv[])
17 : {
18 : CS_CONTEXT *ctx;
19 : CS_CONNECTION *conn;
20 : CS_COMMAND *cmd;
21 8 : int verbose = 0;
22 :
23 : CS_RETCODE ret;
24 : CS_RETCODE results_ret;
25 : CS_INT result_type;
26 : CS_INT num_cols, compute_id;
27 :
28 : CS_DATAFMT datafmt;
29 : CS_INT datalength;
30 : CS_SMALLINT ind;
31 8 : CS_INT count, row_count = 0;
32 :
33 : CS_CHAR select[1024];
34 :
35 : CS_INT col1;
36 : CS_CHAR col2[2];
37 : CS_CHAR col3[32];
38 :
39 : CS_INT compute_col1;
40 : CS_CHAR compute_col3[32];
41 :
42 8 : unsigned rows[3] = { 0, 0, 0 };
43 :
44 8 : printf("%s: Retrieve compute results processing\n", __FILE__);
45 : if (verbose) {
46 : printf("Trying login\n");
47 : }
48 8 : ret = try_ctlogin(&ctx, &conn, &cmd, verbose);
49 8 : if (ret != CS_SUCCEED) {
50 0 : fprintf(stderr, "Login failed\n");
51 0 : return 1;
52 : }
53 :
54 8 : ret = run_command(cmd, "CREATE TABLE #ctlib0009 (col1 int not null, col2 char(1) not null, col3 datetime not null)");
55 8 : if (ret != CS_SUCCEED)
56 : return 1;
57 :
58 8 : ret = run_command(cmd, "insert into #ctlib0009 values (1, 'A', 'Jan 1 2002 10:00:00AM')");
59 8 : if (ret != CS_SUCCEED)
60 : return 1;
61 :
62 8 : ret = run_command(cmd, "insert into #ctlib0009 values (2, 'A', 'Jan 2 2002 10:00:00AM')");
63 8 : if (ret != CS_SUCCEED)
64 : return 1;
65 :
66 8 : ret = run_command(cmd, "insert into #ctlib0009 values (3, 'A', 'Jan 3 2002 10:00:00AM')");
67 8 : if (ret != CS_SUCCEED)
68 : return 1;
69 :
70 8 : ret = run_command(cmd, "insert into #ctlib0009 values (8, 'B', 'Jan 4 2002 10:00:00AM')");
71 8 : if (ret != CS_SUCCEED)
72 : return 1;
73 :
74 8 : ret = run_command(cmd, "insert into #ctlib0009 values (9, 'B', 'Jan 5 2002 10:00:00AM')");
75 8 : if (ret != CS_SUCCEED)
76 : return 1;
77 :
78 :
79 8 : strcpy(select, "select col1, col2, col3 from #ctlib0009 order by col2 ");
80 8 : strcat(select, "compute sum(col1) by col2 ");
81 8 : strcat(select, "compute max(col3)");
82 :
83 8 : ret = ct_command(cmd, CS_LANG_CMD, select, CS_NULLTERM, CS_UNUSED);
84 :
85 8 : if (ret != CS_SUCCEED) {
86 0 : fprintf(stderr, "ct_command(%s) failed\n", select);
87 0 : return 1;
88 : }
89 :
90 8 : ret = ct_send(cmd);
91 8 : if (ret != CS_SUCCEED) {
92 0 : fprintf(stderr, "ct_send() failed\n");
93 0 : return 1;
94 : }
95 :
96 8 : ct_callback(ctx, NULL, CS_SET, CS_SERVERMSG_CB, (CS_VOID *) ex_servermsg_cb);
97 64 : while ((results_ret = ct_results(cmd, &result_type)) == CS_SUCCEED) {
98 48 : printf("ct_results returned %s type\n", res_type_str(result_type));
99 48 : switch ((int) result_type) {
100 : case CS_CMD_SUCCEED:
101 : break;
102 : case CS_CMD_DONE:
103 : break;
104 0 : case CS_CMD_FAIL:
105 0 : if (!compute_supported) {
106 0 : try_ctlogout(ctx, conn, cmd, verbose);
107 0 : return 0;
108 : }
109 0 : fprintf(stderr, "ct_results() result_type CS_CMD_FAIL.\n");
110 0 : return 1;
111 16 : case CS_ROW_RESULT:
112 16 : ret = ct_res_info(cmd, CS_NUMDATA, &num_cols, CS_UNUSED, NULL);
113 16 : if (ret != CS_SUCCEED) {
114 0 : fprintf(stderr, "ct_res_info() failed");
115 0 : return 1;
116 : }
117 16 : if (num_cols != 3) {
118 0 : fprintf(stderr, "num_cols %d != 3", num_cols);
119 0 : return 1;
120 : }
121 :
122 16 : ret = ct_describe(cmd, 1, &datafmt);
123 16 : if (ret != CS_SUCCEED) {
124 0 : fprintf(stderr, "ct_describe() failed");
125 0 : return 1;
126 : }
127 16 : datafmt.format = CS_FMT_UNUSED;
128 16 : if (datafmt.maxlength > 1024) {
129 0 : datafmt.maxlength = 1024;
130 : }
131 16 : ret = ct_bind(cmd, 1, &datafmt, &col1, &datalength, &ind);
132 16 : if (ret != CS_SUCCEED) {
133 0 : fprintf(stderr, "ct_bind() failed\n");
134 0 : return 1;
135 : }
136 :
137 16 : ret = ct_describe(cmd, 2, &datafmt);
138 16 : if (ret != CS_SUCCEED) {
139 0 : fprintf(stderr, "ct_describe() failed");
140 0 : return 1;
141 : }
142 :
143 16 : datafmt.format = CS_FMT_NULLTERM;
144 16 : datafmt.maxlength = 2;
145 :
146 16 : ret = ct_bind(cmd, 2, &datafmt, col2, &datalength, &ind);
147 16 : if (ret != CS_SUCCEED) {
148 0 : fprintf(stderr, "ct_bind() failed\n");
149 0 : return 1;
150 : }
151 :
152 16 : ret = ct_describe(cmd, 3, &datafmt);
153 16 : if (ret != CS_SUCCEED) {
154 0 : fprintf(stderr, "ct_describe() failed");
155 0 : return 1;
156 : }
157 :
158 16 : datafmt.datatype = CS_CHAR_TYPE;
159 16 : datafmt.format = CS_FMT_NULLTERM;
160 16 : datafmt.maxlength = 32;
161 :
162 16 : ret = ct_bind(cmd, 3, &datafmt, col3, &datalength, &ind);
163 16 : if (ret != CS_SUCCEED) {
164 0 : fprintf(stderr, "ct_bind() failed\n");
165 0 : return 1;
166 : }
167 :
168 56 : while (((ret = ct_fetch(cmd, CS_UNUSED, CS_UNUSED, CS_UNUSED, &count)) == CS_SUCCEED)
169 16 : || (ret == CS_ROW_FAIL)) {
170 40 : row_count += count;
171 40 : if (ret == CS_ROW_FAIL) {
172 0 : fprintf(stderr, "ct_fetch() CS_ROW_FAIL on row %d.\n", row_count);
173 0 : return 1;
174 : } else { /* ret == CS_SUCCEED */
175 40 : printf("col1 = %d col2= '%s', col3 = '%s'\n", col1, col2, col3);
176 40 : ++rows[0];
177 : }
178 : }
179 :
180 :
181 16 : switch ((int) ret) {
182 : case CS_END_DATA:
183 : break;
184 0 : case CS_FAIL:
185 0 : fprintf(stderr, "ct_fetch() returned CS_FAIL.\n");
186 0 : return 1;
187 0 : default:
188 0 : fprintf(stderr, "ct_fetch() unexpected return.\n");
189 0 : return 1;
190 : }
191 : break;
192 :
193 24 : case CS_COMPUTE_RESULT:
194 :
195 24 : printf("testing compute_result\n");
196 :
197 24 : ret = ct_compute_info(cmd, CS_COMP_ID, CS_UNUSED, &compute_id, CS_UNUSED, NULL);
198 24 : if (ret != CS_SUCCEED) {
199 0 : fprintf(stderr, "ct_compute_info() failed");
200 0 : return 1;
201 : }
202 :
203 24 : if (compute_id != 1 && compute_id != 2) {
204 0 : fprintf(stderr, "invalid compute_id value");
205 0 : return 1;
206 : }
207 :
208 24 : if (compute_id == 1) {
209 16 : ret = ct_res_info(cmd, CS_NUMDATA, &num_cols, CS_UNUSED, NULL);
210 16 : if (ret != CS_SUCCEED) {
211 0 : fprintf(stderr, "ct_res_info() failed");
212 0 : return 1;
213 : }
214 16 : if (num_cols != 1) {
215 0 : fprintf(stderr, "compute_id %d num_cols %d != 1", compute_id, num_cols);
216 0 : return 1;
217 : }
218 :
219 16 : ret = ct_describe(cmd, 1, &datafmt);
220 16 : if (ret != CS_SUCCEED) {
221 0 : fprintf(stderr, "ct_describe() failed");
222 0 : return 1;
223 : }
224 16 : datafmt.format = CS_FMT_UNUSED;
225 16 : if (datafmt.maxlength > 1024) {
226 0 : datafmt.maxlength = 1024;
227 : }
228 16 : compute_col1 = -1;
229 16 : ret = ct_bind(cmd, 1, &datafmt, &compute_col1, &datalength, &ind);
230 16 : if (ret != CS_SUCCEED) {
231 0 : fprintf(stderr, "ct_bind() failed\n");
232 0 : return 1;
233 : }
234 : }
235 24 : if (compute_id == 2) {
236 8 : ret = ct_res_info(cmd, CS_NUMDATA, &num_cols, CS_UNUSED, NULL);
237 8 : if (ret != CS_SUCCEED) {
238 0 : fprintf(stderr, "ct_res_info() failed");
239 0 : return 1;
240 : }
241 8 : if (num_cols != 1) {
242 0 : fprintf(stderr, "compute_id %d num_cols %d != 1", compute_id, num_cols);
243 0 : return 1;
244 : }
245 :
246 8 : ret = ct_describe(cmd, 1, &datafmt);
247 8 : if (ret != CS_SUCCEED) {
248 0 : fprintf(stderr, "ct_describe() failed");
249 0 : return 1;
250 : }
251 :
252 8 : datafmt.datatype = CS_CHAR_TYPE;
253 8 : datafmt.format = CS_FMT_NULLTERM;
254 8 : datafmt.maxlength = 32;
255 :
256 8 : compute_col3[0] = 0;
257 8 : ret = ct_bind(cmd, 1, &datafmt, compute_col3, &datalength, &ind);
258 8 : if (ret != CS_SUCCEED) {
259 0 : fprintf(stderr, "ct_bind() failed\n");
260 0 : return 1;
261 : }
262 : }
263 :
264 :
265 48 : while (((ret = ct_fetch(cmd, CS_UNUSED, CS_UNUSED, CS_UNUSED, &count)) == CS_SUCCEED)
266 24 : || (ret == CS_ROW_FAIL)) {
267 24 : row_count += count;
268 24 : if (ret == CS_ROW_FAIL) {
269 0 : fprintf(stderr, "ct_fetch() CS_ROW_FAIL on row %d.\n", row_count);
270 0 : return 1;
271 : } else { /* ret == CS_SUCCEED */
272 24 : if (compute_id == 1) {
273 16 : printf("compute_col1 = %d \n", compute_col1);
274 16 : if (compute_col1 != 6 && compute_col1 != 17) {
275 0 : fprintf(stderr, "(should be 6 or 17)\n");
276 0 : return 1;
277 : }
278 : }
279 24 : if (compute_id == 2) {
280 8 : printf("compute_col3 = '%s'\n", compute_col3);
281 8 : if (strcmp("Jan 5 2002 10:00:00AM", compute_col3)
282 8 : && strcmp("Jan 05 2002 10:00AM", compute_col3)
283 0 : && strcmp("Jan 5 2002 10:00AM", compute_col3)) {
284 0 : fprintf(stderr, "(should be \"Jan 5 2002 10:00:00AM\")\n");
285 0 : return 1;
286 : }
287 : }
288 24 : ++rows[compute_id];
289 : }
290 : }
291 :
292 :
293 24 : switch ((int) ret) {
294 : case CS_END_DATA:
295 : break;
296 0 : case CS_FAIL:
297 0 : fprintf(stderr, "ct_fetch() returned CS_FAIL.\n");
298 0 : return 1;
299 0 : default:
300 0 : fprintf(stderr, "ct_fetch() unexpected return.\n");
301 0 : return 1;
302 : }
303 : break;
304 :
305 0 : default:
306 0 : fprintf(stderr, "ct_results() unexpected result_type.\n");
307 0 : return 1;
308 : }
309 : }
310 8 : switch ((int) results_ret) {
311 : case CS_END_RESULTS:
312 : break;
313 0 : case CS_FAIL:
314 0 : fprintf(stderr, "ct_results() failed.\n");
315 0 : return 1;
316 : break;
317 0 : default:
318 0 : fprintf(stderr, "ct_results() unexpected return.\n");
319 0 : return 1;
320 : }
321 :
322 8 : if (rows[0] != 5 || rows[1] != 2 || rows[2] != 1) {
323 0 : fprintf(stderr, "wrong number of rows: normal %u compute_1 %u compute_2 %u, expected 5 2 1\n",
324 : rows[0], rows[1], rows[2]);
325 0 : return 1;
326 : }
327 :
328 : if (verbose) {
329 : printf("Trying logout\n");
330 : }
331 8 : ret = try_ctlogout(ctx, conn, cmd, verbose);
332 8 : if (ret != CS_SUCCEED) {
333 0 : fprintf(stderr, "Logout failed\n");
334 0 : return 1;
335 : }
336 :
337 : return 0;
338 : }
339 :
340 : static CS_RETCODE
341 0 : ex_servermsg_cb(CS_CONTEXT * context, CS_CONNECTION * connection, CS_SERVERMSG * srvmsg)
342 : {
343 0 : if (strstr(srvmsg->text, "compute")) {
344 0 : compute_supported = 0;
345 0 : printf("Server does not support compute\n");
346 0 : return CS_SUCCEED;
347 : }
348 0 : return servermsg_cb(context, connection, srvmsg);
349 : }
|
