setInterval()使Node.js进程保持运行

时间:2015-05-05 16:25:48

标签: javascript node.js coffeescript

我在CoffeeScript中写了一个像令牌桶一样的东西。

class TokenBucket
  constructor: (@speed, @threshold) ->
    @currentVolume = 0
    @interval = setInterval =>
      @currentVolume += 1
      if @threshold < @currentVolume
        @currentVolume = @threshold
    , @speed * 1000

  stop: ->
    clearInterval @interval
    return

  get: ->
    if @currentVolume == 0
      false
    else
      @currentVolume -= 1
      true

对于不熟悉CoffeeScript的人,让我简单解释一下。在构造TokenBucket之后,它会使用一个回调来调用setInterval,该回调每隔几秒就会增加一个“桶”。

问题是,如果用户忘记在现有之前调用stopsetInterval将保持Node.js进程运行,因为事件循环永远不会为空。我想要做的是告诉Node.js这个setInterval不重要,如果其他一切都完成,你可以终止它。我怎样才能做到这一点?

1 个答案:

答案 0 :(得分:8)

  

我想要做的是告诉Node.js这个setInterval不重要,你可以在其他一切完成时终止它。我怎样才能做到这一点?

您正在寻找unref method,其中“将允许您创建一个活动的计时器,但如果它是事件循环中剩下的唯一项目将无法使程序继续运行。< / EM>”。

class TokenBucket
  constructor: (@speed, @threshold) ->
    …
    @interval = setInterval =>
      …
    , @speed * 1000
    @interval.unref()

然而,我想知道这是否是最好的解决方案。可能最好让@curretnVolume只是在需要时动态计算其值(取决于当前时间)的getter。像

这样的东西
class TokenBucket
  constructor: (@speed, @threshold) ->
    @_start = Date.now()

  Object.defineProperty @::, "currentVolume",
    enumerable: true,
    configurable: true,
    get: ->
      dt = (Date.now() - @_start) / 1000
      Math.min @threshold, Math.floor dt/@speed

  get: ->
    if @currentVolume == 0
      false
    else
      @_start = Math.max @_start, Date.now()-@threshold*@speed*1000 # not 100% accurate
      @_start += @peed*1000
      true