Class: Puppet::Util::GraphDBRepositoryManager

Inherits:
Object
  • Object
show all
Defined in:
lib/puppet/util/graphdb_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) ⇒ GraphDBRepositoryManager

Returns a new instance of GraphDBRepositoryManager.



27
28
29
30
 
# File 'lib/puppet/util/graphdb_repository_manager.rb', line 27

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

Instance Attribute Details

#endpoint ⇒ Object (readonly)

Returns the value of attribute endpoint.



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

def endpoint
  @endpoint
end

#repository_id ⇒ Object (readonly)

Returns the value of attribute repository_id.



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

def repository_id
  @repository_id
end

Instance Method Details

#ask(query, expected_response, timeout) ⇒ Object 



130
131
132
133
134
135
136
137
138
139
140
141
142
 
# File 'lib/puppet/util/graphdb_repository_manager.rb', line 130

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::GraphDBRequestManager.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 



32
33
34
35
36
37
38
39
 
# File 'lib/puppet/util/graphdb_repository_manager.rb', line 32

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

#create_repository(repository_template, repository_context, timeout) ⇒ Object 



62
63
64
65
66
67
68
69
70
71
72
73
74
75
 
# File 'lib/puppet/util/graphdb_repository_manager.rb', line 62

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::GraphDBRequestManager.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

#define_node_id(node_id) ⇒ Object 



77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
 
# File 'lib/puppet/util/graphdb_repository_manager.rb', line 77

def define_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::GraphDBRequestManager.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

#define_repository_replication_port(replication_port) ⇒ Object 



97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
 
# File 'lib/puppet/util/graphdb_repository_manager.rb', line 97

def define_repository_replication_port(replication_port)
  tmp_expr = "#{endpoint}/repositories/#{repository_id}"
  Puppet.debug "Trying to set repository replication port [#{tmp_expr}] 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::GraphDBRequestManager.perform_http_request(uri,
                                                           { method: :post,
                                                             content_type: 'application/json',
                                                             body_data: body.to_json },
                                                           { codes: [200] }, 0)

  Puppet.notice("Repository [#{tmp_expr}] replication port set to [#{replication_port}].")
end

#delete_repository(timeout) ⇒ Object 



118
119
120
121
122
123
124
125
126
127
128
 
# File 'lib/puppet/util/graphdb_repository_manager.rb', line 118

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

  Puppet.debug "Trying to delete repository [#{endpoint}/repositories/#{repository_id}]"
  Puppet::Util::GraphDBRequestManager.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 



157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
 
# File 'lib/puppet/util/graphdb_repository_manager.rb', line 157

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::GraphDBRequestManager.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) 


41
42
43
44
45
46
47
48
49
50
51
 
# File 'lib/puppet/util/graphdb_repository_manager.rb', line 41

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

#repository_up?(timeout) ⇒ Boolean

Returns:

  • (Boolean) 


53
54
55
56
57
58
59
60
 
# File 'lib/puppet/util/graphdb_repository_manager.rb', line 53

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

#resolve_content_type(data_format) ⇒ Object

Raises:

  • (ArgumentError) 


174
175
176
177
178
 
# File 'lib/puppet/util/graphdb_repository_manager.rb', line 174

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

#update_query(query, timeout) ⇒ Object 



144
145
146
147
148
149
150
151
152
153
154
155
 
# File 'lib/puppet/util/graphdb_repository_manager.rb', line 144

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::GraphDBRequestManager.perform_http_request(uri,
                                                           { method: :post,
                                                             body_params: { 'update' => query },
                                                             content_type: 'application/x-www-form-urlencoded' },
                                                           { codes: [204] },
                                                           timeout)
end