Module: Riddle::Query

Defined in:
lib/riddle/query.rb

Defined Under Namespace

Classes: Delete, Insert, Select

Class Method Summary collapse

Class Method Details

.beginObject



43
44
45
# File 'lib/riddle/query.rb', line 43

def self.begin
  'BEGIN'
end

.collationObject



35
36
37
# File 'lib/riddle/query.rb', line 35

def self.collation
  'SHOW COLLATION'
end

.commitObject



47
48
49
# File 'lib/riddle/query.rb', line 47

def self.commit
  'COMMIT'
end

.connection(address = '127.0.0.1', port = 9312) ⇒ Object



2
3
4
5
6
7
8
9
10
11
12
13
# File 'lib/riddle/query.rb', line 2

def self.connection(address = '127.0.0.1', port = 9312)
  require 'mysql2'

  # If you use localhost, MySQL insists on a socket connection, but Sphinx
  # requires a TCP connection. Using 127.0.0.1 fixes that.
  address = '127.0.0.1' if address == 'localhost'

  Mysql2::Client.new(
    :host => address,
    :port => port
  )
end

.create_function(name, type, file) ⇒ Object



73
74
75
76
# File 'lib/riddle/query.rb', line 73

def self.create_function(name, type, file)
  type = type.to_s.upcase
  "CREATE FUNCTION #{name} RETURNS #{type} SONAME #{quote file}"
end

.describe(index) ⇒ Object



39
40
41
# File 'lib/riddle/query.rb', line 39

def self.describe(index)
  "DESCRIBE #{index}"
end

.drop_function(name) ⇒ Object



78
79
80
# File 'lib/riddle/query.rb', line 78

def self.drop_function(name)
  "DROP FUNCTION #{name}"
end

.escape(string) ⇒ Object



101
102
103
# File 'lib/riddle/query.rb', line 101

def self.escape(string)
  string.gsub(/[\(\)\|\-!@~\/"\/\^\$\\]/) { |match| "\\#{match}" }
end

.metaObject



15
16
17
# File 'lib/riddle/query.rb', line 15

def self.meta
  'SHOW META'
end

.quote(string) ⇒ Object



105
106
107
# File 'lib/riddle/query.rb', line 105

def self.quote(string)
  "'#{sql_escape string}'"
end

.rollbackObject



51
52
53
# File 'lib/riddle/query.rb', line 51

def self.rollback
  'ROLLBACK'
end

.set(variable, values, global = true) ⇒ Object



55
56
57
58
# File 'lib/riddle/query.rb', line 55

def self.set(variable, values, global = true)
  values = "(#{values.join(', ')})" if values.is_a?(Array)
  "SET#{ ' GLOBAL' if global } #{variable} = #{values}"
end

.snippets(data, index, query, options = nil) ⇒ Object



60
61
62
63
64
65
66
67
68
69
70
71
# File 'lib/riddle/query.rb', line 60

def self.snippets(data, index, query, options = nil)
  data, index, query = quote(data), quote(index), quote(query)

  options = ', ' + options.keys.collect { |key|
    value = translate_value options[key]
    value = quote value if value.is_a?(String)

    "#{value} AS #{key}"
  }.join(', ') unless options.nil?

  "CALL SNIPPETS(#{data}, #{index}, #{query}#{options})"
end

.sql_escape(string) ⇒ Object



109
110
111
112
113
# File 'lib/riddle/query.rb', line 109

def self.sql_escape(string)
  return Mysql2::Client.escape(string) if defined?(Mysql2)

  string.gsub(/['"\\]/) { |character| "\\#{character}" }
end

.statusObject



23
24
25
# File 'lib/riddle/query.rb', line 23

def self.status
  'SHOW STATUS'
end

.tablesObject



27
28
29
# File 'lib/riddle/query.rb', line 27

def self.tables
  'SHOW TABLES'
end

.translate_value(value) ⇒ Object



90
91
92
93
94
95
96
97
98
99
# File 'lib/riddle/query.rb', line 90

def self.translate_value(value)
  case value
  when TrueClass
    1
  when FalseClass
    0
  else
    value
  end
end

.update(index, id, values = {}) ⇒ Object



82
83
84
85
86
87
88
# File 'lib/riddle/query.rb', line 82

def self.update(index, id, values = {})
  values = values.keys.collect { |key|
    "#{key} = #{translate_value values[key]}"
  }.join(', ')

  "UPDATE #{index} SET #{values} WHERE id = #{id}"
end

.variablesObject



31
32
33
# File 'lib/riddle/query.rb', line 31

def self.variables
  'SHOW VARIABLES'
end

.warningsObject



19
20
21
# File 'lib/riddle/query.rb', line 19

def self.warnings
  'SHOW WARNINGS'
end