Skip to content

Commit 821a9bb

Browse files
nitishkumar71nitishkumar71
committed
enable httperror for server status code
Signed-off-by: Nitishkumar Singh <nitishkumarsingh71@gmail.com>
1 parent 8e069b1 commit 821a9bb

File tree

2 files changed

+155
-34
lines changed

2 files changed

+155
-34
lines changed

‎client.go‎

Lines changed: 80 additions & 28 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,19 @@ func NewClient(gatewayURL *url.URL, auth ClientAuth, client *http.Client) *Clien
3535
}
3636
}
3737

38+
type HttpError struct {
39+
Err error
40+
Status int
41+
}
42+
43+
func (e *HttpError) Error() string {
44+
return e.Err.Error()
45+
}
46+
47+
func createHttpError(err error, statusCode int) *HttpError {
48+
return &HttpError{Err: err, Status: statusCode}
49+
}
50+
3851
// GetNamespaces get openfaas namespaces
3952
func (s *Client) GetNamespaces(ctx context.Context) ([]string, error) {
4053
u := s.GatewayURL
@@ -54,7 +67,7 @@ func (s *Client) GetNamespaces(ctx context.Context) ([]string, error) {
5467

5568
res, err := s.Client.Do(req)
5669
if err != nil {
57-
return namespaces, fmt.Errorf("unable to make request: %w", err)
70+
return namespaces, fmt.Errorf("unable to make HTTP request to OpenFaaS on URL: %s, error: %s", s.GatewayURL, err)
5871
}
5972

6073
if res.Body != nil {
@@ -66,8 +79,15 @@ func (s *Client) GetNamespaces(ctx context.Context) ([]string, error) {
6679
return namespaces, err
6780
}
6881

69-
if res.StatusCode == http.StatusUnauthorized {
70-
return namespaces, fmt.Errorf("check authorization, status code: %d", res.StatusCode)
82+
switch res.StatusCode {
83+
case http.StatusOK:
84+
break
85+
86+
case http.StatusUnauthorized:
87+
return namespaces, createHttpError(fmt.Errorf("unauthorized action, please setup authentication for this server"), res.StatusCode)
88+
89+
default:
90+
return namespaces, createHttpError(fmt.Errorf("server returned unexpected status code %d, message: %q", res.StatusCode, string(bytesOut)), res.StatusCode)
7191
}
7292

7393
if len(bytesOut) == 0 {
@@ -106,7 +126,7 @@ func (s *Client) GetFunctions(ctx context.Context, namespace string) ([]types.Fu
106126

107127
res, err := s.Client.Do(req)
108128
if err != nil {
109-
return []types.FunctionStatus{}, fmt.Errorf("unable to make HTTP request: %w", err)
129+
return []types.FunctionStatus{}, fmt.Errorf("unable to make HTTP request to OpenFaaS on URL: %s, error: %s", s.GatewayURL, err)
110130
}
111131

112132
if res.Body != nil {
@@ -115,6 +135,17 @@ func (s *Client) GetFunctions(ctx context.Context, namespace string) ([]types.Fu
115135

116136
body, _ := io.ReadAll(res.Body)
117137

138+
switch res.StatusCode {
139+
case http.StatusAccepted, http.StatusOK:
140+
break
141+
142+
case http.StatusUnauthorized:
143+
return nil, createHttpError(fmt.Errorf("unauthorized action, please setup authentication for this server"), res.StatusCode)
144+
145+
default:
146+
return nil, createHttpError(fmt.Errorf("server returned unexpected status code %d, message: %q", res.StatusCode, string(body)), res.StatusCode)
147+
}
148+
118149
functions := []types.FunctionStatus{}
119150
if err := json.Unmarshal(body, &functions); err != nil {
120151
return []types.FunctionStatus{},
@@ -142,7 +173,7 @@ func (s *Client) GetInfo(ctx context.Context) (SystemInfo, error) {
142173

143174
res, err := s.Client.Do(req)
144175
if err != nil {
145-
return SystemInfo{}, fmt.Errorf("unable to make HTTP request: %w", err)
176+
return SystemInfo{}, fmt.Errorf("unable to make HTTP request to OpenFaaS on URL: %s, error: %s", s.GatewayURL, err)
146177
}
147178

148179
if res.Body != nil {
@@ -185,7 +216,7 @@ func (s *Client) GetFunction(ctx context.Context, name, namespace string) (types
185216

186217
res, err := s.Client.Do(req)
187218
if err != nil {
188-
return types.FunctionDeployment{}, fmt.Errorf("unable to make HTTP request: %w", err)
219+
return types.FunctionDeployment{}, fmt.Errorf("unable to make HTTP request to OpenFaaS on URL: %s, error: %s", s.GatewayURL, err)
189220
}
190221

191222
if res.Body != nil {
@@ -194,6 +225,17 @@ func (s *Client) GetFunction(ctx context.Context, name, namespace string) (types
194225

195226
body, _ := io.ReadAll(res.Body)
196227

228+
switch res.StatusCode {
229+
case http.StatusOK:
230+
break
231+
232+
case http.StatusUnauthorized:
233+
return types.FunctionDeployment{}, createHttpError(fmt.Errorf("unauthorized action, please setup authentication for this server"), res.StatusCode)
234+
235+
default:
236+
return types.FunctionDeployment{}, createHttpError(fmt.Errorf("server returned unexpected status code %d, message: %q", res.StatusCode, string(body)), res.StatusCode)
237+
}
238+
197239
functions := types.FunctionDeployment{}
198240
if err := json.Unmarshal(body, &functions); err != nil {
199241
return types.FunctionDeployment{},
@@ -203,20 +245,30 @@ func (s *Client) GetFunction(ctx context.Context, name, namespace string) (types
203245
return functions, nil
204246
}
205247

206-
func (s *Client) Deploy(ctx context.Context, spec types.FunctionDeployment) (int, error) {
207-
return s.deploy(ctx, http.MethodPost, spec)
248+
// func (s *Client) Deploy(ctx context.Context, spec types.FunctionDeployment) (int, error) {
249+
// return s.deploy(ctx, http.MethodPost, spec)
208250

209-
}
251+
// }
210252

211-
func (s *Client) Update(ctx context.Context, spec types.FunctionDeployment) (int, error) {
253+
// func (s *Client) Update(ctx context.Context, spec types.FunctionDeployment) (int, error) {
254+
// return s.deploy(ctx, http.MethodPut, spec)
255+
// }
256+
257+
// func (s *Client) deploy(ctx context.Context, method string, spec types.FunctionDeployment) (int, error) {
258+
func (s *Client) Deploy(ctx context.Context, spec types.FunctionDeployment) error {
212259
return s.deploy(ctx, http.MethodPut, spec)
260+
213261
}
214262

215-
func (s *Client) deploy(ctx context.Context, method string, spec types.FunctionDeployment) (int, error) {
263+
func (s *Client) Update(ctx context.Context, spec types.FunctionDeployment) error {
264+
return s.deploy(ctx, http.MethodPost, spec)
265+
}
266+
267+
func (s *Client) deploy(ctx context.Context, method string, spec types.FunctionDeployment) error {
216268

217269
bodyBytes, err := json.Marshal(spec)
218270
if err != nil {
219-
return http.StatusBadRequest, err
271+
return createHttpError(err, http.StatusBadRequest)
220272
}
221273

222274
bodyReader := bytes.NewReader(bodyBytes)
@@ -226,18 +278,18 @@ func (s *Client) deploy(ctx context.Context, method string, spec types.FunctionD
226278

227279
req, err := http.NewRequestWithContext(ctx, method, u.String(), bodyReader)
228280
if err != nil {
229-
return http.StatusBadGateway, err
281+
return fmt.Errorf("unable to create request for %s, error: %w", u.String(), err)
230282
}
231283

232284
if s.ClientAuth != nil {
233285
if err := s.ClientAuth.Set(req); err != nil {
234-
return http.StatusInternalServerError, fmt.Errorf("unable to set Authorization header: %w", err)
286+
return fmt.Errorf("unable to set Authorization header: %w", err)
235287
}
236288
}
237289

238290
res, err := s.Client.Do(req)
239291
if err != nil {
240-
return http.StatusBadGateway, err
292+
return fmt.Errorf("unable to make HTTP request to OpenFaaS on URL: %s, error: %s", s.GatewayURL, err)
241293
}
242294

243295
var body []byte
@@ -248,13 +300,13 @@ func (s *Client) deploy(ctx context.Context, method string, spec types.FunctionD
248300

249301
switch res.StatusCode {
250302
case http.StatusAccepted, http.StatusOK, http.StatusCreated:
251-
return res.StatusCode, nil
303+
return nil
252304

253305
case http.StatusUnauthorized:
254-
return res.StatusCode, fmt.Errorf("unauthorized action, please setup authentication for this server")
306+
return createHttpError(fmt.Errorf("unauthorized action, please setup authentication for this server"), res.StatusCode)
255307

256308
default:
257-
return res.StatusCode, fmt.Errorf("unexpected status code: %d, message: %q", res.StatusCode, string(body))
309+
return createHttpError(fmt.Errorf("unexpected status code: %d, message: %q", res.StatusCode, string(body)), res.StatusCode)
258310
}
259311
}
260312

@@ -280,7 +332,7 @@ func (s *Client) ScaleFunction(ctx context.Context, functionName, namespace stri
280332

281333
req, err := http.NewRequestWithContext(ctx, http.MethodPost, u.String(), bodyReader)
282334
if err != nil {
283-
return fmt.Errorf("cannot connect to OpenFaaS on URL: %s, error: %s", u.String(), err)
335+
return fmt.Errorf("unable to create request for %s, error: %w", u.String(), err)
284336
}
285337

286338
if s.ClientAuth != nil {
@@ -290,7 +342,7 @@ func (s *Client) ScaleFunction(ctx context.Context, functionName, namespace stri
290342
}
291343
res, err := http.DefaultClient.Do(req)
292344
if err != nil {
293-
return fmt.Errorf("cannot connect to OpenFaaS on URL: %s, error: %s", s.GatewayURL, err)
345+
return fmt.Errorf("unable to make HTTP request to OpenFaaS on URL: %s, error: %s", s.GatewayURL, err)
294346

295347
}
296348

@@ -303,10 +355,10 @@ func (s *Client) ScaleFunction(ctx context.Context, functionName, namespace stri
303355
break
304356

305357
case http.StatusNotFound:
306-
return fmt.Errorf("function %s not found", functionName)
358+
return createHttpError(fmt.Errorf("function %s not found", functionName), res.StatusCode)
307359

308360
case http.StatusUnauthorized:
309-
return fmt.Errorf("unauthorized action, please setup authentication for this server")
361+
return createHttpError(fmt.Errorf("unauthorized action, please setup authentication for this server"), res.StatusCode)
310362

311363
default:
312364
var err error
@@ -315,7 +367,7 @@ func (s *Client) ScaleFunction(ctx context.Context, functionName, namespace stri
315367
return err
316368
}
317369

318-
return fmt.Errorf("server returned unexpected status code %d, message: %q", res.StatusCode, string(bytesOut))
370+
return createHttpError(fmt.Errorf("server returned unexpected status code %d, message: %q", res.StatusCode, string(bytesOut)), res.StatusCode)
319371
}
320372
return nil
321373
}
@@ -338,7 +390,7 @@ func (s *Client) DeleteFunction(ctx context.Context, functionName, namespace str
338390

339391
req, err := http.NewRequestWithContext(ctx, http.MethodDelete, u.String(), bodyReader)
340392
if err != nil {
341-
return fmt.Errorf("cannot connect to OpenFaaS on URL: %s, error: %s", u.String(), err)
393+
return fmt.Errorf("unable to create request for %s, error: %w", u.String(), err)
342394
}
343395

344396
if s.ClientAuth != nil {
@@ -348,7 +400,7 @@ func (s *Client) DeleteFunction(ctx context.Context, functionName, namespace str
348400
}
349401
res, err := http.DefaultClient.Do(req)
350402
if err != nil {
351-
return fmt.Errorf("cannot connect to OpenFaaS on URL: %s, error: %s", s.GatewayURL, err)
403+
return fmt.Errorf("unable to make HTTP request to OpenFaaS on URL: %s, error: %s", s.GatewayURL, err)
352404

353405
}
354406

@@ -361,10 +413,10 @@ func (s *Client) DeleteFunction(ctx context.Context, functionName, namespace str
361413
break
362414

363415
case http.StatusNotFound:
364-
return fmt.Errorf("function %s not found", functionName)
416+
return createHttpError(fmt.Errorf("function %s not found", functionName), res.StatusCode)
365417

366418
case http.StatusUnauthorized:
367-
return fmt.Errorf("unauthorized action, please setup authentication for this server")
419+
return createHttpError(fmt.Errorf("unauthorized action, please setup authentication for this server"), res.StatusCode)
368420

369421
default:
370422
var err error
@@ -373,7 +425,7 @@ func (s *Client) DeleteFunction(ctx context.Context, functionName, namespace str
373425
return err
374426
}
375427

376-
return fmt.Errorf("server returned unexpected status code %d, message: %q", res.StatusCode, string(bytesOut))
428+
return createHttpError(fmt.Errorf("server returned unexpected status code %d, message: %q", res.StatusCode, string(bytesOut)), res.StatusCode)
377429
}
378430
return nil
379431
}

‎client_test.go‎

Lines changed: 75 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -109,7 +109,7 @@ func TestSdk_DeployFunction(t *testing.T) {
109109
handler: func(rw http.ResponseWriter, req *http.Request) {
110110
rw.WriteHeader(http.StatusUnauthorized)
111111
},
112-
err: fmt.Errorf("unauthorized action, please setup authentication for this server"),
112+
err: createHttpError(fmt.Errorf("unauthorized action, please setup authentication for this server"), http.StatusUnauthorized),
113113
},
114114
{
115115
name: "unknown error",
@@ -118,7 +118,7 @@ func TestSdk_DeployFunction(t *testing.T) {
118118
handler: func(rw http.ResponseWriter, req *http.Request) {
119119
http.Error(rw, "unknown error", http.StatusInternalServerError)
120120
},
121-
err: fmt.Errorf("unexpected status code: %d, message: %q", http.StatusInternalServerError, "unknown error\n"),
121+
err: createHttpError(fmt.Errorf("unexpected status code: %d, message: %q", http.StatusInternalServerError, "unknown error\n"), http.StatusInternalServerError),
122122
},
123123
}
124124

@@ -130,7 +130,8 @@ func TestSdk_DeployFunction(t *testing.T) {
130130

131131
client := NewClient(sU, nil, http.DefaultClient)
132132

133-
_, err := client.Deploy(context.Background(), types.FunctionDeployment{
133+
// _, err := client.Deploy(context.Background(), types.FunctionDeployment{
134+
err := client.Deploy(context.Background(), types.FunctionDeployment{
134135
Service: funcName,
135136
Image: fmt.Sprintf("docker.io/openfaas/%s:latest", funcName),
136137
Namespace: nsName,
@@ -168,7 +169,7 @@ func TestSdk_DeleteFunction(t *testing.T) {
168169
handler: func(rw http.ResponseWriter, req *http.Request) {
169170
rw.WriteHeader(http.StatusNotFound)
170171
},
171-
err: fmt.Errorf("function %s not found", funcName),
172+
err: createHttpError(fmt.Errorf("function %s not found", funcName), http.StatusNotFound),
172173
},
173174
{
174175
name: "client not authorized",
@@ -177,7 +178,7 @@ func TestSdk_DeleteFunction(t *testing.T) {
177178
handler: func(rw http.ResponseWriter, req *http.Request) {
178179
rw.WriteHeader(http.StatusUnauthorized)
179180
},
180-
err: fmt.Errorf("unauthorized action, please setup authentication for this server"),
181+
err: createHttpError(fmt.Errorf("unauthorized action, please setup authentication for this server"), http.StatusUnauthorized),
181182
},
182183
{
183184
name: "unknown error",
@@ -186,7 +187,7 @@ func TestSdk_DeleteFunction(t *testing.T) {
186187
handler: func(rw http.ResponseWriter, req *http.Request) {
187188
http.Error(rw, "unknown error", http.StatusInternalServerError)
188189
},
189-
err: fmt.Errorf("server returned unexpected status code %d, message: %q", http.StatusInternalServerError, string("unknown error\n")),
190+
err: createHttpError(fmt.Errorf("server returned unexpected status code %d, message: %q", http.StatusInternalServerError, string("unknown error\n")), http.StatusUnauthorized),
190191
},
191192
}
192193

@@ -205,3 +206,71 @@ func TestSdk_DeleteFunction(t *testing.T) {
205206
})
206207
}
207208
}
209+
210+
func TestSdk_ScaleFunction(t *testing.T) {
211+
funcName := "funct1"
212+
nsName := "ns1"
213+
tests := []struct {
214+
name string
215+
functionName string
216+
namespace string
217+
replicas uint64
218+
err error
219+
handler func(rw http.ResponseWriter, req *http.Request)
220+
}{
221+
{
222+
name: "scale request accepted",
223+
functionName: funcName,
224+
namespace: nsName,
225+
replicas: 0,
226+
handler: func(rw http.ResponseWriter, req *http.Request) {
227+
rw.WriteHeader(http.StatusAccepted)
228+
},
229+
},
230+
{
231+
name: "function not found",
232+
functionName: funcName,
233+
namespace: nsName,
234+
replicas: 0,
235+
handler: func(rw http.ResponseWriter, req *http.Request) {
236+
rw.WriteHeader(http.StatusNotFound)
237+
},
238+
err: createHttpError(fmt.Errorf("function %s not found", funcName), http.StatusNotFound),
239+
},
240+
{
241+
name: "client not authorized",
242+
functionName: funcName,
243+
namespace: nsName,
244+
replicas: 0,
245+
handler: func(rw http.ResponseWriter, req *http.Request) {
246+
rw.WriteHeader(http.StatusUnauthorized)
247+
},
248+
err: createHttpError(fmt.Errorf("unauthorized action, please setup authentication for this server"), http.StatusUnauthorized),
249+
},
250+
{
251+
name: "unknown error",
252+
functionName: funcName,
253+
namespace: nsName,
254+
replicas: 0,
255+
handler: func(rw http.ResponseWriter, req *http.Request) {
256+
http.Error(rw, "unknown error", http.StatusInternalServerError)
257+
},
258+
err: createHttpError(fmt.Errorf("server returned unexpected status code %d, message: %q", http.StatusInternalServerError, string("unknown error\n")), http.StatusUnauthorized),
259+
},
260+
}
261+
262+
for _, test := range tests {
263+
t.Run(test.name, func(t *testing.T) {
264+
s := httptest.NewServer(http.HandlerFunc(test.handler))
265+
266+
sU, _ := url.Parse(s.URL)
267+
268+
client := NewClient(sU, nil, http.DefaultClient)
269+
err := client.ScaleFunction(context.Background(), test.functionName, test.namespace, test.replicas)
270+
271+
if !errors.Is(err, test.err) && err.Error() != test.err.Error() {
272+
t.Fatalf("wanted %s, but got: %s", test.err, err)
273+
}
274+
})
275+
}
276+
}

0 commit comments

Comments
 (0)