jgaskins/neo4j.cr

module Neo4j
  module Bolt
    class Connection
      private def run(statement, parameters = {} of String => Type, retries = 0)
        write_message do |msg|
          msg.write_structure_start 2
          msg.write_byte Commands::Run
          msg.write statement
          msg.write parameters
        end

        result = read_result
        case result
        when Failure
          raise ::Neo4j::QueryException.new(result.attrs["message"].as(String), result.attrs["code"].as(String))
        when Success, Ignored
          result
        else
          raise ::Neo4j::UnknownResult.new("Cannot identify this result: #{result.inspect}")
        end
      rescue ex : IO::EOFError
        initialize @uri, @ssl
        if retries < 10
          sleep 0.1
          run statement, parameters, retries + 1
        else
          raise ex
        end
      end
    end
  end
end