module RedisClient::ConnectionMixin

def call(command, timeout)
  @pending_reads += 1
  write(command)
  result = read(timeout)
  @pending_reads -= 1
  if result.is_a?(Error)
    result._set_command(command)
    raise result
  else
    result
  end
end

def call_pipelined(commands, timeouts)
  exception = nil
  size = commands.size
  results = Array.new(commands.size)
  @pending_reads += size
  write_multi(commands)
  size.times do |index|
    timeout = timeouts && timeouts[index]
    result = read(timeout)
    @pending_reads -= 1
    if result.is_a?(Error)
      result._set_command(commands[index])
      exception ||= result
    end
    results[index] = result
  end
  if exception
    raise exception
  else
    results
  end
end

def close
  @pending_reads = 0
  nil
end

def initialize
  @pending_reads = 0
end

def reconnect
  close
  connect
end

def revalidate
  if @pending_reads > 0
    close
    false
  else
    connected?
  end
end