-
Notifications
You must be signed in to change notification settings - Fork 3.8k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Release note (cli change): Added a new HTTP sink to the logging system. This can be configured similarly to other log sinks with the new 'http-servers' and 'http-defaults' sections of the logging config passed via the "--log" or "--log-config-file" command line flags.
- Loading branch information
1 parent
d8ba460
commit df5af9d
Showing
12 changed files
with
719 additions
and
287 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,149 @@ | ||
| // Copyright 2020 The Cockroach Authors. | ||
| // | ||
| // Use of this software is governed by the Business Source License | ||
| // included in the file licenses/BSL.txt. | ||
| // | ||
| // As of the Change Date specified in that file, in accordance with | ||
| // the Business Source License, use of this software will be governed | ||
| // by the Apache License, Version 2.0, included in the file | ||
| // licenses/APL.txt. | ||
|
|
||
| package log | ||
|
|
||
| import ( | ||
| "bytes" | ||
| "crypto/tls" | ||
| "fmt" | ||
| "net" | ||
| "net/http" | ||
| "net/url" | ||
| "time" | ||
|
|
||
| "github.com/cockroachdb/cockroach/pkg/cli/exit" | ||
| ) | ||
|
|
||
| var insecureTransport http.RoundTripper = &http.Transport{ | ||
| // Same as DefaultTransport... | ||
| Proxy: http.ProxyFromEnvironment, | ||
| DialContext: (&net.Dialer{ | ||
| Timeout: 30 * time.Second, | ||
| KeepAlive: 30 * time.Second, | ||
| }).DialContext, | ||
| ForceAttemptHTTP2: true, | ||
| MaxIdleConns: 100, | ||
| IdleConnTimeout: 90 * time.Second, | ||
| TLSHandshakeTimeout: 10 * time.Second, | ||
| ExpectContinueTimeout: 1 * time.Second, | ||
| // ...except insecure TLS. | ||
| TLSClientConfig: &tls.Config{InsecureSkipVerify: true}, | ||
| } | ||
|
|
||
| type httpSinkOptions struct { | ||
| unsafeTLS bool | ||
| timeout time.Duration | ||
| method string | ||
| } | ||
|
|
||
| func newHTTPSink(url string, opt httpSinkOptions) *httpSink { | ||
| hs := &httpSink{ | ||
| client: http.Client{ | ||
| Transport: http.DefaultTransport, | ||
| Timeout: opt.timeout}, | ||
| address: url, | ||
| doRequest: doPost} | ||
|
|
||
| if opt.unsafeTLS { | ||
| hs.client.Transport = insecureTransport | ||
| } | ||
|
|
||
| if opt.method == http.MethodGet { | ||
| hs.doRequest = doGet | ||
| } | ||
|
|
||
| return hs | ||
| } | ||
|
|
||
| type httpSink struct { | ||
| client http.Client | ||
| address string | ||
| doRequest func(*httpSink, []byte) (*http.Response, error) | ||
| } | ||
|
|
||
| // output emits some formatted bytes to this sink. | ||
| // the sink is invited to perform an extra flush if indicated | ||
| // by the argument. This is set to true for e.g. Fatal | ||
| // entries. | ||
| // | ||
| // The parent logger's outputMu is held during this operation: log | ||
| // sinks must not recursively call into logging when implementing | ||
| // this method. | ||
| func (hs *httpSink) output(extraSync bool, b []byte) (err error) { | ||
| resp, err := hs.doRequest(hs, b) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| if resp.StatusCode >= 400 { | ||
| return HTTPLogError{ | ||
| StatusCode: resp.StatusCode, | ||
| Address: hs.address} | ||
| } | ||
| return nil | ||
| } | ||
|
|
||
| // emergencyOutput attempts to emit some formatted bytes, and | ||
| // ignores any errors. | ||
| // | ||
| // The parent logger's outputMu is held during this operation: log | ||
| // sinks must not recursively call into logging when implementing | ||
| // this method. | ||
| func (hs *httpSink) emergencyOutput(b []byte) { | ||
| hs.doRequest(hs, b) | ||
| } | ||
|
|
||
| func doPost(hs *httpSink, b []byte) (*http.Response, error) { | ||
| resp, err := hs.client.Post(hs.address, "text/plain", bytes.NewReader(b)) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
| resp.Body.Close() // don't care about content | ||
| return resp, nil | ||
| } | ||
|
|
||
| func doGet(hs *httpSink, b []byte) (*http.Response, error) { | ||
| resp, err := hs.client.Get(hs.address + "?" + url.QueryEscape(string(b))) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
| resp.Body.Close() // don't care about content | ||
| return resp, nil | ||
| } | ||
|
|
||
| // active returns true if this sink is currently active. | ||
| func (*httpSink) active() bool { | ||
| return true | ||
| } | ||
|
|
||
| // attachHints attaches some hints about the location of the message | ||
| // to the stack message. | ||
| func (*httpSink) attachHints(stacks []byte) []byte { | ||
| return stacks | ||
| } | ||
|
|
||
| // exitCode returns the exit code to use if the logger decides | ||
| // to terminate because of an error in output(). | ||
| func (*httpSink) exitCode() exit.Code { | ||
| return exit.LoggingNetCollectorUnavailable() | ||
| } | ||
|
|
||
| // HTTPLogError represents an HTTP error status code from a logging request. | ||
| type HTTPLogError struct { | ||
| StatusCode int | ||
| Address string | ||
| } | ||
|
|
||
| func (e HTTPLogError) Error() string { | ||
| return fmt.Sprintf( | ||
| "received %v response attempting to log to [%v]", | ||
| e.StatusCode, e.Address) | ||
| } |
Oops, something went wrong.