在python中将AWS Cognito用户ID编码为有效的ObjectId，并对其进行解码

Question

我想将AWS Cognito用户id编码成一个有效的ObjectId，它可以在MongoDB中用作文档id，这样每个用户文档都可以通过用户的用户id进行标识。

有效的ObjectId必须是12字节的输入或24个字符的十六进制字符串

下面是一个示例AWS Cognito用户ID（str）：

b5bb610b-de5a-480e-9363-84905ae33d23

我尝试将Cognito用户ID转换为字节，以及十六进制，但它不满足要求。这是密码

import bson
u = "b5bb610b-de5a-480e-9363-84905ae33d23"
bson.ObjectId(u.encode("utf-8"))
bson.ObjectId(u.encode("utf-8").hex())

如果按如下所示使用sha224，

from hashlib import sha224
from bson.html" target="_blank">objectid import ObjectId
hasher = sha224()
cognito_id = cognito_id.encode("utf-8")
prefix = cognito_id[:6]
hasher.update(cognito_id)
hashed = hasher.digest()
partial_hash = hashed[:9]
mongo_id = prefix+partial_hash
mongo_id = ObjectId(str(mongo_id))

它们都抛出了提到上述需求的错误。

使用十六进制字符串进行编码工作，

from hashlib import sha224
from bson.objectid import ObjectId
hasher = sha224()
cognito_id = cognito_id.encode("utf-8")
prefix = cognito_id.hex()[:6]
hasher.update(cognito_id)
hashed = hasher.hexdigest()[:18]
mongo_id = prefix + hashed
mongo_id = ObjectId(mongo_id)

但我无法把它解码回来。我只能通过

cognito_prefix = str(mongo_id)[:6]

会有这样一种情况，我需要从mongo对象ID派生出整个Cognito ID。

我需要将Cognito用户ID编码成一个有效的对象ID，它可以存储在mongoDB中，并解码回来。

请帮忙。

Answer 1

使用hashlib计算Cognito ID的sha224。将前12个字节作为唯一的文档ID馈送给Mongo。

这种方法的缺点是JOIN的排序顺序变得随机化，并且您只能在获取Mongo文档后了解原始Cognito ID的任何信息。为了弥补这一点，你可以选择3/9分割。使用示例ID，您可以逐字复制三个字节0xb5 0xbb 0x61作为文档ID的前缀，然后将sha224结果的初始九个字节追加到uniquify。