You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
85 lines
2.3 KiB
85 lines
2.3 KiB
// GoToSocial |
|
// Copyright (C) GoToSocial Authors admin@gotosocial.org |
|
// SPDX-License-Identifier: AGPL-3.0-or-later |
|
// |
|
// This program is free software: you can redistribute it and/or modify |
|
// it under the terms of the GNU Affero General Public License as published by |
|
// the Free Software Foundation, either version 3 of the License, or |
|
// (at your option) any later version. |
|
// |
|
// This program is distributed in the hope that it will be useful, |
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|
// GNU Affero General Public License for more details. |
|
// |
|
// You should have received a copy of the GNU Affero General Public License |
|
// along with this program. If not, see <http://www.gnu.org/licenses/>. |
|
|
|
package httpclient |
|
|
|
import ( |
|
"net/http" |
|
"strconv" |
|
"time" |
|
|
|
"github.com/superseriousbusiness/gotosocial/internal/log" |
|
) |
|
|
|
const ( |
|
// starting backoff duration. |
|
baseBackoff = 2 * time.Second |
|
) |
|
|
|
// Request wraps an HTTP request |
|
// to add our own retry / backoff. |
|
type Request struct { |
|
|
|
// Current backoff dur. |
|
backoff time.Duration |
|
|
|
// Delivery attempts. |
|
attempts uint |
|
|
|
// log fields. |
|
log.Entry |
|
|
|
// underlying request. |
|
*http.Request |
|
} |
|
|
|
// WrapRequest wraps an existing http.Request within |
|
// our own httpclient.Request with retry / backoff tracking. |
|
func WrapRequest(r *http.Request) *Request { |
|
rr := new(Request) |
|
rr.Request = r |
|
entry := log.WithContext(r.Context()) |
|
entry = entry.WithField("method", r.Method) |
|
entry = entry.WithField("url", r.URL.String()) |
|
if r.Body != nil { |
|
// Only add content-type header if a request body exists. |
|
entry = entry.WithField("contentType", r.Header.Get("Content-Type")) |
|
} |
|
entry = entry.WithField("attempt", uintPtr{&rr.attempts}) |
|
rr.Entry = entry |
|
return rr |
|
} |
|
|
|
// GetBackOff returns the currently set backoff duration, |
|
// (using a default according to no. attempts if needed). |
|
func (r *Request) BackOff() time.Duration { |
|
if r.backoff <= 0 { |
|
// No backoff dur found, set our predefined |
|
// backoff according to a multiplier of 2^n. |
|
r.backoff = baseBackoff * 1 << (r.attempts + 1) |
|
} |
|
return r.backoff |
|
} |
|
|
|
type uintPtr struct{ u *uint } |
|
|
|
func (f uintPtr) String() string { |
|
if f.u == nil { |
|
return "<nil>" |
|
} |
|
return strconv.FormatUint(uint64(*f.u), 10) |
|
}
|
|
|