Sha256: 61a83b29b908aed8ff6d676c6a726598142f61ed60d3148896cbf5681007bd63

Contents?: true

Size: 1.89 KB

Versions: 5

Compression:

Stored size: 1.89 KB

Contents

#
# 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.
#

require 'rack'

module Thrift
  class RackApplication
    THRIFT_HEADER = "application/x-thrift"

    def self.for(path, processor, protocol_factory)
      Rack::Builder.new do
        map path do
          run lambda { |env|
            request = Rack::Request.new(env)
            if RackApplication.valid_thrift_request?(request)
              RackApplication.successful_request(request, processor, protocol_factory)
            else
              RackApplication.failed_request
            end
          }
        end
      end
    end

    def self.successful_request(rack_request, processor, protocol_factory)
      response = Rack::Response.new([], 200, {'Content-Type' => THRIFT_HEADER})
      transport = IOStreamTransport.new rack_request.body, response
      protocol = protocol_factory.get_protocol transport
      processor.process protocol, protocol
      response
    end

    def self.failed_request
      Rack::Response.new(['Not Found'], 404, {'Content-Type' => THRIFT_HEADER})
    end

    def self.valid_thrift_request?(rack_request)
      rack_request.post? && rack_request.env["CONTENT_TYPE"] == THRIFT_HEADER
    end
  end
end

Version data entries

5 entries across 5 versions & 1 rubygems

Version Path
upfluence-thrift-1.0.12 lib/thrift/server/rack_application.rb
upfluence-thrift-1.0.11 lib/thrift/server/rack_application.rb
upfluence-thrift-1.0.10 lib/thrift/server/rack_application.rb
upfluence-thrift-1.0.9 lib/thrift/server/rack_application.rb
upfluence-thrift-1.0.8 lib/thrift/server/rack_application.rb