经常需要保存些程序执行的状态，使用redis等服务有点太重了，发现了轻量级的kv数据库lmdb，但是直接使用有点麻烦，简单实现了个包装类。
用法:
1
2
3
4
5
6
7
8
9
10
11
12
# 写数据
with KVDB() as kvdb:
kvdb.set('abc', 'bingal')
# 读数据
with KVDB() as kvdb:
print(kvdb.get('abc'))
# 游标循环读取所有数据
with KVDB() as kvdb:
for key, value in kvdb.cursor():
print(key, value)
代码如下：
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
#!/usr/bin/python
# encoding: utf-8
# author: bingal
# descryption: 更优雅的方式使用lmdb
import lmdb
class KVDB:
def __init__(self, path=None, size=100*1024*1024):
# 默认100MB
self.env = lmdb.open(os.path.join(os.path.split(os.path.realpath(__file__))[0], 'lmdb') if not path else path,
map_size=size)
def __enter__(self):
return self
def __exit__(self, exc_type, exc_val, exc_tb):
return self.close()
def set(self, sid, name):
txn = self.env.begin(write=True)
txn.put(str(sid).encode(), name.encode())
txn.commit()
def delete(self, sid):
txn = self.env.begin(write=True)
txn.delete(str(sid).encode())
txn.commit()
def get(self, sid):
txn = self.env.begin()
name = txn.get(str(sid).encode())
return name.decode() if name else ''
def cursor(self):
txn = self.env.begin()
cur = txn.cursor()
return cur
def close(self):
self.env.close()
if __name__ == '__main__':
with KVDB() as kvdb:
kvdb.set('abc', 'bingal')
with KVDB() as kvdb:
print(kvdb.get('abc'))
with KVDB() as kvdb:
for key, value in kvdb.cursor():
print(key, value)
真诚邀请您走进我的知识小宇宙，关注我个人的公众号，在这里，我将不时为您献上独家原创且极具价值的技术内容分享。每一次推送，都倾注了我对技术领域的独特见解与实战心得，旨在与您共享成长过程中的每一份收获和感悟。您的关注和支持，是我持续提供优质内容的最大动力，让我们在学习的道路上并肩同行，共同进步，一起书写精彩的成长篇章!
