Подготовить HY2XS к production-сборке
This commit is contained in:
@@ -0,0 +1,116 @@
|
||||
package controller
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"github.com/gin-gonic/gin"
|
||||
"hy2xs-admin/model/constant"
|
||||
"hy2xs-admin/model/dto"
|
||||
"hy2xs-admin/model/vo"
|
||||
"hy2xs-admin/util"
|
||||
"time"
|
||||
)
|
||||
|
||||
func LogSystem(c *gin.Context) {
|
||||
logSystemDto, err := validateField(c, dto.LogDto{})
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
exists := util.Exists(constant.SystemLogPath)
|
||||
logSystemVos := make([]vo.LogSystemVo, 0)
|
||||
if !exists {
|
||||
vo.Success(logSystemVos, c)
|
||||
return
|
||||
}
|
||||
numLine := 0
|
||||
if logSystemDto.NumLine != nil || *logSystemDto.NumLine > 0 {
|
||||
numLine = *logSystemDto.NumLine
|
||||
}
|
||||
logLines, total, err := util.ReadLinesFromBottom(constant.SystemLogPath, numLine)
|
||||
if err != nil {
|
||||
vo.Fail("Unable to read log file", c)
|
||||
return
|
||||
}
|
||||
|
||||
for _, line := range logLines {
|
||||
if line == "" {
|
||||
continue
|
||||
}
|
||||
logSystemVo := vo.LogSystemVo{}
|
||||
err := json.Unmarshal([]byte(line), &logSystemVo)
|
||||
if err != nil {
|
||||
vo.Fail("Unable to unmarshal log data", c)
|
||||
continue
|
||||
}
|
||||
logSystemVos = append(logSystemVos, logSystemVo)
|
||||
}
|
||||
vo.Success(vo.LogSystemPage[vo.LogSystemVo]{
|
||||
LogSystemVos: logSystemVos,
|
||||
Total: int64(total),
|
||||
}, c)
|
||||
}
|
||||
|
||||
func LogHysteria2(c *gin.Context) {
|
||||
logSystemDto, err := validateField(c, dto.LogDto{})
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
exists := util.Exists(constant.Hysteria2LogPath)
|
||||
logHysteria2Vos := make([]vo.LogHysteria2Vo, 0)
|
||||
if !exists {
|
||||
vo.Success(logHysteria2Vos, c)
|
||||
return
|
||||
}
|
||||
numLine := 0
|
||||
if logSystemDto.NumLine != nil || *logSystemDto.NumLine > 0 {
|
||||
numLine = *logSystemDto.NumLine
|
||||
}
|
||||
logLines, total, err := util.ReadLinesFromBottom(constant.Hysteria2LogPath, numLine)
|
||||
if err != nil {
|
||||
vo.Fail("Unable to read log file", c)
|
||||
return
|
||||
}
|
||||
|
||||
for _, line := range logLines {
|
||||
if line == "" {
|
||||
continue
|
||||
}
|
||||
logHysteria2Vo := vo.LogHysteria2Vo{}
|
||||
err := json.Unmarshal([]byte(line), &logHysteria2Vo)
|
||||
if err != nil {
|
||||
vo.Fail("Unable to unmarshal log data", c)
|
||||
continue
|
||||
}
|
||||
logHysteria2Vos = append(logHysteria2Vos, logHysteria2Vo)
|
||||
}
|
||||
vo.Success(vo.LogSystemPage[vo.LogHysteria2Vo]{
|
||||
LogSystemVos: logHysteria2Vos,
|
||||
Total: int64(total),
|
||||
}, c)
|
||||
}
|
||||
|
||||
func ExportLog(c *gin.Context) {
|
||||
logExportDto, err := validateField(c, dto.LogExportDto{})
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
var fileName string
|
||||
var filePath string
|
||||
if *logExportDto.Option == 0 {
|
||||
fileName = fmt.Sprintf("hy2xs-admin-%s.log", time.Now().Format("20060102150405"))
|
||||
filePath = constant.SystemLogPath
|
||||
} else if *logExportDto.Option == 1 {
|
||||
fileName = fmt.Sprintf("hysteria2-%s.log", time.Now().Format("20060102150405"))
|
||||
filePath = constant.Hysteria2LogPath
|
||||
}
|
||||
|
||||
if !util.Exists(filePath) {
|
||||
vo.Fail("log file not exist", c)
|
||||
return
|
||||
}
|
||||
c.Header("Content-Type", "application/octet-stream")
|
||||
c.Header("Content-Transfer-Encoding", "binary")
|
||||
c.Header("Content-Disposition", fmt.Sprintf("attachment; filename=%s", fileName))
|
||||
c.File(filePath)
|
||||
}
|
||||
Reference in New Issue
Block a user