Add generic ToOrNil

AlekSi · Aug 21, 2021 · 47b3772 · 47b3772
1 parent 2eade2a
commit 47b3772
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/generic.go b/generic.go
@@ -4,12 +4,25 @@ func To[T any](t T) *T {
 	return &t
 }
 
+func ToOrNil[T comparable](t T) *T {
+	if z, ok := interface{}(t).(interface{ IsZero() bool }); ok {
+		if z.IsZero() {
+			return nil
+		}
+		return &t
+	}
+
+	var zero T
+	if t == zero {
+		return nil
+	}
+	return &t
+}
+
 func Get[T any](t *T) T {
 	if t == nil {
 		var zero T
 		return zero
 	}
 	return *t
 }
-
-// No `func ToOrNil[T any](t T) *T` due to time.Time.Equal