dbutils/db.c

/*
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
 *
 * Copyright 2024 Olaf Wintermann. All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 *   1. Redistributions of source code must retain the above copyright
 *      notice, this list of conditions and the following disclaimer.
 *
 *   2. Redistributions in binary form must reproduce the above copyright
 *      notice, this list of conditions and the following disclaimer in the
 *      documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 */

#include "db.h"
#include "dbutils/db.h"

int dbuConnectionExec(DBUConnection *conn, const char *sql) {
    DBUQuery *q = conn->createQuery(conn, NULL);
    int err = 1;
    if(!dbuQuerySetSQL(q, sql)) {
        err = dbuQueryExec(q);
    }
    dbuQueryFree(q);
    return err;
}

int dbuConnectionIsActive(DBUConnection *conn) {
    return conn->isActive ? conn->isActive(conn) : 1;
}

void dbuConnectionFree(DBUConnection *conn) {
    conn->free(conn);
}

DBUQuery* dbuConnectionQuery(DBUConnection *conn, const CxAllocator *a) {
    return conn->createQuery(conn, a);
}

DBUQuery* dbuQueryCreate(DBUConnection *conn, const CxAllocator *a, const char *sql) {
    DBUQuery *query = dbuConnectionQuery(conn, a);
    if(query) {
        if(dbuQuerySetSQL(query, sql)) {
            dbuQueryFree(query);
            return NULL;
        }
    }
    return query;
}

int dbuQuerySetSQL(DBUQuery *q, const char *sql) {
    return q->setSQL(q, sql);
}

int dbuQuerySetParamString(DBUQuery *q, int index, cxstring str) {
    return q->setParamString(q, index, str);
}

int dbuQuerySetParamInt(DBUQuery *q, int index, int i) {
    return q->setParamInt(q, index, i);
}

int dbuQuerySetParamInt64(DBUQuery *q, int index, int64_t i) {
    return q->setParamInt64(q, index, i);
}

int dbuQuerySetParamDouble(DBUQuery *q, int index, double d) {
    return q->setParamDouble(q, index, d);
}

int dbuQuerySetParamNull(DBUQuery *q, int index) {
    return q->setParamNull(q, index);
}

int dbuQuerySetParamBytes(DBUQuery *q, int index, void *bytes, int len) {
    return q->setParamBytes(q, index, bytes, len);
}

int dbuQueryExec(DBUQuery *q) {
    return q->exec(q);
}

DBUResult* dbuQueryGetResult(DBUQuery *q) {
    return q->getResult(q);
}

void dbuQueryFree(DBUQuery *q) {
    q->free(q);
}


int dbuResultHasData(DBUResult *result) {
    return result->hasData(result);
}

int dbuResultIsOk(DBUResult *result) {
    return result->isOk(result);
}

int dbuResultNextRow(DBUResult *result) {
    return result->nextRow(result);
}

cxstring dbuResultGetText(DBUResult *result, int field) {
    return result->getText(result, field);
}

int64_t dbuResultGetInt(DBUResult *result, int field) {
    if(result->optional_getInt) {
        return result->optional_getInt(result, field);
    } else {
        cxstring s = result->getText(result, field);
        int64_t value = 0;
        cx_strtoi64(s, &value, 10);
        return value;
    }
}

DBUBytes dbuResultGetBinary(DBUResult *result, int field) {
    if(result->optional_getBinary) {
        return result->optional_getBinary(result, field);
    } else {
        cxstring bin = result->getText(result, field);
        return (DBUBytes) { (const unsigned char*)bin.ptr, bin.length };
    }
}

void dbuResultFree(DBUResult *result) {
    result->free(result);
}


int dbuResultAsInt32(DBUResult *r, int32_t *result) {
    int64_t i;
    int ret = dbuResultAsInt64(r, &i);
    if(!ret) {
        *result = (int32_t)i;
    }
    return ret;
}

int dbuResultAsUInt32(DBUResult *r, uint32_t *result) {
    int64_t i;
    int ret = dbuResultAsInt64(r, &i);
    if(!ret) {
        *result = (uint32_t)i;
    }
    return ret;
}

int dbuResultAsInt64(DBUResult *r, int64_t *result) {
    if(r->hasData(r)) {
        if(r->fieldType(r, 0) == DBU_FIELD_INT) {
            if(r->optional_getInt) {
                *result = r->optional_getInt(r, 0);
            } else {
                cxstring s = r->getText(r, 0);
                r->free(r);
                return cx_strtoi64(s, result, 10);
            }
            r->free(r);
            return 0;
        }
    }
    r->free(r);
    return -1;
}

int dbuResultAsUInt64(DBUResult *r, uint64_t *result) {
    int64_t i;
    int ret = dbuResultAsInt64(r, &i);
    if(!ret) {
        *result = (uint64_t)i;
    }
    return ret;
}

int dbuResultAsString(DBUResult *r, char **result) {
    cxmutstr s;
    int ret = dbuResultAsCxString(r, &s);
    if(!ret) {
        *result = s.ptr;
    }
    return ret;
}

int dbuResultAsCxString(DBUResult *r, cxmutstr *result) {
    return dbuResultAsCxStringA(r, cxDefaultAllocator, result);
}

int dbuResultAsCxStringA(DBUResult *r, const CxAllocator *a, cxmutstr *result) {
    if(r->hasData(r)) {
        if(r->fieldType(r, 0) != DBU_FIELD_NULL) {
            cxstring s = r->getText(r, 0);
            if(s.ptr) {
                *result = cx_strdup_a(a, s);
                r->free(r);
                return 0;
            }
        }
    }
    r->free(r);
    return -1;
}

static DBUQuery* execInt64ParamQuery(DBUConnection *conn, const char *sql, int64_t param) {
    DBUQuery *q = conn->createQuery(conn, cxDefaultAllocator);
    if(dbuQuerySetSQL(q, sql)) {
        dbuQueryFree(q);
        return NULL;
    }
    dbuQuerySetParamInt64(q, 1, param);
    if(dbuQueryExec(q)) {
        dbuQueryFree(q);
        return NULL;
    }
    return q;
}

static DBUQuery* execStringParamQuery(DBUConnection *conn, const char *sql, cxstring param) {
    DBUQuery *q = conn->createQuery(conn, cxDefaultAllocator);
    if(dbuQuerySetSQL(q, sql)) {
        dbuQueryFree(q);
        return NULL;
    }
    dbuQuerySetParamString(q, 1, param);
    if(dbuQueryExec(q)) {
        dbuQueryFree(q);
        return NULL;
    }
    return q;
}

DBUResult* dbuSqlExecQuery(DBUConnection *conn, const CxAllocator *a, const char *sql) {
    DBUQuery *q = dbuQueryCreate(conn, a, sql);
    if(!q) {
        return NULL;
    }
    if(dbuQueryExec(q)) {
        dbuQueryFree(q);
        return NULL;
    }
    DBUResult *result = dbuQueryGetResult(q);
    dbuQueryFree(q);
    return result;
}

DBUResult* dbuSqlExecQueryParamInt32(DBUConnection *conn, const CxAllocator *a, const char *sql, int32_t param) {
    DBUQuery *q = execInt64ParamQuery(conn, sql, param);
    if(!q) return NULL;
    DBUResult *r = dbuQueryGetResult(q);
    dbuQueryFree(q);
    return r;
}

DBUResult* dbuSqlExecQueryParamInt64(DBUConnection *conn, const CxAllocator *a, const char *sql, int64_t param) {
    DBUQuery *q = execInt64ParamQuery(conn, sql, param);
    if(!q) return NULL;
    DBUResult *r = dbuQueryGetResult(q);
    dbuQueryFree(q);
    return r;
}

DBUResult* dbuSqlExecQueryParamUInt32(DBUConnection *conn, const CxAllocator *a, const char *sql, uint32_t param) {
    DBUQuery *q = execInt64ParamQuery(conn, sql, param);
    if(!q) return NULL;
    DBUResult *r = dbuQueryGetResult(q);
    dbuQueryFree(q);
    return r;
}

DBUResult* dbuSqlExecQueryParamUInt64(DBUConnection *conn, const CxAllocator *a, const char *sql, uint32_t param) {
    DBUQuery *q = execInt64ParamQuery(conn, sql, param);
    if(!q) return NULL;
    DBUResult *r = dbuQueryGetResult(q);
    dbuQueryFree(q);
    return r;
}

DBUResult* dbuSqlExecQueryParamString(DBUConnection *conn, const CxAllocator *a, const char *sql, const char *param) {
    DBUQuery *q = execStringParamQuery(conn, sql, cx_str(param));
    if(!q) return NULL;
    DBUResult *r = dbuQueryGetResult(q);
    dbuQueryFree(q);
    return r;
}

DBUResult* dbuSqlExecQueryParamCxString(DBUConnection *conn, const CxAllocator *a, const char *sql, cxstring param) {
    DBUQuery *q = execStringParamQuery(conn, sql, param);
    if(!q) return NULL;
    DBUResult *r = dbuQueryGetResult(q);
    dbuQueryFree(q);
    return r;
}

DBUResult* dbuSqlExecQueryParamCxMutStr(DBUConnection *conn, const CxAllocator *a, const char *sql, cxmutstr param) {
    DBUQuery *q = execStringParamQuery(conn, sql, cx_strcast(param));
    if(!q) return NULL;
    DBUResult *r = dbuQueryGetResult(q);
    dbuQueryFree(q);
    return r;
}


int dbuSimpleGetInt64WithIntParam(DBUConnection *conn, const char *sql, int64_t param, int64_t *result) {
    DBUQuery *q = execInt64ParamQuery(conn, sql, param);
    if(!q) return 1;
    int err = dbuQueryExec(q);
    dbuQueryFree(q);
    if(!err) {
        DBUResult *r = dbuQueryGetResult(q);
        err = dbuResultAsInt64(r, result);
    }
    return err;
}

int dbuSimpleGetInt64WithStringParam(DBUConnection *conn, const char *sql, cxstring param, int64_t *result) {
    DBUQuery *q = execStringParamQuery(conn, sql, param);
    if(!q) return 1;
    DBUResult *r = dbuQueryGetResult(q);
    dbuQueryFree(q);
    return dbuResultAsInt64(r, result);;
}

int dbuSimpleGetStringWithIntParam(DBUConnection *conn, const char *sql, int64_t param, const CxAllocator *a, cxmutstr *result) {
    DBUQuery *q = execInt64ParamQuery(conn, sql, param);
    if(!q) return 1;
    DBUResult *r = dbuQueryGetResult(q);
    dbuQueryFree(q);
    return dbuResultAsCxStringA(r, a, result);
}

int dbuSimpleGetStringWithStringParam(DBUConnection *conn, const char *sql, cxstring param, const CxAllocator *a, cxmutstr *result) {
    DBUQuery *q = execStringParamQuery(conn, sql, param);
    if(!q) return 1;
    DBUResult *r = dbuQueryGetResult(q);
    dbuQueryFree(q);
    return dbuResultAsCxStringA(r, a, result);
}

CxList* dbuSimpleGetListWithIntParam(DBUConnection *conn, const char *sql, int64_t param, const CxAllocator *a, DBUClass *type) {
    DBUQuery *q = conn->createQuery(conn, cxDefaultAllocator);
    if(dbuQuerySetSQL(q, sql)) {
        dbuQueryFree(q);
        return NULL;
    }
    dbuQuerySetParamInt64(q, 1, param);
    DBUObjectBuilder *builder = dbuObjectBuilder(type, q, a);
    CxList *list = dbuObjectBuilderGetList(builder);
    dbuObjectBuilderDestroy(builder);
    return list;
}

CxList* dbuSimpleGetListWithStringParam(DBUConnection *conn, const char *sql, cxstring param, const CxAllocator *a, DBUClass *type) {
    DBUQuery *q = conn->createQuery(conn, cxDefaultAllocator);
    if(dbuQuerySetSQL(q, sql)) {
        dbuQueryFree(q);
        return NULL;
    }
    dbuQuerySetParamString(q, 1, param);
    DBUObjectBuilder *builder = dbuObjectBuilder(type, q, a);
    CxList *list = dbuObjectBuilderGetList(builder);
    dbuObjectBuilderDestroy(builder);
    return list;
}

int dbuSqlExec(DBUConnection *conn, const char *sql) {
    DBUResult *result = dbuSqlExecQuery(conn, NULL, sql);
    int ret = dbuResultIsOk(result);
    dbuResultFree(result);
    return ret;
}

int dbuSqlExecParamInt32(DBUConnection *conn, const char *sql, int32_t param) {
    DBUResult *result = dbuSqlExecQueryParam(conn, NULL, sql, param);
    int ret = dbuResultIsOk(result);
    dbuResultFree(result);
    return ret;
}

int dbuSqlExecParamInt64(DBUConnection *conn, const char *sql, int64_t param) {
    DBUResult *result = dbuSqlExecQueryParam(conn, NULL, sql, param);
    int ret = dbuResultIsOk(result);
    dbuResultFree(result);
    return ret;
}

int dbuSqlExecParamUInt32(DBUConnection *conn, const char *sql, uint32_t param) {
    DBUResult *result = dbuSqlExecQueryParam(conn, NULL, sql, param);
    int ret = dbuResultIsOk(result);
    dbuResultFree(result);
    return ret;
}

int dbuSqlExecParamUInt64(DBUConnection *conn, const char *sql, uint32_t param) {
    DBUResult *result = dbuSqlExecQueryParam(conn, NULL, sql, param);
    int ret = dbuResultIsOk(result);
    dbuResultFree(result);
    return ret;
}

int dbuSqlExecParamString(DBUConnection *conn, const char *sql, const char *param) {
    DBUResult *result = dbuSqlExecQueryParam(conn, NULL, sql, param);
    int ret = dbuResultIsOk(result);
    dbuResultFree(result);
    return ret;
}

int dbuSqlExecParamCxString(DBUConnection *conn, const char *sql, cxstring param) {
    DBUResult *result = dbuSqlExecQueryParam(conn, NULL, sql, param);
    int ret = dbuResultIsOk(result);
    dbuResultFree(result);
    return ret;
}

int dbuSqlExecParamCxMutStr(DBUConnection *conn, const char *sql, cxmutstr param) {
    DBUResult *result = dbuSqlExecQueryParam(conn, NULL, sql, param);
    int ret = dbuResultIsOk(result);
    dbuResultFree(result);
    return ret;
}