Merge pull request #66 from UpCloudLtd/client-timeout

fix(client/service): better timeout handling

Author: Trdls

upcloud/client/client.go

@@ -42,7 +42,10 @@ func NewWithHTTPClient(userName string, password string, httpClient *http.Client
 	client.userName = userName
 	client.password = password
 	client.httpClient = httpClient
-	client.SetTimeout(time.Second * DefaultTimeout)
+	// Set the default timeout if the caller hasn't set its own
+	if client.httpClient.Timeout == 0 {
+		client.SetTimeout(time.Second * DefaultTimeout)
+	}
 
 	return &client
 }

upcloud/service/service.go

@@ -222,13 +222,20 @@ func (s *Service) StartServer(r *request.StartServerRequest) (*upcloud.ServerDet
 
 // StopServer stops the specified server
 func (s *Service) StopServer(r *request.StopServerRequest) (*upcloud.ServerDetails, error) {
+	// Save previous timeout
+	prevTimeout := s.client.GetTimeout()
+
 	// Increase the client timeout to match the request timeout
-	s.client.SetTimeout(r.Timeout)
+	// Allow ten seconds to give the API a chance to respond with an error
+	s.client.SetTimeout(r.Timeout + 10*time.Second)
 
 	serverDetails := upcloud.ServerDetails{}
 	requestBody, _ := json.Marshal(r)
 	response, err := s.client.PerformJSONPostRequest(s.client.CreateRequestURL(r.RequestURL()), requestBody)
 
+	// Restore previous timeout
+	s.client.SetTimeout(prevTimeout)
+
 	if err != nil {
 		return nil, parseJSONServiceError(err)
 	}
@@ -240,13 +247,20 @@ func (s *Service) StopServer(r *request.StopServerRequest) (*upcloud.ServerDetai
 
 // RestartServer restarts the specified server
 func (s *Service) RestartServer(r *request.RestartServerRequest) (*upcloud.ServerDetails, error) {
+	// Save previous timeout
+	prevTimeout := s.client.GetTimeout()
+
 	// Increase the client timeout to match the request timeout
-	s.client.SetTimeout(r.Timeout)
+	// Allow ten seconds to give the API a chance to respond with an error
+	s.client.SetTimeout(r.Timeout + 10*time.Second)
 
 	serverDetails := upcloud.ServerDetails{}
 	requestBody, _ := json.Marshal(r)
 	response, err := s.client.PerformJSONPostRequest(s.client.CreateRequestURL(r.RequestURL()), requestBody)
 
+	// Restore previous timeout
+	s.client.SetTimeout(prevTimeout)
+
 	if err != nil {
 		return nil, parseJSONServiceError(err)
 	}