69 lines
1.5 KiB
Go
69 lines
1.5 KiB
Go
|
package log
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"fmt"
|
||
|
|
"strconv"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"go-common/library/conf/dsn"
|
||
|
|
|
||
|
|
"github.com/pkg/errors"
|
||
|
|
)
|
||
|
|
|
||
|
|
type verboseModule map[string]int32
|
||
|
|
|
||
|
|
type logFilter []string
|
||
|
|
|
||
|
|
func (f *logFilter) String() string {
|
||
|
|
return fmt.Sprint(*f)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Set sets the value of the named command-line flag.
|
||
|
|
// format: -log.filter key1,key2
|
||
|
|
func (f *logFilter) Set(value string) error {
|
||
|
|
for _, i := range strings.Split(value, ",") {
|
||
|
|
*f = append(*f, strings.TrimSpace(i))
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m verboseModule) String() string {
|
||
|
|
// FIXME strings.Builder
|
||
|
|
var buf bytes.Buffer
|
||
|
|
for k, v := range m {
|
||
|
|
buf.WriteString(k)
|
||
|
|
buf.WriteString(strconv.FormatInt(int64(v), 10))
|
||
|
|
buf.WriteString(",")
|
||
|
|
}
|
||
|
|
return buf.String()
|
||
|
|
}
|
||
|
|
|
||
|
|
// Set sets the value of the named command-line flag.
|
||
|
|
// format: -log.module file=1,file2=2
|
||
|
|
func (m verboseModule) Set(value string) error {
|
||
|
|
for _, i := range strings.Split(value, ",") {
|
||
|
|
kv := strings.Split(i, "=")
|
||
|
|
if len(kv) == 2 {
|
||
|
|
if v, err := strconv.ParseInt(kv[1], 10, 64); err == nil {
|
||
|
|
m[strings.TrimSpace(kv[0])] = int32(v)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// parseDSN parse log agent dsn.
|
||
|
|
// unixgram:///var/run/lancer/collector.sock?timeout=100ms&chan=1024
|
||
|
|
func parseDSN(rawdsn string) *AgentConfig {
|
||
|
|
ac := new(AgentConfig)
|
||
|
|
d, err := dsn.Parse(rawdsn)
|
||
|
|
if err != nil {
|
||
|
|
panic(errors.WithMessage(err, fmt.Sprintf("log: invalid dsn: %s", rawdsn)))
|
||
|
|
}
|
||
|
|
if _, err = d.Bind(ac); err != nil {
|
||
|
|
panic(errors.WithMessage(err, fmt.Sprintf("log: invalid dsn: %s", rawdsn)))
|
||
|
|
}
|
||
|
|
return ac
|
||
|
|
}
|