Blame view
logger/main.go
2.98 KB
28a6cc60c
Integrated logger...
|
1 |
package logger |
|
e2880d3fb
integraded golloger
|
2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 |
import (
"bytes"
"encoding/json"
"fmt"
"os"
"path/filepath"
"strings"
"sync"
"time"
)
const dateTimeFormat = "2006-01-02 15:04:05"
// Block ...
const (
MaxLogSize5MB int64 = 5 * 1024 * 1024
MaxLogSize1MB int64 = 1 * 1024 * 1024
MaxLogSize500KB int64 = 500 * 1024
MaxLogSize100KB int64 = 100 * 1024
MaxLogSize512B int64 = 512
)
// Logger ...
type Logger struct {
mu *sync.Mutex
outputFile *os.File
outputFileName string
fullName string
maxFileSize int64
splitCount int
directory string
}
// New ...
func New(name, dir string, maxFileSize int64) (logger *Logger, err error) {
logger = &Logger{}
logger.outputFileName = name
logger.mu = &sync.Mutex{}
logger.maxFileSize = maxFileSize
logger.directory = dir
err = os.Mkdir(dir, os.ModePerm)
if err != nil {
if !os.IsExist(err) {
return nil, err
}
}
date := strings.Replace(time.Now().Format(dateTimeFormat), ":", ".", -1)
logger.outputFileName += " " + date
logger.fullName = logger.outputFileName + ".txt"
path := filepath.Join(dir, logger.fullName)
if logger.outputFile, err = os.Create(path); err != nil {
return nil, err
}
return logger, nil
}
// Log ...
func (l *Logger) Log(format string, v ...interface{}) {
if l.outputFile == nil {
return
}
l.mu.Lock()
defer l.mu.Unlock()
msg := fmt.Sprintf(format, v...)
s := time.Now().Format(dateTimeFormat) + ": " + msg + "
"
if l.shouldSplit(len(s)) {
l.split()
}
l.outputFile.WriteString(s)
}
// Print ...
func (l *Logger) Print(format string, v ...interface{}) {
msg := fmt.Sprintf(format, v...)
fmt.Printf("%s: %s
", time.Now().Format(dateTimeFormat), msg)
}
// Trace ...
func (l *Logger) Trace(format string, v ...interface{}) string {
if l.outputFile == nil {
return ""
}
l.mu.Lock()
defer l.mu.Unlock()
s := getTrace(format, v...) + "
"
if l.shouldSplit(len(s)) {
l.split()
}
l.outputFile.WriteString(s)
return s
}
// PrintTrace ...
func (l *Logger) PrintTrace(format string, v ...interface{}) {
s := getTrace(format, v...)
fmt.Printf("%s
", s)
}
// PrintAndTrace ...
func (l *Logger) PrintAndTrace(format string, v ...interface{}) {
t := l.Trace(format, v...)
fmt.Println(t)
}
// Close ...
func (l *Logger) Close() error {
if l.outputFile == nil {
return nil
}
return l.outputFile.Close()
}
|
|
28a6cc60c
Integrated logger...
|
132 133 134 135 |
// GetOutDir ...
func (l *Logger) GetOutDir() string {
return l.directory
}
|
|
e2880d3fb
integraded golloger
|
136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 |
func (l *Logger) split() error {
if l.outputFile == nil {
return nil
}
// close old file
err := l.outputFile.Close()
if err != nil {
return err
}
// open new file
l.splitCount++
path := filepath.Join(l.directory, l.outputFileName+fmt.Sprintf("(%d)", l.splitCount)+".txt")
l.outputFile, err = os.Create(path)
return err
}
func (l *Logger) shouldSplit(nextEntrySize int) bool {
stats, _ := l.outputFile.Stat()
return int64(nextEntrySize) >= (l.maxFileSize - stats.Size())
}
func printJSON(in []byte) (out []byte, err error) {
var buf bytes.Buffer
err = json.Indent(&buf, in, "", " ")
return buf.Bytes(), err
}
|