From 7f87c3aa198f7a74b10d4566d048db0866c17c52 Mon Sep 17 00:00:00 2001
From: zhangzengfei <zhangzengfei@smartai.com>
Date: 星期一, 24 一月 2022 19:52:02 +0800
Subject: [PATCH] 删除多余的日志级别

---
 logc.go |  226 ++++++++++++++++++++++++++++++++------------------------
 1 files changed, 130 insertions(+), 96 deletions(-)

diff --git a/logc.go b/logc.go
index 494c91f..515dfc2 100644
--- a/logc.go
+++ b/logc.go
@@ -1,10 +1,8 @@
 package logc
 
 import (
-	"basic.com/valib/bhomebus.git"
-	"encoding/json"
+	"basic.com/valib/c_bhomebus.git/proto/source/bhome_msg"
 	"fmt"
-	uuid "github.com/satori/go.uuid"
 	"sync"
 	"time"
 )
@@ -18,14 +16,19 @@
 )
 
 const (
-	TypeManual = iota + 1  //浜哄伐鎿嶄綔鏃ュ織,绯荤粺鍗囩骇,鎽勫儚鏈轰慨鏀�,绠楁硶淇敼...
-	TypeLoginOut //鐧诲綍鐧诲嚭鏃ュ織
-	TypePollInfo //杞贰鎽勫儚鏈哄強鍏跺搴旂殑绠楁硶
-	TypeStackInfo //鏁版嵁鏍堝鐞嗘儏鍐�
-	TypeWarnInfo //鏁呴殰淇℃伅
-	TypeRunInfo //杩愯鎯呭喌,gpu,mem,cpu
-	TypeSysInfo //绯荤粺鍙傛暟鍙樻洿,ip,server name,寮�鍏虫満淇℃伅
+	TypeManual    = iota + 1 //浜哄伐鎿嶄綔鏃ュ織,绯荤粺鍗囩骇,鎽勫儚鏈轰慨鏀�,绠楁硶淇敼...
+	TypeLoginOut             //鐧诲綍鐧诲嚭鏃ュ織
+	TypePollInfo             //杞贰鎽勫儚鏈哄強鍏跺搴旂殑绠楁硶
+	TypeStackInfo            //鏁版嵁鏍堝鐞嗘儏鍐�
+	TypeWarnInfo             //鏁呴殰淇℃伅
+	TypeRunInfo              //杩愯鎯呭喌,gpu,mem,cpu
+	TypeSysInfo              //绯荤粺鍙傛暟鍙樻洿,ip,server name,寮�鍏虫満淇℃伅
 )
+
+type LogPrinter interface {
+	Marshal() ([]byte, error)
+	Topic() string
+}
 
 type LogInfo struct {
 	ID        string `gorm:"column:id;primary_key;unique" json:"id"`
@@ -40,25 +43,29 @@
 }
 
 type LogRegister struct {
-	Nodes      []bhomebus.NetNode
-	Topic      string
-	Payload     []byte
+	Nodes   []bhome_msg.BHAddress
+	Topic   string
+	Payload []byte
 }
 
 type LogReportCallback func(*LogRegister)
 
 var (
-	msgChan chan []byte
-	//bhSock *bhomebus.Socket
-	//pubFn func(nodes []bhomebus.NetNode, topic string, data []byte, milliseconds int) int
+	logCh    chan LogPrinter
+	logger   *Log
 	ProcName string
-	ProcID string
-
+	ProcID   string
 )
 
+func Init(flogWriter LogReportCallback, log *Log, procId string, procName string, wg *sync.WaitGroup, done chan struct{}) bool {
+	logCh = make(chan LogPrinter, 300)
 
-func Init(flogWriter LogReportCallback, procId string, procName string, wg *sync.WaitGroup, done  chan struct{}) bool {
-	msgChan = make(chan []byte, 100)
+	if nil != log {
+		logger = log
+	} else {
+		logger = &Log{}
+	}
+
 	ProcName = procName
 	ProcID = procId
 
@@ -67,77 +74,102 @@
 	return true
 }
 
-//func InitBySock(sock *bhomebus.Socket, procId string, procName string) bool {
-//	msgChan = make(chan []byte, 100)
-//	ProcName = procName
-//	ProcID = procId
-//	bhSock = sock
-//	if bhSock == nil {
-//		return false
-//	}
-//
-//	go saveLoop()
-//
-//	return true
-//}
-
-//func InitByPubFn(fn func(nodes []bhomebus.NetNode, topic string, data []byte, milliseconds int) int,
-//				procId string, procName string) bool {
-//	msgChan = make(chan []byte, 100)
-//	ProcName = procName
-//	ProcID = procId
-//	pubFn = fn
-//	if pubFn == nil {
-//		return false
-//	}
-//
-//	go saveLoop()
-//
-//	return true
-//}
-
-func Save(level int, logType int, v ...interface{}) {
-	cache(level, logType, "", "", v)
-}
-
-func SaveManual(level int, logType int, userID string, userName string, v ...interface{}) {
-	cache(level, logType, userID, userName, v)
-}
-
-func cache(level int, logType int, userID string, userName string, fmtArgs []interface{}) {
-	// Format with Sprint, Sprintf, or neither.
-	msg := ""
-	if len(fmtArgs) > 0 {
-		msg = fmt.Sprint(fmtArgs...)
-	}
-	fmt.Println(msg)
-
-	log := LogInfo {
-		ID:        uuid.NewV4().String(),
-		Timestamp: time.Now().Format("2006-01-02 15:04:05"),
-		ProcName:  ProcName,
-		ProcID:    ProcID,
-		Level:     level,
-		Type:      logType,
-		UserID:    userID,
-		UserName:  userName,
-		Info:      msg,
-	}
-
-	data,err := json.Marshal(log)
-	if err != nil {
-		fmt.Println("json.Marshal(log) error:", log)
+func SaveOperationLog(log *OperationLog, timeout time.Duration) {
+	if nil == log {
 		return
 	}
 
-	msgChan <- data
+	log.ProcName = ProcName
+	log.ProcID = ProcID
+
+	deliverLog(log, timeout)
 }
 
+func SaveScheduleLog(category, level int, timeout time.Duration, v ...interface{}) {
+	msg := ""
+	if len(v) > 0 {
+		msg = fmt.Sprint(v...)
+	}
+
+	if msg == "" {
+		return
+	}
+
+	log := &ScheduleLog{
+		Timestamp: time.Now().Unix(),
+		ProcName:  ProcName,
+		ProcID:    ProcID,
+		Level:     level,
+		Type:      category,
+		Info:      msg,
+	}
+
+	deliverLog(log, timeout)
+}
+
+func SaveRuleServerLog(ruleServerPushLog RuleServerPushLog, timeout time.Duration) {
+	deliverLog(&ruleServerPushLog, timeout)
+}
+
+func deliverLog(l LogPrinter, timeout time.Duration) {
+	select {
+	case logCh <- l:
+		return
+	case <-time.After(timeout):
+		var info string
+		b, err := l.Marshal()
+		if nil == err {
+			info = string(b)
+		}
+		logger.Fatal("SaveScheduleLog failed to save log", info, l.Topic())
+	}
+}
+
+//func Save(level int, logType int, v ...interface{}) {
+//	cache(level, logType, "", "", v)
+//}
+//
+//func SaveManual(level int, logType int, userID string, userName string, v ...interface{}) {
+//	cache(level, logType, userID, userName, v)
+//}
+//
+//func cache(level int, logType int, userID string, userName string, fmtArgs []interface{}) {
+//	// Format with Sprint, Sprintf, or neither.
+//	msg := ""
+//	if len(fmtArgs) > 0 {
+//		msg = fmt.Sprint(fmtArgs...)
+//	}
+//	fmt.Println(msg)
+//
+//	log := LogInfo {
+//		ID:        uuid.NewV4().String(),
+//		Timestamp: time.Now().Format("2006-01-02 15:04:05"),
+//		ProcName:  ProcName,
+//		ProcID:    ProcID,
+//		Level:     level,
+//		Type:      logType,
+//		UserID:    userID,
+//		UserName:  userName,
+//		Info:      msg,
+//	}
+//
+//	data,err := json.Marshal(log)
+//	if err != nil {
+//		fmt.Println("json.Marshal(log) error:", log)
+//		return
+//	}
+//
+//	msgChan <- data
+//}
+
 const (
-	LogSaveTopic = "logSaveTopic"
+	OperationLogTopic  = "operationLogSaveTopic"
+	ScheduleLogTopic   = "scheduleLogSaveTopic"
+	RuleServerLogTopic = "ruleServerLogSaveTopic"
+	VaSystemLogTopic   = "VaSystemLogSaveTopic"
 )
 
-func saveLoop(logCallback LogReportCallback, wg *sync.WaitGroup, done  chan struct{}) {
+func saveLoop(logCallback LogReportCallback, wg *sync.WaitGroup, done chan struct{}) {
 	defer wg.Done()
 
 	if nil == logCallback {
@@ -146,22 +178,24 @@
 
 	for {
 		select {
-		case <- done:
+		case <-done:
 			return
-		case data := <- msgChan:
-			var nodes []bhomebus.NetNode
-			nodes = append(nodes, bhomebus.NetNode{})
+		case log := <-logCh:
+			payload, err := log.Marshal()
+			if err != nil {
+				logger.Error("failed to Marshal", log)
+			} else {
+				var nodes []bhome_msg.BHAddress
+				nodes = append(nodes, bhome_msg.BHAddress{})
 
-			reg := &LogRegister {
-				nodes,
-				LogSaveTopic,
-				data,
+				reg := &LogRegister{
+					nodes,
+					log.Topic(),
+					payload,
+				}
+
+				logCallback(reg)
 			}
-
-			logCallback(reg)
-
-		default:
-			time.Sleep(10*time.Millisecond)
 		}
 	}
 }

--
Gitblit v1.8.0