#include "reg.h"
int main(int argc, const char *argv[])
{
// 打开数据库
sqlite3 *sqldb = NULL;
if (sqlite3_open(DB, &sqldb) != SQLITE_OK)
{
fprintf(stderr, "数据库打开错误 : %d : %s\t__%d__\n", sqlite3_errcode(sqldb), sqlite3_errmsg(sqldb), __LINE__);
return -1;
}
// 创建表
char *create_table = "CREATE TABLE if not exists register(uname char,passwd char)";
char *err_msg = NULL;
if (sqlite3_exec(sqldb, create_table, NULL, NULL, &err_msg) != SQLITE_OK)
{
fprintf(stderr, "create table : %d : %s : __%d__\n", sqlite3_errcode(sqldb), sqlite3_errmsg(sqldb), __LINE__);
return -1;
}
char choose;
// 操作选项
while (1)
{
printf("---------------------------------------\n");
printf("------------1. 用户注册----------------\n");
// printf("------------2. 根据用户名注销用户----------------\n");
// printf("------------3. 根据用户名修改密码----------------\n");
// printf("------------4. 根据用户名查询信息----------------\n");
printf("------------5. 查询全部信息-------------\n");
printf("------------6. 退出--------------------\n");
printf("---------------------------------------\n");
printf("请选择操作 >>> ");
choose = getchar();
printf("choose = %c\n", choose);
while (getchar() != 10)
;
switch (choose)
{
case '1':
// 增
printf("注册\n");
insert_into(sqldb);
break;
case '2':
// 根据用户名删除
break;
case '3':
// 根据用户名更改
break;
case '4':
// 根据用户名查询
break;
case '5':
// 查询全部用户
find_all(sqldb);
break;
case '6':
// 退出
goto END;
}
}
END:
if (sqlite3_close(sqldb) != SQLITE_OK)
{
fprintf(stderr, "close database : %d : %s : __%d__\n", sqlite3_errcode(sqldb), sqlite3_errmsg(sqldb), __LINE__);
return -1;
}
printf("数据库关闭成功\n");
return 0;
}
#include "reg.h"
// 插入
int insert_into(sqlite3 *sqldb)
{
char uname[128] = "";
char passwd[128] = "";
printf("请输入准备注册的用户名 >>> ");
fgets(uname, sizeof(uname), stdin);
uname[strlen(uname) - 1] = 0;
// while (getchar() != 10)
// ;
printf("请输入密码 >>> ");
fgets(passwd, sizeof(passwd), stdin);
passwd[strlen(passwd) - 1] = 0;
// while (getchar() != 10)
// ;
while (find_by_uname(sqldb, uname))
{
printf("请重新输入准备注册的用户名,输入exit取消注册 >>> ");
fgets(uname, sizeof(uname), stdin);
uname[strlen(uname) - 1] = 0;
// while (getchar() != 10)
// ;
if (!strcmp(uname, "exit"))
{
printf("退出注册\n");
return 0;
}
}
char sql_str[256] = "";
sprintf(sql_str, "insert into register (uname,passwd) values('%s','%s');", uname, passwd);
char *errmsg = NULL;
printf("sql_str = %s\n", sql_str);
if (sqlite3_exec(sqldb, sql_str, NULL, NULL, &errmsg) != SQLITE_OK)
{
fprintf(stderr, "insert into register : %d : %s __%d__\n", sqlite3_errcode(sqldb), errmsg, __LINE__);
return -1;
}
printf("用户 [%s] 注册成功\n", uname);
return 1;
}
// 根据用户名删除
int del_by_uname()
{
}
// 根据用户名更改密码
int updata_by_uname()
{
}
// 根据用户名查询
int find_by_uname(sqlite3 *sqldb, char *uname)
{
char sql_str[256] = "";
char *errmsg = NULL;
char **pres = NULL;
int row, colum;
sprintf(sql_str, "select * from register where uname='%s';", uname);
if (sqlite3_get_table(sqldb, sql_str, &pres, &row, &colum, &errmsg) != SQLITE_OK)
{
fprintf(stderr, "find_by_uname error : %d : %s : __%d__\n", sqlite3_errcode(sqldb), errmsg, __LINE__);
return 0;
}
return row;
}
// 查询所有
int find_all(sqlite3 *sqldb)
{
char sql_str[256] = "";
char *errmsg = NULL;
char **pres = NULL;
int row, colum;
sprintf(sql_str, "select * from register;");
if (sqlite3_get_table(sqldb, sql_str, &pres, &row, &colum, &errmsg) != SQLITE_OK)
{
fprintf(stderr, "find_by_uname error : %d : %s : __%d__\n", sqlite3_errcode(sqldb), errmsg, __LINE__);
return 0;
}
for (int i = 0; i <= row; i++)
{
for (int j = 0; j < colum; j++)
{
printf("%s\t", pres[(row * i) + j]);
}
putchar(10);
}
return row;
}