Line data Source code
1 : /*
2 : * SPDX-License-Identifier: MPL-2.0
3 : *
4 : * This Source Code Form is subject to the terms of the Mozilla Public
5 : * License, v. 2.0. If a copy of the MPL was not distributed with this
6 : * file, You can obtain one at http://mozilla.org/MPL/2.0/.
7 : *
8 : * Copyright 2024, 2025 MonetDB Foundation;
9 : * Copyright August 2008 - 2023 MonetDB B.V.;
10 : * Copyright 1997 - July 2008 CWI.
11 : */
12 :
13 : #include "monetdb_config.h"
14 : #include "res_table.h"
15 : #include "bat_utils.h"
16 : #include "sql_types.h"
17 :
18 : static void
19 285317 : bat_incref(bat bid)
20 : {
21 285317 : BBPretain(bid);
22 285312 : }
23 :
24 : static void
25 285304 : bat_decref(bat bid)
26 : {
27 285304 : BBPrelease(bid);
28 285317 : }
29 :
30 : res_table *
31 130302 : res_table_create(sql_trans *tr, int res_id, oid query_id, int nr_cols, mapi_query_t type, res_table *next)
32 : {
33 130302 : res_table *t = MNEW(res_table);
34 130302 : res_col *tcols = ZNEW_ARRAY(res_col, nr_cols);
35 :
36 130302 : (void) tr;
37 130302 : if (!t || !tcols) {
38 0 : _DELETE(t);
39 0 : _DELETE(tcols);
40 0 : return NULL;
41 : }
42 :
43 130302 : *t = (res_table) {
44 : .id = res_id,
45 : .query_id = query_id,
46 : .query_type = type,
47 : .cols = tcols,
48 : .nr_cols = nr_cols,
49 : .next = next,
50 : .nr_rows = BUN_NONE,
51 : .output_col = 0,
52 : };
53 :
54 130302 : return t;
55 : }
56 :
57 : res_col *
58 357751 : res_col_create(sql_trans *tr, res_table *t, const char *tn, const char *name, const char *typename, int digits, int scale, int multiset, bool isbat, char mtype, void *val, bool cached)
59 : {
60 357751 : res_col *c = t->cols + t->cur_col;
61 357751 : BAT *b;
62 :
63 357751 : if (!sql_find_subtype(&c->type, typename, digits, scale))
64 368 : sql_init_subtype(&c->type, sql_trans_bind_type(tr, NULL, typename), digits, scale);
65 357744 : c->nrfields = 1;
66 357744 : c->virt = (multiset&4)?true:false;
67 357744 : c->type.multiset = multiset&3;
68 357744 : c->multiset = c->type.multiset;
69 357744 : c->composite = c->type.type->composite;
70 357744 : t->complex_type = (t->complex_type || c->virt || c->composite || (c->multiset != MS_VALUE));
71 :
72 357744 : c->tn = _STRDUP(tn);
73 357752 : c->name = _STRDUP(name);
74 357754 : if (c->tn == NULL || c->name == NULL) {
75 0 : _DELETE(c->tn);
76 0 : _DELETE(c->name);
77 0 : return NULL;
78 : }
79 357754 : c->b = 0;
80 357754 : c->p = NULL;
81 357754 : c->mtype = mtype;
82 357754 : if (isbat) {
83 294661 : b = (BAT*)val;
84 294661 : if (b && t->nr_rows == BUN_NONE && !c->virt && t->cur_col == t->output_col)
85 67966 : t->nr_rows = BATcount(b);
86 : /*
87 : else if (c->virt) {
88 : if (c->multiset && c->type.type->composite)
89 : t->output_col = c->multiset + c->type.type->composite + 1;
90 : else if (c->multiset)
91 : t->output_col = c->multiset + 2;
92 : else
93 : t->output_col++;
94 : }
95 : */
96 : } else { // wrap scalar values in BATs for result consistency
97 63093 : b = COLnew(0, mtype, 1, TRANSIENT);
98 63091 : if (b == NULL) {
99 0 : _DELETE(c->tn);
100 0 : _DELETE(c->name);
101 0 : return NULL;
102 : }
103 63091 : if (BUNappend(b, val, false) != GDK_SUCCEED) {
104 0 : BBPreclaim(b);
105 0 : _DELETE(c->tn);
106 0 : _DELETE(c->name);
107 0 : return NULL;
108 : }
109 63085 : if (t->nr_rows == BUN_NONE && !c->virt)
110 60907 : t->nr_rows = 1;
111 : cached = true; /* simply keep memory pointer for this small bat */
112 : }
113 357746 : c->b = b->batCacheid;
114 357746 : c->cached = cached;
115 357746 : if (cached)
116 72429 : c->p = (void*)b;
117 : else
118 285317 : bat_incref(c->b);
119 357741 : t->cur_col++;
120 357741 : assert(t->cur_col <= t->nr_cols);
121 357741 : t->nr_output_cols = t->nr_cols;
122 357741 : return c;
123 : }
124 :
125 : static void
126 357733 : res_col_destroy(res_col *c)
127 : {
128 357733 : if (c->b && !c->cached) {
129 285304 : bat_decref(c->b);
130 72429 : } else if (c->b) {
131 72429 : bat_destroy((BAT*)c->p);
132 : } else {
133 0 : _DELETE(c->p);
134 : }
135 357728 : _DELETE(c->name);
136 357754 : _DELETE(c->tn);
137 357760 : }
138 :
139 : void
140 130201 : res_table_destroy(res_table *t)
141 : {
142 130201 : int i;
143 :
144 488028 : for (i = 0; i < t->nr_cols; i++) {
145 357728 : res_col *c = t->cols + i;
146 :
147 357728 : if (c)
148 357728 : res_col_destroy(c);
149 : }
150 130300 : _DELETE(t->cols);
151 130303 : _DELETE(t);
152 130303 : }
153 :
154 : res_table *
155 126790 : res_tables_remove(res_table *results, res_table *t)
156 : {
157 126790 : res_table *r = results;
158 :
159 126790 : if (r == t) {
160 126790 : results = t->next;
161 : } else {
162 0 : for (; r; r = r->next) {
163 0 : if (r->next == t) {
164 0 : r->next = t->next;
165 0 : break;
166 : }
167 : }
168 : }
169 126790 : res_table_destroy(t);
170 126838 : return results;
171 : }
172 :
173 : void
174 37634 : res_tables_destroy(res_table *tab)
175 : {
176 37634 : if (tab) {
177 183 : res_table *r = tab, *t;
178 :
179 183 : for (t = r; t; t = r) {
180 149 : r = t->next;
181 149 : res_table_destroy(t);
182 : }
183 : }
184 37634 : }
185 :
186 : res_table *
187 261074 : res_tables_find(res_table *results, int res_id)
188 : {
189 261074 : res_table *r = results;
190 :
191 274760 : for (; r; r = r->next) {
192 270708 : if (r->id == res_id)
193 257022 : return r;
194 : }
195 : return NULL;
196 : }
|
