mirror of
https://github.com/astaxie/beego.git
synced 2024-12-25 20:40:49 +00:00
177 lines
4.4 KiB
Go
177 lines
4.4 KiB
Go
// 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 (
|
|
"io"
|
|
"runtime"
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
type logWriter struct {
|
|
sync.Mutex
|
|
writer io.Writer
|
|
}
|
|
|
|
func newLogWriter(wr io.Writer) *logWriter {
|
|
return &logWriter{writer: wr}
|
|
}
|
|
|
|
func (lg *logWriter) writeln(when time.Time, msg string) (int, error) {
|
|
lg.Lock()
|
|
h, _, _ := formatTimeHeader(when)
|
|
n, err := lg.writer.Write(append(append(h, msg...), '\n'))
|
|
lg.Unlock()
|
|
return n, err
|
|
}
|
|
|
|
const (
|
|
y1 = `0123456789`
|
|
y2 = `0123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789`
|
|
y3 = `0000000000111111111122222222223333333333444444444455555555556666666666777777777788888888889999999999`
|
|
y4 = `0123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789`
|
|
mo1 = `000000000111`
|
|
mo2 = `123456789012`
|
|
d1 = `0000000001111111111222222222233`
|
|
d2 = `1234567890123456789012345678901`
|
|
h1 = `000000000011111111112222`
|
|
h2 = `012345678901234567890123`
|
|
mi1 = `000000000011111111112222222222333333333344444444445555555555`
|
|
mi2 = `012345678901234567890123456789012345678901234567890123456789`
|
|
s1 = `000000000011111111112222222222333333333344444444445555555555`
|
|
s2 = `012345678901234567890123456789012345678901234567890123456789`
|
|
ns1 = `0123456789`
|
|
)
|
|
|
|
func formatTimeHeader(when time.Time) ([]byte, int, int) {
|
|
y, mo, d := when.Date()
|
|
h, mi, s := when.Clock()
|
|
ns := when.Nanosecond() / 1000000
|
|
//len("2006/01/02 15:04:05.123 ")==24
|
|
var buf [24]byte
|
|
|
|
buf[0] = y1[y/1000%10]
|
|
buf[1] = y2[y/100]
|
|
buf[2] = y3[y-y/100*100]
|
|
buf[3] = y4[y-y/100*100]
|
|
buf[4] = '/'
|
|
buf[5] = mo1[mo-1]
|
|
buf[6] = mo2[mo-1]
|
|
buf[7] = '/'
|
|
buf[8] = d1[d-1]
|
|
buf[9] = d2[d-1]
|
|
buf[10] = ' '
|
|
buf[11] = h1[h]
|
|
buf[12] = h2[h]
|
|
buf[13] = ':'
|
|
buf[14] = mi1[mi]
|
|
buf[15] = mi2[mi]
|
|
buf[16] = ':'
|
|
buf[17] = s1[s]
|
|
buf[18] = s2[s]
|
|
buf[19] = '.'
|
|
buf[20] = ns1[ns/100]
|
|
buf[21] = ns1[ns%100/10]
|
|
buf[22] = ns1[ns%10]
|
|
|
|
buf[23] = ' '
|
|
|
|
return buf[0:], d, h
|
|
}
|
|
|
|
var (
|
|
green = string([]byte{27, 91, 57, 55, 59, 52, 50, 109})
|
|
white = string([]byte{27, 91, 57, 48, 59, 52, 55, 109})
|
|
yellow = string([]byte{27, 91, 57, 55, 59, 52, 51, 109})
|
|
red = string([]byte{27, 91, 57, 55, 59, 52, 49, 109})
|
|
blue = string([]byte{27, 91, 57, 55, 59, 52, 52, 109})
|
|
magenta = string([]byte{27, 91, 57, 55, 59, 52, 53, 109})
|
|
cyan = string([]byte{27, 91, 57, 55, 59, 52, 54, 109})
|
|
|
|
w32Green = string([]byte{27, 91, 52, 50, 109})
|
|
w32White = string([]byte{27, 91, 52, 55, 109})
|
|
w32Yellow = string([]byte{27, 91, 52, 51, 109})
|
|
w32Red = string([]byte{27, 91, 52, 49, 109})
|
|
w32Blue = string([]byte{27, 91, 52, 52, 109})
|
|
w32Magenta = string([]byte{27, 91, 52, 53, 109})
|
|
w32Cyan = string([]byte{27, 91, 52, 54, 109})
|
|
|
|
reset = string([]byte{27, 91, 48, 109})
|
|
)
|
|
|
|
var once sync.Once
|
|
var colorMap map[string]string
|
|
|
|
func initColor() {
|
|
if runtime.GOOS == "windows" {
|
|
green = w32Green
|
|
white = w32White
|
|
yellow = w32Yellow
|
|
red = w32Red
|
|
blue = w32Blue
|
|
magenta = w32Magenta
|
|
cyan = w32Cyan
|
|
}
|
|
colorMap = map[string]string{
|
|
//by color
|
|
"green": green,
|
|
"white": white,
|
|
"yellow": yellow,
|
|
"red": red,
|
|
//by method
|
|
"GET": blue,
|
|
"POST": cyan,
|
|
"PUT": yellow,
|
|
"DELETE": red,
|
|
"PATCH": green,
|
|
"HEAD": magenta,
|
|
"OPTIONS": white,
|
|
}
|
|
}
|
|
|
|
// ColorByStatus return color by http code
|
|
// 2xx return Green
|
|
// 3xx return White
|
|
// 4xx return Yellow
|
|
// 5xx return Red
|
|
func ColorByStatus(code int) string {
|
|
once.Do(initColor)
|
|
switch {
|
|
case code >= 200 && code < 300:
|
|
return colorMap["green"]
|
|
case code >= 300 && code < 400:
|
|
return colorMap["white"]
|
|
case code >= 400 && code < 500:
|
|
return colorMap["yellow"]
|
|
default:
|
|
return colorMap["red"]
|
|
}
|
|
}
|
|
|
|
// ColorByMethod return color by http code
|
|
func ColorByMethod(method string) string {
|
|
once.Do(initColor)
|
|
if c := colorMap[method]; c != "" {
|
|
return c
|
|
}
|
|
return reset
|
|
}
|
|
|
|
// ResetColor return reset color
|
|
func ResetColor() string {
|
|
return reset
|
|
}
|