如何使用PyOrient在OrientDB中创建函数(存储过程)?

时间:2016-07-27 17:22:33

标签: python stored-procedures orientdb pyorient

我正在尝试使用PyOrient创建一个OrientDB图形数据库,但我找不到足够的文档来让我使函数正常工作。我已经能够使用record_createofunction群集中创建一个函数,但是虽然它没有崩溃,但它似乎也不起作用。 这是我的代码:

#!/usr/bin/python

import pyorient

ousername="user"
opassword="pass"

client = pyorient.OrientDB("localhost", 2424)
session_id = client.connect( ousername, opassword )

db_name="database"

client.db_create( db_name, pyorient.DB_TYPE_GRAPH, pyorient.STORAGE_TYPE_PLOCAL )

# Set up the schema of the database
client.command( "create class URL extends V" )
client.command( "CREATE PROPERTY URL.url STRING")
client.command( "CREATE PROPERTY URL.id INTEGER")
client.command( "CREATE SEQUENCE urlseq")
client.command( "CREATE INDEX urls ON URL (url) UNIQUE")

# Get the id numbers of all the clusters
info=client.db_reload()
clusters={}
for c in info:
  clusters[c.name]=c.id
print(clusters)

# Construct a test function
# All this should do is create a new URL vertex. Eventually it will check for uniqueness of url, etc.
code="INSERT INTO URL SET id = sequence('urlseq').next(), url='?'"
addURL_func = { '@OFunction': { 'name': 'addURL', 'code':'orient.getGraph().command("sql","%s",[urlparam]);' % code, 'language':'javascript', 'parameters':'urlparam', 'idempotent':False } }
client.record_create( clusters['ofunction'], addURL_func )

# Assume  allURLs  contains the list of URLs I want to store
for url in allURLs:
  client.command("select addURL('%s')" % url)

vs = client.command("select * from URL")
for v in vs:
    print(v.url)

执行所有select addURL位运行愉快,但执行select * from URL只是超时。大概是因为(正如我通过检查Studio中的数据库所发现的那样)仍然没有URL个顶点。虽然为什么会超时而不是返回空列表或给出有用的错误信息,但我不确定。

我做错了什么,是否有更简单的方法通过PyOrient创建函数?

我不想只是在Studio中编写函数,因为我正在进行原型设计并希望它们是从Python代码编写的,而不是每次丢弃损坏的实验图时都丢失!

我主要使用OrientDB wiki page来查找OrientDB函数,而PyOrient github page几乎是我唯一的文档来源。

<小时/> 编辑:我已经能够在SQL中创建一个工作函数(请参阅下面我自己的答案),但我仍然无法创建一个可以创建顶点的工作Javascript函数。我目前最好的尝试是:

code2="""var g=orient.getGraph();g.command('sql','CREATE VERTEX URL SET id = sequence(\\"urlseq\\").next(), url = \\"'+urlparam+'\\"',[urlparam]);"""
myFunction2 = 'CREATE FUNCTION addURL2 "' + code2 + '" parameters [urlparam] idempotent false language javascript'
client.command(myFunction2)

从PyOrient调用时不会崩溃,但实际上并不创建任何顶点。但如果我从Studio调用它,它可以工作!?!我不知道发生了什么。

2 个答案:

答案 0 :(得分:2)

好的,经过大量的黑客攻击和谷歌搜索,我已经开始工作了:

code="CREATE VERTEX URL SET id = sequence('urlseq').next(), url = :urlparam;"
myFunction = 'CREATE FUNCTION addURL "' + code + '" parameters [urlparam] idempotent false language sql'
client.command(myFunction)

这里的关键似乎是在OrientDB的SQL版本中使用参数名之前的冒号。我无法在OrientDB文档中的任何地方找到任何对此的引用,但是有人在网上以某种方式发现了它。 我回答了我自己的问题,希望这能帮助其他人在ODB糟糕的文件中挣扎!

答案 1 :(得分:1)

您可以尝试以下方式:

code="var g=orient.getGraph();\ng.command(\\'sql\\',\\'%s\\',[urlparam]);"
myFunction = "CREATE FUNCTION addURL '" + code + "' parameters [urlparam] idempotent false language javascrip"
client.command(myFunction);

更新

我使用了这段代码(版本2.2.5),它对我有用

code="var g=orient.getGraph().command(\\'sql\\',\\'%s\\',[urlparam]);"
myFunction = "CREATE FUNCTION addURL '" + code + "' parameters [urlparam] idempotent false language javascrip"
client.command(myFunction);

希望有所帮助