sqlops: functions to get null result value as empty or zero
authorDaniel-Constantin Mierla <miconda@gmail.com>
Sun, 3 May 2020 07:44:01 +0000 (09:44 +0200)
committerDaniel-Constantin Mierla <miconda@gmail.com>
Sun, 3 May 2020 09:38:59 +0000 (11:38 +0200)
src/modules/sqlops/sqlops.c

index 7172bcc..b3f21f6 100644 (file)
@@ -540,7 +540,8 @@ static sr_kemi_xval_t _ksr_kemi_sqlops_xval = {0};
 /**
  *
  */
-static sr_kemi_xval_t* ki_sqlops_result_get(sip_msg_t *msg, str *resid, int row, int col)
+static sr_kemi_xval_t* ki_sqlops_result_get_mode(sip_msg_t *msg, str *resid,
+               int row, int col, int rmode)
 {
        sql_result_t *res = NULL;
 
@@ -548,31 +549,31 @@ static sr_kemi_xval_t* ki_sqlops_result_get(sip_msg_t *msg, str *resid, int row,
 
        if (resid == NULL || resid->s == NULL || resid->len == 0) {
                LM_ERR("invalid result name\n");
-               sr_kemi_xval_null(&_ksr_kemi_sqlops_xval, SR_KEMI_XVAL_NULL_NONE);
+               sr_kemi_xval_null(&_ksr_kemi_sqlops_xval, rmode);
                return &_ksr_kemi_sqlops_xval;
        }
 
        res = sql_get_result(resid);
        if(res==NULL) {
                LM_ERR("invalid result container [%.*s]\n", resid->len, resid->s);
-               sr_kemi_xval_null(&_ksr_kemi_sqlops_xval, SR_KEMI_XVAL_NULL_NONE);
+               sr_kemi_xval_null(&_ksr_kemi_sqlops_xval, rmode);
                return &_ksr_kemi_sqlops_xval;
        }
 
        if(row >= res->nrows) {
-               sr_kemi_xval_null(&_ksr_kemi_sqlops_xval, SR_KEMI_XVAL_NULL_NONE);
+               sr_kemi_xval_null(&_ksr_kemi_sqlops_xval, rmode);
                return &_ksr_kemi_sqlops_xval;
        }
        if(col >= res->ncols) {
-               sr_kemi_xval_null(&_ksr_kemi_sqlops_xval, SR_KEMI_XVAL_NULL_NONE);
+               sr_kemi_xval_null(&_ksr_kemi_sqlops_xval, rmode);
                return &_ksr_kemi_sqlops_xval;
        }
        if(res->vals[row][col].flags&PV_VAL_NULL) {
-               sr_kemi_xval_null(&_ksr_kemi_sqlops_xval, SR_KEMI_XVAL_NULL_NONE);
+               sr_kemi_xval_null(&_ksr_kemi_sqlops_xval, rmode);
                return &_ksr_kemi_sqlops_xval;
        }
        if(res->vals[row][col].flags&PV_VAL_INT) {
-               _ksr_kemi_sqlops_xval.vtype = SR_KEMIP_STR;
+               _ksr_kemi_sqlops_xval.vtype = SR_KEMIP_INT;
                _ksr_kemi_sqlops_xval.v.n = res->vals[row][col].value.n;
                return &_ksr_kemi_sqlops_xval;
        }
@@ -581,6 +582,36 @@ static sr_kemi_xval_t* ki_sqlops_result_get(sip_msg_t *msg, str *resid, int row,
        return &_ksr_kemi_sqlops_xval;
 }
 
+/**
+ *
+ */
+static sr_kemi_xval_t* ki_sqlops_result_get(sip_msg_t *msg, str *resid,
+               int row, int col)
+{
+       return ki_sqlops_result_get_mode(msg, resid, row, col,
+                       SR_KEMI_XVAL_NULL_NONE);
+}
+
+/**
+ *
+ */
+static sr_kemi_xval_t* ki_sqlops_result_gete(sip_msg_t *msg, str *resid,
+               int row, int col)
+{
+       return ki_sqlops_result_get_mode(msg, resid, row, col,
+                       SR_KEMI_XVAL_NULL_EMPTY);
+}
+
+/**
+ *
+ */
+static sr_kemi_xval_t* ki_sqlops_result_getz(sip_msg_t *msg, str *resid,
+               int row, int col)
+{
+       return ki_sqlops_result_get_mode(msg, resid, row, col,
+                       SR_KEMI_XVAL_NULL_ZERO);
+}
+
 /**
  *
  */
@@ -626,6 +657,16 @@ static sr_kemi_t sr_kemi_sqlops_exports[] = {
                { SR_KEMIP_STR, SR_KEMIP_INT, SR_KEMIP_INT,
                        SR_KEMIP_NONE, SR_KEMIP_NONE, SR_KEMIP_NONE }
        },
+       { str_init("sqlops"), str_init("sql_result_gete"),
+               SR_KEMIP_XVAL, ki_sqlops_result_gete,
+               { SR_KEMIP_STR, SR_KEMIP_INT, SR_KEMIP_INT,
+                       SR_KEMIP_NONE, SR_KEMIP_NONE, SR_KEMIP_NONE }
+       },
+       { str_init("sqlops"), str_init("sql_result_getz"),
+               SR_KEMIP_XVAL, ki_sqlops_result_getz,
+               { SR_KEMIP_STR, SR_KEMIP_INT, SR_KEMIP_INT,
+                       SR_KEMIP_NONE, SR_KEMIP_NONE, SR_KEMIP_NONE }
+       },
 
        { {0, 0}, {0, 0}, 0, NULL, { 0, 0, 0, 0, 0, 0 } }
 };