1 files changed, 59 insertions, 0 deletions
diff --git a/sortNew.go b/sortNew.go
new file mode 100644
index 0000000..d2c05b1
--- /dev/null
+++ b/sortNew.go
@@ -0,0 +1,59 @@
+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, "")
+}