Skip to content

Commit 7b23f03

Browse files
committed
use a helper in gocode (TryLog) instead, simplify gen
1 parent 251e4ff commit 7b23f03

File tree

3 files changed

+20
-31
lines changed

3 files changed

+20
-31
lines changed

generator/go_client.stoneg.py

Lines changed: 7 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,6 @@
1111
fmt_type,
1212
fmt_var,
1313
generate_doc,
14-
generate_if_else,
1514
)
1615

1716

@@ -99,12 +98,7 @@ def _generate_request(self, namespace, route):
9998

10099
body = 'nil'
101100
if not is_void_type(route.arg_data_type):
102-
with self.block('if dbx.Config.Verbose'):
103-
generate_if_else(self,
104-
'dbx.Config.Logger != nil',
105-
'dbx.Config.Logger.Printf("arg: %v", arg)',
106-
'log.Printf("arg: %v", arg)')
107-
out()
101+
out('dbx.Config.TryLog("arg: %v", arg)')
108102

109103
out('b, err := json.Marshal(arg)')
110104
with self.block('if err != nil'):
@@ -139,28 +133,21 @@ def _generate_request(self, namespace, route):
139133
with self.block('if err != nil'):
140134
out('return')
141135

142-
with self.block('if dbx.Config.Verbose'):
143-
generate_if_else(self,
144-
'dbx.Config.Logger != nil',
145-
'dbx.Config.Logger.Printf("req: %v", req)',
146-
'log.Printf("req: %v", req)')
136+
out('dbx.Config.TryLog("req: %v", req)')
137+
147138
out()
148139

149140
def _generate_post(self):
150141
out = self.emit
151142

152143
out('resp, err := cli.Do(req)')
153-
with self.block('if dbx.Config.Verbose'):
154-
generate_if_else(self,
155-
'dbx.Config.Logger != nil',
156-
'dbx.Config.Logger.Printf("resp: %v", resp)',
157-
'log.Printf("resp: %v", resp)')
158-
out()
159144

160145
with self.block('if err != nil'):
161146
out('return')
162147
out()
163148

149+
out('dbx.Config.TryLog("resp: %v", resp)')
150+
164151
def _generate_response(self, route):
165152
out = self.emit
166153
style = route.attrs.get('style', 'rpc')
@@ -173,12 +160,8 @@ def _generate_response(self, route):
173160
'if err != nil'):
174161
out('return')
175162
out()
176-
with self.block('if dbx.Config.Verbose'):
177-
generate_if_else(self,
178-
'dbx.Config.Logger != nil',
179-
'dbx.Config.Logger.Printf("body: %v", body)',
180-
'log.Printf("body: %v", body)')
181-
out()
163+
164+
out('dbx.Config.TryLog("body: %v", body)')
182165

183166
def _generate_error_handling(self, route):
184167
out = self.emit

generator/go_helpers.py

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -118,13 +118,6 @@ def generate_doc(code_generator, t):
118118
d = 'Package %s : %s' % (t.name, doc)
119119
code_generator.emit_wrapped_text(d, prefix='// ')
120120

121-
def generate_if_else(code_generator, ifcond, ifbody, elsebody):
122-
with code_generator.block('if %s' % ifcond, after=" else {"):
123-
code_generator.emit(ifbody)
124-
with code_generator.indent():
125-
code_generator.emit(elsebody)
126-
code_generator.emit('}')
127-
128121
def _needs_base_type(data_type):
129122
if is_struct_type(data_type) and data_type.has_enumerated_subtypes():
130123
return True

generator/go_rsrc/sdk.go

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -64,6 +64,19 @@ type Config struct {
6464
URLGenerator func(hostType string, style string, namespace string, route string) string
6565
}
6666

67+
// TryLog will, if Verbose is set, log to the config's logger
68+
// or the default log (stderr) if Config.Logger is nil.
69+
func (c *Config) TryLog(format string, v ...interface{}) {
70+
if !c.Verbose {
71+
return
72+
}
73+
if c.Logger != nil {
74+
c.Logger.Printf(format, v...)
75+
} else {
76+
log.Printf(format, v...)
77+
}
78+
}
79+
6780
// Context is the base client context used to implement per-namespace clients.
6881
type Context struct {
6982
Config Config

0 commit comments

Comments
 (0)