test.c

嵌入式数据Berkeley DB 4.5.20的应用例程.该例是在at91rm9200下运行,实现对数据库表的创建,数据的存取

#include <db.h>
#include <stdio.h>
#include <stdlib.h>
#include <pthread.h>

/* DB的函数执行完成后，返回0代表成功，否则失败 */
void print_error(int ret)
{
  if(ret != 0)
    printf("ERROR: %s\n",db_strerror(ret));
}

/* 数据结构DBT在使用前，应首先初始化，否则编译可通过但运行时报参数错误 */
void init_DBT(DBT * key, DBT * data)
{
  memset(key, 0, sizeof(DBT));
  memset(data, 0, sizeof(DBT));
}

void main(void)
{
  DB *dbp;      
  DBT key, data;
  u_int32_t flags;
  int ret;

  char *fruit = "apple";
  int number = 15;
 
  typedef struct customer
  {
    int c_id;
    char name[10];
    char address[20];
    int age;
  } CUSTOMER;
  CUSTOMER cust;
  int key_cust_c_id = 1;

  cust.c_id = 1;
  strncpy(cust.name, "javer", 9);
  strncpy(cust.address, "chengdu", 19);
  cust.age = 32;

  /* 首先创建数据库句柄 */
  ret = db_create(&dbp, NULL, 0);
  print_error(ret);
 
  /* 创建数据库标志 */
  flags = DB_CREATE;  

  /* 创建一个名为single.db的数据库，使用B+树访问算法，本段代码演示对简单数据类型的处理 */
  ret = dbp->open(dbp, NULL, "single.db", NULL, DB_BTREE, flags, 0);
  print_error(ret);

  init_DBT(&key, &data);
 
  /* 分别对关键字和数据赋值和规定长度 */
  key.data = fruit;
  key.size = strlen(fruit) + 1;
  data.data = &number;
  data.size = sizeof(int);

  /* 把记录写入数据库中，不允许覆盖关键字相同的记录 */
  ret = dbp->put(dbp, NULL, &key, &data,DB_NOOVERWRITE);
  print_error(ret);
 
/* 手动把缓存中的数据刷新到硬盘文件中，实际上在关闭数据库时，数据会被自动刷新 */
dbp->sync()；
 
init_DBT(&key, &data);

  key.data = fruit;
  key.size = strlen(fruit) + 1;
 
  /* 从数据库中查询关键字为apple的记录 */
  ret = dbp->get(dbp, NULL, &key, &data, 0);
  print_error(ret);

  /* 特别要注意数据结构DBT的字段data为void *型，所以在对data赋值和取值时，要做必要的类型转换。 */
  printf("The number = %d\n", *(int*)(data.data));
 
  if(dbp != NULL)
        dbp->close(dbp, 0);

  ret = db_create(&dbp, NULL, 0);
  print_error(ret);

  flags = DB_CREATE;  

  /* 创建一个名为complex.db的数据库，使用HASH访问算法，本段代码演示对复杂数据结构的处理 */
  ret = dbp->open(dbp, NULL, "complex.db", NULL, DB_HASH, flags, 0);
  print_error(ret);

  init_DBT(&key, &data);

  key.size = sizeof(int);
  key.data = &(cust.c_id);

  data.size = sizeof(CUSTOMER);
  data.data = &cust;

  ret = dbp->put(dbp, NULL, &key, &data,DB_NOOVERWRITE);
  print_error(ret);
 
  memset(&cust, 0, sizeof(CUSTOMER));
 
  key.size = sizeof(int);
  key.data = &key_cust_c_id;

  data.data = &cust;
  data.ulen = sizeof(CUSTOMER);
  data.flags = DB_DBT_USERMEM;

  dbp->get(dbp, NULL, &key, &data, 0);
  print_error(ret);
 
  printf("c_id = %d name = %s address = %s age = %d\n",
    cust.c_id, cust.name, cust.address, cust.age);
 
  if(dbp != NULL)
        dbp->close(dbp, 0);
}