bfx_ws2_aggressive.js

bfx_ws2_aggressive.js

/*
mkdir bfx_test
cd bfx_test
// save this script as run.js, and add your api key/secret
npm install ws
npm install async
node run.js
*/

const WebSocket = require('ws');
const crypto = require('crypto')
const async = require('async')
const fs = require('fs')

const w1_apiKey = '...'
const w1_apiSecret = '...'

const ATTEMPTS = 50
const STATS = {
  success: 0,
  error: 0,
  errorsByType: {}
}

const test = (cb) => {
  const w1 = new WebSocket('wss://api.bitfinex.com/ws/2')
  const authNonce = (new Date()).getTime() * 1000
  const payload = 'AUTH' + authNonce
  const signature = crypto.createHmac('sha384', w1_apiSecret).update(payload).digest('hex')

  let isExecuted = false
  let isOpened = false
  let isAuthed = false

  const mto = setTimeout (() => {
    if (isExecuted) {
      // already handled
      return
    }

    setExecuted(isOpened && isAuthed ? 'error no wallet data' : 'error connection')
  }, 5000)

  function setExecuted (err) {
    if (isExecuted) {
      return  
    }

    isExecuted = true
    try {
      w1.close()
    } catch (e) {}
    clearTimeout(mto)
    cb(err)
  }

  w1.on('message', function(data) {
    data = JSON.parse(data)
    //console.log(data)

    if (data.event === 'auth') {
      if (data.status === 'OK') {
        isAuthed = true
      } else {
        return setExecuted('error auth')
      }
    } else if (data[0] === 0) {
      if (data[1] === 'ws') {
        // SUCCESS
        return setExecuted()
      }
    }
  })
 
  w1.on('close', function() {
    if (isExecuted) {
      // already handled
      return
    }

    setExecuted('error connection')
  })

  w1.on('open', function() {
    isOpened = true

    w1.send(JSON.stringify({
      event: 'auth',
      apiKey: w1_apiKey,
      authSig: signature,
      authPayload: payload,
      authNonce: +authNonce
    }))
  })
}

let count = 0
const d1 = Date.now()

async.whilst(
  () => { 
    return count < ATTEMPTS
  },
  (cb) => {
    count++
    test((err) => {
      STATS[err ? 'error' : 'success']++
      if (err) {
        STATS['errorsByType'][err] = (STATS['errorsByType'][err] || 0) + 1
      }
      cb()
    })
  },
  (err, n) => {
    console.log(STATS, (Date.now() - d1))
  }
)

I may be did something wrong in code, but please try it.