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 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      276758 : bat_incref(bat bid)
      20             : {
      21      276758 :         BBPretain(bid);
      22      276759 : }
      23             : 
      24             : static void
      25      276757 : bat_decref(bat bid)
      26             : {
      27      276757 :         BBPrelease(bid);
      28      276756 : }
      29             : 
      30             : res_table *
      31      126030 : res_table_create(sql_trans *tr, int res_id, oid query_id, int nr_cols, mapi_query_t type, res_table *next)
      32             : {
      33      126030 :         res_table *t = MNEW(res_table);
      34      126030 :         res_col *tcols = ZNEW_ARRAY(res_col, nr_cols);
      35             : 
      36      126030 :         (void) tr;
      37      126030 :         if (!t || !tcols) {
      38           0 :                 _DELETE(t);
      39           0 :                 _DELETE(tcols);
      40           0 :                 return NULL;
      41             :         }
      42             : 
      43      126030 :         *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             :         };
      51             : 
      52      126030 :         return t;
      53             : }
      54             : 
      55             : res_col *
      56      345637 : res_col_create(sql_trans *tr, res_table *t, const char *tn, const char *name, const char *typename, int digits, int scale, bool isbat, char mtype, void *val, bool cached)
      57             : {
      58      345637 :         res_col *c = t->cols + t->cur_col;
      59      345637 :         BAT *b;
      60             : 
      61      345637 :         if (!sql_find_subtype(&c->type, typename, digits, scale))
      62         343 :                 sql_init_subtype(&c->type, sql_trans_bind_type(tr, NULL, typename), digits, scale);
      63      345640 :         c->tn = _STRDUP(tn);
      64      345641 :         c->name = _STRDUP(name);
      65      345642 :         if (c->tn == NULL || c->name == NULL) {
      66           0 :                 _DELETE(c->tn);
      67           0 :                 _DELETE(c->name);
      68           0 :                 return NULL;
      69             :         }
      70      345642 :         c->b = 0;
      71      345642 :         c->p = NULL;
      72      345642 :         c->mtype = mtype;
      73      345642 :         if (isbat) {
      74      284213 :                 b = (BAT*)val;
      75      284213 :                 if (b && t->cur_col == 0)
      76       65291 :                         t->nr_rows = BATcount(b);
      77             :         } else { // wrap scalar values in BATs for result consistency
      78       61429 :                 b = COLnew(0, mtype, 1, TRANSIENT);
      79       61429 :                 if (b == NULL) {
      80           0 :                         _DELETE(c->tn);
      81           0 :                         _DELETE(c->name);
      82           0 :                         return NULL;
      83             :                 }
      84       61429 :                 if (BUNappend(b, val, false) != GDK_SUCCEED) {
      85           0 :                         BBPreclaim(b);
      86           0 :                         _DELETE(c->tn);
      87           0 :                         _DELETE(c->name);
      88           0 :                         return NULL;
      89             :                 }
      90       61429 :                 if (t->cur_col == 0)
      91       60738 :                         t->nr_rows = 1;
      92             :                 cached = true; /* simply keep memory pointer for this small bat */
      93             :         }
      94      345642 :         c->b = b->batCacheid;
      95      345642 :         c->cached = cached;
      96      345642 :         if (cached)
      97       68884 :                 c->p = (void*)b;
      98             :         else
      99      276758 :                 bat_incref(c->b);
     100      345643 :         t->cur_col++;
     101      345643 :         assert(t->cur_col <= t->nr_cols);
     102             :         return c;
     103             : }
     104             : 
     105             : static void
     106      345641 : res_col_destroy(res_col *c)
     107             : {
     108      345641 :         if (c->b && !c->cached) {
     109      276757 :                 bat_decref(c->b);
     110       68884 :         } else if (c->b) {
     111       68884 :                 bat_destroy((BAT*)c->p);
     112             :         } else {
     113           0 :                 _DELETE(c->p);
     114             :         }
     115      345640 :         _DELETE(c->name);
     116      345645 :         _DELETE(c->tn);
     117      345643 : }
     118             : 
     119             : void
     120      126030 : res_table_destroy(res_table *t)
     121             : {
     122      126030 :         int i;
     123             : 
     124      471673 :         for (i = 0; i < t->nr_cols; i++) {
     125      345644 :                 res_col *c = t->cols + i;
     126             : 
     127      345644 :                 if (c)
     128      345644 :                         res_col_destroy(c);
     129             :         }
     130      126029 :         _DELETE(t->cols);
     131      126030 :         _DELETE(t);
     132      126030 : }
     133             : 
     134             : res_table *
     135      123427 : res_tables_remove(res_table *results, res_table *t)
     136             : {
     137      123427 :         res_table *r = results;
     138             : 
     139      123427 :         if (r == t) {
     140      123427 :                 results = t->next;
     141             :         } else {
     142           0 :                 for (; r; r = r->next) {
     143           0 :                         if (r->next == t) {
     144           0 :                                 r->next = t->next;
     145           0 :                                 break;
     146             :                         }
     147             :                 }
     148             :         }
     149      123427 :         res_table_destroy(t);
     150      123428 :         return results;
     151             : }
     152             : 
     153             : void
     154       37417 : res_tables_destroy(res_table *tab)
     155             : {
     156       37417 :         if (tab) {
     157         174 :                 res_table *r = tab, *t;
     158             : 
     159         174 :                 for (t = r; t; t = r) {
     160         145 :                         r = t->next;
     161         145 :                         res_table_destroy(t);
     162             :                 }
     163             :         }
     164       37417 : }
     165             : 
     166             : res_table *
     167      253595 : res_tables_find(res_table *results, int res_id)
     168             : {
     169      253595 :         res_table *r = results;
     170             : 
     171      267361 :         for (; r; r = r->next) {
     172      263337 :                 if (r->id == res_id)
     173      249571 :                         return r;
     174             :         }
     175             :         return NULL;
     176             : }