Make webhook http connections resuable (#6976)

* make http connections resuable

* add error handler

* fix lint
This commit is contained in:
Lunny Xiao 2019-05-21 15:20:17 +08:00 committed by GitHub
parent 02542a2c15
commit 1f84970de0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -13,11 +13,12 @@ import (
"encoding/json" "encoding/json"
"fmt" "fmt"
"io/ioutil" "io/ioutil"
"net"
"net/http" "net/http"
"net/url"
"strings" "strings"
"time" "time"
"code.gitea.io/gitea/modules/httplib"
"code.gitea.io/gitea/modules/log" "code.gitea.io/gitea/modules/log"
"code.gitea.io/gitea/modules/setting" "code.gitea.io/gitea/modules/setting"
api "code.gitea.io/gitea/modules/structs" api "code.gitea.io/gitea/modules/structs"
@ -753,47 +754,66 @@ func prepareWebhooks(e Engine, repo *Repository, event HookEventType, p api.Payl
return nil return nil
} }
func (t *HookTask) deliver() { func (t *HookTask) deliver() error {
t.IsDelivered = true t.IsDelivered = true
timeout := time.Duration(setting.Webhook.DeliverTimeout) * time.Second var req *http.Request
var err error
var req *httplib.Request
switch t.HTTPMethod { switch t.HTTPMethod {
case "": case "":
log.Info("HTTP Method for webhook %d empty, setting to POST as default", t.ID) log.Info("HTTP Method for webhook %d empty, setting to POST as default", t.ID)
fallthrough fallthrough
case http.MethodPost: case http.MethodPost:
req = httplib.Post(t.URL)
switch t.ContentType { switch t.ContentType {
case ContentTypeJSON: case ContentTypeJSON:
req = req.Header("Content-Type", "application/json").Body(t.PayloadContent) req, err = http.NewRequest("POST", t.URL, strings.NewReader(t.PayloadContent))
if err != nil {
return err
}
req.Header.Set("Content-Type", "application/json")
case ContentTypeForm: case ContentTypeForm:
req.Param("payload", t.PayloadContent) var forms = url.Values{
"payload": []string{t.PayloadContent},
}
req, err = http.NewRequest("POST", t.URL, strings.NewReader(forms.Encode()))
if err != nil {
return err
}
} }
case http.MethodGet: case http.MethodGet:
req = httplib.Get(t.URL).Param("payload", t.PayloadContent) u, err := url.Parse(t.URL)
if err != nil {
return err
}
vals := u.Query()
vals["payload"] = []string{t.PayloadContent}
u.RawQuery = vals.Encode()
req, err = http.NewRequest("GET", u.String(), nil)
if err != nil {
return err
}
default: default:
log.Error("Invalid http method for webhook: [%d] %v", t.ID, t.HTTPMethod) return fmt.Errorf("Invalid http method for webhook: [%d] %v", t.ID, t.HTTPMethod)
return
} }
req = req.SetTimeout(timeout, timeout). req.Header.Add("X-Gitea-Delivery", t.UUID)
Header("X-Gitea-Delivery", t.UUID). req.Header.Add("X-Gitea-Event", string(t.EventType))
Header("X-Gitea-Event", string(t.EventType)). req.Header.Add("X-Gitea-Signature", t.Signature)
Header("X-Gitea-Signature", t.Signature). req.Header.Add("X-Gogs-Delivery", t.UUID)
Header("X-Gogs-Delivery", t.UUID). req.Header.Add("X-Gogs-Event", string(t.EventType))
Header("X-Gogs-Event", string(t.EventType)). req.Header.Add("X-Gogs-Signature", t.Signature)
Header("X-Gogs-Signature", t.Signature). req.Header["X-GitHub-Delivery"] = []string{t.UUID}
HeaderWithSensitiveCase("X-GitHub-Delivery", t.UUID). req.Header["X-GitHub-Event"] = []string{string(t.EventType)}
HeaderWithSensitiveCase("X-GitHub-Event", string(t.EventType)).
SetTLSClientConfig(&tls.Config{InsecureSkipVerify: setting.Webhook.SkipTLSVerify})
// Record delivery information. // Record delivery information.
t.RequestInfo = &HookRequest{ t.RequestInfo = &HookRequest{
Headers: map[string]string{}, Headers: map[string]string{},
} }
for k, vals := range req.Headers() { for k, vals := range req.Header {
t.RequestInfo.Headers[k] = strings.Join(vals, ",") t.RequestInfo.Headers[k] = strings.Join(vals, ",")
} }
@ -830,10 +850,10 @@ func (t *HookTask) deliver() {
} }
}() }()
resp, err := req.Response() resp, err := webhookHTTPClient.Do(req)
if err != nil { if err != nil {
t.ResponseInfo.Body = fmt.Sprintf("Delivery: %v", err) t.ResponseInfo.Body = fmt.Sprintf("Delivery: %v", err)
return return err
} }
defer resp.Body.Close() defer resp.Body.Close()
@ -847,9 +867,10 @@ func (t *HookTask) deliver() {
p, err := ioutil.ReadAll(resp.Body) p, err := ioutil.ReadAll(resp.Body)
if err != nil { if err != nil {
t.ResponseInfo.Body = fmt.Sprintf("read body: %s", err) t.ResponseInfo.Body = fmt.Sprintf("read body: %s", err)
return return err
} }
t.ResponseInfo.Body = string(p) t.ResponseInfo.Body = string(p)
return nil
} }
// DeliverHooks checks and delivers undelivered hooks. // DeliverHooks checks and delivers undelivered hooks.
@ -864,7 +885,10 @@ func DeliverHooks() {
// Update hook task status. // Update hook task status.
for _, t := range tasks { for _, t := range tasks {
t.deliver() if err = t.deliver(); err != nil {
log.Error("deliver: %v", err)
continue
}
} }
// Start listening on new hook requests. // Start listening on new hook requests.
@ -884,12 +908,34 @@ func DeliverHooks() {
continue continue
} }
for _, t := range tasks { for _, t := range tasks {
t.deliver() if err = t.deliver(); err != nil {
log.Error("deliver: %v", err)
}
} }
} }
} }
var webhookHTTPClient *http.Client
// InitDeliverHooks starts the hooks delivery thread // InitDeliverHooks starts the hooks delivery thread
func InitDeliverHooks() { func InitDeliverHooks() {
timeout := time.Duration(setting.Webhook.DeliverTimeout) * time.Second
webhookHTTPClient = &http.Client{
Transport: &http.Transport{
TLSClientConfig: &tls.Config{InsecureSkipVerify: setting.Webhook.SkipTLSVerify},
Dial: func(netw, addr string) (net.Conn, error) {
conn, err := net.DialTimeout(netw, addr, timeout)
if err != nil {
return nil, err
}
conn.SetDeadline(time.Now().Add(timeout))
return conn, nil
},
},
}
go DeliverHooks() go DeliverHooks()
} }