internal: Fixed the cluster status cannot be changed after the POD exception exits. #353

runkecheng · runkecheng · commit c1b775a74227 · 2022-01-13T18:08:23.000+08:00
diff --git a/internal/http_executor.go b/internal/http_executor.go
@@ -17,9 +17,11 @@ limitations under the License.
 package internal
 
 import (
+	"context"
 	"fmt"
 	"io"
 	"net/http"
+	"time"
 )
 
 type Request struct {
@@ -63,7 +65,9 @@ func NewHttpExecutor(httpClient HttpClient) Executor {
 }
 
 func (executor *httpExecutor) Execute(req *Request) (*Response, error) {
-	resp, err := executor.Client.Do(req.Req)
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+	resp, err := executor.Client.Do(req.Req.WithContext(ctx))
 	if err != nil {
 		return nil, err
 	}

Original file line number	Diff line number	Diff line change
`@@ -17,9 +17,11 @@ limitations under the License.`
`17`	`17`	`package internal`
`18`	`18`
`19`	`19`	`import (`
	`20`	`+ "context"`
`20`	`21`	`"fmt"`
`21`	`22`	`"io"`
`22`	`23`	`"net/http"`
	`24`	`+ "time"`
`23`	`25`	`)`
`24`	`26`
`25`	`27`	`type Request struct {`
`@@ -63,7 +65,9 @@ func NewHttpExecutor(httpClient HttpClient) Executor {`
`63`	`65`	`}`
`64`	`66`
`65`	`67`	`func (executor httpExecutor) Execute(req Request) (*Response, error) {`
`66`		`- resp, err := executor.Client.Do(req.Req)`
	`68`	`+ ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)`
	`69`	`+ defer cancel()`
	`70`	`+ resp, err := executor.Client.Do(req.Req.WithContext(ctx))`
`67`	`71`	`if err != nil {`
`68`	`72`	`return nil, err`
`69`	`73`	`}`