2021-12-31 22:04:08 : stat update

Author: jamiesun

README.md

@@ -1,8 +1,33 @@
-# dnssrc
+# datahub
 
 *datahub* - a CoreDNS Data Manage Plugin
 
+##  相关项目
 
-# Example
+[]
 
+## 配置案例
 
+```
+.:53 {
+    datahub {
+        bootstrap 114.114.114.114:53 # 引导 DNS
+        jwt_secret 9b6de5cc-vcty-4bf1-zpms-0f568ac9da37
+        geoip_path conf/geoip.dat
+        geosite_path conf/geosite.dat
+        geoip_cache cn hk jp google apple
+        geosite_cache cn hk jp private apple
+        geodat_upgrade_url http://teamsacs.mydomain.cn/geodat
+        geodat_upgrade_cron 0 30 0 * * *
+        # datatables conf/datatables.txt
+        keyword_table  cn,google  conf/keywords.txt
+        domain_table  cn,aliyun,ads  conf/domains.txt
+        netlist_table  cn,aliyun,local,office  conf/networks.txt
+        ecs_table  global  conf/ecs_table.txt
+        datapub_listen :9800
+        notify_server  https://teamsacs.appsway.cn
+        reload @every 3s
+    }
+}
+
+```

plugin/datahub/dataapi_stat.go

@@ -16,6 +16,14 @@ func (dh *Datahub) GetMetricSValue(name string) int64 {
 	return dh.metricsStat.GetValue(name)
 }
 
+func (dh *Datahub) GetDomainTop(count int) []stats.Counter {
+	return dh.queryStat.GetTopValues(count)
+}
+
+func (dh *Datahub) GetClientTop(count int) []stats.Counter {
+	return dh.clientStat.GetTopValues(count)
+}
+
 // GetDomainDayLineStat 查询域名标签匹配 24小时趋势图
 func (dh *Datahub) GetDomainDayLineStat() *stats.LineChartData {
 	return dh.dayaDomainChartStat.LineChartData("最近 24 小时域名匹配统计").ChartData()

plugin/datahub/sched.go

@@ -15,7 +15,7 @@ func (dh *Datahub) startSched() {
 		dh.cronUpdateKeywordTableMap()
 	})
 
-	_, _ = dh.sched.AddFunc("@every 10s", func() {
+	_, _ = dh.sched.AddFunc("@every 60s", func() {
 		dh.dayaDomainChartStat.Update(dh.domainMatchStat)
 		dh.dayNetworkChartStat.Update(dh.networkMatchStat)
 	})

plugin/pkg/stats/stats.go

@@ -1,6 +1,7 @@
 package stats
 
 import (
+	"sort"
 	"sync"
 	"sync/atomic"
 )
@@ -80,3 +81,14 @@ func (c *CounterStat) GetValue(name string) int64 {
 		return 0
 	}
 }
+
+func (c *CounterStat) GetTopValues(count int) []Counter {
+	values := c.Values()
+	sort.Slice(values, func(i, j int) bool {
+		return values[i].Value > values[j].Value
+	})
+	if len(values) < count {
+		return values
+	}
+	return values[0:count]
+}