void
insert_tuple(struct tnt_tuple *tuple)
{
if (tnt_insert(tnt, 0, TNT_FLAG_RETURN, tuple) < 0)
fail_tnt_perror("tnt_insert");
if (tnt_flush(tnt) < 0)
fail_tnt_perror("tnt_flush");
recv_command("insert");
}
static bool
tnt_sql_stmt(struct tnt_sql *sql)
{
struct tnt_tuple tu;
struct tnt_list tuples;
struct tnt_stream update;
tnt_tuple_init(&tu);
tnt_list_init(&tuples);
tnt_buf(&update);
int flags = 0;
struct tnt_tk *tk = NULL, *tn = NULL;
bool rc = false;
switch (tnt_lex(sql->l, &tk)) {
/* <INSERT|REPLACE> [INTO] TABLE VALUES ( list ) */
case TNT_TK_INSERT:
case TNT_TK_REPLACE:
tnt_sqltry(sql, TNT_TK_INTO);
if (sql->error)
goto error;
tnt_expect(tnt_sqltkv(sql, TNT_TK_TABLE, &tn));
tnt_expect(tnt_sqltk(sql, TNT_TK_VALUES));
tnt_expect(tnt_sqltk(sql, '('));
while (1) {
tnt_expect(tnt_sql_kv(sql, &tu, false));
if (tnt_sqltry(sql, ','))
continue;
if (sql->error)
goto error;
break;
}
flags = TNT_FLAG_ADD;
if (tk->tk == TNT_TK_REPLACE)
flags = TNT_FLAG_REPLACE;
tnt_expect(tnt_sqltk(sql, ')'));
tnt_expect(tnt_sqltk(sql, TNT_TK_EOF));
if (tnt_insert(sql->s, TNT_TK_I32(tn), flags, &tu) == -1) {
tnt_sql_error(sql, tk, "insert failed");
goto error;
}
break;
/* UPDATE TABLE SET operations WHERE predicate */
case TNT_TK_UPDATE:
if (!tnt_sql_stmt_update(sql, &tu, &update))
goto error;
break;
/* DELETE FROM TABLE WHERE predicate */
case TNT_TK_DELETE:
tnt_expect(tnt_sqltk(sql, TNT_TK_FROM));
tnt_expect(tnt_sqltkv(sql, TNT_TK_TABLE, &tn));
tnt_expect(tnt_sqltk(sql, TNT_TK_WHERE));
/* predicate */
tnt_expect(tnt_sql_kv(sql, &tu, true));
tnt_expect(tnt_sqltk(sql, TNT_TK_EOF));
if (tnt_delete(sql->s, TNT_TK_I32(tn), 0, &tu) == -1) {
tnt_sql_error(sql, tk, "delete failed");
goto error;
}
break;
/* SELECT * FROM TABLE WHERE predicate OR predicate... LIMIT NUM */
case TNT_TK_SELECT: {
tnt_expect(tnt_sqltk(sql, '*'));
tnt_expect(tnt_sqltk(sql, TNT_TK_FROM));
tnt_expect(tnt_sqltkv(sql, TNT_TK_TABLE, &tn));
tnt_expect(tnt_sqltk(sql, TNT_TK_WHERE));
int32_t index = -1;
while (1) {
struct tnt_tuple *tup = tnt_list_at(&tuples, NULL);
while (1) {
tnt_expect(tnt_sql_kv_select(sql, tup, &index));
if (tnt_sqltry(sql, TNT_TK_AND))
continue;
if (sql->error)
goto error;
break;
}
if (tnt_sqltry(sql, TNT_TK_OR))
continue;
if (sql->error)
goto error;
break;
}
uint32_t limit = UINT32_MAX;
if (tnt_sqltry(sql, TNT_TK_LIMIT)) {
struct tnt_tk *ltk;
tnt_expect(tnt_sqltkv(sql, TNT_TK_NUM32, <k));
limit = TNT_TK_I32(ltk);
} else
if (sql->error)
goto error;
tnt_expect(tnt_sqltk(sql, TNT_TK_EOF));
if (tnt_select(sql->s, TNT_TK_I32(tn), index, 0, limit, &tuples) == -1) {
tnt_sql_error(sql, tk, "select failed");
goto error;
}
break;
}
/* CALL NAME[{.NAME}+](STRING [{,STRING}+]) */
case TNT_TK_CALL: {
char proc[512];
int len = 0;
while (1) {
struct tnt_tk *name = NULL;
tnt_lex_idonly(sql->l, true);
tnt_expect(tnt_sqltkv(sql, TNT_TK_ID, &name));
tnt_lex_idonly(sql->l, false);
len += snprintf(proc + len, sizeof(proc) - len, "%.*s",
(int)TNT_TK_S(name)->size, TNT_TK_S(name)->data);
if (!tnt_sqltry(sql, '.'))
break;
if (sql->error)
goto error;
len += snprintf(proc + len, sizeof(proc) - len, "%s", ".");
}
tnt_expect(tnt_sqltk(sql, '('));
if (tnt_sqltry(sql, ')'))
goto noargs;
if (sql->error)
goto error;
while (1) {
tnt_expect(tnt_sql_kv(sql, &tu, false));
if (tnt_sqltry(sql, ','))
continue;
if (sql->error)
goto error;
break;
}
tnt_expect(tnt_sqltk(sql, ')'));
noargs:
tnt_expect(tnt_sqltk(sql, TNT_TK_EOF));
if (tnt_call(sql->s, 0, proc, &tu) == -1) {
tnt_sql_error(sql, tk, "call failed");
goto error;
}
break;
}
/* PING */
case TNT_TK_PING:
tnt_expect(tnt_sqltk(sql, TNT_TK_EOF));
if (tnt_ping(sql->s) == -1) {
tnt_sql_error(sql, tk, "ping failed");
goto error;
}
break;
case TNT_TK_EOF:
break;
case TNT_TK_ERROR:
return tnt_sql_error(sql, tk, "%s", sql->l->error);
default:
return tnt_sql_error(sql, tk,
"insert, replace, update, delete, select, call, ping are expected");
}
rc = true;
error:
tnt_tuple_free(&tu);
tnt_list_free(&tuples);
tnt_stream_free(&update);
return rc;
}