Chinaunix首页 | 论坛 | 博客

SuShouzheng的ChinaUnix博客

首页　| 　博文目录　| 　关于我

SuShouzheng

  • 博客访问： 9176
  • 博文数量： 22
  • 博客积分： 10
  • 博客等级： 民兵
  • 技术积分： 125
  • 用 户 组： 普通用户
  • 注册时间： 2012-12-04 07:52
文章分类

全部博文（22）

文章存档

2012年（22）

我的朋友
最近访客
推荐博文
linux 下sqlite的 C编程

分类：

2012-12-04 07:53:50

原文地址：linux 下sqlite的 C编程 作者：luozhiyong131

点击(此处)折叠或打开

  1. /**
  2. * 交易记录数据库模块
  3.  * 模块功能:对交易记录数据库系统进行各种操作
  4.  * Author:Lzy
  5.  * Greate Dat:2012.11.01
  6.  */
  7. #include <stdio.h>
  8. #include <sys/ioctl.h>
  9. #include <string.h>
  10. #include <fcntl.h>
  11. #include <sys/types.h>
  12. #include <sys/stat.h>
  13. #include <sys/time.h>
  14. #include <signal.h>
  15. #include <stdlib.h>

  17. #include "sqlite3.h"

  19. #define SQLITE_MENU "/home/"
  20. #define MAX_DB 32
  21. #define TABLE_NAME "SensorData"

  23. typedef struct
  24. {
  25.     unsigned int id;
  26.     sqlite3 *db;
  27. } sqlite_db;
  28. sqlite_db user_db[MAX_DB];

  30. int volid_db(void)
  31. {
  32.     int i;
  33.     for (i = 0; i < MAX_DB; i++)
  34.     {
  35.         if (user_db[i].id == 0)
  36.             break;

  38.     }
  39.     if (i >= MAX_DB)
  40.         return -1;
  41.     return i;

  43. }

  45. int volid_db_item(int item)
  46. {
  47.     int i;
  48.     for (i = 0; i < MAX_DB; i++)
  49.     {
  50.         if (user_db[i].id == item)
  51.             break;
  52.     }
  53.     if (i >= MAX_DB)
  54.         return -1;
  55.     return i;
  56. }

  58. /**
  59.  * 函数功能：创建交易记录数据库（数据库不能重名）并且名字不能超过16字节
  60.  * *pucDBName:数据库名指针
  61.  * uiRecLen:16~4090字节,该数据库的记录长度
  62.  * 返回值    　    0x0A    内存变量校验出错        0x00    成功        0x0B    数据库系统崩溃        0x04    输入错误        0x0C    数据库已经存在        0x0E    数据库已经打开，不能删除
  63.  */
  64. int ST_DBCreate(char *pucDBName, unsigned int uiRecLen)
  65. {
  66.     sqlite3 *db = NULL;
  67.     char sql[512];
  68.     char *zErrMsg = NULL;
  69.     int rc;

  71.     printf("ST_DBCreate\n");
  72.     char fullname[100];
  73.     strcpy(fullname, SQLITE_MENU);
  74.     strcat(fullname, pucDBName);
  75.     rc = sqlite3_open(fullname, &db); /* 打开数据库 */
  76.     if (rc)
  77.     {
  78.         printf("get db %s error\n", pucDBName);
  79.         sqlite3_close(db);
  80.         return 0x0b;
  81.     }
  82.     else
  83.         printf("get db sucess\n");
  84.     sprintf(sql, "CREATE TABLE %s(ID INTEGER PRIMARY KEY,RECORD VARCHAR(%d));", TABLE_NAME, uiRecLen);
  85.     rc = sqlite3_exec(db, sql, NULL, NULL, &zErrMsg);
  86.     if (rc != SQLITE_OK)
  87.     {
  88.         printf("zErrMsg = %s \n", zErrMsg);
  89.         return 0x0c;
  90.     }
  91.     else
  92.         return 0;
  93. }

  95. /*＊
  96.  * 函数功能：删除交易记录数据库
  97.  * pszDBName:记录数据库名
  98.  * 返回值:0x0A-->内存变量校验出错 :0x00-->成功 :0x0B-->数据库系统崩溃 :0x02-->数据库不存在 :0x04-->输入错误 :0x08-->打开失败，已经打开了32个数据库 :0x0E-->数据库已经打开
  99.  */
  100. int ST_DBDel(char * pszDBName)
  101. {
  102.     char fullname[40];
  103.     char cmd[256];

  105.     printf("ST_DBDel\n");
  106.     strcpy(fullname, SQLITE_MENU);
  107.     strcat(fullname, pszDBName);
  108.     sprintf(cmd, "rm %s", fullname);
  109.     if (system(cmd) == -1)
  110.     {
  111.         printf("rm error\n");
  112.         return 0x04;
  113.     }
  114.     return 0x00;

  116. }

  118. /**
  119.  * 函数功能：打开交易记录数据库
  120.  * pcDBName:数据库名指针, 最多9字节
  121.  * pucDBId:1~32,返回的数据库指针
  122.  * 返回值:0x0A-->内存变量校验出错 :0x00-->成功:0x0B-->数据库系统崩溃 :0x02-->数据库不存在 :0x04-->输入错误 :0x08-->打开失败，已经打开了32个数据库 :0x0E-->数据库已经打开
  123.  */
  124. int ST_DBOpen(char *pcDBName, unsigned char *pucDBId)
  125. {
  126.     int item;
  127.     sqlite3 *db = NULL;
  128.     int rc;
  129.     char fullname[40];
  130.     //    printf("ST_DBOpen %s\n",*pcDBName);
  131.     item = volid_db();
  132.     if (item == -1)
  133.         return 0x08;

  135.     strcpy(fullname, SQLITE_MENU);
  136.     strcat(fullname, pcDBName);
  137.     rc = sqlite3_open(fullname, &db);
  138.     printf("open db %s \n", fullname);
  139.     if (!db)
  140.     {
  141.         printf("open db %s error\n", fullname);
  142.         return 0x0b;
  143.     }
  144.     memcpy(&user_db[item].db, &db, sizeof(db));
  145.     user_db[item].id = item + 1;
  146.     *pucDBId = user_db[item].id;
  147.     printf("id=%d\n", *pucDBId);
  148.     return 0;
  149. }

  151. /**
  152.  * 函数功能： 执行读交易记录操作
  153.  * ucDBId:1~32,数据库文件号
  154.  * nRecNo:表示读的起始记录号,记录号
  155.  * nRecCount:表示要读的记录数,记录条数
  156.  * szString:表示读入指针,记录指针
  157.  * pnLen:读出记录的长度指针,记录长度
  158.  * 返回值:0x0A-->内存变量校验出错 :0x00-->成功 :0x0B-->数据库系统崩溃 :0x03-->数据库没有打开 :0x04-->输入错误
  159.  */
  160. int ST_DBReadRecord(unsigned char ucDBId, int nRecNo, int nRecCount, char *szString, int *pnLen)
  161. {
  162.     char sql[2056];
  163.     int len = 0;
  164.     int nrow = 0;
  165.     int ncolumn = 0;
  166.     int rc;
  167.     char *zErrMsg = NULL;
  168.     char **azResult;
  169.     printf("get record\n");
  170.     printf("ucDBId=%d\n", ucDBId);
  171.     if (ucDBId > MAX_DB || ucDBId < 0)
  172.         return 0x04;
  173.     if ((user_db[ucDBId - 1].id < 0) || (user_db[ucDBId - 1].id > MAX_DB + 1))
  174.         return 0x04;
  175.     if (user_db[ucDBId - 1].id == 0)
  176.         return 0x03;
  177.     printf("ucDBId=%d\n", ucDBId);
  178.     sqlite3 *db = (sqlite3 *) user_db[ucDBId - 1].db;
  179.     sprintf(sql, "SELECT * FROM %s WHERE ID BETWEEN %d AND %d;", TABLE_NAME, nRecNo, nRecNo + nRecCount - 1);
  180.     printf("%s\n", sql);
  181.     rc = sqlite3_get_table(db, sql, &azResult, &nrow, &ncolumn, &zErrMsg);
  182.     printf("get table ok\n");
  183.     if (rc != SQLITE_OK)
  184.     {
  185.         printf("zErrMsg = %s \n", zErrMsg);
  186.         return 0x04;
  187.     }

  189.     int i = 0;
  190.     printf("row:%d column=%d \n", nrow, ncolumn);
  191.     if ((nrow == 0) && (ncolumn == 0))
  192.         return 0x04;
  193.     printf("\nThe result of querying is : \n");
  194.     for (i = 3; i < (nrow + 1) * ncolumn; i += 2)
  195.     {
  196.         printf("azResult[%d] = %s\n", i, azResult[i]);
  197.         memcpy(szString + len, azResult[i], strlen(azResult[i]));
  198.         len += strlen(azResult[i]);
  199.     }
  200.     *pnLen = len;
  201.     printf("len=%d\n", len);
  202.     for (i = 0; i < len; i++)
  203.     {
  204.         if (*(szString + i) == 0x01)
  205.             *(szString + i) = 0x00;
  206.     }
  207.     return 0;
  208. }

  210. /**
  211.  * 函数功能：执行写交易记录操作
  212.  * ucDBId:1~32,数据库文件号, 最多8字节
  213.  * nRecNo:0表示添加在文件尾，1-N表示改写该记录;记录号
  214.  * szString:如果是写记录，表示要写的记录指针;记录指针
  215.  * pnLen:如果是写记录，表示记录长度指针;记录长度
  216.  * 返回值:0x0A-->内存变量校验出错 :0x00-->成功 :0x0B-->数据库系统崩溃 :0x03-->数据库没有打开 :0x04-->输入错误 :0x06-->无剩余空间
  217.  */
  218. int ST_DBWriteRecord(unsigned char ucDBId, int nRecNo, int nRecCount, char *szString, int *pnLen)
  219. {
  220.     char sql[2056];
  221.     char db_string[2056];
  222.     int ret;
  223.     char *string = (char *) db_string;
  224.     memcpy(db_string, szString, *pnLen);
  225.     int i;
  226.     printf("ucDBId=%d\n", ucDBId);
  227.     printf("ST_DBWriteRecord\n");
  228.     if (ucDBId > MAX_DB || ucDBId < 0)
  229.         return 0x04;
  230.     for (i = 0; i < *pnLen; i++)
  231.     {
  232.         if (*(string + i) == '\0')
  233.             *(string + i) = 0x01;
  234.     }
  235.     char* g_caErrorMsg = NULL;
  236.     if ((user_db[ucDBId - 1].id < 0) || (user_db[ucDBId - 1].id > MAX_DB + 1))
  237.         return 0x04;
  238.     if (user_db[ucDBId - 1].id == 0)
  239.         return 0x03;
  240.     if (nRecNo != 0)
  241.     {
  242.         sprintf(sql, "UPDATE SensorData SET RECORD ='%s' WHERE ID=%d", szString, nRecNo);
  243.         ret = sqlite3_exec(user_db[ucDBId - 1].db, sql, NULL, NULL, &g_caErrorMsg);
  244.         if (ret != SQLITE_OK)
  245.         {
  246.             printf("changed record error%d\n", ret);
  247.             return -1;
  248.         }
  249.         else
  250.         {
  251.             sqlite3_changes(user_db[ucDBId - 1].db);
  252.             return 0x00;
  253.         }
  254.     }
  255.     else
  256.     {
  257.         printf("len=%d\n", *pnLen);
  258.         printf("string:%s\n", string);
  259.         sprintf(sql, "INSERT INTO %s VALUES(NULL,'%s');", TABLE_NAME, string);
  260.         printf("%s\n", sql);
  261.         if (sqlite3_exec(user_db[ucDBId - 1].db, sql, NULL, NULL, &g_caErrorMsg) != SQLITE_OK)
  262.         {
  263.             printf("zErrMsg = %s \n", g_caErrorMsg);
  264.             return 0x0b;
  265.         }
  266.         else
  267.             return 0x00;
  268.     }
  269. }

  271. /**
  272.  * 函数功能：执行删除交易记录操作
  273.  * ucDBId:1~32,数据库文件号, 最多8字节
  274.  * nRecNo:表示删除的记录号;记录号
  275.  * 返回值:0x0A-->内存变量校验出错 :0x00-->成功 :0x0B-->数据库系统崩溃 :0x03-->数据库没有打开 :0x04-->输入错误
  276.  */
  277. int ST_DBDelRecord(unsigned char ucDBId, int nRecNo)
  278. {
  279.     char sql[2056];
  280.     int ret = -1;
  281.     char *zErrMsg = NULL;
  282.     printf("ST_DBDelRecord\n");
  283.     printf("ucDBId=%d\n", ucDBId);
  284.     if (ucDBId > MAX_DB || ucDBId < 0)
  285.         return 0x04;
  286.     if ((user_db[ucDBId - 1].id < 0) || (user_db[ucDBId - 1].id > MAX_DB + 1))
  287.         return 0x04;
  288.     if (user_db[ucDBId - 1].id == 0)
  289.         return 0x03;
  290.     sqlite3 *db = (sqlite3 *) user_db[ucDBId - 1].db;
  291.     sprintf(sql, "DELETE FROM SensorData WHERE ID=%d;", nRecNo);
  292.     ret = sqlite3_exec(db, sql, 0, 0, &zErrMsg);
  293.     if (ret != SQLITE_OK)
  294.     {
  295.         printf("zErrMsg = %s \n", zErrMsg);
  296.         return 0x0b;
  297.     }
  298.     return 0;

  300. }

  302. /**
  303.  * 函数功能：关闭交易记录数据库
  304.  * ucDBID:0~32,要关闭的文件号，0表示关闭所有打开数据库文件，建议不要使用
  305.  * 返回值:0x0A-->内存变量校验出错 :0x00-->成功 :0x0B-->数据库系统崩溃 :0x03-->数据库没有打开 :0x04-->输入错误
  306.  */
  307. int ST_DBClose(unsigned char ucDBID)
  308. {
  309.     printf("ST_DBClose\n");
  310.     printf("ucDBId=%d\n", ucDBID);
  311.     if (ucDBID > MAX_DB || ucDBID < 0)
  312.         return 0x04;
  313.     if ((user_db[ucDBID - 1].id < 0) || (user_db[ucDBID - 1].id > MAX_DB + 1))
  314.         return 0x04;
  315.     if (user_db[ucDBID - 1].id == 0)
  316.         return 0x03;
  317.     printf("close db \n");
  318.     sqlite3 *db = user_db[ucDBID - 1].db;
  319.     sqlite3_close(db);
  320.     user_db[ucDBID - 1].id = 0;
  321.     app_sleep(2000);
  322.     return 0;
  323. }

  325. /**
  326.  * 函数功能： 根据文件号查询文件信息，当ucItem为0~3时，需要先打开文件，当ucItem为4时，不需要打开文件
  327.  * ucDBID:1~32,数据库文件号
  328.  * ucItem: 0-->查询当前有效记录数 :1-->查询所有记录数（包括被删除记录） :2-->查询剩余可写记录数
  329.  *              :3-->查询记录长度     :4-->查询有效数据库个数
  330.  * puiRecNum( 出口):    查询到的结果
  331.  * 返回值:0x0A-->内存变量校验出错     :0x00-->成功     :0x0B-->数据库系统崩溃     :0x03-->数据库没有打开 :0x04-->输入错误
  332.  */
  333. int ST_DBInfo(unsigned char ucDBID, unsigned char ucItem, unsigned int *puiRecNum)
  334. {
  335.     int ret;
  336.     printf("ST_DBInfo\n");
  337.     printf("ucDBId=%d\n", ucDBID);
  338.     if (ucDBID > MAX_DB || ucDBID < 0)
  339.     {
  340.         *puiRecNum = 0;
  341.         return 0x04;
  342.     }
  343.     if ((user_db[ucDBID - 1].id < 0) || (user_db[ucDBID - 1].id > MAX_DB + 1))
  344.         return 0x04;
  345.     if (user_db[ucDBID - 1].id == 0)
  346.         return 0x03;
  347.     if (ucItem == 2)
  348.     {
  349.         *puiRecNum = 1000;
  350.         return 0x00;
  351.     }
  352.     if (ucItem == 4)
  353.     {
  354.         int len = 0;
  355.         int i;
  356.         for (i = 0; i < MAX_DB; i++)
  357.         {
  358.             if (user_db[ucDBID - 1].id != 0)
  359.                 len++;
  360.         }
  361.         *puiRecNum = len;
  362.         return 0;
  363.     }
  364.     if ((ucItem == 3) || (ucItem == 1) || (ucItem == 0))
  365.     {
  366.         const char* pTail;
  367.         sqlite3_stmt* stmt;
  368.         char sql[512];
  369.         sprintf(sql, "SELECT COUNT(*) FROM %s", TABLE_NAME);
  370.         ret = sqlite3_prepare(user_db[ucDBID - 1].db, sql, strlen(sql), &stmt, &pTail);
  371.         if (ret != SQLITE_OK)
  372.         {
  373.             printf("get record number error\n");
  374.             return -1;
  375.         }
  376.         else
  377.         {
  378.             int recordCount = 0;
  379.             ret = sqlite3_step(stmt);
  380.             if (ret != SQLITE_ROW)
  381.             {
  382.                 printf("search record number error\n");
  383.                 return -1;
  384.             }
  385.             recordCount = sqlite3_column_int(stmt, 0);
  386.             printf("da have %d records\n", recordCount);
  387.             *puiRecNum = recordCount;
  388.         }
  389.         sqlite3_finalize(stmt);
  390.         return 0x00;
  391.     }

  393.     return 0x00;
  394. }

源码： db.zip   

 

阅读(134) | 评论(0) | 转发(0) |
0

上一篇：Linux_C定时器实现

下一篇：Linux_C计时器

给主人留下些什么吧！~~
关于我们 | 关于IT168 | 联系方式 | 广告合作 | 法律声明 | 免费注册

Copyright 2001-2010 ChinaUnix.net All Rights Reserved 北京皓辰网域网络信息技术有限公司. 版权所有

感谢所有关心和支持过ChinaUnix的朋友们

16024965号-6