-
Notifications
You must be signed in to change notification settings - Fork 0
/
template.go
82 lines (70 loc) · 3.04 KB
/
template.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
package main
const (
goTemplate = `
// Code generated by protoc-gen-go-methodoptions. DO NOT EDIT.
// source file: {{ .FileName }}
// proto package: {{ .Package }}
package {{ .GoPackage }}
{{ if .Imports }}
import (
{{- range $alias, $path := .Imports }}
{{ $alias }} "{{ $path }}"
{{- end }}
)
{{ end }}
{{ if .Services }}
{{ range $service := .Services }}
{{ if $service.Methods }}
// {{ $service.Name }}
const (
{{- range $method := $service.Methods }}
{{- range $key, $option := $method.Options }}
{{- if eq $option.Type "enum" }}
{{ $service.Name }}_{{ $method.Name }}_{{ $key }} = {{ $option.Package }}.{{ $key }}_{{ $option.Value }}
{{- else if eq $option.Type "string" }}
{{ $service.Name }}_{{ $method.Name }}_{{ $key }} = "{{ $option.Value }}"
{{- else if eq $option.Type "int32" }}
{{ $service.Name }}_{{ $method.Name }}_{{ $key }} = {{ $option.Value }}
{{- else }}
// {{ $option.Type }} is not currently supported ({{ $service.Name }}_{{ $method.Name }}_{{ $key }} = {{ $option.Value }})
{{- end }}
{{- end }}
{{- end}}
)
{{ end }}
{{ range $method := $service.Methods }}
{{- range $key, $option := $method.Options }}
{{ if eq $option.Type "enum" }}
// {{ $method.Name }}{{ $option.Name }} returns the {{ $option.Name }} for {{ $method.Name }}.
func (Unimplemented{{ $service.Name }}Server) {{ $method.Name }}{{ $option.Name }}() {{ $option.Package }}.{{ $option.Name }} {
return {{ $service.Name }}_{{ $method.Name }}_{{ $key }}
}
// {{ $option.Name }} returns the {{ $option.Name }} for {{ $method.Name }} request.
func (*{{ if $method.RequestPackage }}{{ $method.RequestPackage }}.{{ end }}{{ $method.RequestName }}) {{ $option.Name }}() {{ $option.Package }}.{{ $option.Name }} {
return {{ $service.Name }}_{{ $method.Name }}_{{ $key }}
}
{{- else if eq $option.Type "string" }}
// {{ $method.Name }}{{ $option.Name }} returns the {{ $option.Name }} for {{ $method.Name }}.
func (Unimplemented{{ $service.Name }}Server) {{ $method.Name }}{{ $option.Name }}() {{ $option.Type }} {
return {{ $service.Name }}_{{ $method.Name }}_{{ $key }}
}
// {{ $option.Name }} returns the {{ $option.Name }} for {{ $method.Name }} request.
func (*{{ if $method.RequestPackage }}{{ $method.RequestPackage }}.{{ end }}{{ $method.RequestName }}) {{ $option.Name }}() {{ $option.Type }} {
return {{ $service.Name }}_{{ $method.Name }}_{{ $key }}
}
{{- else if eq $option.Type "int32" }}
// {{ $method.Name }}{{ $option.Name }} returns the {{ $option.Name }} for {{ $method.Name }}.
func (Unimplemented{{ $service.Name }}Server) {{ $method.Name }}{{ $option.Name }}() {{ $option.Type }} {
return {{ $service.Name }}_{{ $method.Name }}_{{ $key }}
}
// {{ $option.Name }} returns the {{ $option.Name }} for {{ $method.Name }} request.
func (*{{ if $method.RequestPackage }}{{ $method.RequestPackage }}.{{ end }}{{ $method.RequestName }}) {{ $option.Name }}() {{ $option.Type }} {
return {{ $service.Name }}_{{ $method.Name }}_{{ $key }}
}
{{- end }}
{{- end }}
{{ end }}
{{ end }}
{{ end }}
`
)