mirror of
https://github.com/tailscale/tailscale.git
synced 2025-08-11 13:18:53 +00:00
log/logheap: change to POST to a URL instead of logging
It's too big to log.
This commit is contained in:
@@ -7,9 +7,9 @@ package logheap
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/json"
|
||||
"io"
|
||||
"os"
|
||||
"context"
|
||||
"log"
|
||||
"net/http"
|
||||
"runtime"
|
||||
"runtime/pprof"
|
||||
"time"
|
||||
@@ -17,29 +17,26 @@ import (
|
||||
|
||||
// LogHeap writes a JSON logtail record with the base64 heap pprof to
|
||||
// os.Stderr.
|
||||
func LogHeap() {
|
||||
logHeap(os.Stderr)
|
||||
}
|
||||
|
||||
type logTail struct {
|
||||
ClientTime string `json:"client_time"`
|
||||
}
|
||||
|
||||
type pprofRec struct {
|
||||
Heap []byte `json:"heap,omitempty"`
|
||||
}
|
||||
|
||||
type logLine struct {
|
||||
LogTail logTail `json:"logtail"`
|
||||
Pprof pprofRec `json:"pprof"`
|
||||
}
|
||||
|
||||
func logHeap(w io.Writer) error {
|
||||
func LogHeap(postURL string) {
|
||||
if postURL == "" {
|
||||
return
|
||||
}
|
||||
runtime.GC()
|
||||
buf := new(bytes.Buffer)
|
||||
pprof.WriteHeapProfile(buf)
|
||||
return json.NewEncoder(w).Encode(logLine{
|
||||
LogTail: logTail{ClientTime: time.Now().Format(time.RFC3339Nano)},
|
||||
Pprof: pprofRec{Heap: buf.Bytes()},
|
||||
})
|
||||
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
||||
defer cancel()
|
||||
req, err := http.NewRequestWithContext(ctx, "POST", postURL, buf)
|
||||
if err != nil {
|
||||
log.Printf("LogHeap: %v", err)
|
||||
return
|
||||
}
|
||||
res, err := http.DefaultClient.Do(req)
|
||||
if err != nil {
|
||||
log.Printf("LogHeap: %v", err)
|
||||
return
|
||||
}
|
||||
defer res.Body.Close()
|
||||
return
|
||||
}
|
||||
|
@@ -1,40 +0,0 @@
|
||||
// Copyright (c) 2020 Tailscale Inc & AUTHORS All rights reserved.
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package logheap
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"compress/gzip"
|
||||
"encoding/json"
|
||||
"io/ioutil"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestLogHeap(t *testing.T) {
|
||||
var buf bytes.Buffer
|
||||
if err := logHeap(&buf); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
t.Logf("Got line: %s", buf.Bytes())
|
||||
|
||||
var ll logLine
|
||||
if err := json.Unmarshal(buf.Bytes(), &ll); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
zr, err := gzip.NewReader(bytes.NewReader(ll.Pprof.Heap))
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
rawProto, err := ioutil.ReadAll(zr)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
// Just sanity check it. Too lazy to properly decode the protobuf. But see that
|
||||
// it contains an expected sample name.
|
||||
if !bytes.Contains(rawProto, []byte("alloc_objects")) {
|
||||
t.Errorf("raw proto didn't contain `alloc_objects`: %q", rawProto)
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user