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
|
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
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
end
class RestClientWrapper
# PENDING: remove as soon as redirect tasks are remove from partner webservices
def self.redirect_task( uri )
raise "no redirect task uri: "+uri.to_s unless uri.to_s =~ /194.141.0.136|ambit.*task|tu-muenchen.*task/
while (uri.to_s =~ /194.141.0.136|ambit.*task|tu-muenchen.*task/)
#HACK handle redirect
LOGGER.debug "REDIRECT TASK: "+uri.to_s
redirect = ""
while (redirect.size == 0)
IO.popen("bin/redirect.sh "+uri.to_s) do |f|
while line = f.gets
redirect += line.chomp
end
end
raise redirect!=nil && redirect.size>0 ? redirect : "TASK ERROR" if $?!=0
sleep 0.3
end
uri = redirect
LOGGER.debug "REDIRECT TO: "+uri.to_s
end
return uri
end
def self.get(uri, headers=nil)
execute( "get", uri, headers)
end
def self.post(uri, headers, payload=nil, wait=true)
execute( "post", uri, headers, payload, wait )
end
def self.put(uri, headers, payload=nil )
execute( "put", uri, headers, payload )
end
def self.delete(uri, headers=nil)
execute( "delete", uri, headers, nil)
end
def self.raise_uri_error(error_msg, uri, headers=nil, payload=nil)
do_halt( "-", error_msg, uri, headers, payload )
end
private
def self.execute( rest_call, uri, headers, payload=nil, wait=true )
do_halt 400,"uri is null",uri,headers,payload unless uri
do_halt 400,"not a uri",uri,headers,payload unless Utils.is_uri?(uri)
do_halt 400,"headers are no hash",uri,headers,payload unless headers==nil or headers.is_a?(Hash)
do_halt 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, :user => @@users[:users].keys[0], :password => @@users[:users].values[0]})
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 filed content_type
res = WrapperResult.new(result.to_s)
res.content_type = result.headers[:content_type]
# get result cannot be a task
return res if rest_call=="get" or !wait
return res if res.strip.size==0
# try to load task from result (maybe task-uri, or task-object)
task = nil
case res.content_type
when /application\/rdf\+xml|text\/x-yaml/
task = OpenTox::Task.from_data(res, res.content_type, uri)
when /text\//
return res 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
# HACK for redirect tasks
if res =~ /ambit.*task|tu-muenchen.*task/
res = WrapperResult.new(redirect_task(res))
res.content_type = "text/uri-list"
return res
end
task = OpenTox::Task.find(res) if Utils.task_uri?(res)
else
raise "unknown content-type when checking for task: "+res.content_type+" content: "+res[0..200]
end
# task could be loaded, wait for task to finish
if task
LOGGER.debug "result is a task "+task.uri.to_s+", wait for completion"
task.wait_for_completion
raise task.description if task.error?
res = WrapperResult.new(task.resultURI)
LOGGER.debug "task resultURI "+res.to_s
res.content_type = "text/uri-list"
end
return res
rescue RestClient::RequestFailed => ex
do_halt ex.http_code,ex.http_body,uri,headers,payload
rescue RestClient::RequestTimeout => ex
do_halt 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
do_halt code,msg,uri,headers,payload
end
end
def self.do_halt( 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)
# return error (by halting, halts should be logged)
# PENDING always return yaml for now
begin
if defined?(halt)
halt(502,error.to_yaml)
elsif defined?($sinatra)
$sinatra.halt(502,error.to_yaml)
else
raise ""
end
rescue
raise error.to_yaml
end
end
end
end
|
