提交 v1.3.0 beta
This commit is contained in:
dushixiang
@ -2,10 +2,11 @@ package api
|
||||
|
||||
import (
|
||||
"context"
|
||||
|
||||
"next-terminal/server/constant"
|
||||
"next-terminal/server/common/nt"
|
||||
"next-terminal/server/dto"
|
||||
"next-terminal/server/global/stat"
|
||||
"next-terminal/server/repository"
|
||||
"time"
|
||||
|
||||
"github.com/labstack/echo/v4"
|
||||
)
|
||||
@ -14,21 +15,27 @@ type OverviewApi struct{}
|
||||
|
||||
func (api OverviewApi) OverviewCounterEndPoint(c echo.Context) error {
|
||||
var (
|
||||
countUser int64
|
||||
countOnlineSession int64
|
||||
credential int64
|
||||
asset int64
|
||||
totalUser int64
|
||||
onlineUser int64
|
||||
countOfflineSession int64
|
||||
totalAsset int64
|
||||
activeAsset int64
|
||||
failLoginCount int64
|
||||
)
|
||||
countUser, _ = repository.UserRepository.CountOnlineUser(context.TODO())
|
||||
countOnlineSession, _ = repository.SessionRepository.CountOnlineSession(context.TODO())
|
||||
credential, _ = repository.CredentialRepository.Count(context.TODO())
|
||||
asset, _ = repository.AssetRepository.Count(context.TODO())
|
||||
totalUser, _ = repository.UserRepository.Count(context.TODO())
|
||||
onlineUser, _ = repository.UserRepository.CountOnlineUser(context.TODO())
|
||||
countOfflineSession, _ = repository.SessionRepository.CountOfflineSession(context.TODO())
|
||||
totalAsset, _ = repository.AssetRepository.Count(context.TODO())
|
||||
activeAsset, _ = repository.AssetRepository.CountByActive(context.TODO(), true)
|
||||
failLoginCount, _ = repository.LoginLogRepository.CountByState(context.TODO(), "0")
|
||||
|
||||
counter := dto.Counter{
|
||||
User: countUser,
|
||||
OnlineSession: countOnlineSession,
|
||||
Credential: credential,
|
||||
Asset: asset,
|
||||
TotalUser: totalUser,
|
||||
OnlineUser: onlineUser,
|
||||
OfflineSession: countOfflineSession,
|
||||
TotalAsset: totalAsset,
|
||||
ActiveAsset: activeAsset,
|
||||
FailLoginCount: failLoginCount,
|
||||
}
|
||||
|
||||
return Success(c, counter)
|
||||
@ -43,11 +50,11 @@ func (api OverviewApi) OverviewAssetEndPoint(c echo.Context) error {
|
||||
kubernetes int64
|
||||
)
|
||||
|
||||
ssh, _ = repository.AssetRepository.CountByProtocol(context.TODO(), constant.SSH)
|
||||
rdp, _ = repository.AssetRepository.CountByProtocol(context.TODO(), constant.RDP)
|
||||
vnc, _ = repository.AssetRepository.CountByProtocol(context.TODO(), constant.VNC)
|
||||
telnet, _ = repository.AssetRepository.CountByProtocol(context.TODO(), constant.Telnet)
|
||||
kubernetes, _ = repository.AssetRepository.CountByProtocol(context.TODO(), constant.K8s)
|
||||
ssh, _ = repository.AssetRepository.CountByProtocol(context.TODO(), nt.SSH)
|
||||
rdp, _ = repository.AssetRepository.CountByProtocol(context.TODO(), nt.RDP)
|
||||
vnc, _ = repository.AssetRepository.CountByProtocol(context.TODO(), nt.VNC)
|
||||
telnet, _ = repository.AssetRepository.CountByProtocol(context.TODO(), nt.Telnet)
|
||||
kubernetes, _ = repository.AssetRepository.CountByProtocol(context.TODO(), nt.K8s)
|
||||
|
||||
m := echo.Map{
|
||||
"ssh": ssh,
|
||||
@ -55,14 +62,168 @@ func (api OverviewApi) OverviewAssetEndPoint(c echo.Context) error {
|
||||
"vnc": vnc,
|
||||
"telnet": telnet,
|
||||
"kubernetes": kubernetes,
|
||||
"all": ssh + rdp + vnc + telnet + kubernetes,
|
||||
}
|
||||
return Success(c, m)
|
||||
}
|
||||
|
||||
func (api OverviewApi) OverviewAccessEndPoint(c echo.Context) error {
|
||||
access, err := repository.SessionRepository.OverviewAccess(context.TODO())
|
||||
func (api OverviewApi) OverviewDateCounterEndPoint(c echo.Context) error {
|
||||
d := c.QueryParam("d")
|
||||
var days = 7
|
||||
if d == "month" {
|
||||
days = 30
|
||||
}
|
||||
now := time.Now()
|
||||
lastDate := now.AddDate(0, 0, -days)
|
||||
// 最近一月登录次数
|
||||
loginLogCounters, err := repository.LoginLogRepository.CountWithGroupByLoginTime(context.TODO(), lastDate)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return Success(c, access)
|
||||
// 最近一月活跃用户
|
||||
userCounters, err := repository.LoginLogRepository.CountWithGroupByLoginTimeAndUsername(context.TODO(), lastDate)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
// 最近一月活跃资产
|
||||
sessionCounters, err := repository.SessionRepository.CountWithGroupByLoginTime(context.TODO(), lastDate)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
var counters []dto.DateCounter
|
||||
for i := 0; i < days; i++ {
|
||||
day := lastDate.AddDate(0, 0, i).Format("2006-01-02")
|
||||
|
||||
var exist = false
|
||||
for _, counter := range loginLogCounters {
|
||||
if counter.Date == day {
|
||||
exist = true
|
||||
counters = append(counters, dto.DateCounter{
|
||||
Type: "登录次数",
|
||||
Date: day,
|
||||
Value: counter.Value,
|
||||
})
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
if !exist {
|
||||
counters = append(counters, dto.DateCounter{
|
||||
Type: "登录次数",
|
||||
Date: day,
|
||||
Value: 0,
|
||||
})
|
||||
}
|
||||
|
||||
exist = false
|
||||
for _, counter := range userCounters {
|
||||
if counter.Date == day {
|
||||
exist = true
|
||||
counters = append(counters, dto.DateCounter{
|
||||
Type: "活跃用户",
|
||||
Date: day,
|
||||
Value: counter.Value,
|
||||
})
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
if !exist {
|
||||
counters = append(counters, dto.DateCounter{
|
||||
Type: "活跃用户",
|
||||
Date: day,
|
||||
Value: 0,
|
||||
})
|
||||
}
|
||||
|
||||
exist = false
|
||||
for _, counter := range sessionCounters {
|
||||
if counter.Date == day {
|
||||
exist = true
|
||||
counters = append(counters, dto.DateCounter{
|
||||
Type: "活跃资产",
|
||||
Date: day,
|
||||
Value: counter.Value,
|
||||
})
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
if !exist {
|
||||
counters = append(counters, dto.DateCounter{
|
||||
Type: "活跃资产",
|
||||
Date: day,
|
||||
Value: 0,
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
return Success(c, counters)
|
||||
}
|
||||
|
||||
func (api OverviewApi) OverviewPS(c echo.Context) error {
|
||||
//memoryStat, err := mem.VirtualMemory()
|
||||
//if err != nil {
|
||||
// return err
|
||||
//}
|
||||
//avgStat, err := load.Avg()
|
||||
//if err != nil {
|
||||
// return err
|
||||
//}
|
||||
//
|
||||
//cpuCount, err := cpu.Counts(true)
|
||||
//if err != nil {
|
||||
// return err
|
||||
//}
|
||||
//
|
||||
//percent, err := cpu.Percent(time.Second, false)
|
||||
//if err != nil {
|
||||
// return err
|
||||
//}
|
||||
//
|
||||
//var bytesRead uint64 = 0
|
||||
//var bytesWrite uint64 = 0
|
||||
//
|
||||
//diskIO, err := disk.IOCounters()
|
||||
//if err != nil {
|
||||
// return err
|
||||
//}
|
||||
//for _, v := range diskIO {
|
||||
// bytesRead += v.ReadBytes
|
||||
// bytesWrite += v.WriteBytes
|
||||
//}
|
||||
//
|
||||
//var bytesSent uint64 = 0
|
||||
//var bytesRecv uint64 = 0
|
||||
//netIO, err := net.IOCounters(true)
|
||||
//if err != nil {
|
||||
// return err
|
||||
//}
|
||||
//for _, v := range netIO {
|
||||
// bytesSent += v.BytesSent
|
||||
// bytesRecv += v.BytesRecv
|
||||
//}
|
||||
|
||||
//return Success(c, Map{
|
||||
// "mem": Map{
|
||||
// "total": memoryStat.Total,
|
||||
// "usedPercent": memoryStat.UsedPercent,
|
||||
// },
|
||||
// "cpu": Map{
|
||||
// "count": cpuCount,
|
||||
// "loadAvg": avgStat,
|
||||
// "usedPercent": percent[0],
|
||||
// },
|
||||
// "diskIO": Map{
|
||||
// "bytesRead": bytesRead,
|
||||
// "bytesWrite": bytesWrite,
|
||||
// },
|
||||
// "netIO": Map{
|
||||
// "bytesSent": bytesSent,
|
||||
// "bytesRecv": bytesRecv,
|
||||
// },
|
||||
//})
|
||||
|
||||
return Success(c, stat.SystemLoad)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user