jyapayne/Nim
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Support transactions

Browse source
This commit is contained in:
Jason Livesay 2014-04-04 23:29:50 -07:00
commit e6d2490b9c
1 changed files with 32 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

47
lib/pure/redis.nim
View file

@ -85,13 +85,7 @@ proc recv(sock: TSocket, size: int): TaintedString =
if sock.recv(cstring(result), size) != size: if sock.recv(cstring(result), size) != size:
raise newException(EInvalidReply, "recv failed") raise newException(EInvalidReply, "recv failed")
proc parseBulk(r: TRedis, allowMBNil = False): TRedisString = proc parseSingle(r: TRedis, line:string, allowMBNil = False): TRedisString =
var line = ""
r.socket.readLine(line.TaintedString)
if line == "+QUEUED" or line == "+OK": # inside of a transaction (multi)
return nil
# Error. # Error.
if line[0] == '-': if line[0] == '-':
raise newException(ERedis, strip(line)) raise newException(ERedis, strip(line))
@ -101,6 +95,9 @@ proc parseBulk(r: TRedis, allowMBNil = False): TRedisString =
if line == "*-1": if line == "*-1":
return RedisNil return RedisNil
if line == "+QUEUED" or line == "+OK" : # inside of a transaction (multi)
return nil
if line[0] != '$': if line[0] != '$':
raiseInvalidReply('$', line[0]) raiseInvalidReply('$', line[0])
@ -111,6 +108,32 @@ proc parseBulk(r: TRedis, allowMBNil = False): TRedisString =
var s = r.socket.recv(numBytes+2) var s = r.socket.recv(numBytes+2)
result = strip(s.string) result = strip(s.string)
proc parseMultiLines(r: TRedis, countLine:string): TRedisList =
if countLine.string[0] != '*':
raiseInvalidReply('*', countLine.string[0])
var numElems = parseInt(countLine.string.substr(1))
if numElems == -1: return nil
result = @[]
for i in 1..numElems:
var line = ""
r.socket.readLine(line.TaintedString)
if line[0] == '*': # after exec() may contain more multi-bulk replies
var parsed = r.parseMultiLines(line)
for item in parsed:
result.add(item)
else:
result.add(r.parseSingle(line))
proc parseBulk(r: TRedis, allowMBNil = False): TRedisString =
var line = ""
r.socket.readLine(line.TaintedString)
if line == "+QUEUED" or line == "+OK": # inside of a transaction (multi)
return nil
return r.parseSingle(line, allowMBNil)
proc parseMultiBulk(r: TRedis): TRedisList = proc parseMultiBulk(r: TRedis): TRedisList =
var line = TaintedString"" var line = TaintedString""
r.socket.readLine(line) r.socket.readLine(line)
@ -118,14 +141,8 @@ proc parseMultiBulk(r: TRedis): TRedisList =
if line == "+QUEUED": # inside of a transaction (multi) if line == "+QUEUED": # inside of a transaction (multi)
return nil return nil
if line.string[0] != '*': return r.parseMultiLines(line)
raiseInvalidReply('*', line.string[0])
var numElems = parseInt(line.string.substr(1))
if numElems == -1: return nil
result = @[]
for i in 1..numElems:
result.add(r.parseBulk())
proc sendCommand(r: TRedis, cmd: string, args: varargs[string]) = proc sendCommand(r: TRedis, cmd: string, args: varargs[string]) =
var request = "*" & $(1 + args.len()) & "\c\L" var request = "*" & $(1 + args.len()) & "\c\L"