git add

Hunter0x7c7

2022-08-11 a82f9cb69f63aaeba40c024960deda7d75b9fece

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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
package command 
 
//go:generate go run github.com/v2fly/v2ray-core/v5/common/errors/errorgen 
 
import ( 
    "context" 
 
    grpc "google.golang.org/grpc" 
 
    core "github.com/v2fly/v2ray-core/v5" 
    "github.com/v2fly/v2ray-core/v5/app/log" 
    "github.com/v2fly/v2ray-core/v5/common" 
    cmlog "github.com/v2fly/v2ray-core/v5/common/log" 
) 
 
// LoggerServer is the implemention of LoggerService 
type LoggerServer struct { 
    V *core.Instance 
} 
 
// RestartLogger implements LoggerService. 
func (s *LoggerServer) RestartLogger(ctx context.Context, request *RestartLoggerRequest) (*RestartLoggerResponse, error) { 
    logger := s.V.GetFeature((*log.Instance)(nil)) 
    if logger == nil { 
        return nil, newError("unable to get logger instance") 
    } 
    if err := logger.Close(); err != nil { 
        return nil, newError("failed to close logger").Base(err) 
    } 
    if err := logger.Start(); err != nil { 
        return nil, newError("failed to start logger").Base(err) 
    } 
    return &RestartLoggerResponse{}, nil 
} 
 
// FollowLog implements LoggerService. 
func (s *LoggerServer) FollowLog(_ *FollowLogRequest, stream LoggerService_FollowLogServer) error { 
    logger := s.V.GetFeature((*log.Instance)(nil)) 
    if logger == nil { 
        return newError("unable to get logger instance") 
    } 
    follower, ok := logger.(cmlog.Follower) 
    if !ok { 
        return newError("logger not support following") 
    } 
    done := make(chan struct{}) 
    f := func(msg cmlog.Message) { 
        err := stream.Send(&FollowLogResponse{ 
            Message: msg.String(), 
        }) 
        if err != nil { 
            close(done) 
        } 
    } 
    follower.AddFollower(f) 
    defer follower.RemoveFollower(f) 
    <-done 
    return nil 
} 
 
func (s *LoggerServer) mustEmbedUnimplementedLoggerServiceServer() {} 
 
type service struct { 
    v *core.Instance 
} 
 
func (s *service) Register(server *grpc.Server) { 
    RegisterLoggerServiceServer(server, &LoggerServer{ 
        V: s.v, 
    }) 
} 
 
func init() { 
    common.Must(common.RegisterConfig((*Config)(nil), func(ctx context.Context, cfg interface{}) (interface{}, error) { 
        s := core.MustFromContext(ctx) 
        return &service{v: s}, nil 
    })) 
}