8

#include <event2/event_struct.h>

    13

struct evsql_test_ctx {

    14

    struct evsql *db;

    15

    struct evsql_trans *trans;

    16

};

    17

    18

// forward-declare

    19

void query_send (struct evsql *db, struct evsql_trans *trans);

    20

    21

    22

void query_timer (int fd, short what, void *arg) {

    23

    struct evsql *db = arg;

    24

    25

    INFO("[evsql_test.timer] *tick*");

    26

    27

    query_send(db, NULL);

    28

}

    29

    30

void query_start (struct event_base *base, struct evsql *db) {

    31

    static struct event ev;

    32

    struct timeval tv = { 5, 0 };

    33

    34

    evperiodic_assign(&ev, base, &tv, &query_timer, db);

    35

    event_add(&ev, &tv);

    36

    37

    INFO("[evsql_test.timer_start] started timer");

    38

}

    39

    41

    struct evsql *db = arg;

    44

    (void) db;

    45

    82

    if ((query = evsql_query_exec(db, trans, &query_info, query_results, db,

    84

    )) == NULL)

    85

        WARNING("evsql_query_exec failed");

    90

void trans_insert_result (struct evsql_result *res, void *arg) {

    91

    struct evsql_test_ctx *ctx = arg;

    92

    err_t err;

    93

    94

    // the result info

    95

    uint32_t id;

    96

    const char *str;

    97

    98

    static struct evsql_result_info result_info = {

    99

        0, {

   100

            {   EVSQL_FMT_BINARY,   EVSQL_TYPE_UINT32   },

   101

            {   EVSQL_FMT_BINARY,   EVSQL_TYPE_STRING   },

   102

            {   0,                  0                   }

   103

        }

   104

    };

   105

   106

    // begin

   107

    if ((err = evsql_result_begin(&result_info, res)))

   108

        EFATAL(err, "query failed%s", err == EIO ? evsql_result_error(res) : "");

   109

   110

    INFO("[evsql_test.insert] got %zu rows:", evsql_result_rows(res));

   111

   112

    // iterate over rows

   113

    while ((err = evsql_result_next(res, &id, &str)) > 0) {

   114

        INFO("\t%-4lu %s", (unsigned long) id, str);

   115

    }

   116

   117

    if (err)

   118

        EFATAL(err, "evsql_result_next failed");

   119

   120

    INFO("\t(done)");

   121

   122

    // done

   123

    evsql_result_end(res);

   124

}

   125

   126

void trans_insert (struct evsql_test_ctx *ctx) {

   127

    struct evsql_query *query = NULL;

   128

   129

    // the query info

   130

    static struct evsql_query_info query_info = {

   131

        .sql    = "INSERT INTO evsql_test (str) VALUES ($1::varchar), ($2::varchar) RETURNING id, str",

   132

   133

        .params = {

   134

            {   EVSQL_FMT_BINARY,   EVSQL_TYPE_STRING   },

   135

            {   EVSQL_FMT_BINARY,   EVSQL_TYPE_STRING   },

   136

            {   0,                  0                   }

   137

        }

   138

    };

   139

   140

    // run the query

   141

    assert((query = evsql_query_exec(ctx->db, ctx->trans, &query_info, trans_insert_result, ctx,

   142

        (const char *) "row A",

   143

        (const char *) "row B"

   144

    )) != NULL);

   145

   146

    INFO("[evsql_test.insert] enqueued query: %p", query);

   147

}

   148

   150

    struct evsql_test_ctx *ctx = arg;

   151

   158

    // free result

   160

   161

    // insert

   162

    trans_insert(ctx);

   165

void trans_create_query (struct evsql_test_ctx *ctx) {

   170

        .sql    = "CREATE TEMPORARY TABLE evsql_test ( id serial4, str varchar(32) DEFAULT 'foobar' ) ON COMMIT DROP",

   173

//            {   EVSQL_FMT_BINARY,   EVSQL_TYPE_STRING   },

   179

    assert((query = evsql_query_exec(ctx->db, ctx->trans, &query_info, trans_create_result, ctx

   180

//        (const char *) "foobar"

   187

    struct evsql_test_ctx *ctx = arg;

   193

    struct evsql_test_ctx *ctx = arg;

   197

    trans_create_query(ctx);

   201

    struct evsql_test_ctx *ctx = arg;

   206

void begin_transaction (struct evsql_test_ctx *ctx) {

   207

    assert((ctx->trans = evsql_trans(ctx->db, EVSQL_TRANS_DEFAULT, 

   209

        ctx

   216

    struct evsql_test_ctx ctx;

   234

    if ((ctx.db = evsql_new_pq(ev_base, db_conninfo, NULL, NULL)) == NULL)

   238

    query_send(ctx.db, NULL);

   241

    begin_transaction(&ctx);

   244

    query_send(ctx.db, NULL);

   245

   246

    // start query timer

   247

    query_start(ev_base, ctx.db);

   250

    INFO("[evsql_test.main] running libevent loop");

   258

    if (ctx.db) {