mirror of
https://codeberg.org/forgejo/forgejo.git
synced 2024-11-09 03:11:51 +01:00
5422f23ed8
* Rename BaseLogger to WriterLogger to help the creation of other providers * Don't export ColorBytes and ResetBytes from ColoredValues * Make integration tests only print logs if they fail * check can color before coloring * I always forget about MSSQL * Oh and use LEVEL in sqlite.ini * Make the test logger log at info - as it means you see the router * Remove empty expected changes * Make the migrations quieter too * Don't display SQL on error - it can be looked at in the file logs if necessary * Fix skip when using onGiteaRun
121 lines
2.4 KiB
Go
121 lines
2.4 KiB
Go
// Copyright 2014 The Gogs Authors. All rights reserved.
|
|
// Copyright 2019 The Gitea Authors. All rights reserved.
|
|
// Use of this source code is governed by a MIT-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package log
|
|
|
|
import (
|
|
"encoding/json"
|
|
"io"
|
|
"net"
|
|
)
|
|
|
|
type connWriter struct {
|
|
innerWriter io.WriteCloser
|
|
ReconnectOnMsg bool `json:"reconnectOnMsg"`
|
|
Reconnect bool `json:"reconnect"`
|
|
Net string `json:"net"`
|
|
Addr string `json:"addr"`
|
|
}
|
|
|
|
// Close the inner writer
|
|
func (i *connWriter) Close() error {
|
|
if i.innerWriter != nil {
|
|
return i.innerWriter.Close()
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// Write the data to the connection
|
|
func (i *connWriter) Write(p []byte) (int, error) {
|
|
if i.neededConnectOnMsg() {
|
|
if err := i.connect(); err != nil {
|
|
return 0, err
|
|
}
|
|
}
|
|
|
|
if i.ReconnectOnMsg {
|
|
defer i.innerWriter.Close()
|
|
}
|
|
|
|
return i.innerWriter.Write(p)
|
|
}
|
|
|
|
func (i *connWriter) neededConnectOnMsg() bool {
|
|
if i.Reconnect {
|
|
i.Reconnect = false
|
|
return true
|
|
}
|
|
|
|
if i.innerWriter == nil {
|
|
return true
|
|
}
|
|
|
|
return i.ReconnectOnMsg
|
|
}
|
|
|
|
func (i *connWriter) connect() error {
|
|
if i.innerWriter != nil {
|
|
i.innerWriter.Close()
|
|
i.innerWriter = nil
|
|
}
|
|
|
|
conn, err := net.Dial(i.Net, i.Addr)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if tcpConn, ok := conn.(*net.TCPConn); ok {
|
|
tcpConn.SetKeepAlive(true)
|
|
}
|
|
|
|
i.innerWriter = conn
|
|
return nil
|
|
}
|
|
|
|
// ConnLogger implements LoggerProvider.
|
|
// it writes messages in keep-live tcp connection.
|
|
type ConnLogger struct {
|
|
WriterLogger
|
|
ReconnectOnMsg bool `json:"reconnectOnMsg"`
|
|
Reconnect bool `json:"reconnect"`
|
|
Net string `json:"net"`
|
|
Addr string `json:"addr"`
|
|
}
|
|
|
|
// NewConn creates new ConnLogger returning as LoggerProvider.
|
|
func NewConn() LoggerProvider {
|
|
conn := new(ConnLogger)
|
|
conn.Level = TRACE
|
|
return conn
|
|
}
|
|
|
|
// Init inits connection writer with json config.
|
|
// json config only need key "level".
|
|
func (log *ConnLogger) Init(jsonconfig string) error {
|
|
err := json.Unmarshal([]byte(jsonconfig), log)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
log.NewWriterLogger(&connWriter{
|
|
ReconnectOnMsg: log.ReconnectOnMsg,
|
|
Reconnect: log.Reconnect,
|
|
Net: log.Net,
|
|
Addr: log.Addr,
|
|
}, log.Level)
|
|
return nil
|
|
}
|
|
|
|
// Flush does nothing for this implementation
|
|
func (log *ConnLogger) Flush() {
|
|
}
|
|
|
|
// GetName returns the default name for this implementation
|
|
func (log *ConnLogger) GetName() string {
|
|
return "conn"
|
|
}
|
|
|
|
func init() {
|
|
Register("conn", NewConn)
|
|
}
|