main.cpp

来自「能够运行的c调用sqlite的一个程序-程序3(s3)」· C++ 代码 · 共 109 行

#include <stddef.h>
#include <stdlib.h>
#include <stdio.h>
#include <sqlite3.h>
#include <string.h>

#define BUF_LEN (1000)
#define SQL_INSERT "INSERT INTO list (id, name) valueS (?, ?);"

int
main (int argc, char **argv)
{
sqlite3 *db;
int rc, len, i, cols, type;
char  buf[BUF_LEN];
const char *next;
sqlite3_stmt *st;
char *err_msg;
int id;

if (argc < 2) {
fprintf(stderr, "Usage: %s database\n", argv[0]);
exit(EXIT_FAILURE);
}
rc = sqlite3_open(argv[1], &db);
if (rc != SQLITE_OK) {
fprintf(stderr, "Can't open database: %s\n", sqlite3_errmsg(db));
sqlite3_close(db);
exit(EXIT_FAILURE);
}
rc = sqlite3_exec(db, "BEGIN TRANSACTION", NULL, NULL, &err_msg);
if (rc != SQLITE_OK) {
fprintf(stderr, "Error on BEGIN TRANSACTION: %s\n", err_msg);
sqlite3_close(db);
exit(EXIT_FAILURE);
}
len = strlen(SQL_INSERT);
rc = sqlite3_prepare(db, SQL_INSERT, len, &st, &next);
if (rc != SQLITE_OK) {
fprintf(stderr, "Error on sqlite3_prepare: %s\n", sqlite3_errmsg(db));
sqlite3_close(db);
exit(EXIT_FAILURE);
}
while (fgets(buf, BUF_LEN, stdin) != NULL) {
len = strlen(buf);
if (len > 1) {
if (buf[len - 1] == '\n') {
buf[len - 1] = '\0';
len--;
}
}
rc = sqlite3_reset(st);
if (rc != SQLITE_OK) {
fprintf(stderr, "Error on sqlite3_reset: %s\n", sqlite3_errmsg(db));
sqlite3_close(db);
exit(EXIT_FAILURE);
}
id = strtol(buf, NULL, 10);
rc = sqlite3_bind_int(st, 1, id);
if (rc != SQLITE_OK) {
fprintf(stderr, "Error on sqlite3_bind_int: %s\n",
sqlite3_errmsg(db));
sqlite3_close(db);
exit(EXIT_FAILURE);
}
next = strchr(buf, ',');
if (next == NULL) {
rc = sqlite3_bind_null(st, 2);
if (rc != SQLITE_OK) {
fprintf(stderr, "Error on sqlite3_bind_null: %s\n",
sqlite3_errmsg(db));
sqlite3_close(db);
exit(EXIT_FAILURE);
}
} else {
next++;
len = strlen(next);
rc = sqlite3_bind_text(st, 2, next, len, SQLITE_TRANSIENT);
if (rc != SQLITE_OK) {
fprintf(stderr, "Error on sqlite3_bind_text: %s\n",
sqlite3_errmsg(db));
sqlite3_close(db);
exit(EXIT_FAILURE);
}
}
rc = sqlite3_step(st);
if (rc != SQLITE_DONE) {
fprintf(stderr, "Error on sqlite3_step: %s\n", sqlite3_errmsg(db));
sqlite3_close(db);
exit(EXIT_FAILURE);
}
}
rc = sqlite3_finalize(st);
if (rc != SQLITE_OK) {
fprintf(stderr, "Error on sqlite3_finalize: %s\n", sqlite3_errmsg(db));
sqlite3_close(db);
exit(EXIT_FAILURE);
}
rc = sqlite3_exec(db, "COMMIT TRANSACTION", NULL, NULL, &err_msg);
if (rc != SQLITE_OK) {
fprintf(stderr, "Error on COMMIT TRANSACTION: %s\n", err_msg);
sqlite3_close(db);
exit(EXIT_FAILURE);
}
sqlite3_close(db);
exit(EXIT_SUCCESS);
return 0;
}