我正在执行我想象的索引图数据库的常见模式:我的数据是边列表,我想“流式传输”这些数据的上传。即,对于每条边,我想在每一边创建两个节点,然后在它们之间创建边;我不想先上传所有节点,然后再链接它们。一个简单的实现显然会导致大量重复节点。因此,我想实现某种“get_or_create”以避免重复。
我当前的实现如下,使用 pyArango:
def get_or_create_graph(self):
db = self._get_db()
if db.hasGraph('citator'):
self.g = db.graphs["citator"]
self.judgment = db["judgment"]
self.citation = db["citation"]
else:
self.judgment = db.createCollection("judgment")
self.citation = db.createCollection("citation")
self.g = db.createGraph("citator")
def get_or_create_node_object(self, name, vertex_data):
object_list = self.judgment.fetchFirstExample(
{"name": name}
)
if object_list:
node = object_list[0]
else:
node = self.g.createVertex('judgment', vertex_data)
node.save()
return node
我对这个解决方案的问题是:
- 由于应用程序而不是数据库正在检查存在性,因此在存在性检查和创建之间可能存在插入。我在实践中发现了重复的节点我怀疑这是为什么?
- 它不是很快。可能是因为它可能会两次命中数据库。
我在徘徊是否有更快和/或更原子的方式来做到这一点,理想情况下是原生 ArangoDB 查询?建议?谢谢你。
更新 按要求,调用代码如下所示。它在 Django 上下文中,其中 Link 是 Django 模型(即数据库中的数据):
... # Class definitions etc
links = Link.objects.filter(dirty=True)
for i, batch in enumerate(batch_iterator(links, limit=LIMIT, batch_size=ITERATOR_BATCH_SIZE)):
for link in batch:
source_name = cleaner.clean(link.case.mnc)
target_name = cleaner.clean(link.citation.case.mnc)
if source_name == target_name: continue
source_data = _serialize_node(link.case)
target_data = _serialize_node(link.citation.case)
populate_pair(citation_manager, source_name, source_data, target_name, target_data, link)
def populate_pair(citation_manager, source_name, source_data, target_name, target_data, link):
source_node = citation_manager.get_or_create_node_object(
source_name,
source_data
)
target_node = citation_manager.get_or_create_node_object(
target_name,
target_data
)
description = source_name + " to " + target_name
citation_manager.populate_link(source_node, target_node, description)
link.dirty = False
link.save()
以下是清理和序列化后数据的样例:
source_data: {'name': 'P v R A Fu', 'court': 'ukw', 'collection': 'uf', 'number': 'CA 139/2009', 'tag': 'NA', 'node_id': 'uf89638', 'multiplier': '5.012480529547776', 'setdown_year': 0, 'judgment_year': 0, 'phantom': 'false'}
target_data: {'name': 'Ck v R A Fu', 'court': 'ukw', 'collection': 'uf', 'number': '10/22147', 'tag': 'NA', 'node_id': 'uf67224', 'multiplier': '1.316227766016838', 'setdown_year': 0, 'judgment_year': 0, 'phantom': 'false'}
source_name: [2010] ZAECGHC 9
target_name: [2012] ZAGPJHC 189