OMC/be.ems
1
0
Fork 0
Code Issues Releases 3 Activity

fix: 禁用TaskWriteSystemLog

Browse Source
This commit is contained in:
TsMask
2024-08-16 10:11:40 +08:00
parent 5c6857e896
commit d90c26a28e
1 changed files with 75 additions and 76 deletions
Download Patch File Download Diff File Expand all files Collapse all files

151
crontask/tasks.go
View File

@@ -18,7 +18,6 @@ import (
"be.ems/lib/log" "be.ems/lib/log"
"github.com/go-resty/resty/v2" "github.com/go-resty/resty/v2"
"github.com/shirou/gopsutil/process"
// cron "github.com/go-co-op/gocron" // cron "github.com/go-co-op/gocron"
"github.com/jasonlvhit/gocron" "github.com/jasonlvhit/gocron"
@@ -1456,82 +1455,82 @@ func init() {
func (t *TaskFunc) TaskWriteSystemLog(uri, params, body string) { func (t *TaskFunc) TaskWriteSystemLog(uri, params, body string) {
processes, err := process.Processes() // processes, err := process.Processes()
if err != nil { // if err != nil {
log.Error("Failed to get processes:", err) // log.Error("Failed to get processes:", err)
return // return
} // }
for _, proc := range processes { // for _, proc := range processes {
name, err := proc.Name() // name, err := proc.Name()
if err != nil { // if err != nil {
log.Error("Failed to get process name:", err) // log.Error("Failed to get process name:", err)
continue // continue
} // }
for i, procInfo := range procInfos { // for i, procInfo := range procInfos {
if name == procInfo.Name { // if name == procInfo.Name {
pid := proc.Pid // pid := proc.Pid
timestamp, _ := proc.CreateTime() // timestamp, _ := proc.CreateTime()
userName, _ := proc.Username() // userName, _ := proc.Username()
startTime := time.Unix((timestamp / 1000), 0).Format(time.DateTime) // startTime := time.Unix((timestamp / 1000), 0).Format(time.DateTime)
newProcInfo := &ProcInfo{ // newProcInfo := &ProcInfo{
Name: procInfo.Name, // Name: procInfo.Name,
Pid: pid, // Pid: pid,
StartTime: startTime, // StartTime: startTime,
} // }
log.Trace("newProcInfo:", *newProcInfo) // log.Trace("newProcInfo:", *newProcInfo)
log.Trace("procInfo:", procInfo) // log.Trace("procInfo:", procInfo)
if *newProcInfo != procInfo { // if *newProcInfo != procInfo {
procInfos[i] = *newProcInfo // procInfos[i] = *newProcInfo
var operation string // var operation string
if newProcInfo.Pid != 0 { // if newProcInfo.Pid != 0 {
operation = "start" // operation = "start"
} else { // } else {
operation = "stop" // operation = "stop"
} // }
sysLogs := &[]dborm.SystemLog{} // sysLogs := &[]dborm.SystemLog{}
sysLog := &dborm.SystemLog{ // sysLog := &dborm.SystemLog{
UserName: userName, // UserName: userName,
ProcessName: newProcInfo.Name, // ProcessName: newProcInfo.Name,
ProcessID: newProcInfo.Pid, // ProcessID: newProcInfo.Pid,
Operation: operation, // Operation: operation,
StartTime: newProcInfo.StartTime, // StartTime: newProcInfo.StartTime,
LogTime: time.Now().Format(time.DateTime), // LogTime: time.Now().Format(time.DateTime),
} // }
var response *resty.Response // var response *resty.Response
requestURI := fmt.Sprintf("%s?%s", uri, params) // requestURI := fmt.Sprintf("%s?%s", uri, params)
requestURL := fmt.Sprintf("%s%s", yamlConfig.OMC.HostUri, requestURI) // requestURL := fmt.Sprintf("%s%s", yamlConfig.OMC.HostUri, requestURI)
*sysLogs = append(*sysLogs, *sysLog) // *sysLogs = append(*sysLogs, *sysLog)
data := make(map[string]interface{}) // data := make(map[string]interface{})
data["system_log"] = *sysLogs // data["system_log"] = *sysLogs
body, _ := json.Marshal(data) // body, _ := json.Marshal(data)
log.Trace("data:", data) // log.Trace("data:", data)
log.Debug("requestURL: POST ", requestURL) // log.Debug("requestURL: POST ", requestURL)
client := resty.New() // client := resty.New()
response, err = client.R(). // response, err = client.R().
EnableTrace(). // EnableTrace().
SetHeaders(map[string]string{"User-Agent": GetDefaultUserAgent()}). // SetHeaders(map[string]string{"User-Agent": GetDefaultUserAgent()}).
SetHeaders(map[string]string{"Content-Type": "application/json;charset=UTF-8"}). // SetHeaders(map[string]string{"Content-Type": "application/json;charset=UTF-8"}).
SetBody(body). // SetBody(body).
Post(requestURL) // Post(requestURL)
if err != nil { // if err != nil {
log.Error("Failed to post:", err) // log.Error("Failed to post:", err)
} // }
log.Debug("StatusCode: ", response.StatusCode()) // log.Debug("StatusCode: ", response.StatusCode())
switch response.StatusCode() { // switch response.StatusCode() {
case http.StatusOK, http.StatusCreated, http.StatusNoContent, http.StatusAccepted: // case http.StatusOK, http.StatusCreated, http.StatusNoContent, http.StatusAccepted:
log.Debug("response body:", string(response.Body())) // log.Debug("response body:", string(response.Body()))
body := new(map[string]interface{}) // body := new(map[string]interface{})
_ = json.Unmarshal(response.Body(), &body) // _ = json.Unmarshal(response.Body(), &body)
default: // default:
log.Debug("response body:", string(response.Body())) // log.Debug("response body:", string(response.Body()))
body := new(map[string]interface{}) // body := new(map[string]interface{})
_ = json.Unmarshal(response.Body(), &body) // _ = json.Unmarshal(response.Body(), &body)
} // }
} // }
} // }
} // }
} // }
} }