@@ -52,22 +52,22 @@ type (
52
52
53
53
var (
54
54
services = []service {
55
- service {
55
+ {
56
56
name : "frontend" ,
57
57
clientType : reflect .TypeOf ((* workflowservice .WorkflowServiceClient )(nil )),
58
58
clientGenerator : generateFrontendOrAdminClient ,
59
59
},
60
- service {
60
+ {
61
61
name : "admin" ,
62
62
clientType : reflect .TypeOf ((* adminservice .AdminServiceClient )(nil )),
63
63
clientGenerator : generateFrontendOrAdminClient ,
64
64
},
65
- service {
65
+ {
66
66
name : "history" ,
67
67
clientType : reflect .TypeOf ((* historyservice .HistoryServiceClient )(nil )),
68
68
clientGenerator : generateHistoryClient ,
69
69
},
70
- service {
70
+ {
71
71
name : "matching" ,
72
72
clientType : reflect .TypeOf ((* matchingservice .MatchingServiceClient )(nil )),
73
73
clientGenerator : generateMatchingClient ,
@@ -102,12 +102,17 @@ var (
102
102
}
103
103
)
104
104
105
+ func panicIfErr (err error ) {
106
+ if err != nil {
107
+ panic (err )
108
+ }
109
+ }
110
+
105
111
func writeTemplatedCode (w io.Writer , service service , text string ) {
106
- t := template .Must (template .New ("code" ).Parse (text ))
107
- t .Execute (w , map [string ]string {
112
+ panicIfErr (template .Must (template .New ("code" ).Parse (text )).Execute (w , map [string ]string {
108
113
"ServiceName" : service .name ,
109
114
"ServicePackagePath" : service .clientType .Elem ().PkgPath (),
110
- })
115
+ }))
111
116
}
112
117
113
118
func pathToField (t reflect.Type , name string , path string , maxDepth int ) string {
@@ -231,8 +236,7 @@ func writeTemplatedMethod(w io.Writer, service service, impl string, m reflect.M
231
236
}
232
237
}
233
238
234
- t := template .Must (template .New ("code" ).Parse (text ))
235
- t .Execute (w , fields )
239
+ panicIfErr (template .Must (template .New ("code" ).Parse (text )).Execute (w , fields ))
236
240
}
237
241
238
242
func writeTemplatedMethods (w io.Writer , service service , impl string , text string ) {
@@ -412,12 +416,13 @@ func callWithFile(f func(io.Writer, service), service service, filename string,
412
416
if err != nil {
413
417
panic (err )
414
418
}
415
- fmt . Fprintf ( w , "%s \n // Code generated by cmd/tools/rpcwrappers. DO NOT EDIT. \n " , licenseText )
416
- f ( w , service )
417
- err = w . Close ()
418
- if err != nil {
419
+ defer func () {
420
+ panicIfErr ( w . Close () )
421
+ } ()
422
+ if _ , err := fmt . Fprintf ( w , "%s \n // Code generated by cmd/tools/rpcwrappers. DO NOT EDIT. \n " , licenseText ); err != nil {
419
423
panic (err )
420
424
}
425
+ f (w , service )
421
426
}
422
427
423
428
func readLicenseFile (path string ) string {
0 commit comments