This repository has been archived by the owner on Jan 5, 2019. It is now read-only.
forked from emicklei/proto
/
visitor.go
88 lines (80 loc) · 3.42 KB
/
visitor.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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
// Copyright (c) 2017 Ernest Micklei
//
// MIT License
//
// Permission is hereby granted, free of charge, to any person obtaining
// a copy of this software and associated documentation files (the
// "Software"), to deal in the Software without restriction, including
// without limitation the rights to use, copy, modify, merge, publish,
// distribute, sublicense, and/or sell copies of the Software, and to
// permit persons to whom the Software is furnished to do so, subject to
// the following conditions:
//
// The above copyright notice and this permission notice shall be
// included in all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
// EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
// NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
// LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
// OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
// WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
package proto
// Visitor is for dispatching Proto elements.
type Visitor interface {
VisitMessage(m *Message)
VisitService(v *Service)
VisitSyntax(s *Syntax)
VisitPackage(p *Package)
VisitOption(o *Option)
VisitImport(i *Import)
VisitNormalField(i *NormalField)
VisitEnumField(i *EnumField)
VisitEnum(e *Enum)
VisitComment(e *Comment)
VisitOneof(o *Oneof)
VisitOneofField(o *OneOfField)
VisitReserved(r *Reserved)
VisitRPC(r *RPC)
VisitMapField(f *MapField)
// proto2
VisitGroup(g *Group)
VisitExtensions(e *Extensions)
}
// Visitee is implemented by all Proto elements.
type Visitee interface {
Accept(v Visitor)
}
// Documented is for types that may have an associated comment (not inlined).
type Documented interface {
Doc() *Comment
}
// reflector is a Visitor that can tell the short type name of a Visitee.
type reflector struct {
name string
}
// sole instance of reflector
var namer = new(reflector)
func (r *reflector) VisitMessage(m *Message) { r.name = "Message" }
func (r *reflector) VisitService(v *Service) { r.name = "Service" }
func (r *reflector) VisitSyntax(s *Syntax) { r.name = "Syntax" }
func (r *reflector) VisitPackage(p *Package) { r.name = "Package" }
func (r *reflector) VisitOption(o *Option) { r.name = "Option" }
func (r *reflector) VisitImport(i *Import) { r.name = "Import" }
func (r *reflector) VisitNormalField(i *NormalField) { r.name = "NormalField" }
func (r *reflector) VisitEnumField(i *EnumField) { r.name = "EnumField" }
func (r *reflector) VisitEnum(e *Enum) { r.name = "Enum" }
func (r *reflector) VisitComment(e *Comment) { r.name = "Comment" }
func (r *reflector) VisitOneof(o *Oneof) { r.name = "Oneof" }
func (r *reflector) VisitOneofField(o *OneOfField) { r.name = "OneOfField" }
func (r *reflector) VisitReserved(rs *Reserved) { r.name = "Reserved" }
func (r *reflector) VisitRPC(rpc *RPC) { r.name = "RPC" }
func (r *reflector) VisitMapField(f *MapField) { r.name = "MapField" }
func (r *reflector) VisitGroup(g *Group) { r.name = "Group" }
func (r *reflector) VisitExtensions(e *Extensions) { r.name = "Extensions" }
// nameOfVisitee returns the short type name of a Visitee.
func nameOfVisitee(e Visitee) string {
e.Accept(namer)
return namer.name
}