lorm.Set

generic.go

@@ -13,6 +13,14 @@ import (
 
 func UUID() string { return gouuidv6.New().String() }
 
+func DoPks[PK, R interface{ Pk() PK }](l []R) []PK {
+	pks := make([]PK, len(l))
+	for i, r := range l {
+		pks[i] = r.Pk()
+	}
+	return pks
+}
+
 func DoCount(f Filter, table op.Table) int {
 	var res int
 	if !f.IsEmpty() {

go.mod

@@ -1,6 +1,6 @@
 module github.com/tada-team/lorm
 
-go 1.14
+go 1.18
 
 require (
 	github.com/bradleypeabody/gouuidv6 v0.0.0-20210725191436-99413e4b686d

pk_set.go

@@ -0,0 +1,20 @@
+package lorm
+
+type iPK interface {
+	int | int64 | string
+}
+
+type Set[PK iPK] map[PK]struct{}
+
+func (s Set[PK]) Add(pk PK) { s[pk] = struct{}{} }
+
+func (s Set[PK]) Contains(pk PK) bool { _, ok := s[pk]; return ok }
+
+func (s Set[PK]) AsList() []PK {
+	//pks := make([]PK, 0, len(s)) <-- len(s) doesn't work yet
+	pks := make([]PK, 0)
+	for pk := range s {
+		pks = append(pks, pk)
+	}
+	return pks
+}