Class: DB::MariaDB::Native::Connection

Inherits:
FFI::Pointer
  • Object
show all
Defined in:
lib/db/mariadb/native/connection.rb

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(address, io, types, **options) ⇒ Connection

Returns a new instance of Connection.



117
118
119
120
121
122
123
124
 
# File 'lib/db/mariadb/native/connection.rb', line 117

def initialize(address, io, types, **options)
	super(address)
	
	@io = io
	@result = nil
	
	@types = types
end

Class Method Details

.connect(wrapper: IO, host: 'localhost', username: nil, password: nil, database: nil, port: 0, unix_socket: nil, client_flags: 0, compression: false, types: DEFAULT_TYPES, **options) ⇒ Object 



76
77
78
79
80
81
82
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
 
# File 'lib/db/mariadb/native/connection.rb', line 76

def self.connect(wrapper: IO, host: 'localhost', username: nil, password: nil, database: nil, port: 0, unix_socket: nil, client_flags: 0, compression: false, types: DEFAULT_TYPES, **options)
	pointer = Native.mysql_init(nil)
	Native.mysql_options(pointer, MYSQL_OPT_NONBLOCK, nil)
	
	# if protocol
	# 	Native.mysql_options(pointer, MYSQL_OPT_PROTOCOL, FFI::MemoryPointer.new(:uint, protocol))
	# end
	
	client_flags |= CLIENT_MULTI_STATEMENT | CLIENT_MULTI_RESULTS
	
	if compression
		client_flags |= CLIENT_COMPRESSION
	end
	
	result = FFI::MemoryPointer.new(:pointer)
	
	status = Native.mysql_real_connect_start(result, pointer, host, username, password, database, port, unix_socket, client_flags);
	
	io = wrapper.new(Native.mysql_get_socket(pointer), "r+")
	
	if status > 0
		while status > 0
			if status & MYSQL_WAIT_READ
				io.wait_readable
			elsif status & MYSQL_WAIT_WRITE
				io.wait_writable
			else
				io.wait_any
			end
			
			status = Native.mysql_real_connect_cont(result, pointer, status)
		end
	end
	
	if result.read_pointer.null?
		raise Error, "Could not connect: #{Native.mysql_error(pointer)}!"
	end
	
	return self.new(pointer, io, types, **options)
end

Instance Method Details

#affected_rows ⇒ Object 



208
209
210
 
# File 'lib/db/mariadb/native/connection.rb', line 208

def affected_rows
	Native.mysql_affected_rows(self)
end

#check_error!(message) ⇒ Object 



134
135
136
137
138
 
# File 'lib/db/mariadb/native/connection.rb', line 134

def check_error!(message)
	if Native.mysql_errno(self) != 0
		raise Error, "#{message}: #{Native.mysql_error(self)}!"
	end
end

#close ⇒ Object 



152
153
154
155
156
157
158
 
# File 'lib/db/mariadb/native/connection.rb', line 152

def close
	self.free_result
	
	Native.mysql_close(self)
	
	@io.close
end

#discard_results ⇒ Object

Silently discard any results that application didn’t read.



201
202
203
204
205
206
 
# File 'lib/db/mariadb/native/connection.rb', line 201

def discard_results
	while result = self.get_result
	end
	
	return nil
end

#escape(value) ⇒ Object 



160
161
162
163
164
165
166
167
168
169
 
# File 'lib/db/mariadb/native/connection.rb', line 160

def escape(value)
	value = value.to_s
	
	maximum_length = value.bytesize * 2 + 1
	out = FFI::MemoryPointer.new(:char, maximum_length)
	
	Native.mysql_real_escape_string(self, out, value, value.bytesize)
	
	return out.read_string
end

#free_result ⇒ Object 



144
145
146
147
148
149
150
 
# File 'lib/db/mariadb/native/connection.rb', line 144

def free_result
	if @result
		Native.mysql_free_result(@result)
		
		@result = nil
	end
end

#info ⇒ Object 



216
217
218
 
# File 'lib/db/mariadb/native/connection.rb', line 216

def info
	Native.mysql_info(self)
end

#insert_id ⇒ Object 



212
213
214
 
# File 'lib/db/mariadb/native/connection.rb', line 212

def insert_id
	Native.mysql_insert_id(self)
end

#more_results? ⇒ Boolean

Returns:

  • (Boolean) 


190
191
192
 
# File 'lib/db/mariadb/native/connection.rb', line 190

def more_results?
	Native.mysql_more_results(self) == 1
end

#next_result(types: @types) ⇒ Object 



194
195
196
197
198
 
# File 'lib/db/mariadb/native/connection.rb', line 194

def next_result(types: @types)
	if result = self.get_result
		return Result.new(self, types, result)
	end
end

#send_query(statement) ⇒ Object 



171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
 
# File 'lib/db/mariadb/native/connection.rb', line 171

def send_query(statement)
	self.free_result
	
	error = FFI::MemoryPointer.new(:int)
	
	status = Native.mysql_real_query_start(error, self, statement, statement.bytesize)
	
	while status != 0
		self.wait_for(status)
		
		status = Native.mysql_real_query_cont(error, self, status)
	end
	
	if error.read_int != 0
		raise Error, "Could not send query: #{Native.mysql_error(self)}!"
	end
end

#status ⇒ Object 



140
141
142
 
# File 'lib/db/mariadb/native/connection.rb', line 140

def status
	Native.mysql_stat(self)
end

#wait_for(status) ⇒ Object 



126
127
128
129
130
131
132
 
# File 'lib/db/mariadb/native/connection.rb', line 126

def wait_for(status)
	if status & MYSQL_WAIT_READ
		@io.wait_readable
	elsif status & MYSQL_WAIT_WRITE
		@io.wait_writable
	end
end