Subversion Repositories SmartDukaan

# 

# Licensed to the Apache Software Foundation (ASF) under one

# or more contributor license agreements. See the NOTICE file

# distributed with this work for additional information

# regarding copyright ownership. The ASF licenses this file

# to you under the Apache License, Version 2.0 (the

# "License"); you may not use this file except in compliance

# with the License. You may obtain a copy of the License at

# 

#   http://www.apache.org/licenses/LICENSE-2.0

# 

# Unless required by applicable law or agreed to in writing,

# software distributed under the License is distributed on an

# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY

# KIND, either express or implied. See the License for the

# specific language governing permissions and limitations

# under the License.

# 

module Thrift

  module Client

    def initialize(iprot, oprot=nil)

      @iprot = iprot

      @oprot = oprot || iprot

      @seqid = 0

    end

    def send_message(name, args_class, args = {})

      @oprot.write_message_begin(name, MessageTypes::CALL, @seqid)

      data = args_class.new

      args.each do |k, v|

        data.send("#{k.to_s}=", v)

      end

      begin

        data.write(@oprot)

      rescue StandardError => e

        @oprot.trans.close

        raise e

      end

      @oprot.write_message_end

      @oprot.trans.flush

    end

    def receive_message(result_klass)

      fname, mtype, rseqid = @iprot.read_message_begin

      handle_exception(mtype)

      result = result_klass.new

      result.read(@iprot)

      @iprot.read_message_end

      result

    end

    def handle_exception(mtype)

      if mtype == MessageTypes::EXCEPTION

        x = ApplicationException.new

        x.read(@iprot)

        @iprot.read_message_end

        raise x

      end

    end

  end

end