frp/vendor/github.com/astaxie/beego/logs/conn.go

118 lines
2.5 KiB
Go
Raw Normal View History

2016-01-27 21:24:36 +08:00
// Copyright 2014 beego Author. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package logs
import (
"encoding/json"
"io"
"net"
2016-08-11 16:32:05 +08:00
"time"
2016-01-27 21:24:36 +08:00
)
2016-08-11 16:32:05 +08:00
// connWriter implements LoggerInterface.
2016-01-27 21:24:36 +08:00
// it writes messages in keep-live tcp connection.
2016-08-11 16:32:05 +08:00
type connWriter struct {
lg *logWriter
2016-01-27 21:24:36 +08:00
innerWriter io.WriteCloser
ReconnectOnMsg bool `json:"reconnectOnMsg"`
Reconnect bool `json:"reconnect"`
Net string `json:"net"`
Addr string `json:"addr"`
Level int `json:"level"`
}
2016-08-11 16:32:05 +08:00
// NewConn create new ConnWrite returning as LoggerInterface.
func NewConn() Logger {
conn := new(connWriter)
2016-01-27 21:24:36 +08:00
conn.Level = LevelTrace
return conn
}
2016-08-11 16:32:05 +08:00
// Init init connection writer with json config.
2016-01-27 21:24:36 +08:00
// json config only need key "level".
2016-08-11 16:32:05 +08:00
func (c *connWriter) Init(jsonConfig string) error {
return json.Unmarshal([]byte(jsonConfig), c)
2016-01-27 21:24:36 +08:00
}
2016-08-11 16:32:05 +08:00
// WriteMsg write message in connection.
2016-01-27 21:24:36 +08:00
// if connection is down, try to re-connect.
2016-08-11 16:32:05 +08:00
func (c *connWriter) WriteMsg(when time.Time, msg string, level int) error {
2016-01-27 21:24:36 +08:00
if level > c.Level {
return nil
}
2016-08-11 16:32:05 +08:00
if c.needToConnectOnMsg() {
2016-01-27 21:24:36 +08:00
err := c.connect()
if err != nil {
return err
}
}
if c.ReconnectOnMsg {
defer c.innerWriter.Close()
}
2016-08-11 16:32:05 +08:00
c.lg.println(when, msg)
2016-01-27 21:24:36 +08:00
return nil
}
2016-08-11 16:32:05 +08:00
// Flush implementing method. empty.
func (c *connWriter) Flush() {
2016-01-27 21:24:36 +08:00
}
2016-08-11 16:32:05 +08:00
// Destroy destroy connection writer and close tcp listener.
func (c *connWriter) Destroy() {
2016-01-27 21:24:36 +08:00
if c.innerWriter != nil {
c.innerWriter.Close()
}
}
2016-08-11 16:32:05 +08:00
func (c *connWriter) connect() error {
2016-01-27 21:24:36 +08:00
if c.innerWriter != nil {
c.innerWriter.Close()
c.innerWriter = nil
}
conn, err := net.Dial(c.Net, c.Addr)
if err != nil {
return err
}
if tcpConn, ok := conn.(*net.TCPConn); ok {
tcpConn.SetKeepAlive(true)
}
c.innerWriter = conn
2016-08-11 16:32:05 +08:00
c.lg = newLogWriter(conn)
2016-01-27 21:24:36 +08:00
return nil
}
2016-08-11 16:32:05 +08:00
func (c *connWriter) needToConnectOnMsg() bool {
2016-01-27 21:24:36 +08:00
if c.Reconnect {
c.Reconnect = false
return true
}
if c.innerWriter == nil {
return true
}
return c.ReconnectOnMsg
}
func init() {
Register(AdapterConn, NewConn)
2016-01-27 21:24:36 +08:00
}