60 lines
1.8 KiB
Go
60 lines
1.8 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
)
|
|
|
|
func (msg *MsgName) getLockname(s string) string {
|
|
if argv.Mutex {
|
|
// use the mutex lock from the modified protoc.pb.go file
|
|
return s + ".Lock"
|
|
}
|
|
// a single global lock by struct name
|
|
return msg.Lockname
|
|
}
|
|
|
|
func (msg *MsgName) appendUnique(w io.Writer, FRUIT, APPLES, APPLE string, COLORS []string) {
|
|
LOCK := msg.getLockname("x")
|
|
|
|
fmt.Fprintln(w, "// TESTING")
|
|
fmt.Fprintln(w, "// enforces "+APPLE+" is unique in "+FRUIT+"."+APPLES)
|
|
fmt.Fprintln(w, "func (x *"+FRUIT+") AppendUnique(newP *"+APPLE+") bool {")
|
|
fmt.Fprintln(w, " "+LOCK+".Lock()")
|
|
fmt.Fprintln(w, " defer "+LOCK+".Unlock()")
|
|
fmt.Fprintln(w, "")
|
|
for _, COLOR := range COLORS {
|
|
fmt.Fprintln(w, " for _, p := range x."+APPLES+" {")
|
|
fmt.Fprintln(w, " if p."+COLOR+" == newP."+COLOR+" {")
|
|
fmt.Fprintln(w, " return false")
|
|
fmt.Fprintln(w, " }")
|
|
fmt.Fprintln(w, " }")
|
|
}
|
|
fmt.Fprintln(w, "")
|
|
fmt.Fprintln(w, " x."+APPLES+" = append(x."+APPLES+", newP)")
|
|
fmt.Fprintln(w, " return true")
|
|
fmt.Fprintln(w, "}")
|
|
fmt.Fprintln(w, "")
|
|
}
|
|
|
|
func (msg *MsgName) appendUniqueBy(w io.Writer, FRUIT, APPLES, APPLE, COLOR string) {
|
|
LOCK := msg.getLockname("x")
|
|
|
|
fmt.Fprintln(w, "// TESTING")
|
|
fmt.Fprintln(w, "// enforces "+APPLE+"."+COLOR+" is unique in "+FRUIT+"."+APPLES)
|
|
fmt.Fprintln(w, "func (x *"+FRUIT+") AppendUniqueBy"+COLOR+"(newP *"+APPLE+") bool {")
|
|
fmt.Fprintln(w, " "+LOCK+".Lock()")
|
|
fmt.Fprintln(w, " defer "+LOCK+".Unlock()")
|
|
fmt.Fprintln(w, "")
|
|
fmt.Fprintln(w, " for _, p := range x."+APPLES+" {")
|
|
fmt.Fprintln(w, " if p."+COLOR+" == newP."+COLOR+" {")
|
|
fmt.Fprintln(w, " return false")
|
|
fmt.Fprintln(w, " }")
|
|
fmt.Fprintln(w, " }")
|
|
fmt.Fprintln(w, "")
|
|
fmt.Fprintln(w, " x."+APPLES+" = append(x."+APPLES+", newP)")
|
|
fmt.Fprintln(w, " return true")
|
|
fmt.Fprintln(w, "}")
|
|
fmt.Fprintln(w, "")
|
|
}
|