封装lmdb数据库接口

LMDB是一个高效的键值存储数据库，被广泛用于各种应用中。为了方便使用LMDB数据库，我们可以封装一个LMDB数据库接口，使得用户可以方便地使用LMDB数据库进行数据存储和读取。

下面是一个简单的LMDB数据库接口的Python实现：

import lmdb
import pickle

class LMDBDatabase:
    def __init__(self, path):
        self.path = path
        self.env = lmdb.open(path, map_size=1099511627776, max_dbs=1000, writemap=True)
        self.dbs = {}

    def get_db(self, name):
        if name in self.dbs:
            return self.dbs[name]
        else:
            db = self.env.open_db(name.encode())
            self.dbs[name] = db
            return db

    def put(self, db_name, key, value):
        with self.env.begin(write=True) as txn:
            db = self.get_db(db_name)
            txn.put(key.encode(), pickle.dumps(value), db=db)

    def get(self, db_name, key):
        with self.env.begin() as txn:
            db = self.get_db(db_name)
            value = txn.get(key.encode(), db=db)
            if value is not None:
                return pickle.loads(value)

    def delete(self, db_name, key):
        with self.env.begin(write=True) as txn:
            db = self.get_db(db_name)
            txn.delete(key.encode(), db=db)

    def iter(self, db_name):
        with self.env.begin() as txn:
            db = self.get_db(db_name)
            cursor = txn.cursor(db=db)
            for key, value in cursor:
                yield key.decode(), pickle.loads(value)

使用LMDB数据库接口非常简单，只需要按照以下步骤即可：

1. 创建一个LMDBDatabase对象，指定LMDB数据库的路径；
2. 调用put方法将数据存储到数据库中；
3. 调用get方法从数据库中读取数据；
4. 调用delete方法从数据库中删除数据；
5. 调用iter方法遍历数据库中的所有数据。

下面是一个示例程序，演示了如何使用LMDB数据库接口存储和读取数据：

db = LMDBDatabase('mydb')

db.put('users', 'user1', {'name': 'Alice', 'age': 20})
db.put('users', 'user2', {'name': 'Bob', 'age': 30})

print(db.get('users', 'user1'))
print(db.get('users', 'user2'))

db.delete('users', 'user1')

for key, value in db.iter('users'):
    print(key, value)

LMDB数据库接口封装了LMDB库的细节，使得用户可以方便地使用LMDB数据库进行数据存储和读取。