Initial QSfera import

Server/vendor/go-micro.dev/v4/debug/profile/http/http.go

@@ -0,0 +1,78 @@
+// Package http enables the http profiler
+package http
+
+import (
+	"context"
+	"net/http"
+	"net/http/pprof"
+	"sync"
+
+	"go-micro.dev/v4/debug/profile"
+)
+
+type httpProfile struct {
+	server *http.Server
+	sync.Mutex
+	running bool
+}
+
+var (
+	DefaultAddress = ":6060"
+)
+
+// Start the profiler.
+func (h *httpProfile) Start() error {
+	h.Lock()
+	defer h.Unlock()
+
+	if h.running {
+		return nil
+	}
+
+	go func() {
+		if err := h.server.ListenAndServe(); err != nil {
+			h.Lock()
+			h.running = false
+			h.Unlock()
+		}
+	}()
+
+	h.running = true
+
+	return nil
+}
+
+// Stop the profiler.
+func (h *httpProfile) Stop() error {
+	h.Lock()
+	defer h.Unlock()
+
+	if !h.running {
+		return nil
+	}
+
+	h.running = false
+
+	return h.server.Shutdown(context.TODO())
+}
+
+func (h *httpProfile) String() string {
+	return "http"
+}
+
+func NewProfile(opts ...profile.Option) profile.Profile {
+	mux := http.NewServeMux()
+
+	mux.HandleFunc("/debug/pprof/", pprof.Index)
+	mux.HandleFunc("/debug/pprof/cmdline", pprof.Cmdline)
+	mux.HandleFunc("/debug/pprof/profile", pprof.Profile)
+	mux.HandleFunc("/debug/pprof/symbol", pprof.Symbol)
+	mux.HandleFunc("/debug/pprof/trace", pprof.Trace)
+
+	return &httpProfile{
+		server: &http.Server{
+			Addr:    DefaultAddress,
+			Handler: mux,
+		},
+	}
+}

Server/vendor/go-micro.dev/v4/debug/profile/pprof/pprof.go

@@ -0,0 +1,98 @@
+// Package pprof provides a pprof profiler
+package pprof
+
+import (
+	"os"
+	"path/filepath"
+	"runtime"
+	"runtime/pprof"
+	"sync"
+
+	"go-micro.dev/v4/debug/profile"
+)
+
+type profiler struct {
+
+	// where the cpu profile is written
+	cpuFile *os.File
+	// where the mem profile is written
+	memFile *os.File
+	opts    profile.Options
+
+	sync.Mutex
+	running bool
+}
+
+func (p *profiler) Start() error {
+	p.Lock()
+	defer p.Unlock()
+
+	if p.running {
+		return nil
+	}
+
+	cpuFile := filepath.Join(os.TempDir(), "cpu.pprof")
+	memFile := filepath.Join(os.TempDir(), "mem.pprof")
+
+	if len(p.opts.Name) > 0 {
+		cpuFile = filepath.Join(os.TempDir(), p.opts.Name+".cpu.pprof")
+		memFile = filepath.Join(os.TempDir(), p.opts.Name+".mem.pprof")
+	}
+
+	f1, err := os.Create(cpuFile)
+	if err != nil {
+		return err
+	}
+
+	f2, err := os.Create(memFile)
+	if err != nil {
+		return err
+	}
+
+	// start cpu profiling
+	if err := pprof.StartCPUProfile(f1); err != nil {
+		return err
+	}
+
+	// set cpu file
+	p.cpuFile = f1
+	// set mem file
+	p.memFile = f2
+
+	p.running = true
+
+	return nil
+}
+
+func (p *profiler) Stop() error {
+	p.Lock()
+	defer p.Unlock()
+
+	if !p.running {
+		return nil
+	}
+
+	pprof.StopCPUProfile()
+	p.cpuFile.Close()
+	runtime.GC()
+	pprof.WriteHeapProfile(p.memFile)
+	p.memFile.Close()
+	p.running = false
+	p.cpuFile = nil
+	p.memFile = nil
+	return nil
+}
+
+func (p *profiler) String() string {
+	return "pprof"
+}
+
+func NewProfile(opts ...profile.Option) profile.Profile {
+	var options profile.Options
+	for _, o := range opts {
+		o(&options)
+	}
+	p := new(profiler)
+	p.opts = options
+	return p
+}

Server/vendor/go-micro.dev/v4/debug/profile/profile.go

@@ -0,0 +1,43 @@
+// Package profile is for profilers
+package profile
+
+type Profile interface {
+	// Start the profiler
+	Start() error
+	// Stop the profiler
+	Stop() error
+	// Name of the profiler
+	String() string
+}
+
+var (
+	DefaultProfile Profile = new(noop)
+)
+
+type noop struct{}
+
+func (p *noop) Start() error {
+	return nil
+}
+
+func (p *noop) Stop() error {
+	return nil
+}
+
+func (p *noop) String() string {
+	return "noop"
+}
+
+type Options struct {
+	// Name to use for the profile
+	Name string
+}
+
+type Option func(o *Options)
+
+// Name of the profile.
+func Name(n string) Option {
+	return func(o *Options) {
+		o.Name = n
+	}
+}