Aerospike实战之UDF（用户自定义函数）

Aerospike可以用lua在服务端创建自定义函数，可以提高服务端的操作效率，使用自定义函数的前提上对lua语法有个基本的了解。

如下是创建的UDF，主要的作用就是判断一个用户的key是否存在，不存在则创建新的key和新的属性；

如果key存在则判断intt属性是否存在，判断intt属性的value进行具体的业务存在。

函数定义如下：

local function Split(szFullString, szSeparator) --此方法为Split字符串  local nFindStartIndex = 1  local nSplitIndex = 1  local nSplitArray = {}  while true do    local nFindLastIndex = string.find(szFullString, szSeparator, nFindStartIndex)    if not nFindLastIndex then      nSplitArray[nSplitIndex] = string.sub(szFullString, nFindStartIndex, string.len(szFullString))      break    end    nSplitArray[nSplitIndex] = string.sub(szFullString, nFindStartIndex, nFindLastIndex - 1)    nFindStartIndex = nFindLastIndex + string.len(szSeparator)    nSplitIndex = nSplitIndex + 1  end  return nSplitArrayendfunction updateIntt(rec,appendIntt,sqid)--rec参数为Aerospike当前记录本身，默认就传递进来的，后面两参数是具体的业务参数  local ret = map()  if not aerospike:exists(rec) then--判断rec记录是否存在      ret['status'] = 'DOES NOT EXIST'      rec['sqid'] = sqid['sqid']      rec['intt'] = appendIntt['appendIntt']      aerospike:create(rec)--不存在则创建一个新的key, 属性为sqid和intt  else--key存在则更新intt属性的值      --ret['status'] = 'DOES EXIST'      local intt = rec['intt']      if intt == nil then          --rec['sqid'] = userid['userid']          rec['intt'] = appendIntt['appendIntt']          ret['status'] = 'INTT DOES NOT EXIST'    ret['intt'] = appendIntt['appendIntt']      else          ret['status'] = 'INTT DOES EXIST'    local intt = rec['intt']    local appendIntt = appendIntt['appendIntt']    local list = Split(appendIntt, ':')    local l1 = list[1] .. ':'    if string.find(intt, l1) ~= nil then    ret['status'] = 'appendIntt is old'    local l2 = list[2]    --local list2 = Split(intt, ':')    ret['l2'] = l2    else    ret['status'] = 'appendIntt is new'    rec['intt'] = intt .. ',' .. appendIntt    ret['intt'] = intt .. ',' .. appendIntt    end         end      aerospike:update(rec)--修改  end  --aerospike:update(rec)  return retend

把上面的代码保存到updateIntt.lua文件中，然后通过ascli udf-put命令把这个自定义文件上传到Aerospike服务端：

ascli udf-put updateIntt.lua 可以在as部署的集群上执行，如果是远程上传的话，可以用如下命令：

ascli -h ip -p port udf-put updateIntt.lua 上传之后可以通过ascli udf-list查看是否存在updateIntt.lua，出现则说明上传成功。

接下来就可以调用我们自定义的函数了，调用方法如下：

ascli udf-record-apply <ns> <set> <key> <module> <function> ARGS

<ns>命名空间

<set>setname

<key> key

<module>lua脚本文件名，不带后缀

<function>函数名

ARGS 参数列表

ascli -h 192.168.0.15 -p 3000 udf-record-apply upf testuser ${var1} updateIntt updateIntt "{\"appendIntt\": \"${var2}\"}" "{\"sqid\": \"${var1}\

其中${var1}为key的位置，appendIntt和sqid为我定义函数时需要传的参数；

需要注意以下几点：

1，ARGS 参数传递的时候一定要按照定义函数时的顺序，要不有些参数会取不到；

2，ARGS 每个参数都是一个单独的json格式，比如'{"a":"b"}'，不能把多个参数写到同一个json里面进行传递。

Aerospike交流QQ群 ：419183757