mirror of
https://github.com/tailscale/tailscale.git
synced 2024-12-05 16:05:36 +00:00
688f923db1
LogHeap no longer logs to os.Stderr and instead uploads the heap profile by means of an HTTP POST request to the target URL endpoint. While here, also ensured that an error from pprof.WriteHeapProfile isn't ignored and will prevent the HTTP request from being made if non-nil. Signed-off-by: Emmanuel T Odeke <emmanuel@orijtech.com>
45 lines
1015 B
Go
45 lines
1015 B
Go
// 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 logs a heap pprof profile.
|
|
package logheap
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"log"
|
|
"net/http"
|
|
"runtime"
|
|
"runtime/pprof"
|
|
"time"
|
|
)
|
|
|
|
// LogHeap uploads a JSON logtail record with the base64 heap pprof by means
|
|
// of an HTTP POST request to the endpoint referred to in postURL.
|
|
func LogHeap(postURL string) {
|
|
if postURL == "" {
|
|
return
|
|
}
|
|
runtime.GC()
|
|
buf := new(bytes.Buffer)
|
|
if err := pprof.WriteHeapProfile(buf); err != nil {
|
|
log.Printf("LogHeap: %v", err)
|
|
return
|
|
}
|
|
|
|
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()
|
|
}
|