【4】自己写数据库函数库 — 获取一条记录

最新推荐文章于 2020-12-01 18:05:31 发布
最新推荐文章于 2020-12-01 18:05:31 发布
阅读量1.1k 收藏
点赞数
分类专栏： 数据库函数库
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/nestler/article/details/27498579
版权

函数db_fetch为用户接口，根据传入的键值寻找对应的数据，如果找到，则返回指向数据的指针，否则返回NULL，代码如下：

/* 根据给定的键读取一条记录 */
char *db_fetch(DBHANDLE h, const char *key)
{
	DB   *db = h;
	char *ptr;

	/* 参数三：0表示读，1表示写，根据此函数决定加什么锁 */
	if (_db_find_and_lock(db, key, 0) < 0)
	{
		/* 返回-1表示未找到记录 */
		ptr = NULL;
		db->cnt_fetcherr++;
	}
	else	/* 返回0表示查找成功 */
	{
		ptr = _db_readdat(db);	/* 返回找到的数据 */
		db->cnt_fetchok++;
	}

	if (un_lock(db->idxfd, db->chainoff, SEEK_SET, 1) < 0)
	{
		printf("%d\n", __LINE__);
		exit(-1);
	}

	return ptr;
}

此函数的最后用到了un_lock宏函数来解锁，因为在_db_find_and_lock函数中获得了一个锁。db_fetch函数的大致流程是：根据传入的key调用_db_find_and_lock查找对应的索引记录，如果找到，则继续调用_db_readdat读取对应的数据记录，未找到，则不会调用_db_readdat函数，而是直接返回NULL。 所以，用户在调用db_fetch时一定要判断返回值。我在刚开使用时由于未判断返回值而直接打印返回的指针，结果导致在未找到的情况下出现Segmentation fault。


函数_db_find_and_lock是关键所在，下面是它的代码：

/* writelock：0表示读，1表示写
 * 返回值：0查找成功，-1查找失败
 */
static int _db_find_and_lock(DB *db, const char *key, int writelock)
{
	off_t offset, nextoffset;

	/* 根据散列函数定位桶的起始字符，这里要跳过第一个空闲链表 */
	db->chainoff = (_db_hash(db, key) * PTR_SZ) + db->hashoff;
	db->ptroff   = db->chainoff;	/* 同样指向某个桶 */

	if (writelock)	/* 写锁 */
	{
		/* 这里只锁一个链的开头一个字节，其它链仍然可用 */
		if (writew_lock(db->idxfd, db->chainoff, SEEK_SET, 1) < 0)
		{
			printf("%d\n", __LINE__);
			exit(-1);
		}
	}
	else	/* 读锁 */
	{
		/* 这里只锁一个链的开头一个字节，其它链仍然可用 */
		if (readw_lock(db->idxfd, db->chainoff, SEEK_SET, 1) < 0)
		{
			printf("%d\n", __LINE__);
			exit(-1);
		}
	}

	/* 知道偏移量，知道指针宽度，atol将指针转换成数字类型 */
	offset = _db_readptr(db, db->ptroff);		/* 直接读取桶中的元素，即第一个节点的偏移量 */

	/* 沿着桶向下查找 */
	while (offset != 0)							/* 如果为0则该散列链为空 */
	{
		nextoffset = _db_readidx(db, offset);	/* 读取一条索引记录存入db->idxbuf */

		/* db->idxbuf = 键值 '\0' 数据记录偏移量 '\0' 数据记录长度 '\0' */

		if (strcmp(db->idxbuf, key) == 0)
			break;
		db->ptroff = offset;		/* ptroff记录前一索引记录的地址 */
		offset = nextoffset;
	}

	return (offset == 0 ? -1 : 0);	/* offset = 0则没有找到记录 */
}

_db_find_and_lock函数根据传入的key计算出散列值，然后定位到散列表中的某个链表，从链表头向后遍历每条记录，直到找出某条记录保存的key和传入的key相等或到链表结尾为止。散列函数如下： 

/* 散列函数 */
static DBHASH _db_hash(DB *db, const char *key)
{
	DBHASH hval = 0;
	char c;
	int i;

	/* 以下为散列函数 */
	for (i = 1; (c = *key++) != 0; i++)
		hval += c * i;
	return (hval % db->nhash);	/* db->nhash = 137 */
}

下面介绍_db_find_and_lock调用到的相关函数。首先是_db_readptr，代码如下： 

/* 读取一个指针并转换为数值型，凡是要获取指针值都要调用这个函数
 * 根据offset知道指针所在位置，指针所占字节宽度也已知，就能读取一个指针的值了
 * 注意：此函数未进行加锁操作，需要手动进行
 */
static off_t _db_readptr(DB *db, off_t offset)
{
	char asciiptr[PTR_SZ + 1];

	if (lseek(db->idxfd, offset, SEEK_SET) == -1)
	{
		printf("%d\n", __LINE__);
		exit(-1);
	}
	if (read(db->idxfd, asciiptr, PTR_SZ) != PTR_SZ)	/* PTR_SZ = 6，一次读6个字符 */
	{
		printf("%d\n", __LINE__);
		exit(-1);
	}
	asciiptr[PTR_SZ] = 0;	/* 追加一个空字符，为了atol函数辨识结尾处 */

	/* atol会跳过前面的空格，遇到非数字或字符串结束时停止转换 */
	return (atol(asciiptr));
}

由于所有的记录都是以字符的形式保存在文件中的，所以如果想要获得指针值，则必须将字符串类型转换成数值类型，上述函数就是这个作用。根据偏移量，就能知道指针所在位置，然后读取这个字符串，把字符串转换成数值类型，然后返回。


接下来是_db_readidx函数：

/* 根据索引记录的偏移值offset读取一条索引记录，填充DB结构许多成员
 * 返回值：下一条记录的偏移量
 */
static off_t _db_readidx(DB *db, off_t offset)
{
	ssize_t i;
	char *ptr1, *ptr2;
	char asciiptr[PTR_SZ + 1], asciilen[IDXLEN_SZ + 1];
	struct iovec iov[2];

	/* 保存当前索引记录偏移量
	 * 如果offset为0，表示从当前偏移量处读
	 */
	if ((db->idxoff = lseek(db->idxfd, offset, offset == 0 ? SEEK_CUR : SEEK_SET)) == -1)
	{
		printf("%d\n", __LINE__);
		exit(-1);
	}

	/* 散布读，第一部分放asciiptr中，第二部分放asciilen中 */
	iov[0].iov_base = asciiptr;
	iov[0].iov_len  = PTR_SZ;		/* 6 */
	iov[1].iov_base = asciilen;
	iov[1].iov_len  = IDXLEN_SZ;	/* 4 */
	if ((i = readv(db->idxfd, &iov[0], 2)) != PTR_SZ + IDXLEN_SZ)
	{
		/* 返回读取的字节数，结尾返回-1 */
		if (i == 0 && offset == 0)
			return -1;	/* 这个返回值是给函数db_nextrec使用的，_db_find_and_lock永远不可能返回-1 */
		else
		{
			printf("%d\n", __LINE__);
			exit(-1);
		}
	}

	/* 下一条索引记录偏移量 */
	asciiptr[PTR_SZ] = 0;
	db->ptrval = atol(asciiptr);

	/* 当前索引记录长度 */
	asciilen[IDXLEN_SZ] = 0;
	if ((db->idxlen = atoi(asciilen)) < IDXLEN_MIN || db->idxlen > IDXLEN_MAX)
	{
		/* 索引记录长度必须在6~1024字节之间 */
		printf("%d\n", __LINE__);
		exit(-1);
	}

	/* 下面读取实际的索引记录，文件指针已经在调用readv后指向正确位置，即key开头 */
	if ((i = read(db->idxfd, db->idxbuf, db->idxlen)) != db->idxlen)
	{
		printf("%d\n", __LINE__);
		exit(-1);
	}
	if (db->idxbuf[db->idxlen -1] != NEWLINE)
	{
		printf("%d\n", __LINE__);
		exit(-1);
	}
	db->idxbuf[db->idxlen - 1] = 0;	/* 把换行符替换为空字符，为了atol函数做准备 */

	/* 找出分隔符 */
	if ((ptr1 = strchr(db->idxbuf, SEP)) == NULL)
	{
		printf("%d\n", __LINE__);
		exit(-1);
	}
	*ptr1++ = 0;	/* 分隔符替换为空字符，ptr1现在指向数据记录的偏移量 */

	if ((ptr2 = strchr(ptr1, SEP)) == NULL)
	{
		printf("%d\n", __LINE__);
		exit(-1);
	}
	*ptr2++ = 0;	/* 分隔符替换为空字符，ptr2现在指向数据记录的长度 */

	if (strchr(ptr2, SEP) != NULL)	/* 只有两个分隔符 */
	{
		printf("%d\n", __LINE__);
		exit(-1);
	}

	/* 保存实际数据的偏移量 */
	if ((db->datoff = atol(ptr1)) < 0)	/* atol遇到空字符结束 */
	{
		printf("%d\n", __LINE__);
		exit(-1);
	}

	/* 保存实际数据的长度，范围必须在2~1024字节之间 */
	if ((db->datlen = atol(ptr2)) <= 0 || db->datlen > DATLEN_MAX)
	{
		printf("%d\n", __LINE__);
		exit(-1);
	}

	return (db->ptrval);
}

由_db_readptr知道索引记录偏移量，在调用_db_readidx函数从文件中读取一条索引记录到一个buffer中，这个buffer实际上就是db->idxbuf。在调用完这个函数后，DB结构体中的许多成员都记录了某条记录的信息，例如数据偏移量，数据长度，索引记录长度等。知道了数据偏移量和数据长度，就可以读取数据文件中实际的数据了，此即函数_db_readdat的职责： 

/* 读取实际数据到缓存db->datbuf中
 * 返回值：NULL结尾的实际数据
 */
static char* _db_readdat(DB *db)
{
	if (lseek(db->datfd, db->datoff, SEEK_SET) == -1)			/* 定位 */
	{
		printf("%d\n", __LINE__);
		exit(-1);
	}
	if (read(db->datfd, db->datbuf, db->datlen) != db->datlen)	/* 读入缓存 */
	{
		printf("%d\n", __LINE__);
		exit(-1);
	}
	if (db->datbuf[db->datlen - 1] != NEWLINE)	/* 数据文件中的一个条必以换行符结尾 */
	{
		printf("%d\n", __LINE__);
		exit(-1);
	}
	db->datbuf[db->datlen - 1] = 0;				/* 换行符替换成空字符 */
	return (db->datbuf);
}

由于存放在文件中时，每条数据记录都是以换行符结尾，此函数读取一条数据记录后，把换行符修改为结束符，这方便了用户代码直接打印返回的数据。

Nestler
关注
  • 0
    点赞
  • 踩
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
数据库的函数
demonXwire的博客
11-15 2869
一文了解数据库函数
第二章 SQL命令参考-FETCH
kygoal的博客
10-28 9660
FETCH 使用游标从一个查询中获取数据 概要 FETCH [ forward direction { FROM | IN }] I cursorname where forward_direction can be empty or one of:   NEXT FIRST LAST
再来一发！DB2 应用程序如何从数据库取数据 Fetch
weixin_30565199的博客
08-11 508
The FETCH statement positions a cursor on the next row of its result table and assigns the values of that row to target variables. Invocation Although an interactive SQL facility might prov...
如何使用fetch
dbgpm676149的博客
04-02 201
Fetch API 提供了一个 JavaScript接口，用于访问和操纵HTTP管道的部分，例如请求和响应。它还提供了一个全局fetch()方法，该方法提供了一种简单，合乎逻辑的方式来跨网络异步获取资源。 1.fetch()返回Promise对象 一般情况下返回一个已经resolve的Promise对象，只有在网络故障。或则请求被阻止时才会是reject状态的Promise。所以说...
fetchRow的DB_FETCHMODE_ASSOC参数
fjnjxr的博客|PHP每周一贴
11-06 725
参考文章：http://blog.csdn.net/wssxy/article/details/2192287 fetchRow 的第一个参数就是所谓读取模式 (fetch mode, 结果以那种方式传回) ，可以有下列值： DB_FETCHMODE_ORDERED (缺省值)DB_FETCHMODE_ASSOCDB_FETCHMODE_OBJECT 如没有设定读取模
《unix环境高级编程》--- 数据库函数库
新博客：https://aping-dev.com/
06-13 708
讲解：https://blog.csdn.net/cinmyheart/article/details/42094953 大多数访问数据库的函数库使用两个文件存储信息：索引文件和数据文件。索引文件包含索引值（键）和指向数据文件中对应数据记录的指针。为提高按键查询的速度和效率，可用散列法和B+树组织索引文件，本文采用固定大小的散列表， 并采用链表法解决散列冲突。 由于只有一个索引文件，所以每条...
如何从数据库中随机取出10条记录的方法
01-02
从数据库中随机取出10条记录的方法是数据库操作中常见的需求，尤其是在处理大数据或进行数据分析时。该操作可以使用不同的数据库管理系统（DBMS），如MySQL、PostgreSQL、SQLite等，它们各自提供不同的SQL语句用于...
数据库删除完全重复和部分关键字段重复的记录
09-14
- 使用`IDENTITY`函数创建一个临时表`#Tmp`，并为每条记录分配唯一的ID。 - 创建另一个临时表`#Tmp2`，获取每个Name-Address组合的最小`autoID`。 - 删除原始表`tableName`。 - 选择`#Tmp`中`autoID`在`#Tmp2`中...
php实现的在线人员函数库
10-30
这里的在线人员函数库包含了几个核心功能，包括增加用户、更新在线用户名单、更新在线状态、删除用户、检查用户是否在线以及获取在线名单。下面将详细解释这些函数的工作原理和用途。 1. **增加用户**： - `...
mysql测试库(100万数据库和世界人口数据库).rar
04-22
这个数据库可能设计为单一的大表，每行代表一条数据，用于测试在大量数据下的SQL查询性能、索引效率以及数据库的扩展性。这种大规模的数据集有助于模拟实际生产环境中的高并发访问和大数据处理场景，对于理解和优化...
fetch用法
Ronychen的博客
07-14 1201
序言： 在传统Ajax 时代，进行 API 等网络请求都是通过XMLHttpRequest或者封装后的框架进行网络请求,然而配置和调用方式非常混乱，对于刚入门的新手并不友好。今天我们介绍的Fetch提供了一个更好的替代方法，它不仅提供了一种简单，合乎逻辑的方式来跨网络异步获取资源，而且可以很容易地被其他技术使用；fetch是web提供的一个可以获取异步资源的api，它提供的api返回的是Promise对象； 与传统Ajax对比： Ajax代码如下： var xhr = new XMLHttpRequest(
PEAR::DB fetchRow
尚兴跃的备忘录
03-17 1926
$result->fetchRow 及 $dbh->setFetchMode 第一个例子已告诉了大家可以用 $result->fetchRow 去读取查询的结果。用 PHP 写过数据库程序的朋友都知道 PHP 提供了三种方式来传回查询： *_fetch_row 、 *_fetch_array 及 *_fetch_object 。 PEAR::DB 当然也不会漏了这功能。 fetchRow 的第
python 提取固定列名数据_python获取数据库列名
weixin_39806288的博客
12-01 6091
{"moduleinfo":{"card_count":[{"count_phone":1,"count":1}],"search_count":[{"count_phone":4,"count":4}]},"card":[{"des":"阿里云数据库专家保驾护航，为用户的数据库应用系统进行性能和风险评估，参与配合进行数据压测演练，提供数据库优化方面专业建议，在业务高峰期与用户共同保障数据库系统平...
数据库查询 - max()获取最新的一条数据
张念磊的博客
11-11 8214
场景自动生成代码，要根据上一条记录的代码，代码要为上一条代码+1 select * FROM z_cashier_data where id = (SELECT max(z_cashier_data.id) from z_cashier_data where abstract_code = "1001"); ...
APUE-数据库函数库
weixin_34219944的博客
12-21 264
一、函数库 1、创建、关闭 1 DBHANDLE db_open(const char *pathname, int oflag, .../* int mode */);/*成功返回数据库句柄，出错返回NULL*/ 2 void db_close(DBHANDLE); 　　若db_open成功返回，则建立两个文件：pathname.idx(索引文件)和pathname...
【1】自己写数据库函数库 — 整体概况
Nestle的专栏
05-28 1550
本工程是根据《unix环境高级编程》中第20章的例子写的一个数据库函数库
【5】自己写数据库函数库 — 删除一条记录
Nestle的专栏
06-03 1508
删除一条记录的ji
使用C#从MySQL数据库获取单条数据
"该资源提供了一段C#代码示例，用于从MySQL数据库中提取特定数据，特别是查询表中的第一条num数据。" 在标题和描述中提到的知识点主要涉及MySQL数据库的操作，具体包括以下几个方面： 1. **MySQL数据库连接**: ...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值