log/修改包名 jsonlog->logjson; 修改 logjson.config 的初始化方式;
parent
625f12a793
commit
1840681201
@ -0,0 +1,98 @@
|
|||||||
|
package logjson
|
||||||
|
|
||||||
|
import (
|
||||||
|
"io"
|
||||||
|
"os"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"git.blauwelle.com/go/crate/synchelper"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Option 配置日志处理对象
|
||||||
|
type Option interface {
|
||||||
|
apply(cfg *config)
|
||||||
|
}
|
||||||
|
|
||||||
|
// WithBufferPool 配置缓冲池
|
||||||
|
func WithBufferPool(pool synchelper.BytesBufferPool) Option {
|
||||||
|
return optionFunc(func(cfg *config) {
|
||||||
|
cfg.pool = pool
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// WithOutput 配置输出
|
||||||
|
func WithOutput(w io.Writer) Option {
|
||||||
|
return optionFunc(func(cfg *config) {
|
||||||
|
cfg.output = w
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// WithTimeFormat 配置时间格式
|
||||||
|
func WithTimeFormat(format string) Option {
|
||||||
|
return optionFunc(func(cfg *config) {
|
||||||
|
cfg.timestampFormat = format
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// WithDisableTime 配置仅用时间输出
|
||||||
|
func WithDisableTime(disable bool) Option {
|
||||||
|
return optionFunc(func(cfg *config) {
|
||||||
|
cfg.disableTime = disable
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// WithDisableHTMLEscape 配置禁止 HTML 转义
|
||||||
|
func WithDisableHTMLEscape(disable bool) Option {
|
||||||
|
return optionFunc(func(cfg *config) {
|
||||||
|
cfg.disableHTMLEscape = disable
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// WithPrettyPrint 配置 JSON 多行缩进输出
|
||||||
|
func WithPrettyPrint(pretty bool) Option {
|
||||||
|
return optionFunc(func(cfg *config) {
|
||||||
|
cfg.prettyPrint = pretty
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func newConfig(opts ...Option) *config {
|
||||||
|
cfg := defaultConfig()
|
||||||
|
for _, opt := range opts {
|
||||||
|
opt.apply(cfg)
|
||||||
|
}
|
||||||
|
if !cfg.hasPool {
|
||||||
|
cfg.pool = synchelper.NewBytesBufferPool(512, 4096)
|
||||||
|
}
|
||||||
|
if cfg.output == nil {
|
||||||
|
cfg.output = synchelper.NewSyncWriter(os.Stderr)
|
||||||
|
}
|
||||||
|
if cfg.timestampFormat == "" {
|
||||||
|
cfg.timestampFormat = time.RFC3339Nano
|
||||||
|
}
|
||||||
|
return cfg
|
||||||
|
}
|
||||||
|
|
||||||
|
func defaultConfig() *config {
|
||||||
|
return &config{
|
||||||
|
hasPool: false,
|
||||||
|
disableTime: false,
|
||||||
|
disableHTMLEscape: false,
|
||||||
|
prettyPrint: false,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
type config struct {
|
||||||
|
pool synchelper.BytesBufferPool
|
||||||
|
output io.Writer
|
||||||
|
timestampFormat string
|
||||||
|
hasPool bool
|
||||||
|
disableTime bool
|
||||||
|
disableHTMLEscape bool
|
||||||
|
prettyPrint bool
|
||||||
|
}
|
||||||
|
|
||||||
|
type optionFunc func(cfg *config)
|
||||||
|
|
||||||
|
func (fn optionFunc) apply(cfg *config) {
|
||||||
|
fn(cfg)
|
||||||
|
}
|
Loading…
Reference in New Issue