class Fluent::Counter::Connection

def initialize(io, on_message)
  super(io)
  @connection = false
  @buffer = ''
  @on_message = on_message
end

def on_close
  @connection = false
end

def on_connect
  @connection = true
  write @buffer
  @buffer = ''
end

def on_message(data)
  @on_message.call(data)
end

def send_data(data)
  if @connection
    packed_write data
  else
    @buffer += pack(data)
  end
end