/
writer.go
64 lines (54 loc) · 1.23 KB
/
writer.go
1
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
package grpc
import (
"context"
"github.com/sparrow-community/plugins/v4/logger/grpc/proto"
"go-micro.dev/v4/client"
"go-micro.dev/v4/logger"
"sync"
"sync/atomic"
"syscall"
)
type ZapGrpcWriter struct {
serviceName string
client proto.LoggerService
closed int32
closeMutex sync.Mutex
}
func (g *ZapGrpcWriter) Write(msg []byte) (n int, err error) {
rsp, err := g.client.Write(context.Background(), &proto.WriteRequest{
ServiceName: g.serviceName,
Data: msg,
})
if err != nil {
return 0, err
}
return int(rsp.N), nil
}
func (g *ZapGrpcWriter) Sync() error {
return nil
}
func (g *ZapGrpcWriter) Close() error {
g.closeMutex.Lock()
defer g.closeMutex.Unlock()
if g.Closed() {
return syscall.EINVAL
}
atomic.StoreInt32(&g.closed, 1)
return nil
}
func (g *ZapGrpcWriter) Closed() bool {
return atomic.LoadInt32(&g.closed) != 0
}
// InitializeLogger is initialize logger service
func InitializeLogger(serviceName string) {
cs := client.DefaultClient
l, err := NewLogger(
WithServiceNameKey(serviceName),
WithClientKey(proto.NewLoggerService("github.com.sparrow-community.logger-service", cs)),
)
if nil != err {
logger.Error("logger service error: ", err)
} else {
logger.DefaultLogger = l
}
}