diff options
Diffstat (limited to 'generateFind.go')
| -rw-r--r-- | generateFind.go | 17 |
1 files changed, 11 insertions, 6 deletions
diff --git a/generateFind.go b/generateFind.go index d1f660b..7d8637d 100644 --- a/generateFind.go +++ b/generateFind.go @@ -18,9 +18,12 @@ func (msg *MsgName) getLockname(s string) string { return msg.Lockname } -func generateFindBy(w io.Writer, FRUIT, APPLES, APPLE, COLOR, LOCK string) { - fmt.Fprintln(w, "// lookup a", FRUIT, "by the ", COLOR) - fmt.Fprintln(w, "func (x *"+FRUIT+") FindBy"+COLOR+"(s string) *"+APPLE+" {") +func (msg *MsgName) generateFindBy(w io.Writer, FUNCNAME, STRUCT, VARNAME, VARTYPE, COLOR string) string { + LOCK := msg.getLockname("x") + funcdef := "func (x *" + STRUCT + ") " + FUNCNAME + "(s string) *" + VARTYPE + + fmt.Fprintln(w, "// lookup a", STRUCT, "by the ", COLOR) + fmt.Fprintln(w, funcdef, "{") fmt.Fprintln(w, " if x == nil {") fmt.Fprintln(w, " return nil") fmt.Fprintln(w, " }") @@ -28,14 +31,16 @@ func generateFindBy(w io.Writer, FRUIT, APPLES, APPLE, COLOR, LOCK string) { fmt.Fprintln(w, " "+LOCK+".RLock()") fmt.Fprintln(w, " defer "+LOCK+".RUnlock()") fmt.Fprintln(w, "") - fmt.Fprintln(w, " for i, _ := range x."+APPLES+" {") - fmt.Fprintln(w, " if x."+APPLES+"[i]."+COLOR+" == s {") - fmt.Fprintln(w, " return x."+APPLES+"[i]") + fmt.Fprintln(w, " for i, _ := range x."+VARNAME+" {") + fmt.Fprintln(w, " if x."+VARNAME+"[i]."+COLOR+" == s {") + fmt.Fprintln(w, " return x."+VARNAME+"[i]") fmt.Fprintln(w, " }") fmt.Fprintln(w, " }") fmt.Fprintln(w, " return nil") fmt.Fprintln(w, "}") fmt.Fprintln(w, "") + + return funcdef } func (msg *MsgName) deleteBy(w io.Writer, FRUIT, APPLES, APPLE, COLOR, FUNCNAME, VARNAME string) string { |