finish logs module

logs/conn.go

@@ -0,0 +1,111 @@
+package logs
+
+import (
+	"encoding/json"
+	"io"
+	"log"
+	"net"
+)
+
+type ConnWriter struct {
+	lg             *log.Logger
+	innerWriter    io.WriteCloser
+	reconnectOnMsg bool
+	reconnect      bool
+	net            string
+	addr           string
+	level          int
+}
+
+func NewConn() LoggerInterface {
+	conn := new(ConnWriter)
+	conn.level = LevelTrace
+	return conn
+}
+
+func (c *ConnWriter) Init(jsonconfig string) error {
+	var m map[string]interface{}
+	err := json.Unmarshal([]byte(jsonconfig), &m)
+	if err != nil {
+		return err
+	}
+	if rom, ok := m["reconnectOnMsg"]; ok {
+		c.reconnectOnMsg = rom.(bool)
+	}
+	if rc, ok := m["reconnect"]; ok {
+		c.reconnect = rc.(bool)
+	}
+	if nt, ok := m["net"]; ok {
+		c.net = nt.(string)
+	}
+	if addr, ok := m["addr"]; ok {
+		c.addr = addr.(string)
+	}
+	if lv, ok := m["level"]; ok {
+		c.level = int(lv.(float64))
+	}
+	return nil
+}
+
+func (c *ConnWriter) WriteMsg(msg string, level int) error {
+	if level < c.level {
+		return nil
+	}
+	if c.neddedConnectOnMsg() {
+		err := c.connect()
+		if err != nil {
+			return err
+		}
+	}
+
+	if c.reconnectOnMsg {
+		defer c.innerWriter.Close()
+	}
+	c.lg.Println(msg)
+	return nil
+}
+
+func (c *ConnWriter) Destroy() {
+	if c.innerWriter == nil {
+		return
+	}
+	c.innerWriter.Close()
+}
+
+func (c *ConnWriter) connect() error {
+	if c.innerWriter != nil {
+		c.innerWriter.Close()
+		c.innerWriter = nil
+	}
+
+	conn, err := net.Dial(c.net, c.addr)
+	if err != nil {
+		return err
+	}
+
+	tcpConn, ok := conn.(*net.TCPConn)
+	if ok {
+		tcpConn.SetKeepAlive(true)
+	}
+
+	c.innerWriter = conn
+	c.lg = log.New(conn, "", log.Ldate|log.Ltime)
+	return nil
+}
+
+func (c *ConnWriter) neddedConnectOnMsg() bool {
+	if c.reconnect {
+		c.reconnect = false
+		return true
+	}
+
+	if c.innerWriter == nil {
+		return true
+	}
+
+	return c.reconnectOnMsg
+}
+
+func init() {
+	Register("conn", NewConn)
+}