changed style of logs so they can read more easily
This commit is contained in:
parent
76d343e8ac
commit
3d57688c3e
|
@ -71,7 +71,7 @@ func connectToWebSocket(agentName, agentId, agentIp, agentType, hostName string)
|
|||
for {
|
||||
conn, _, err = websocket.DefaultDialer.Dial(wsURL, nil)
|
||||
if err == nil {
|
||||
log.Println("WeSocket connection established")
|
||||
log.Println("WebSocket connection established")
|
||||
// logger.LogEntries = append(logger.LogEntries, fmt.Sprintf("%s websocket established", time.Now().Format(time.RFC3339)))
|
||||
return nil
|
||||
}
|
||||
|
|
|
@ -77,7 +77,8 @@ func (l *Logger) InsertLog(level LogLevel, message string) error {
|
|||
}
|
||||
|
||||
insertQuery := `INSERT INTO logs (message, level) VALUES(?, ?)`
|
||||
message = strings.ReplaceAll(message, `"`, `\"`)
|
||||
/* String escaping is obsolete when logfmt will not be used */
|
||||
// message = strings.ReplaceAll(message, `"`, `\"`)
|
||||
|
||||
_, err = tx.Exec(insertQuery, message, level)
|
||||
if err != nil {
|
||||
|
@ -134,7 +135,6 @@ func (l *Logger) FetchLogs(limit int, levels []string) ([]LogEntry, error) {
|
|||
return logs, nil
|
||||
}
|
||||
|
||||
|
||||
func (l * Logger) Close() error {
|
||||
return l.db.Close()
|
||||
}
|
||||
|
|
|
@ -19,8 +19,8 @@
|
|||
/* background-color: #f8f9fa; /1* optional: subtle background for log readability *1/ */
|
||||
}
|
||||
|
||||
.log-info, .log-warning, .log-error, .log-fatal, .log-debug{ font-family: "Lucida Console", Monaco, monospace;
|
||||
font-size: 13px;
|
||||
.log-info, .log-warning, .log-error, .log-fatal, .log-debug{ font-family: "Roboto Mono", "Consolas", "Courier New", monospace;
|
||||
font-size: 14px;
|
||||
}
|
||||
|
||||
.log-info {
|
||||
|
|
|
@ -1,5 +1,5 @@
|
|||
{{range .}}
|
||||
<div class="log-{{.Level}}">
|
||||
ts={{.Timestamp}} level={{.Level}} msg="{{.Message}}"
|
||||
{{.Timestamp}} [{{.Level}}] {{.Message}}
|
||||
</div>
|
||||
{{end}}
|
||||
|
|
|
@ -65,7 +65,6 @@ func registerAgent(agentName, agentId, agentIp, agentType, addPort, hostname str
|
|||
resp, err := http.PostForm(registerURL, form)
|
||||
if err != nil {
|
||||
log.Printf("Error registering agent: %v", err)
|
||||
// logger.InsertLog(logger.Error, fmt.Sprintf("Error registering agent: %v", err))
|
||||
logService.Info(fmt.Sprintf("Error registering agent: %v", err))
|
||||
return err
|
||||
}
|
||||
|
@ -73,17 +72,14 @@ func registerAgent(agentName, agentId, agentIp, agentType, addPort, hostname str
|
|||
|
||||
if resp.StatusCode == http.StatusCreated {
|
||||
log.Printf("Agent %s successfully registered.", agentName)
|
||||
// logger.InsertLog(logger.Info, fmt.Sprintf("Agent %s successfully registered.", agentName))
|
||||
logService.Info(fmt.Sprintf("Agent %s successfully registered.", agentName))
|
||||
return nil
|
||||
} else if resp.StatusCode == http.StatusOK {
|
||||
log.Printf("Agent %s already registered.", agentName)
|
||||
// logger.InsertLog(logger.Info, fmt.Sprintf("Agent %s already registered.", agentName))
|
||||
logService.Info(fmt.Sprintf("Agent %s already registered.", agentName))
|
||||
return nil
|
||||
} else {
|
||||
log.Printf("Failed to register agent, status: %v", resp.Status)
|
||||
// logger.InsertLog(logger.Error, fmt.Sprintf("Failed to register agent, status: %v", resp.Status))
|
||||
logService.Error(fmt.Sprintf("Failer to register agent, status: %v", resp.Status))
|
||||
return err
|
||||
}
|
||||
|
@ -96,7 +92,6 @@ func getAgentDetails(agentId string) (*api.Agent, error) {
|
|||
resp, err := http.Get(agentURL)
|
||||
if err != nil {
|
||||
log.Printf("Failed to make GET request: %s", err)
|
||||
// logger.InsertLog(logger.Error, fmt.Sprintf("Failed to make GET request: %s", err))
|
||||
logService.Error(fmt.Sprintf("Failed to make GET request: %s", err))
|
||||
return nil, err
|
||||
}
|
||||
|
@ -117,7 +112,6 @@ func getAgentDetails(agentId string) (*api.Agent, error) {
|
|||
agent.AgentID, err = strconv.Atoi(strings.TrimSpace(strings.TrimPrefix(text, "ID:")))
|
||||
if err != nil {
|
||||
log.Printf("Converting string to integer failed in getAgentDetails(): %s", err)
|
||||
// logger.InsertLog(logger.Error, fmt.Sprintf("Converting string to integer failed in getAgentDetails(): %s", err))
|
||||
logService.Error(fmt.Sprintf("Converting string to integer failed in getAgentDetails(): %s", err))
|
||||
}
|
||||
} else if strings.HasPrefix(text, "Name:") {
|
||||
|
@ -144,7 +138,6 @@ func getAgentIds() ([]string, error) {
|
|||
resp, err := http.Get(idURL)
|
||||
if err != nil {
|
||||
log.Printf("Failed to make GET request: %s", err)
|
||||
// logger.InsertLog(logger.Error, fmt.Sprintf("Failed to make GET request: %s", err))
|
||||
logService.Error(fmt.Sprintf("Failed to make GET request: %s", err))
|
||||
return nil, err
|
||||
}
|
||||
|
@ -152,7 +145,6 @@ func getAgentIds() ([]string, error) {
|
|||
|
||||
if resp.StatusCode != http.StatusOK {
|
||||
log.Printf("Unexpected status code: %d", resp.StatusCode)
|
||||
// logger.InsertLog(logger.Info, fmt.Sprintf("Unexpected status code: %d", resp.StatusCode))
|
||||
logService.Info(fmt.Sprintf("Unexpected status code: %d", resp.StatusCode))
|
||||
return nil, nil
|
||||
}
|
||||
|
@ -160,7 +152,6 @@ func getAgentIds() ([]string, error) {
|
|||
body, err := io.ReadAll(resp.Body)
|
||||
if err != nil {
|
||||
log.Printf("Failed to read response body: %s", err)
|
||||
// logger.InsertLog(logger.Error, fmt.Sprintf("Failed to read response body: %s", err))
|
||||
logService.Error(fmt.Sprintf("Failed to read response body: %s", err))
|
||||
return nil, err
|
||||
}
|
||||
|
@ -168,7 +159,6 @@ func getAgentIds() ([]string, error) {
|
|||
var agentIds []string
|
||||
if err := json.Unmarshal(body, &agentIds); err != nil {
|
||||
log.Printf("Failed to parse JSON response: %s", err)
|
||||
// logger.InsertLog(logger.Error, fmt.Sprintf("Failed to parse JSON response: %s", err))
|
||||
logService.Error(fmt.Sprintf("Failed to parse JSON response: %s", err))
|
||||
return nil, err
|
||||
}
|
||||
|
@ -181,7 +171,6 @@ func (wsh webSocketHandler) ServeHTTP(w http.ResponseWriter, r *http.Request){
|
|||
c, err := wsh.upgrader.Upgrade(w, r, nil)
|
||||
if err != nil {
|
||||
log.Printf("Error %s when upgrading connection to websocket", err)
|
||||
// logger.InsertLog(logger.Error, fmt.Sprintf("Error %s when upgrading connection to websocket", err))
|
||||
logService.Error(fmt.Sprintf("Error %s when upgrading connection to websocket", err))
|
||||
return
|
||||
}
|
||||
|
@ -215,14 +204,12 @@ func (wsh webSocketHandler) ServeHTTP(w http.ResponseWriter, r *http.Request){
|
|||
|
||||
if agentName == "" || agentIP == "" {
|
||||
log.Printf("Missing agentName or IPv4Address in query parameters")
|
||||
// logger.InsertLog(logger.Info, fmt.Sprintf("Missing agentName or IPv4Address in query parameters"))
|
||||
logService.Info(fmt.Sprintf("Missing agentName or IPv4Address in query parameters"))
|
||||
c.Close()
|
||||
return
|
||||
}
|
||||
|
||||
log.Printf("Agent %s connected: %s (%s)", agentId, agentName, agentIP)
|
||||
// logger.InsertLog(logger.Info, fmt.Sprintf("Agent %s connected: %s (%s)", agentId, agentName, agentIP))
|
||||
logService.Info(fmt.Sprintf("Agent %s connected: %s (%s)", agentId, agentName, agentIP))
|
||||
|
||||
agentSocketsMutex.Lock()
|
||||
|
@ -235,29 +222,9 @@ func (wsh webSocketHandler) ServeHTTP(w http.ResponseWriter, r *http.Request){
|
|||
agentSocketsMutex.Unlock()
|
||||
c.Close()
|
||||
log.Printf("Agent disconnected: %s (%s)", agentName, agentIP)
|
||||
// logger.InsertLog(logger.Info, fmt.Sprintf("Agent disconnected: %s (%s)", agentName, agentIP))
|
||||
logService.Info(fmt.Sprintf("Agent disconnected: %s (%s)", agentName, agentIP))
|
||||
}()
|
||||
|
||||
// for {
|
||||
// _, message, err := c.ReadMessage()
|
||||
// if err != nil {
|
||||
// log.Printf("Error reading from agent %s: %v", agentName, err)
|
||||
// // logger.InsertLog(logger.Error, fmt.Sprintf("Error reading from agent %s: %v", agentName, err))
|
||||
// logService.Error(fmt.Sprintf("Error reading from agent %s: %v", agentName, err))
|
||||
// break
|
||||
// }
|
||||
// // log.Printf("Message from agent %s: %s", agentName, message)
|
||||
// log.Printf("Message from agent %s received", agentName)
|
||||
// // logger.InsertLog(logger.Debug, fmt.Sprintf("Message from agent %s: %s", agentName, message))
|
||||
// logService.Debug(fmt.Sprintf("Message from agent %s: %s", agentName, message))
|
||||
|
||||
// if ch, ok := responseChannels.Load(agentName); ok {
|
||||
// responseChan := ch.(chan string)
|
||||
// responseChan <- string(message)
|
||||
// }
|
||||
// }
|
||||
|
||||
for {
|
||||
_, message, err := c.ReadMessage()
|
||||
if err != nil {
|
||||
|
@ -316,7 +283,6 @@ var executeCommand http.HandlerFunc = func(w http.ResponseWriter, r *http.Reques
|
|||
err := r.ParseForm()
|
||||
if err != nil {
|
||||
http.Error(w, "Invalid form data", http.StatusBadRequest)
|
||||
// logger.InsertLog(logger.Info, "Invalid form data")
|
||||
logService.Info("Invalid form data")
|
||||
return
|
||||
}
|
||||
|
@ -337,7 +303,6 @@ var executeCommand http.HandlerFunc = func(w http.ResponseWriter, r *http.Reques
|
|||
|
||||
if len(agentNames) == 0 || command == "" {
|
||||
http.Error(w, "Missing agent or command", http.StatusBadRequest)
|
||||
// logger.InsertLog(logger.Error, "Missing agent or command")
|
||||
logService.Error("Missing agent or command")
|
||||
return
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue