Class: MiniSql::InlineParamEncoder

Inherits:
Object
  • Object
show all
Defined in:
lib/mini_sql/inline_param_encoder.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(conn) ⇒ InlineParamEncoder

Returns a new instance of InlineParamEncoder.



7
8
9
# File 'lib/mini_sql/inline_param_encoder.rb', line 7

def initialize(conn)
  @conn = conn
end

Instance Attribute Details

#connObject (readonly)

Returns the value of attribute conn.



5
6
7
# File 'lib/mini_sql/inline_param_encoder.rb', line 5

def conn
  @conn
end

Instance Method Details

#encode(sql, *params) ⇒ Object



11
12
13
14
15
16
17
18
# File 'lib/mini_sql/inline_param_encoder.rb', line 11

def encode(sql, *params)
  if Hash === (hash = params[0])
    raise ArgumentError, "Only one hash param is allowed, multiple were sent" if params.length > 1
    encode_hash(sql, hash)
  else
    encode_array(sql, params)
  end
end

#encode_array(sql, array) ⇒ Object



37
38
39
40
41
42
43
# File 'lib/mini_sql/inline_param_encoder.rb', line 37

def encode_array(sql, array)
  i = -1
  sql.gsub("?") do |p|
    i += 1
    quote_val(array[i])
  end
end

#encode_hash(sql, hash) ⇒ Object



20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
# File 'lib/mini_sql/inline_param_encoder.rb', line 20

def encode_hash(sql, hash)
  sql = sql.dup

  hash.each do |k, v|
    sql.gsub!(":#{k}") do
      # ignore ::int and stuff like that
      # $` is previous to match
      if $` && $`[-1] != ":"
        quote_val(v)
      else
        ":#{k}"
      end
    end
  end
  sql
end

#quote_val(value) ⇒ Object



49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
# File 'lib/mini_sql/inline_param_encoder.rb', line 49

def quote_val(value)
  case value
  when String     then "'#{conn.escape_string(value.to_s)}'"
  when Numeric    then value.to_s
  when BigDecimal then value.to_s("F")
  when Date, Time then "'#{quoted_date(value)}'"
  when Symbol     then "'#{conn.escape_string(value.to_s)}'"
  when true       then "true"
  when false      then "false"
  when nil        then "NULL"
  when []         then "NULL"
  when Array
    value.map do |v|
      quote_val(v)
    end.join(', ')
  else raise TypeError, "can't quote #{value.class.name}"
  end
end

#quoted_date(value) ⇒ Object



45
46
47
# File 'lib/mini_sql/inline_param_encoder.rb', line 45

def quoted_date(value)
  value.utc.iso8601
end