Class: Hiera::Backend::Mysql_json_backend

Inherits:
Object
  • Object
show all
Defined in:
lib/hiera/backend/mysql_json_backend.rb

Instance Method Summary collapse

Constructor Details

#initialize(cache = nil) ⇒ Mysql_json_backend

Returns a new instance of Mysql_json_backend.



5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
 
# File 'lib/hiera/backend/mysql_json_backend.rb', line 5

def initialize(cache=nil)
  require 'json'
  if defined?(JRUBY_VERSION)
    require 'jdbc/mysql'
    require 'java'
  else
    begin
      require 'mysql2'
    rescue LoadError
      require 'rubygems'
      require 'mysql2'
    end
  end

  @cache = cache || Filecache.new

  Hiera.debug('Hiera mysql_json initialized')
end

Instance Method Details

#compile_regexes(regexes) ⇒ Object 



37
38
39
40
 
# File 'lib/hiera/backend/mysql_json_backend.rb', line 37

def compile_regexes(regexes)
  res = regexes.map { |re| Regexp.compile(re) }
  Regexp.union(res)
end

#lookup(key, scope, order_override, resolution_type) ⇒ Object 



42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
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
116
117
 
# File 'lib/hiera/backend/mysql_json_backend.rb', line 42

def lookup(key, scope, order_override, resolution_type)
  # default answer is set to nil otherwise the lookup ends up returning
  # an Array of nils and causing a Puppet::Parser::AST::Resource failed with error ArgumentError
  # for any other lookup because their default value is overwriten by [nil,nil,nil,nil]
  # so hiera('myvalue', 'test1') returns [nil,nil,nil,nil]
  results = nil

  Hiera.debug("looking up #{key} in mysql_json Backend")
  Hiera.debug("resolution type is #{resolution_type}")

  return nil unless should_lookup?(Config[:mysql_json][:only_for], scope)

  Backend.datasources(scope, order_override) do |source|
    Hiera.debug("Looking for data source #{source}")
    sqlfile = Backend.datafile(:mysql_json, scope, source, "sql") || next

    next unless File.exist?(sqlfile)
    data = @cache.read(sqlfile, Hash, {}) do |datafile|
      YAML.load(datafile)
    end

    mysql_config = data.fetch(:dbconfig, {})
    mysql_host = mysql_config.fetch(:host, nil) || Config[:mysql_json][:host] || 'localhost'
    mysql_user = mysql_config.fetch(:user, nil) || Config[:mysql_json][:user]
    mysql_pass = mysql_config.fetch(:pass, nil) || Config[:mysql_json][:pass]
    mysql_port = mysql_config.fetch(:port, nil) || Config[:mysql_json][:port] || '3306'
    mysql_database = mysql_config.fetch(:database, nil) || Config[:mysql_json][:database]

    connection_hash = {
      host:      mysql_host,
      username:  mysql_user,
      password:  mysql_pass,
      database:  mysql_database,
      port:      mysql_port,
      reconnect: true
    }

    Hiera.debug("data #{data.inspect}")
    next if data.empty?
    next unless data.include?(key)

    Hiera.debug("Found #{key} in #{source}")

    query = Backend.parse_answer(data[key], scope)

    sql_results = query(connection_hash, query)
    # TODO: make sure we fail if we have more than 1 result, skip if less than 1.
    next if sql_results.length != 1

    new_answer = nil
    sqlvalue = sql_results[0]['value']

    begin
      new_answer = JSON.parse("{\"r\":#{sqlvalue}}")['r']
    rescue
      Hiera.debug("Miserable failure while parsing #{key} as JSON.")
      raise Exception, "Parse error for key '#{key}'. Offending data: #{sql_results[0]['value']}." unless Config[:mysql_json][:ignore_json_parse_errors]
      next
    end

    case resolution_type.is_a?(Hash) ? :hash : resolution_type
    when :array
      raise Exception, "Hiera type mismatch for key '#{key}': expected Array and got #{new_answer.class}" unless new_answer.kind_of? Array or new_answer.kind_of? String
      results ||= []
      results << new_answer
    when :hash
      raise Exception, "Hiera type mismatch for key '#{key}': expected Hash and got #{new_answer.class}" unless new_answer.kind_of? Hash
      results ||= {}
      results = Backend.merge_answer(new_answer, results, resolution_type)
    else
      results = new_answer
      break
    end
  end
  results
end

#query(connection_hash, query) ⇒ Object 



119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
 
# File 'lib/hiera/backend/mysql_json_backend.rb', line 119

def query(connection_hash, query)
  Hiera.debug("Executing SQL Query: #{query}")

  data = []
  mysql_host     = connection_hash[:host]
  mysql_user     = connection_hash[:username]
  mysql_pass     = connection_hash[:password]
  mysql_database = connection_hash[:database]
  mysql_port     = connection_hash[:port]

  if defined?(JRUBY_VERSION)
    Jdbc::MySQL.load_driver
    url = "jdbc:mysql://#{mysql_host}:#{mysql_port}/#{mysql_database}"
    props = java.util.Properties.new
    props.set_property :user, mysql_user
    props.set_property :password, mysql_pass

    conn = com.mysql.jdbc.Driver.new.connect(url, props)
    stmt = conn.create_statement

    res = stmt.execute_query(query)
    md = res.getMetaData
    numcols = md.getColumnCount

    Hiera.debug("Mysql Query returned #{numcols} rows")

    while res.next
      row = {}
      (1..numcols).each do |c|
        row[md.getColumnName(c)] = res.getString(c)
      end
      data << row
    end
    stmt.close
    conn.close
    data
  else
    client = Mysql2::Client.new(connection_hash)
    begin
      data = client.query(query).to_a
      Hiera.debug("Mysql Query returned #{data.size} rows")
    rescue => e
      Hiera.debug e.message
      data = nil
    ensure
      client.close
    end
  end

  data
end

#should_lookup?(constraints, scope) ⇒ Boolean

Returns:

  • (Boolean) 


24
25
26
27
28
29
30
31
32
33
34
35
 
# File 'lib/hiera/backend/mysql_json_backend.rb', line 24

def should_lookup?(constraints, scope)
  return true unless constraints.is_a?(Hash)
  should_lookup = false
  constraints.each do |item, matchers|
    next unless scope[item.to_s]
    if scope[item.to_s] =~ compile_regexes(matchers)
      should_lookup = true
      break
    end
  end
  should_lookup
end