update

Hunter0x7c7

2022-08-11 b8230139fb40edea387617b6accd8371e37eda58

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
//go:build !confonly 
// +build !confonly 
 
package command 
 
//go:generate go run github.com/v2fly/v2ray-core/v5/common/errors/errorgen 
 
import ( 
    "context" 
 
    "github.com/golang/protobuf/proto" 
    "google.golang.org/grpc" 
 
    core "github.com/v2fly/v2ray-core/v5" 
    "github.com/v2fly/v2ray-core/v5/app/observatory" 
    "github.com/v2fly/v2ray-core/v5/common" 
    "github.com/v2fly/v2ray-core/v5/features" 
    "github.com/v2fly/v2ray-core/v5/features/extension" 
) 
 
type service struct { 
    UnimplementedObservatoryServiceServer 
    v *core.Instance 
 
    observatory extension.Observatory 
} 
 
func (s *service) GetOutboundStatus(ctx context.Context, request *GetOutboundStatusRequest) (*GetOutboundStatusResponse, error) { 
    var result proto.Message 
    if request.Tag == "" { 
        observeResult, err := s.observatory.GetObservation(ctx) 
        if err != nil { 
            return nil, newError("cannot get observation").Base(err) 
        } 
        result = observeResult 
    } else { 
        fet, err := s.observatory.(features.TaggedFeatures).GetFeaturesByTag(request.Tag) 
        if err != nil { 
            return nil, newError("cannot get tagged observatory").Base(err) 
        } 
        observeResult, err := fet.(extension.Observatory).GetObservation(ctx) 
        if err != nil { 
            return nil, newError("cannot get observation").Base(err) 
        } 
        result = observeResult 
    } 
    retdata := result.(*observatory.ObservationResult) 
    return &GetOutboundStatusResponse{ 
        Status: retdata, 
    }, nil 
} 
 
func (s *service) Register(server *grpc.Server) { 
    RegisterObservatoryServiceServer(server, s) 
} 
 
func init() { 
    common.Must(common.RegisterConfig((*Config)(nil), func(ctx context.Context, cfg interface{}) (interface{}, error) { 
        s := core.MustFromContext(ctx) 
        sv := &service{v: s} 
        err := s.RequireFeatures(func(Observatory extension.Observatory) { 
            sv.observatory = Observatory 
        }) 
        if err != nil { 
            return nil, err 
        } 
        return sv, nil 
    })) 
}