Sha256: 795dfa69e581de6a10fb3cc17818409a6136378affa7eb573a9b4be1193d439e

Contents?: true

Size: 1.98 KB

Versions: 24

Compression:

Stored size: 1.98 KB

Contents

# frozen_string_literal: true

require "ipaddr"

module HTTPX
  class UDP
    include Loggable

    def initialize(uri, _, options)
      ip = IPAddr.new(uri.host)
      @host = ip.to_s
      @port = uri.port
      @io = UDPSocket.new(ip.family)
      @options = options
    end

    def to_io
      @io.to_io
    end

    def connect; end

    def connected?
      true
    end

    if RUBY_VERSION < "2.3"
      # :nocov:
      def close
        @io.close
      rescue StandardError
        nil
      end
      # :nocov:
    else
      def close
        @io.close
      end
    end

    # :nocov:
    if (RUBY_ENGINE == "truffleruby" && RUBY_ENGINE_VERSION < "21.1.0") ||
       RUBY_VERSION < "2.3"
      def write(buffer)
        siz = @io.sendmsg_nonblock(buffer.to_s, 0, Socket.sockaddr_in(@port, @host.to_s))
        log { "WRITE: #{siz} bytes..." }
        buffer.shift!(siz)
        siz
      rescue ::IO::WaitWritable
        0
      rescue EOFError
        nil
      end

      def read(size, buffer)
        data, _ = @io.recvfrom_nonblock(size)
        buffer.replace(data)
        log { "READ: #{buffer.bytesize} bytes..." }
        buffer.bytesize
      rescue ::IO::WaitReadable
        0
      rescue IOError
      end
    else

      def write(buffer)
        siz = @io.sendmsg_nonblock(buffer.to_s, 0, Socket.sockaddr_in(@port, @host.to_s), exception: false)
        return 0 if siz == :wait_writable
        return if siz.nil?

        log { "WRITE: #{siz} bytes..." }

        buffer.shift!(siz)
        siz
      end

      def read(size, buffer)
        ret = @io.recvfrom_nonblock(size, 0, buffer, exception: false)
        return 0 if ret == :wait_readable
        return if ret.nil?

        buffer.bytesize
      rescue IOError
      end
    end

    # In JRuby, sendmsg_nonblock is not implemented
    def write(buffer)
      siz = @io.send(buffer.to_s, 0, @host, @port)
      log { "WRITE: #{siz} bytes..." }
      buffer.shift!(siz)
      siz
    end if RUBY_ENGINE == "jruby"
    # :nocov:
  end
end

Version data entries

24 entries across 24 versions & 1 rubygems

Version Path
httpx-0.18.7 lib/httpx/io/udp.rb
httpx-0.18.6 lib/httpx/io/udp.rb
httpx-0.18.5 lib/httpx/io/udp.rb
httpx-0.18.4 lib/httpx/io/udp.rb