mirror of
https://github.com/codeninjasllc/codecombat.git
synced 2024-12-18 03:23:42 -05:00
43 lines
1.6 KiB
CoffeeScript
43 lines
1.6 KiB
CoffeeScript
config = require '../../server_config'
|
|
redis = require 'redis'
|
|
log = require 'winston'
|
|
|
|
class LockManager
|
|
constructor: ->
|
|
unless config.isProduction
|
|
throw "You shouldn't be instantiating distributed locks unless in production."
|
|
@redisNotAvailable = true
|
|
@redisClient = redis.createClient config.redis.port, config.redis.host
|
|
@redisClient.on "ready", =>
|
|
log.info "Redis ready!"
|
|
@redisNotAvailable = false
|
|
@redisClient.on "error", (err) =>
|
|
@redisNotAvailable = true
|
|
log.error "Redis connection error! Err: #{err}"
|
|
@redisClient.on "end", =>
|
|
@redisNotAvailable = true
|
|
log.error "Redis connection ended!"
|
|
@lockValues = {}
|
|
@unlockScript = "if redis.call(\"get\",KEYS[1]) == ARGV[1] then return redis.call(\"del\",KEYS[1]) else return 0 end"
|
|
|
|
setLock: (lockName, timeoutMs, cb) =>
|
|
if @redisNotAvailable is true then return cb "Redis not available!"
|
|
randomNumber = Math.floor(Math.random() * 1000000000)
|
|
@redisClient.set [lockName,randomNumber, "NX", "PX", timeoutMs], (err, res) =>
|
|
if err? then return cb err, null
|
|
if res is "OK"
|
|
@lockValues[lockName] = randomNumber
|
|
return cb null, "Lock set!"
|
|
unless res
|
|
return cb "Lock already set!", null
|
|
|
|
releaseLock: (lockName, cb) =>
|
|
if @redisNotAvailable is true then return cb "Redis not available!"
|
|
@redisClient.eval [@unlockScript, 1, lockName, @lockValues[lockName]], (err, res) ->
|
|
if err? then return cb err, null
|
|
if res
|
|
cb null, "The lock was released!"
|
|
else
|
|
cb "The lock was not released.", null
|
|
|
|
module.exports = new LockManager()
|