Buriburizaemon/nezha_domains
1
0
Fork 0
mirror of https://github.com/Buriburizaem0n/nezha_domains.git synced 2026-02-04 04:30:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: show online user id

Browse Source
This commit is contained in:
naiba
2024-12-24 23:28:37 +08:00
parent d50605d668
commit 5128bfff61
3 changed files with 11 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
cmd/dashboard/controller/waf.go
View File

@@ -33,7 +33,7 @@ func listBlockedAddress(c *gin.Context) (*model.Value[[]*model.WAF], error) {
}
var waf []*model.WAF
if err := singleton.DB.Limit(limit).Offset(offset).Find(&waf).Error; err != nil {
if err := singleton.DB.Order("block_timestamp DESC").Limit(limit).Offset(offset).Find(&waf).Error; err != nil {
return nil, err
}

13
cmd/dashboard/controller/ws.go
View File

@@ -119,7 +119,14 @@ func serverStream(c *gin.Context) (any, error) {
userIp = c.RemoteIP()
}
u, isMember := c.Get(model.CtxKeyAuthorizedUser)
var userId uint64
if isMember {
userId = u.(*model.User).ID
}
singleton.AddOnlineUser(connId, &model.OnlineUser{
UserID: userId,
IP: userIp,
ConnectedAt: time.Now(),
Conn: conn,
@@ -128,7 +135,7 @@ func serverStream(c *gin.Context) (any, error) {
count := 0
for {
stat, err := getServerStat(c, count == 0)
stat, err := getServerStat(count == 0, isMember)
if err != nil {
continue
}
@@ -149,9 +156,7 @@ func serverStream(c *gin.Context) (any, error) {
var requestGroup singleflight.Group
func getServerStat(c *gin.Context, withPublicNote bool) ([]byte, error) {
_, isMember := c.Get(model.CtxKeyAuthorizedUser)
authorized := isMember // TODO || isViewPasswordVerfied
func getServerStat(withPublicNote, authorized bool) ([]byte, error) {
v, err, _ := requestGroup.Do(fmt.Sprintf("serverStats::%t", authorized), func() (interface{}, error) {
singleton.SortedServerLock.RLock()
defer singleton.SortedServerLock.RUnlock()