119 lines
3.1 KiB
Go
119 lines
3.1 KiB
Go
package metrics
|
|
|
|
import (
|
|
"context"
|
|
"expvar"
|
|
"net/http"
|
|
_ "net/http/pprof"
|
|
"strings"
|
|
|
|
"github.com/xtls/xray-core/app/observatory"
|
|
"github.com/xtls/xray-core/app/stats"
|
|
"github.com/xtls/xray-core/common"
|
|
"github.com/xtls/xray-core/common/net"
|
|
"github.com/xtls/xray-core/common/signal/done"
|
|
"github.com/xtls/xray-core/core"
|
|
"github.com/xtls/xray-core/features/extension"
|
|
"github.com/xtls/xray-core/features/outbound"
|
|
feature_stats "github.com/xtls/xray-core/features/stats"
|
|
)
|
|
|
|
type MetricsHandler struct {
|
|
ohm outbound.Manager
|
|
statsManager feature_stats.Manager
|
|
observatory extension.Observatory
|
|
tag string
|
|
}
|
|
|
|
// NewMetricsHandler creates a new MetricsHandler based on the given config.
|
|
func NewMetricsHandler(ctx context.Context, config *Config) (*MetricsHandler, error) {
|
|
c := &MetricsHandler{
|
|
tag: config.Tag,
|
|
}
|
|
common.Must(core.RequireFeatures(ctx, func(om outbound.Manager, sm feature_stats.Manager) {
|
|
c.statsManager = sm
|
|
c.ohm = om
|
|
}))
|
|
expvar.Publish("stats", expvar.Func(func() interface{} {
|
|
manager, ok := c.statsManager.(*stats.Manager)
|
|
if !ok {
|
|
return nil
|
|
}
|
|
var resp = map[string]map[string]map[string]int64{
|
|
"inbound": {},
|
|
"outbound": {},
|
|
"user": {},
|
|
}
|
|
manager.VisitCounters(func(name string, counter feature_stats.Counter) bool {
|
|
nameSplit := strings.Split(name, ">>>")
|
|
typeName, tagOrUser, direction := nameSplit[0], nameSplit[1], nameSplit[3]
|
|
if item, found := resp[typeName][tagOrUser]; found {
|
|
item[direction] = counter.Value()
|
|
} else {
|
|
resp[typeName][tagOrUser] = map[string]int64{
|
|
direction: counter.Value(),
|
|
}
|
|
}
|
|
return true
|
|
})
|
|
return resp
|
|
}))
|
|
expvar.Publish("observatory", expvar.Func(func() interface{} {
|
|
if c.observatory == nil {
|
|
common.Must(core.RequireFeatures(ctx, func(observatory extension.Observatory) error {
|
|
c.observatory = observatory
|
|
return nil
|
|
}))
|
|
if c.observatory == nil {
|
|
return nil
|
|
}
|
|
}
|
|
var resp = map[string]*observatory.OutboundStatus{}
|
|
if o, err := c.observatory.GetObservation(context.Background()); err != nil {
|
|
return err
|
|
} else {
|
|
for _, x := range o.(*observatory.ObservationResult).GetStatus() {
|
|
resp[x.OutboundTag] = x
|
|
}
|
|
}
|
|
return resp
|
|
}))
|
|
return c, nil
|
|
}
|
|
|
|
func (p *MetricsHandler) Type() interface{} {
|
|
return (*MetricsHandler)(nil)
|
|
}
|
|
|
|
func (p *MetricsHandler) Start() error {
|
|
listener := &OutboundListener{
|
|
buffer: make(chan net.Conn, 4),
|
|
done: done.New(),
|
|
}
|
|
|
|
go func() {
|
|
if err := http.Serve(listener, http.DefaultServeMux); err != nil {
|
|
newError("failed to start metrics server").Base(err).AtError().WriteToLog()
|
|
}
|
|
}()
|
|
|
|
if err := p.ohm.RemoveHandler(context.Background(), p.tag); err != nil {
|
|
newError("failed to remove existing handler").WriteToLog()
|
|
}
|
|
|
|
return p.ohm.AddHandler(context.Background(), &Outbound{
|
|
tag: p.tag,
|
|
listener: listener,
|
|
})
|
|
}
|
|
|
|
func (p *MetricsHandler) Close() error {
|
|
return nil
|
|
}
|
|
|
|
func init() {
|
|
common.Must(common.RegisterConfig((*Config)(nil), func(ctx context.Context, cfg interface{}) (interface{}, error) {
|
|
return NewMetricsHandler(ctx, cfg.(*Config))
|
|
}))
|
|
}
|