Class: Puppet::Util::RepositoryManager

Inherits:
Object
  • Object
show all
Defined in:
lib/puppet/util/repository_manager.rb

Overview

GraphDB repository manager

Constant Summary collapse

DATA_TYPE_CONTENT_TYPE = 
{ 'rdfxml' => 'application/rdf+xml; charset=utf-8',
'ntriples' => 'application/n-triples; charset=utf-8',
'turtle' => 'application/x-turtle; charset=utf-8',
'n3' => 'text/n3; charset=utf-8',
'trix' => 'application/trix; charset=utf-8',
'trig' => 'application/x-trig; charset=utf-8',
'binary' => 'application/x-binary-rdf; charset=utf-8',
'nquads' => 'application/n-quads; charset=utf-8',
'jsonld' => 'application/ld+json; charset=utf-8',
'rdfjson' => 'application/rdf+json; charset=utf-8',
'rdfa' => 'application/html; charset=utf-8' }.freeze

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(endpoint, repository_id) ⇒ RepositoryManager

Returns a new instance of RepositoryManager.



24
25
26
27
 
# File 'lib/puppet/util/repository_manager.rb', line 24

def initialize(endpoint, repository_id)
  @endpoint = endpoint
  @repository_id = repository_id
end

Instance Attribute Details

#endpoint ⇒ Object (readonly)

Returns the value of attribute endpoint.



21
22
23
 
# File 'lib/puppet/util/repository_manager.rb', line 21

def endpoint
  @endpoint
end

#repository_id ⇒ Object (readonly)

Returns the value of attribute repository_id.



22
23
24
 
# File 'lib/puppet/util/repository_manager.rb', line 22

def repository_id
  @repository_id
end

Instance Method Details

#ask(query, expected_response, timeout) ⇒ Object 



125
126
127
128
129
130
131
132
133
134
135
136
 
# File 'lib/puppet/util/repository_manager.rb', line 125

def ask(query, expected_response, timeout)
  uri = endpoint.dup
  uri.path = "/repositories/#{repository_id}"

  Puppet.debug "Trying to ask repository [#{endpoint}/#{repository_id}]"
  Puppet::Util::RequestManager.perform_http_request(uri, { method: :get,
                                                           params: { 'query' => query },
                                                           content_type: 'application/x-www-form-urlencoded',
                                                           accept_type: 'text/boolean' },
                                                    { messages: [expected_response.to_s], codes: [200] },
                                                    timeout)
end

#check_repository(timeout) ⇒ Object 



29
30
31
32
33
34
35
36
 
# File 'lib/puppet/util/repository_manager.rb', line 29

def check_repository(timeout)
  Puppet.debug "Checking repository [#{endpoint}/repositories/#{repository_id}]"
  unless repository_exists?
    raise Puppet::Exceptions::RequestFailError,
          "Repository [#{endpoint}/repositories/#{repository_id}] doesn't exists"
  end
  repository_up?(timeout)
end

#create_repository(repository_template, repository_context, timeout) ⇒ Object 



59
60
61
62
63
64
65
66
67
68
69
70
71
72
 
# File 'lib/puppet/util/repository_manager.rb', line 59

def create_repository(repository_template, repository_context, timeout)
  Puppet.debug "Trying to create repository [#{endpoint}/repositories/#{repository_id}]"
  uri = endpoint.dup
  uri.path = '/repositories/SYSTEM/rdf-graphs/service'

  Puppet::Util::RequestManager.perform_http_request(uri,
                                                    { method: :post,
                                                      params: { 'graph' => repository_context },
                                                      body_data: repository_template,
                                                      content_type: 'application/x-turtle' },
                                                    { codes: [204] },
                                                    timeout)
  Puppet.notice("Repository [#{endpoint}/repositories/#{repository_id}] creation passed.")
end

#delete_repository(timeout) ⇒ Object 



114
115
116
117
118
119
120
121
122
123
 
# File 'lib/puppet/util/repository_manager.rb', line 114

def delete_repository(timeout)
  uri = endpoint.dup
  uri.path = "/repositories/#{repository_id}"

  Puppet.debug "Trying to delete repository [#{endpoint}/repositories/#{repository_id}]"
  Puppet::Util::RequestManager.perform_http_request(uri,
                                                    { method: :delete, content_type: 'application/x-turtle' },
                                                    { codes: [204] },
                                                    timeout)
end

#load_data(data, data_format, data_context, overwrite, timeout) ⇒ Object 



151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
 
# File 'lib/puppet/util/repository_manager.rb', line 151

def load_data(data, data_format, data_context, overwrite, timeout)
  method = overwrite == true ? :put : :post
  uri = endpoint.dup
  uri.path = "/repositories/#{repository_id}/statements"

  context = data_context.nil? ? 'null' : data_context

  Puppet.debug "Trying to load data into repository [#{endpoint}/#{repository_id}]"
  Puppet::Util::RequestManager.perform_http_request(uri,
                                                    { method: method,
                                                      params: { 'context' => context },
                                                      body_data: data,
                                                      content_type: resolve_content_type(data_format) },
                                                    { codes: [204] },
                                                    timeout)
end

#repository_exists? ⇒ Boolean

Returns:

  • (Boolean) 


38
39
40
41
42
43
44
45
46
47
48
 
# File 'lib/puppet/util/repository_manager.rb', line 38

def repository_exists?
  uri = endpoint.dup
  uri.path = "/repositories/#{repository_id}/size"
  Puppet::Util::RequestManager.perform_http_request(uri,
                                                    { method: :get },
                                                    { codes: [404] },
                                                    0)
  false
rescue
  true
end

#repository_up?(timeout) ⇒ Boolean

Returns:

  • (Boolean) 


50
51
52
53
54
55
56
57
 
# File 'lib/puppet/util/repository_manager.rb', line 50

def repository_up?(timeout)
  uri = endpoint.dup
  uri.path = "/repositories/#{repository_id}/size"
  Puppet::Util::RequestManager.perform_http_request(uri,
                                                    { method: :get },
                                                    { messages: ['No workers configured', '\d+'],
                                                      codes: [200, 500] }, timeout)
end

#resolve_content_type(data_format) ⇒ Object

Raises:

  • (ArgumentError) 


168
169
170
171
 
# File 'lib/puppet/util/repository_manager.rb', line 168

def resolve_content_type(data_format)
  return DATA_TYPE_CONTENT_TYPE[data_format] if DATA_TYPE_CONTENT_TYPE.key?(data_format)
  raise ArgumentError, "Unknown data format [#{data_format}], please check"
end

#set_node_id(node_id) ⇒ Object 



74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
 
# File 'lib/puppet/util/repository_manager.rb', line 74

def set_node_id(node_id)
  Puppet.debug "Trying to set node id [#{endpoint}/repositories/#{repository_id}] to [#{node_id}]"
  uri = endpoint.dup
  uri.path = '/jolokia'
  body = {
    'type'      => 'write',
    'mbean'     => "ReplicationCluster:name=ClusterInfo/#{repository_id}",
    'attribute' => 'NodeID',
    'value'     => node_id
  }

  Puppet::Util::RequestManager.perform_http_request(uri,
                                                    { method: :post,
                                                      content_type: "application/json",
                                                      body_data: body.to_json },
                                                    { codes: [200] }, 0)

  Puppet.notice("Repository [#{endpoint}/repositories/#{repository_id}] node id set to [#{node_id}].")
end

#set_repository_replication_port(replication_port) ⇒ Object 



94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
 
# File 'lib/puppet/util/repository_manager.rb', line 94

def set_repository_replication_port(replication_port)
  Puppet.debug "Trying to set repository replication port [#{endpoint}/repositories/#{repository_id}] to [#{replication_port}]"
  uri = endpoint.dup
  uri.path = '/jolokia'
  body = {
    'type'      => 'write',
    'mbean'     => "ReplicationCluster:name=ClusterInfo/#{repository_id}",
    'attribute' => 'MasterReplicationPort',
    'value'     => replication_port
  }

  Puppet::Util::RequestManager.perform_http_request(uri,
                                                    { method: :post,
                                                      content_type: "application/json",
                                                      body_data: body.to_json },
                                                    { codes: [200] }, 0)

  Puppet.notice("Repository [#{endpoint}/repositories/#{repository_id}] replication port set to [#{replication_port}].")
end

#update_query(query, timeout) ⇒ Object 



138
139
140
141
142
143
144
145
146
147
148
149
 
# File 'lib/puppet/util/repository_manager.rb', line 138

def update_query(query, timeout)
  uri = endpoint.dup
  uri.path = "/repositories/#{repository_id}/statements"

  Puppet.debug "Trying to post update query to repository [#{endpoint}/#{repository_id}]"
  Puppet::Util::RequestManager.perform_http_request(uri,
                                                    { method: :post,
                                                      body_params: { 'update' => query },
                                                      content_type: 'application/x-www-form-urlencoded' },
                                                    { codes: [204] },
                                                    timeout)
end