mirror of https://github.com/fatedier/frp.git
api: add server web api for statistics
fatedier
8 years ago
parent
9e683fe446
commit
a4fece3f51
@ -1,153 +0,0 @@
|
|||||||
// Copyright 2016 fatedier, fatedier@gmail.com
|
|
||||||
//
|
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
||||||
// you may not use this file except in compliance with the License.
|
|
||||||
// You may obtain a copy of the License at
|
|
||||||
//
|
|
||||||
// http://www.apache.org/licenses/LICENSE-2.0
|
|
||||||
//
|
|
||||||
// Unless required by applicable law or agreed to in writing, software
|
|
||||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
||||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
||||||
// See the License for the specific language governing permissions and
|
|
||||||
// limitations under the License.
|
|
||||||
|
|
||||||
package client
|
|
||||||
|
|
||||||
import (
|
|
||||||
"fmt"
|
|
||||||
"io"
|
|
||||||
"net"
|
|
||||||
"sync"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/fatedier/frp/src/models/msg"
|
|
||||||
"github.com/fatedier/frp/src/utils/conn"
|
|
||||||
"github.com/fatedier/frp/src/utils/pool"
|
|
||||||
)
|
|
||||||
|
|
||||||
type UdpProcesser struct {
|
|
||||||
tcpConn *conn.Conn
|
|
||||||
closeCh chan struct{}
|
|
||||||
|
|
||||||
localAddr string
|
|
||||||
|
|
||||||
// cache local udp connections
|
|
||||||
// key is remoteAddr
|
|
||||||
localUdpConns map[string]*net.UDPConn
|
|
||||||
mutex sync.RWMutex
|
|
||||||
tcpConnMutex sync.RWMutex
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewUdpProcesser(c *conn.Conn, localIp string, localPort int64) *UdpProcesser {
|
|
||||||
return &UdpProcesser{
|
|
||||||
tcpConn: c,
|
|
||||||
closeCh: make(chan struct{}),
|
|
||||||
localAddr: fmt.Sprintf("%s:%d", localIp, localPort),
|
|
||||||
localUdpConns: make(map[string]*net.UDPConn),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (up *UdpProcesser) UpdateTcpConn(c *conn.Conn) {
|
|
||||||
up.tcpConnMutex.Lock()
|
|
||||||
defer up.tcpConnMutex.Unlock()
|
|
||||||
up.tcpConn = c
|
|
||||||
}
|
|
||||||
|
|
||||||
func (up *UdpProcesser) Run() {
|
|
||||||
go up.ReadLoop()
|
|
||||||
}
|
|
||||||
|
|
||||||
func (up *UdpProcesser) ReadLoop() {
|
|
||||||
var (
|
|
||||||
buf string
|
|
||||||
err error
|
|
||||||
)
|
|
||||||
for {
|
|
||||||
udpPacket := &msg.UdpPacket{}
|
|
||||||
|
|
||||||
// read udp package from frps
|
|
||||||
buf, err = up.tcpConn.ReadLine()
|
|
||||||
if err != nil {
|
|
||||||
if err == io.EOF {
|
|
||||||
return
|
|
||||||
} else {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
}
|
|
||||||
err = udpPacket.UnPack([]byte(buf))
|
|
||||||
if err != nil {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
// write to local udp port
|
|
||||||
sendConn, ok := up.GetUdpConn(udpPacket.SrcStr)
|
|
||||||
if !ok {
|
|
||||||
dstAddr, err := net.ResolveUDPAddr("udp", up.localAddr)
|
|
||||||
if err != nil {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
sendConn, err = net.DialUDP("udp", nil, dstAddr)
|
|
||||||
if err != nil {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
up.SetUdpConn(udpPacket.SrcStr, sendConn)
|
|
||||||
}
|
|
||||||
|
|
||||||
_, err = sendConn.Write(udpPacket.Content)
|
|
||||||
if err != nil {
|
|
||||||
sendConn.Close()
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
if !ok {
|
|
||||||
go up.Forward(udpPacket, sendConn)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (up *UdpProcesser) Forward(udpPacket *msg.UdpPacket, singleConn *net.UDPConn) {
|
|
||||||
addr := udpPacket.SrcStr
|
|
||||||
defer up.RemoveUdpConn(addr)
|
|
||||||
|
|
||||||
buf := pool.GetBuf(2048)
|
|
||||||
for {
|
|
||||||
singleConn.SetReadDeadline(time.Now().Add(120 * time.Second))
|
|
||||||
n, remoteAddr, err := singleConn.ReadFromUDP(buf)
|
|
||||||
if err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
// forward to frps
|
|
||||||
forwardPacket := msg.NewUdpPacket(buf[0:n], remoteAddr, udpPacket.Src)
|
|
||||||
up.tcpConnMutex.RLock()
|
|
||||||
err = up.tcpConn.WriteString(string(forwardPacket.Pack()) + "\n")
|
|
||||||
up.tcpConnMutex.RUnlock()
|
|
||||||
if err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (up *UdpProcesser) GetUdpConn(addr string) (singleConn *net.UDPConn, ok bool) {
|
|
||||||
up.mutex.RLock()
|
|
||||||
defer up.mutex.RUnlock()
|
|
||||||
singleConn, ok = up.localUdpConns[addr]
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
func (up *UdpProcesser) SetUdpConn(addr string, conn *net.UDPConn) {
|
|
||||||
up.mutex.Lock()
|
|
||||||
defer up.mutex.Unlock()
|
|
||||||
up.localUdpConns[addr] = conn
|
|
||||||
}
|
|
||||||
|
|
||||||
func (up *UdpProcesser) RemoveUdpConn(addr string) {
|
|
||||||
up.mutex.Lock()
|
|
||||||
defer up.mutex.Unlock()
|
|
||||||
if c, ok := up.localUdpConns[addr]; ok {
|
|
||||||
c.Close()
|
|
||||||
}
|
|
||||||
delete(up.localUdpConns, addr)
|
|
||||||
}
|
|
||||||
@ -1,221 +0,0 @@
|
|||||||
// Copyright 2016 fatedier, fatedier@gmail.com
|
|
||||||
//
|
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
||||||
// you may not use this file except in compliance with the License.
|
|
||||||
// You may obtain a copy of the License at
|
|
||||||
//
|
|
||||||
// http://www.apache.org/licenses/LICENSE-2.0
|
|
||||||
//
|
|
||||||
// Unless required by applicable law or agreed to in writing, software
|
|
||||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
||||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
||||||
// See the License for the specific language governing permissions and
|
|
||||||
// limitations under the License.
|
|
||||||
|
|
||||||
package metric
|
|
||||||
|
|
||||||
/*
|
|
||||||
var (
|
|
||||||
DailyDataKeepDays int = 7
|
|
||||||
ServerMetricInfoMap map[string]*ServerMetric
|
|
||||||
smMutex sync.RWMutex
|
|
||||||
)
|
|
||||||
|
|
||||||
type ServerMetric struct {
|
|
||||||
Name string `json:"name"`
|
|
||||||
Type string `json:"type"`
|
|
||||||
BindAddr string `json:"bind_addr"`
|
|
||||||
ListenPort int64 `json:"listen_port"`
|
|
||||||
CustomDomains []string `json:"custom_domains"`
|
|
||||||
Locations []string `json:"locations"`
|
|
||||||
Status string `json:"status"`
|
|
||||||
UseEncryption bool `json:"use_encryption"`
|
|
||||||
UseGzip bool `json:"use_gzip"`
|
|
||||||
PrivilegeMode bool `json:"privilege_mode"`
|
|
||||||
|
|
||||||
// statistics
|
|
||||||
CurrentConns int64 `json:"current_conns"`
|
|
||||||
Daily []*DailyServerStats `json:"daily"`
|
|
||||||
mutex sync.RWMutex
|
|
||||||
}
|
|
||||||
|
|
||||||
type DailyServerStats struct {
|
|
||||||
Time string `json:"time"`
|
|
||||||
FlowIn int64 `json:"flow_in"`
|
|
||||||
FlowOut int64 `json:"flow_out"`
|
|
||||||
TotalAcceptConns int64 `json:"total_accept_conns"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// for sort
|
|
||||||
type ServerMetricList []*ServerMetric
|
|
||||||
|
|
||||||
func (l ServerMetricList) Len() int { return len(l) }
|
|
||||||
func (l ServerMetricList) Less(i, j int) bool { return l[i].Name < l[j].Name }
|
|
||||||
func (l ServerMetricList) Swap(i, j int) { l[i], l[j] = l[j], l[i] }
|
|
||||||
|
|
||||||
func init() {
|
|
||||||
ServerMetricInfoMap = make(map[string]*ServerMetric)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *ServerMetric) clone() *ServerMetric {
|
|
||||||
copy := *s
|
|
||||||
copy.CustomDomains = make([]string, len(s.CustomDomains))
|
|
||||||
var i int
|
|
||||||
for i = range copy.CustomDomains {
|
|
||||||
copy.CustomDomains[i] = s.CustomDomains[i]
|
|
||||||
}
|
|
||||||
|
|
||||||
copy.Daily = make([]*DailyServerStats, len(s.Daily))
|
|
||||||
for i = range copy.Daily {
|
|
||||||
tmpDaily := *s.Daily[i]
|
|
||||||
copy.Daily[i] = &tmpDaily
|
|
||||||
}
|
|
||||||
return ©
|
|
||||||
}
|
|
||||||
|
|
||||||
func GetAllProxyMetrics() []*ServerMetric {
|
|
||||||
result := make(ServerMetricList, 0)
|
|
||||||
smMutex.RLock()
|
|
||||||
for _, metric := range ServerMetricInfoMap {
|
|
||||||
metric.mutex.RLock()
|
|
||||||
tmpMetric := metric.clone()
|
|
||||||
metric.mutex.RUnlock()
|
|
||||||
result = append(result, tmpMetric)
|
|
||||||
}
|
|
||||||
smMutex.RUnlock()
|
|
||||||
|
|
||||||
// sort for result by proxy name
|
|
||||||
sort.Sort(result)
|
|
||||||
return result
|
|
||||||
}
|
|
||||||
|
|
||||||
// if proxyName isn't exist, return nil
|
|
||||||
func GetProxyMetrics(proxyName string) *ServerMetric {
|
|
||||||
smMutex.RLock()
|
|
||||||
defer smMutex.RUnlock()
|
|
||||||
metric, ok := ServerMetricInfoMap[proxyName]
|
|
||||||
if ok {
|
|
||||||
metric.mutex.RLock()
|
|
||||||
tmpMetric := metric.clone()
|
|
||||||
metric.mutex.RUnlock()
|
|
||||||
return tmpMetric
|
|
||||||
} else {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func SetProxyInfo(proxyName string, proxyType, bindAddr string,
|
|
||||||
useEncryption, useGzip, privilegeMode bool, customDomains []string,
|
|
||||||
locations []string, listenPort int64) {
|
|
||||||
smMutex.Lock()
|
|
||||||
info, ok := ServerMetricInfoMap[proxyName]
|
|
||||||
if !ok {
|
|
||||||
info = &ServerMetric{}
|
|
||||||
info.Daily = make([]*DailyServerStats, 0)
|
|
||||||
}
|
|
||||||
info.Name = proxyName
|
|
||||||
info.Type = proxyType
|
|
||||||
info.UseEncryption = useEncryption
|
|
||||||
info.UseGzip = useGzip
|
|
||||||
info.PrivilegeMode = privilegeMode
|
|
||||||
info.BindAddr = bindAddr
|
|
||||||
info.ListenPort = listenPort
|
|
||||||
info.CustomDomains = customDomains
|
|
||||||
info.Locations = locations
|
|
||||||
ServerMetricInfoMap[proxyName] = info
|
|
||||||
smMutex.Unlock()
|
|
||||||
}
|
|
||||||
|
|
||||||
func SetStatus(proxyName string, status int64) {
|
|
||||||
smMutex.RLock()
|
|
||||||
metric, ok := ServerMetricInfoMap[proxyName]
|
|
||||||
smMutex.RUnlock()
|
|
||||||
if ok {
|
|
||||||
metric.mutex.Lock()
|
|
||||||
metric.Status = consts.StatusStr[status]
|
|
||||||
metric.mutex.Unlock()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
type DealFuncType func(*DailyServerStats)
|
|
||||||
|
|
||||||
func DealDailyData(dailyData []*DailyServerStats, fn DealFuncType) (newDailyData []*DailyServerStats) {
|
|
||||||
now := time.Now().Format("20060102")
|
|
||||||
dailyLen := len(dailyData)
|
|
||||||
if dailyLen == 0 {
|
|
||||||
daily := &DailyServerStats{}
|
|
||||||
daily.Time = now
|
|
||||||
fn(daily)
|
|
||||||
dailyData = append(dailyData, daily)
|
|
||||||
} else {
|
|
||||||
daily := dailyData[dailyLen-1]
|
|
||||||
if daily.Time == now {
|
|
||||||
fn(daily)
|
|
||||||
} else {
|
|
||||||
newDaily := &DailyServerStats{}
|
|
||||||
newDaily.Time = now
|
|
||||||
fn(newDaily)
|
|
||||||
if dailyLen == DailyDataKeepDays {
|
|
||||||
for i := 0; i < dailyLen-1; i++ {
|
|
||||||
dailyData[i] = dailyData[i+1]
|
|
||||||
}
|
|
||||||
dailyData[dailyLen-1] = newDaily
|
|
||||||
} else {
|
|
||||||
dailyData = append(dailyData, newDaily)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return dailyData
|
|
||||||
}
|
|
||||||
|
|
||||||
func OpenConnection(proxyName string) {
|
|
||||||
smMutex.RLock()
|
|
||||||
metric, ok := ServerMetricInfoMap[proxyName]
|
|
||||||
smMutex.RUnlock()
|
|
||||||
if ok {
|
|
||||||
metric.mutex.Lock()
|
|
||||||
metric.CurrentConns++
|
|
||||||
metric.Daily = DealDailyData(metric.Daily, func(stats *DailyServerStats) {
|
|
||||||
stats.TotalAcceptConns++
|
|
||||||
})
|
|
||||||
metric.mutex.Unlock()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func CloseConnection(proxyName string) {
|
|
||||||
smMutex.RLock()
|
|
||||||
metric, ok := ServerMetricInfoMap[proxyName]
|
|
||||||
smMutex.RUnlock()
|
|
||||||
if ok {
|
|
||||||
metric.mutex.Lock()
|
|
||||||
metric.CurrentConns--
|
|
||||||
metric.mutex.Unlock()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func AddFlowIn(proxyName string, value int64) {
|
|
||||||
smMutex.RLock()
|
|
||||||
metric, ok := ServerMetricInfoMap[proxyName]
|
|
||||||
smMutex.RUnlock()
|
|
||||||
if ok {
|
|
||||||
metric.mutex.Lock()
|
|
||||||
metric.Daily = DealDailyData(metric.Daily, func(stats *DailyServerStats) {
|
|
||||||
stats.FlowIn += value
|
|
||||||
})
|
|
||||||
metric.mutex.Unlock()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func AddFlowOut(proxyName string, value int64) {
|
|
||||||
smMutex.RLock()
|
|
||||||
metric, ok := ServerMetricInfoMap[proxyName]
|
|
||||||
smMutex.RUnlock()
|
|
||||||
if ok {
|
|
||||||
metric.mutex.Lock()
|
|
||||||
metric.Daily = DealDailyData(metric.Daily, func(stats *DailyServerStats) {
|
|
||||||
stats.FlowOut += value
|
|
||||||
})
|
|
||||||
metric.mutex.Unlock()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
*/
|
|
||||||
@ -0,0 +1,241 @@
|
|||||||
|
// Copyright 2017 fatedier, fatedier@gmail.com
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
// you may not use this file except in compliance with the License.
|
||||||
|
// You may obtain a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
// See the License for the specific language governing permissions and
|
||||||
|
// limitations under the License.
|
||||||
|
|
||||||
|
package server
|
||||||
|
|
||||||
|
import (
|
||||||
|
"sync"
|
||||||
|
|
||||||
|
"github.com/fatedier/frp/models/config"
|
||||||
|
"github.com/fatedier/frp/utils/metric"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
ReserveDays = 7
|
||||||
|
)
|
||||||
|
|
||||||
|
var globalStats *ServerStatistics
|
||||||
|
|
||||||
|
type ServerStatistics struct {
|
||||||
|
TotalFlowIn metric.DateCounter
|
||||||
|
TotalFlowOut metric.DateCounter
|
||||||
|
CurConns metric.Counter
|
||||||
|
|
||||||
|
ClientCounts metric.Counter
|
||||||
|
ProxyTypeCounts map[string]metric.Counter
|
||||||
|
|
||||||
|
ProxyStatistics map[string]*ProxyStatistics
|
||||||
|
|
||||||
|
mu sync.Mutex
|
||||||
|
}
|
||||||
|
|
||||||
|
type ProxyStatistics struct {
|
||||||
|
ProxyType string
|
||||||
|
FlowIn metric.DateCounter
|
||||||
|
FlowOut metric.DateCounter
|
||||||
|
CurConns metric.Counter
|
||||||
|
}
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
globalStats = &ServerStatistics{
|
||||||
|
TotalFlowIn: metric.NewDateCounter(ReserveDays),
|
||||||
|
TotalFlowOut: metric.NewDateCounter(ReserveDays),
|
||||||
|
CurConns: metric.NewCounter(),
|
||||||
|
|
||||||
|
ClientCounts: metric.NewCounter(),
|
||||||
|
ProxyTypeCounts: make(map[string]metric.Counter),
|
||||||
|
|
||||||
|
ProxyStatistics: make(map[string]*ProxyStatistics),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func StatsNewClient() {
|
||||||
|
if config.ServerCommonCfg.DashboardPort != 0 {
|
||||||
|
globalStats.ClientCounts.Inc(1)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func StatsCloseClient() {
|
||||||
|
if config.ServerCommonCfg.DashboardPort != 0 {
|
||||||
|
globalStats.ClientCounts.Dec(1)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func StatsNewProxy(name string, proxyType string) {
|
||||||
|
if config.ServerCommonCfg.DashboardPort != 0 {
|
||||||
|
globalStats.mu.Lock()
|
||||||
|
defer globalStats.mu.Unlock()
|
||||||
|
counter, ok := globalStats.ProxyTypeCounts[proxyType]
|
||||||
|
if !ok {
|
||||||
|
counter = metric.NewCounter()
|
||||||
|
}
|
||||||
|
counter.Inc(1)
|
||||||
|
globalStats.ProxyTypeCounts[proxyType] = counter
|
||||||
|
|
||||||
|
proxyStats, ok := globalStats.ProxyStatistics[name]
|
||||||
|
if !ok {
|
||||||
|
proxyStats = &ProxyStatistics{
|
||||||
|
ProxyType: proxyType,
|
||||||
|
CurConns: metric.NewCounter(),
|
||||||
|
FlowIn: metric.NewDateCounter(ReserveDays),
|
||||||
|
FlowOut: metric.NewDateCounter(ReserveDays),
|
||||||
|
}
|
||||||
|
globalStats.ProxyStatistics[name] = proxyStats
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func StatsCloseProxy(proxyType string) {
|
||||||
|
if config.ServerCommonCfg.DashboardPort != 0 {
|
||||||
|
globalStats.mu.Lock()
|
||||||
|
defer globalStats.mu.Unlock()
|
||||||
|
if counter, ok := globalStats.ProxyTypeCounts[proxyType]; ok {
|
||||||
|
counter.Dec(1)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func StatsOpenConnection(name string) {
|
||||||
|
if config.ServerCommonCfg.DashboardPort != 0 {
|
||||||
|
globalStats.CurConns.Inc(1)
|
||||||
|
|
||||||
|
globalStats.mu.Lock()
|
||||||
|
defer globalStats.mu.Unlock()
|
||||||
|
proxyStats, ok := globalStats.ProxyStatistics[name]
|
||||||
|
if ok {
|
||||||
|
proxyStats.CurConns.Inc(1)
|
||||||
|
globalStats.ProxyStatistics[name] = proxyStats
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func StatsCloseConnection(name string) {
|
||||||
|
if config.ServerCommonCfg.DashboardPort != 0 {
|
||||||
|
globalStats.CurConns.Dec(1)
|
||||||
|
|
||||||
|
globalStats.mu.Lock()
|
||||||
|
defer globalStats.mu.Unlock()
|
||||||
|
proxyStats, ok := globalStats.ProxyStatistics[name]
|
||||||
|
if ok {
|
||||||
|
proxyStats.CurConns.Dec(1)
|
||||||
|
globalStats.ProxyStatistics[name] = proxyStats
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func StatsAddFlowIn(name string, flowIn int64) {
|
||||||
|
if config.ServerCommonCfg.DashboardPort != 0 {
|
||||||
|
globalStats.TotalFlowIn.Inc(flowIn)
|
||||||
|
|
||||||
|
globalStats.mu.Lock()
|
||||||
|
defer globalStats.mu.Unlock()
|
||||||
|
|
||||||
|
proxyStats, ok := globalStats.ProxyStatistics[name]
|
||||||
|
if ok {
|
||||||
|
proxyStats.FlowIn.Inc(flowIn)
|
||||||
|
globalStats.ProxyStatistics[name] = proxyStats
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func StatsAddFlowOut(name string, flowOut int64) {
|
||||||
|
if config.ServerCommonCfg.DashboardPort != 0 {
|
||||||
|
globalStats.TotalFlowOut.Inc(flowOut)
|
||||||
|
|
||||||
|
globalStats.mu.Lock()
|
||||||
|
defer globalStats.mu.Unlock()
|
||||||
|
|
||||||
|
proxyStats, ok := globalStats.ProxyStatistics[name]
|
||||||
|
if ok {
|
||||||
|
proxyStats.FlowOut.Inc(flowOut)
|
||||||
|
globalStats.ProxyStatistics[name] = proxyStats
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Functions for getting server stats.
|
||||||
|
type ServerStats struct {
|
||||||
|
TotalFlowIn int64
|
||||||
|
TotalFlowOut int64
|
||||||
|
CurConns int64
|
||||||
|
ClientCounts int64
|
||||||
|
ProxyTypeCounts map[string]int64
|
||||||
|
}
|
||||||
|
|
||||||
|
func StatsGetServer() *ServerStats {
|
||||||
|
globalStats.mu.Lock()
|
||||||
|
defer globalStats.mu.Unlock()
|
||||||
|
s := &ServerStats{
|
||||||
|
TotalFlowIn: globalStats.TotalFlowIn.TodayCount(),
|
||||||
|
TotalFlowOut: globalStats.TotalFlowOut.TodayCount(),
|
||||||
|
CurConns: globalStats.CurConns.Count(),
|
||||||
|
ClientCounts: globalStats.ClientCounts.Count(),
|
||||||
|
ProxyTypeCounts: make(map[string]int64),
|
||||||
|
}
|
||||||
|
for k, v := range globalStats.ProxyTypeCounts {
|
||||||
|
s.ProxyTypeCounts[k] = v.Count()
|
||||||
|
}
|
||||||
|
return s
|
||||||
|
}
|
||||||
|
|
||||||
|
type ProxyStats struct {
|
||||||
|
Name string
|
||||||
|
Type string
|
||||||
|
TodayFlowIn int64
|
||||||
|
TodayFlowOut int64
|
||||||
|
CurConns int64
|
||||||
|
}
|
||||||
|
|
||||||
|
func StatsGetProxiesByType(proxyType string) []*ProxyStats {
|
||||||
|
res := make([]*ProxyStats, 0)
|
||||||
|
globalStats.mu.Lock()
|
||||||
|
defer globalStats.mu.Unlock()
|
||||||
|
|
||||||
|
for name, proxyStats := range globalStats.ProxyStatistics {
|
||||||
|
if proxyStats.ProxyType != proxyType {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
ps := &ProxyStats{
|
||||||
|
Name: name,
|
||||||
|
Type: proxyStats.ProxyType,
|
||||||
|
TodayFlowIn: proxyStats.FlowIn.TodayCount(),
|
||||||
|
TodayFlowOut: proxyStats.FlowOut.TodayCount(),
|
||||||
|
CurConns: proxyStats.CurConns.Count(),
|
||||||
|
}
|
||||||
|
res = append(res, ps)
|
||||||
|
}
|
||||||
|
return res
|
||||||
|
}
|
||||||
|
|
||||||
|
type ProxyFlowInfo struct {
|
||||||
|
Name string
|
||||||
|
FlowIn []int64
|
||||||
|
FlowOut []int64
|
||||||
|
}
|
||||||
|
|
||||||
|
func StatsGetProxyFlow(name string) (res *ProxyFlowInfo) {
|
||||||
|
globalStats.mu.Lock()
|
||||||
|
defer globalStats.mu.Unlock()
|
||||||
|
|
||||||
|
proxyStats, ok := globalStats.ProxyStatistics[name]
|
||||||
|
if ok {
|
||||||
|
res = &ProxyFlowInfo{
|
||||||
|
Name: name,
|
||||||
|
}
|
||||||
|
res.FlowIn = proxyStats.FlowIn.GetLastDaysCount(ReserveDays)
|
||||||
|
res.FlowOut = proxyStats.FlowOut.GetLastDaysCount(ReserveDays)
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
@ -0,0 +1,60 @@
|
|||||||
|
// Copyright 2017 fatedier, fatedier@gmail.com
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
// you may not use this file except in compliance with the License.
|
||||||
|
// You may obtain a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
// See the License for the specific language governing permissions and
|
||||||
|
// limitations under the License.
|
||||||
|
|
||||||
|
package metric
|
||||||
|
|
||||||
|
import (
|
||||||
|
"sync/atomic"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Counter interface {
|
||||||
|
Count() int64
|
||||||
|
Inc(int64)
|
||||||
|
Dec(int64)
|
||||||
|
Snapshot() Counter
|
||||||
|
Clear()
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewCounter() Counter {
|
||||||
|
return &StandardCounter{
|
||||||
|
count: 0,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
type StandardCounter struct {
|
||||||
|
count int64
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *StandardCounter) Count() int64 {
|
||||||
|
return atomic.LoadInt64(&c.count)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *StandardCounter) Inc(count int64) {
|
||||||
|
atomic.AddInt64(&c.count, count)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *StandardCounter) Dec(count int64) {
|
||||||
|
atomic.AddInt64(&c.count, -count)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *StandardCounter) Snapshot() Counter {
|
||||||
|
tmp := &StandardCounter{
|
||||||
|
count: atomic.LoadInt64(&c.count),
|
||||||
|
}
|
||||||
|
return tmp
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *StandardCounter) Clear() {
|
||||||
|
atomic.StoreInt64(&c.count, 0)
|
||||||
|
}
|
||||||
@ -0,0 +1,157 @@
|
|||||||
|
// Copyright 2017 fatedier, fatedier@gmail.com
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
// you may not use this file except in compliance with the License.
|
||||||
|
// You may obtain a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
// See the License for the specific language governing permissions and
|
||||||
|
// limitations under the License.
|
||||||
|
|
||||||
|
package metric
|
||||||
|
|
||||||
|
import (
|
||||||
|
"sync"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
type DateCounter interface {
|
||||||
|
TodayCount() int64
|
||||||
|
GetLastDaysCount(lastdays int64) []int64
|
||||||
|
Inc(int64)
|
||||||
|
Dec(int64)
|
||||||
|
Snapshot() DateCounter
|
||||||
|
Clear()
|
||||||
|
Close()
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewDateCounter(reserveDays int64) DateCounter {
|
||||||
|
if reserveDays <= 0 {
|
||||||
|
reserveDays = 1
|
||||||
|
}
|
||||||
|
return newStandardDateCounter(reserveDays)
|
||||||
|
}
|
||||||
|
|
||||||
|
type StandardDateCounter struct {
|
||||||
|
reserveDays int64
|
||||||
|
counts []int64
|
||||||
|
|
||||||
|
closeCh chan struct{}
|
||||||
|
closed bool
|
||||||
|
mu sync.Mutex
|
||||||
|
}
|
||||||
|
|
||||||
|
func newStandardDateCounter(reserveDays int64) *StandardDateCounter {
|
||||||
|
s := &StandardDateCounter{
|
||||||
|
reserveDays: reserveDays,
|
||||||
|
counts: make([]int64, reserveDays),
|
||||||
|
closeCh: make(chan struct{}),
|
||||||
|
}
|
||||||
|
s.startRotateWorker()
|
||||||
|
return s
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *StandardDateCounter) TodayCount() int64 {
|
||||||
|
c.mu.Lock()
|
||||||
|
defer c.mu.Unlock()
|
||||||
|
return c.counts[0]
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *StandardDateCounter) GetLastDaysCount(lastdays int64) []int64 {
|
||||||
|
if lastdays > c.reserveDays {
|
||||||
|
lastdays = c.reserveDays
|
||||||
|
}
|
||||||
|
counts := make([]int64, lastdays)
|
||||||
|
|
||||||
|
c.mu.Lock()
|
||||||
|
defer c.mu.Unlock()
|
||||||
|
for i := 0; i < int(lastdays); i++ {
|
||||||
|
counts[i] = c.counts[i]
|
||||||
|
}
|
||||||
|
return counts
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *StandardDateCounter) Inc(count int64) {
|
||||||
|
c.mu.Lock()
|
||||||
|
defer c.mu.Unlock()
|
||||||
|
c.counts[0] += count
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *StandardDateCounter) Dec(count int64) {
|
||||||
|
c.mu.Lock()
|
||||||
|
defer c.mu.Unlock()
|
||||||
|
c.counts[0] -= count
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *StandardDateCounter) Snapshot() DateCounter {
|
||||||
|
c.mu.Lock()
|
||||||
|
defer c.mu.Unlock()
|
||||||
|
tmp := &StandardDateCounter{
|
||||||
|
reserveDays: c.reserveDays,
|
||||||
|
counts: make([]int64, c.reserveDays),
|
||||||
|
}
|
||||||
|
for i := 0; i < int(c.reserveDays); i++ {
|
||||||
|
tmp.counts[i] = c.counts[i]
|
||||||
|
}
|
||||||
|
return tmp
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *StandardDateCounter) Clear() {
|
||||||
|
c.mu.Lock()
|
||||||
|
defer c.mu.Unlock()
|
||||||
|
for i := 0; i < int(c.reserveDays); i++ {
|
||||||
|
c.counts[i] = 0
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *StandardDateCounter) Close() {
|
||||||
|
c.mu.Lock()
|
||||||
|
defer c.mu.Unlock()
|
||||||
|
if !c.closed {
|
||||||
|
close(c.closeCh)
|
||||||
|
c.closed = true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *StandardDateCounter) rotate() {
|
||||||
|
c.mu.Lock()
|
||||||
|
defer c.mu.Unlock()
|
||||||
|
newCounts := make([]int64, c.reserveDays)
|
||||||
|
|
||||||
|
for i := 1; i < int(c.reserveDays-1); i++ {
|
||||||
|
newCounts[i] = c.counts[i+1]
|
||||||
|
}
|
||||||
|
c.counts = newCounts
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *StandardDateCounter) startRotateWorker() {
|
||||||
|
now := time.Now()
|
||||||
|
nextDayTimeStr := now.Add(24 * time.Hour).Format("20060102")
|
||||||
|
nextDay, _ := time.Parse("20060102", nextDayTimeStr)
|
||||||
|
d := nextDay.Sub(now)
|
||||||
|
|
||||||
|
firstTimer := time.NewTimer(d)
|
||||||
|
rotateTicker := time.NewTicker(24 * time.Hour)
|
||||||
|
|
||||||
|
go func() {
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case <-firstTimer.C:
|
||||||
|
firstTimer.Stop()
|
||||||
|
rotateTicker.Stop()
|
||||||
|
rotateTicker = time.NewTicker(24 * time.Hour)
|
||||||
|
c.rotate()
|
||||||
|
case <-rotateTicker.C:
|
||||||
|
c.rotate()
|
||||||
|
case <-c.closeCh:
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
firstTimer.Stop()
|
||||||
|
rotateTicker.Stop()
|
||||||
|
}()
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue