通讯录

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


void insert();
void updata();
void delete();
void query();


 int ret;
 sqlite3 *db;    
 char *errmsg;


int main()
{
    int a, s;


    ret = sqlite3_open("info.db", &db);
    if (SQLITE_OK != ret)
    {
        perror("open");
        exit(1);
    }
    char *sql = "create table stu(id integer primary key, name text, age int, phone int);";
    ret = sqlite3_exec(db, sql, NULL, NULL, &errmsg);
    if (SQLITE_OK != ret)
    {
        perror("create");
        printf("error : %s\n", errmsg);
        exit(2);
    }

loop:    printf("1(insert), 2(updata), 3(delete), 4(query)\n");
    scanf("%d", &a);
if(1 == a)
{
   printf("insert? y(1)(1)/n(0)(0)\n");
   do
   {
       scanf("%d", &s);
       if(1 == s)
       {
          insert();
       }
          printf("contiune insert? y(1)(1)/n(0)(0)\n");
   }
   while(1 == s);
   goto loop;
}
else if(2 == a)
{
   printf("updata? y(1)/n(0)\n");
            do
   {
       scanf("%d", &s);
       if(1 == s)
       {
          updata();
          printf("contiune updata? y(1)/n(0)\n");
       }
   }
   while(1 == s);
         goto loop;
}
else if(3 == a)
{
   printf("delete? y(1)/n(0)\n");
            do
   {
       scanf("%d", &s);
       if(1 == s)
       {
           delete();
           printf("contiune delete? y(1)/n(0)\n");
       }
   }
   while(1 == s);
   goto loop;
}
else if(4 == a)
{
   printf("query? y(1)/n(0)\n");
            do
   {
       scanf("%d", &s);
       if(1 == s)
       {
           query();
           printf("contiune query? y(1)/n(0)\n");
       }
   }
   while(1 == s);
   goto loop;
}

    ret = sqlite3_close(db);
    if (SQLITE_OK != ret)
    {
        perror("close");
        exit(3);
    }
    return 0;
}


void insert()
{
    int id, phone, age;
    char name[20];
    printf("id   name  age  phone\n");
    scanf("%d %s %d %d", &id, name, &age, &phone);
    char sql[1024] = {0};
    sprintf(sql, "insert into stu values(%d, '%s', %d, %d);", id, name, age, phone);
    ret = sqlite3_exec(db, sql, NULL, NULL, &errmsg);
    if (SQLITE_OK != ret)
    {
        printf("error : %s\n", errmsg);
        exit(4);
    }
}
void updata()
{
    int id, age;
    printf("input id  age\n");
    scanf("%d %d", &id, &age);
    char sql[1024] = {0};
    sprintf(sql, "update stu set age = %d where id = %d;", age, id);
    ret = sqlite3_exec(db, sql, NULL, NULL, &errmsg);
    if (SQLITE_OK != ret)
    {
        printf("error : %s\n", errmsg);
        exit(5);
    }
}
void delete()
{
    char name[20];
    int id;
    printf("input name\n");
    scanf("%s", name);
    char sql[1024] = {0};
    sprintf(sql, "delete from stu where name = '%s';", name);
    ret = sqlite3_exec(db, sql, NULL, NULL, &errmsg);
    if (SQLITE_OK != ret)
    {
        printf("error : %s\n", errmsg);
        exit(6);
    }
}
void query()
{
    int row, column;
    char** result;
    char sql[1024] = "select * from stu;";
    ret = sqlite3_get_table(db, sql, &result, &row, &column, &errmsg);
    if (SQLITE_OK != ret)
    {
        printf("%s\n", errmsg);
        exit(7);
    }    
    printf("row = %d   column = %d\n", row, column);
    
    int i, j;
    for (i = 0; i < column; i++)
    {
        printf("%s\n", result[i]);
    }


    for (i = 0; i <= row; i++)
    {
        for (j = 0; j < column; j++)
        {
            printf("%s|", result[i * column + j]);
        }
        printf("\n");
    }


    sqlite3_free_table(result);
}