半夜来一发C语言gdbm数据库基本操作

看到linux程序设计第317 of 819的数据库程序，手痒啊。赶紧锻炼下小脑，敲个效果看看。附带关键代码注释。

#include<unistd.h>
#include<stdlib.h>
#include<stdio.h>
#include<fcntl.h>

#include<gdbm-ndbm.h>

#include<string.h>

#define TEST_DB_FILE "/tmp/dbm1_test"
#define ITEMS_USED 3

struct test_data {

char misc_chars[15];
int any_integer;
char more_chars[21];

};

int main() {

struct test_data items_to_store[ITEMS_USED];
struct test_data item_retrieved;

char key_to_use[20];
int i, result;

datum key_datum;
datum data_datum;

DBM *dbm_ptr;

dbm_ptr = dbm_open(TEST_DB_FILE, O_RDWR|O_CREAT,0666);
if (!dbm_ptr) {
fprintf(stderr, "Failed to open database\n");
exit(EXIT_FAILURE);
}

memset(items_to_store,'\0',sizeof(items_to_store));
strcpy(items_to_store[0].misc_chars,"First!");
items_to_store[0].any_integer = 47;
strcpy(items_to_store[0].more_chars, "foo");

strcpy(items_to_store[1].misc_chars,"bar");
items_to_store[1].any_integer = 13;
strcpy(items_to_store[1].more_chars, "unlucky?");

strcpy(items_to_store[2].misc_chars,"Third");
items_to_store[2].any_integer = 3;
strcpy(items_to_store[2].more_chars, "baz");

for (i = 0; i < ITEMS_USED; i++) { //这个奇葩的部分就是以每个字段的第一个字节作为一条记录的识别，之后fetch就以这个记号作为一个标准来找到对应的记录。（特指下面的key_datum）
sprintf(key_to_use, "%c%c%d", //想想觉得难怪C快呢，这个应该就是数据库的最底层思路
items_to_store[i].misc_chars[0],
items_to_store[i].more_chars[0],
items_to_store[i].any_integer);

key_datum.dptr = (void *) key_to_use;
key_datum.dsize = strlen(key_to_use);
data_datum.dptr = (void *)&items_to_store[i];
data_datum.dsize = sizeof(struct test_data);

result = dbm_store(dbm_ptr, key_datum,data_datum,DBM_REPLACE);
//如果设置为DBM_REPLACE那么录入相同记录的时候，数据库会执行overwrite然后return 0。如果设置为DBM_INSERT，遇到相同记录，系统跳出return 1。
//然后这个data_datum帮助记录了这个结构体指针的初始位置和这个结构体的Size。

if (result != 0) {
fprintf(stderr, "dbm_store failed on key %s\n", key_to_use);
exit(2);

}

}

sprintf(key_to_use, "bu%d",13);
key_datum.dptr = key_to_use;
key_datum.dsize = strlen(key_to_use);

data_datum = dbm_fetch(dbm_ptr,key_datum);

if(data_datum.dptr) {

printf("Data retrieved\n");
memcpy(&item_retrieved, data_datum.dptr, data_datum.dsize);
printf("Retrieved item - %s %d %s\n",
item_retrieved.misc_chars,
item_retrieved.any_integer,
item_retrieved.more_chars);

}
else {
printf("No data found for key %s\n", key_to_use);
}

dbm_close(dbm_ptr);
exit(EXIT_SUCCESS);
}