1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
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
118
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
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
|
module OpenTox
#PENDING: implement ot error api, move to own file
class Error
attr_accessor :code, :body, :uri, :payload, :headers
def initialize(code, body, uri, payload, headers)
self.code = code
self.body = body.to_s[0..1000]
self.uri = uri
self.payload = payload
self.headers = headers
end
def self.parse(error_array_string)
begin
err = YAML.load(error_array_string)
if err and err.is_a?(Array) and err.size>0 and err[0].is_a?(Error)
return err
else
return nil
end
rescue
return nil
end
end
end
class WrapperResult < String
attr_accessor :content_type, :code
end
class RestClientWrapper
# performs a GET REST call
# raises OpenTox::Error if call fails (rescued in overwrite.rb -> halt 502)
# per default: waits for Task to finish and returns result URI of Task
# @param [String] uri destination URI
# @param [optional,Hash] headers contains params like accept-header
# @param [optional,OpenTox::Task] waiting_task (can be a OpenTox::Subtask as well), progress is updated accordingly
# @param [wait,Boolean] wait set to false to NOT wait for task if result is task
# @return [OpenTox::WrapperResult] a String containing the result-body of the REST call
def self.get(uri, headers=nil, waiting_task=nil, wait=true )
execute( "get", uri, headers, nil, waiting_task, wait)
end
# performs a POST REST call
# raises OpenTox::Error if call fails (rescued in overwrite.rb -> halt 502)
# per default: waits for Task to finish and returns result URI of Task
# @param [String] uri destination URI
# @param [optional,Hash] headers contains params like accept-header
# @param [optional,String] payload data posted to the service
# @param [optional,OpenTox::Task] waiting_task (can be a OpenTox::Subtask as well), progress is updated accordingly
# @param [wait,Boolean] wait set to false to NOT wait for task if result is task
# @return [OpenTox::WrapperResult] a String containing the result-body of the REST call
def self.post(uri, headers, payload=nil, waiting_task=nil, wait=true )
execute( "post", uri, headers, payload, waiting_task, wait )
end
# performs a PUT REST call
# raises OpenTox::Error if call fails (rescued in overwrite.rb -> halt 502)
# @param [String] uri destination URI
# @param [optional,Hash] headers contains params like accept-header
# @param [optional,String] payload data put to the service
# @return [OpenTox::WrapperResult] a String containing the result-body of the REST call
def self.put(uri, headers, payload=nil )
execute( "put", uri, headers, payload )
end
# performs a DELETE REST call
# raises OpenTox::Error if call fails (rescued in overwrite.rb -> halt 502)
# @param [String] uri destination URI
# @param [optional,Hash] headers contains params like accept-header
# @return [OpenTox::WrapperResult] a String containing the result-body of the REST call
def self.delete(uri, headers=nil )
execute( "delete", uri, headers, nil)
end
# raises an Error message (rescued in overwrite.rb -> halt 502)
# usage: if the return value of a call is invalid
# @param [String] error_msg the error message
# @param [String] uri destination URI that is responsible for the error
# @param [optional,Hash] headers sent to the URI
# @param [optional,String] payload data sent to the URI
def self.raise_uri_error(error_msg, uri, headers=nil, payload=nil)
raise_ot_error( "-", error_msg, uri, headers, payload )
end
private
def self.execute( rest_call, uri, headers, payload=nil, waiting_task=nil, wait=true )
raise_ot_error 400,"uri is null",uri,headers,payload unless uri
raise_ot_error 400,"not a uri",uri,headers,payload unless uri.to_s.uri?
raise_ot_error 400,"headers are no hash",uri,headers,payload unless headers==nil or headers.is_a?(Hash)
raise_ot_error 400,"nil headers for post not allowed, use {}",uri,headers,payload if rest_call=="post" and headers==nil
headers.each{ |k,v| headers.delete(k) if v==nil } if headers #remove keys with empty values, as this can cause problems
begin
#LOGGER.debug "RestCall: "+rest_call.to_s+" "+uri.to_s+" "+headers.inspect
resource = RestClient::Resource.new(uri,{:timeout => 60})
if payload
result = resource.send(rest_call, payload, headers)
elsif headers
result = resource.send(rest_call, headers)
else
result = resource.send(rest_call)
end
# result is a string, with the additional fields content_type and code
res = WrapperResult.new(result.body)
res.content_type = result.headers[:content_type]
raise "content-type not set" unless res.content_type
res.code = result.code
# TODO: Ambit returns task representation with 200 instead of result URI
return res if res.code==200 || !wait
while (res.code==201 || res.code==202)
res = wait_for_task(res, uri, waiting_task)
end
raise "illegal status code: '"+res.code.to_s+"'" unless res.code==200
return res
rescue RestClient::RequestTimeout => ex
raise_ot_error 408,ex.message,uri,headers,payload
rescue => ex
#raise ex
#raise "'"+ex.message+"' uri: "+uri.to_s
begin
code = ex.http_code
msg = ex.http_body
rescue
code = 500
msg = ex.to_s
end
raise_ot_error code,msg,uri,headers,payload
end
end
def self.wait_for_task( res, base_uri, waiting_task=nil )
task = nil
case res.content_type
when /application\/rdf\+xml/
task = OpenTox::Task.from_rdfxml(res)
when /yaml/
task = OpenTox::Task.from_yaml(res)
when /text\//
raise "uri list has more than one entry, should be a task" if res.content_type=~/text\/uri-list/ and res.split("\n").size > 1 #if uri list contains more then one uri, its not a task
task = OpenTox::Task.find(res.to_s.chomp) if res.to_s.uri?
else
raise "unknown content-type for task: '"+res.content_type.to_s+"'" #+"' content: "+res[0..200].to_s
end
LOGGER.debug "result is a task '"+task.uri.to_s+"', wait for completion"
task.wait_for_completion waiting_task
raise task.description unless task.completed? # maybe task was cancelled / error
res = WrapperResult.new task.result_uri
res.code = task.http_code
res.content_type = "text/uri-list"
return res
end
def self.raise_ot_error( code, body, uri, headers, payload=nil )
#build error
causing_errors = Error.parse(body)
if causing_errors
error = causing_errors + [Error.new(code, "subsequent error", uri, payload, headers)]
else
error = [Error.new(code, body, uri, payload, headers)]
end
#debug utility: write error to file
error_dir = "/tmp/ot_errors"
FileUtils.mkdir(error_dir) unless File.exist?(error_dir)
raise "could not create error dir" unless File.exist?(error_dir) and File.directory?(error_dir)
file_name = "error"
time=Time.now.strftime("%m.%d.%Y-%H:%M:%S")
count = 1
count+=1 while File.exist?(File.join(error_dir,file_name+"_"+time+"_"+count.to_s))
File.new(File.join(error_dir,file_name+"_"+time+"_"+count.to_s),"w").puts(body)
# handle error
# PENDING: always return yaml for now
# raising OpenTox::Error
# to handle the error yourself, put rest-call in begin, rescue block
# if the error is not caught:
# if we are in a task, the error is caught, logged, and task state is set to error in Task.as_task
# if we are in a default call, the error is handled in overwrite.rb to return 502 (according to OT API)
raise error.to_yaml
end
end
end
|