lib/rb/lib/thrift/transport/tsocket.rb

#!/usr/bin/env ruby
#
# Copyright (c) 2006- Facebook
# Distributed under the Thrift Software License
#
# See accompanying file LICENSE or visit the Thrift site at:
# http://developers.facebook.com/thrift/
#
# Author: Mark Slee <mcslee@facebook.com>
#
require 'thrift/transport/ttransport'
require 'socket'

class TSocket < TTransport
  def initialize(host='localhost', port=9090)
    @host = host
    @port = port
    @handle = nil
  end

  def setHandle(handle)
    @handle = handle
  end

  def open()
    begin
      @handle = TCPSocket.new(@host, @port)
    rescue StandardError
      raise TTransportException.new(TTransportException::NOT_OPEN, "Could not connect to #{@host}:#{@port}")
    end
  end

  def isOpen()
    return !@handle.nil?
  end

  def write(str)
    begin
      @handle.write(str)
    rescue StandardError
      raise TTransportException.new(TTransportException::NOT_OPEN)
    end
  end

  def read(sz)
    begin
      data = @handle.recv(sz)
      if (data.length == 0)
        raise TTransportException.new("TSocket: Could not read #{sz} bytes from #{@host}:#{@port}")
      end
      return data
    rescue StandardError
      raise TTransportException.new(TTransportException::NOT_OPEN)
    end
  end

  def close()
    @handle.close() unless @handle.nil?
    @handle = nil
  end

end

class TServerSocket < TServerTransport
  def initialize(port)
    @port = port
    @handle = nil
  end

  def listen()
    @handle = TCPserver.new(nil, @port)
  end

  def accept()
    if (@handle != nil)
      sock = @handle.accept()
      trans = TSocket.new()
      trans.setHandle(sock)
      return trans
    end
    return nil
  end

  def close()
   @handle.close() unless @handle.nil?
  end

end