Imported FTC Clock

lib/game_clock/net/connection.rb

@@ -0,0 +1,97 @@
+module TAC
+  class PracticeGameClock
+    class ClockNet
+  class Connection
+    TAG = "ClockNet|Connection"
+    attr_reader :hostname, :port, :client, :proxy_object
+
+    def initialize(hostname: "localhost", port: 4567, proxy_object:)
+      @hostname = hostname
+      @port = port
+      @proxy_object = proxy_object
+
+      @client = nil
+
+      @last_sync_time = Gosu.milliseconds
+      @sync_interval = SYNC_INTERVAL
+
+      @last_heartbeat_sent = Gosu.milliseconds
+      @heartbeat_interval = HEARTBEAT_INTERVAL
+
+      @connection_handler = proc do
+        handle_connection
+      end
+
+      @packet_handler = PacketHandler.new(host_is_a_connection: true, proxy_object: @proxy_object)
+    end
+
+    def connect
+      return if @client
+
+      @client = Client.new
+
+      Thread.new do
+        begin
+          @client.socket = Socket.tcp(@hostname, @port, connect_timeout: 5)
+          @client.socket.setsockopt(Socket::IPPROTO_TCP, Socket::TCP_NODELAY, 1)
+          log.i(TAG, "Connected to: #{@hostname}:#{@port}")
+
+          while @client && @client.connected?
+            if Gosu.milliseconds > @last_sync_time + @sync_interval
+              @last_sync_time = Gosu.milliseconds
+
+              @client.sync(@connection_handler)
+            end
+          end
+
+        rescue => error
+          log.e(TAG, error)
+
+          if @client
+            @client.close
+            @client.last_socket_error = error
+            @client.socket_error = true
+          end
+        end
+      end
+    end
+
+    def handle_connection
+      if @client && @client.connected?
+        message = @client.gets
+
+        @packet_handler.handle(message) if message
+
+        if Gosu.milliseconds > @last_heartbeat_sent + @heartbeat_interval
+          @last_heartbeat_sent = Gosu.milliseconds
+
+          @client.puts(PacketHandler.packet_heartbeat)
+        end
+
+        sleep @sync_interval / 1000.0
+      end
+    end
+
+    def puts(packet)
+      @client.puts(packet)
+    end
+
+    def gets
+      @client.gets
+    end
+
+    def connected?
+      @client.connected? if @client
+    end
+
+    def closed?
+      @client.closed? if @client
+    end
+
+    def close
+      @client.close if @client
+    end
+  end
+    end
+  end
+end