Merge pull request #2 from octu0/v1.7.1

v1.7.1

Author: octu0

map.go

@@ -1,6 +1,7 @@
 package armap
 
 import (
+	"fmt"
 	"unsafe"
 
 	"github.com/dolthub/maphash"
@@ -331,6 +332,9 @@ func (m *Map[K, V]) Clear() {
 }
 
 func NewMap[K comparable, V any](arena Arena, funcs ...OptionFunc) *Map[K, V] {
+	checkType[K](arena)
+	checkType[V](arena)
+
 	opt := newOption()
 	for _, fn := range funcs {
 		fn(opt)
@@ -350,3 +354,15 @@ func NewMap[K comparable, V any](arena Arena, funcs ...OptionFunc) *Map[K, V] {
 	m.resize(capacity)
 	return m
 }
+
+func checkType[T any](arena Arena) {
+	defer func() {
+		if r := recover(); r != nil {
+			panic(fmt.Sprintf("armap: type %T cannot be used in Map (likely due to unexported fields preventing clone): %v", *new(T), r))
+		}
+	}()
+
+	ta := NewTypeArena[T](arena)
+	var zero T
+	_ = ta.Clone(zero)
+}

map_test.go

@@ -4,6 +4,7 @@ import (
 	"fmt"
 	"strconv"
 	"testing"
+	"time"
 )
 
 func TestMap(t *testing.T) {
@@ -196,4 +197,55 @@ func TestMap(t *testing.T) {
 			tt.Errorf("count = %d, expected 1", count)
 		}
 	})
+
+	t.Run("string,time.Time", func(tt *testing.T) {
+		a := NewArena(1024)
+		defer a.Release()
+
+		defer func() {
+			if r := recover(); r == nil {
+				tt.Errorf("expected panic for time.Time value")
+			}
+		}()
+
+		NewMap[string, time.Time](a)
+	})
+
+	t.Run("string,PublicStruct", func(tt *testing.T) {
+		type PublicStruct struct {
+			ID   int
+			Name string
+		}
+		a := NewArena(1024)
+		defer a.Release()
+		m := NewMap[string, PublicStruct](a)
+
+		val := PublicStruct{ID: 1, Name: "test"}
+		key := "p1"
+
+		m.Set(key, val)
+
+		if v, ok := m.Get(key); !ok {
+			tt.Errorf("key %s not found", key)
+		} else if v != val {
+			tt.Errorf("value mismatch: got %v, want %v", v, val)
+		}
+	})
+
+	t.Run("string,PrivateStruct", func(tt *testing.T) {
+		type PrivateStruct struct {
+			id   int
+			name string
+		}
+		a := NewArena(1024)
+		defer a.Release()
+
+		defer func() {
+			if r := recover(); r == nil {
+				tt.Errorf("expected panic for PrivateStruct value")
+			}
+		}()
+
+		NewMap[string, PrivateStruct](a)
+	})
 }

set_test.go

@@ -74,4 +74,42 @@ func TestSet(t *testing.T) {
 			tt.Errorf("test1 is deleted")
 		}
 	})
+
+	t.Run("PublicStruct", func(tt *testing.T) {
+		type PublicStruct struct {
+			ID   int
+			Name string
+		}
+		a := NewArena(1024)
+		defer a.Release()
+		s := NewSet[PublicStruct](a)
+
+		val := PublicStruct{ID: 1, Name: "test"}
+		if ok := s.Add(val); ok {
+			tt.Errorf("val is new key")
+		}
+		if ok := s.Contains(val); ok != true {
+			tt.Errorf("val exists")
+		}
+		if ok := s.Delete(val); ok != true {
+			tt.Errorf("val exists")
+		}
+	})
+
+	t.Run("PrivateStruct", func(tt *testing.T) {
+		type PrivateStruct struct {
+			id   int
+			name string
+		}
+		a := NewArena(1024)
+		defer a.Release()
+
+		defer func() {
+			if r := recover(); r == nil {
+				tt.Errorf("expected panic for PrivateStruct key")
+			}
+		}()
+
+		NewSet[PrivateStruct](a)
+	})
 }

version.go

@@ -2,5 +2,5 @@ package armap
 
 const (
 	AppName string = "armap"
-	Version string = "1.7.0"
+	Version string = "1.7.1"
 )