67 lines
1.9 KiB
Go
67 lines
1.9 KiB
Go
// Copyright 2017-2025 WIT.COM Inc. All rights reserved.
|
|
// Use of this source code is governed by the GPL 3.0
|
|
|
|
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
)
|
|
|
|
// generates Find() and some other stuff
|
|
|
|
func (msg *MsgName) generateFindBy(w io.Writer, FUNCNAME, STRUCT string, sortvals *Sort, childVar *MsgVar) string {
|
|
LOCK := msg.getLockname("x")
|
|
VARNAME := sortvals.VarName
|
|
VARTYPE := sortvals.VarType
|
|
COLOR := childVar.VarName
|
|
|
|
if childVar.VarType == "string" {
|
|
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, " }")
|
|
fmt.Fprintln(w, "")
|
|
fmt.Fprintln(w, " "+LOCK+".RLock()")
|
|
fmt.Fprintln(w, " defer "+LOCK+".RUnlock()")
|
|
fmt.Fprintln(w, "")
|
|
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
|
|
}
|
|
|
|
if childVar.VarType == "int64" {
|
|
funcdef := "func (x *" + STRUCT + ") " + FUNCNAME + "(y int64) *" + 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, " }")
|
|
fmt.Fprintln(w, "")
|
|
fmt.Fprintln(w, " "+LOCK+".RLock()")
|
|
fmt.Fprintln(w, " defer "+LOCK+".RUnlock()")
|
|
fmt.Fprintln(w, "")
|
|
fmt.Fprintln(w, " for i, _ := range x."+VARNAME+" {")
|
|
fmt.Fprintln(w, " if x."+VARNAME+"[i]."+COLOR+" == y {")
|
|
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
|
|
}
|
|
|
|
return ""
|
|
}
|