Class: Protocol::Redis::Connection

Inherits:
Object
  • Object
show all
Defined in:
lib/protocol/redis/connection.rb

Constant Summary collapse

CRLF = 
"\r\n".freeze

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(stream) ⇒ Connection

Returns a new instance of Connection.



28
29
30
 
# File 'lib/protocol/redis/connection.rb', line 28

def initialize(stream)
	@stream = stream
end

Instance Attribute Details

#stream ⇒ Object (readonly)

Returns the value of attribute stream.



32
33
34
 
# File 'lib/protocol/redis/connection.rb', line 32

def stream
  @stream
end

Instance Method Details

#close ⇒ Object 



34
35
36
 
# File 'lib/protocol/redis/connection.rb', line 34

def close
	@stream.close
end

#closed? ⇒ Boolean

Returns:

  • (Boolean) 


46
47
48
 
# File 'lib/protocol/redis/connection.rb', line 46

def closed?
	@stream.closed?
end

#flush ⇒ Object 



42
43
44
 
# File 'lib/protocol/redis/connection.rb', line 42

def flush
	@stream.flush
end

#read_data(length) ⇒ Object 



74
75
76
77
78
79
80
81
 
# File 'lib/protocol/redis/connection.rb', line 74

def read_data(length)
	buffer = @stream.read(length) or @stream.eof!
	
	# Eat trailing whitespace because length does not include the CRLF:
	@stream.read(2) or @stream.eof!
	
	return buffer
end

#read_object ⇒ Object Also known as: read_response 



83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
 
# File 'lib/protocol/redis/connection.rb', line 83

def read_object
	line = read_line or raise EOFError
	
	token = line.slice!(0, 1)
	
	case token
	when '$'
		length = line.to_i
		
		if length == -1
			return nil
		else
			return read_data(length)
		end
	when '*'
		count = line.to_i
		
		# Null array (https://redis.io/topics/protocol#resp-arrays):
		return nil if count == -1
		
		array = Array.new(count) {read_object}
		
		return array
	when ':'
		return line.to_i
	
	when '-'
		raise ServerError.new(line)
	
	when '+'
		return line
	
	else
		@stream.flush
		
		raise NotImplementedError, "Implementation for token #{token} missing"
	end
	
	# TODO: If an exception (e.g. Async::TimeoutError) propagates out of this function, perhaps @stream should be closed? Otherwise it might be in a weird state.
end

#write_object(object) ⇒ Object 



61
62
63
64
65
66
67
68
69
70
71
72
 
# File 'lib/protocol/redis/connection.rb', line 61

def write_object(object)
	case object
	when String
		write_lines("$#{object.bytesize}", object)
	when Array
		write_array(object)
	when Integer
		write_lines(":#{object}")
	else
		write_object(object.to_redis)
	end
end

#write_request(arguments) ⇒ Object

The redis server doesn’t want actual objects (e.g. integers) but only bulk strings. So, we inline it for performance.



51
52
53
54
55
56
57
58
59
 
# File 'lib/protocol/redis/connection.rb', line 51

def write_request(arguments)
	write_lines("*#{arguments.size}")
	
	arguments.each do |argument|
		string = argument.to_s
		
		write_lines("$#{string.bytesize}", string)
	end
end