join

map.go

@@ -1,5 +1,7 @@
 package extypes
 
+import "reflect"
+
 type HashMap[K comparable, V any] map[K]V
 
 func NewMapFrom[K comparable, V any](m map[K]V) HashMap[K, V] {
@@ -9,9 +11,7 @@ func NewMapFrom[K comparable, V any](m map[K]V) HashMap[K, V] {
 	}
 	return out
 }
-func (m HashMap[K, V]) Len() int {
-	return len(m)
-}
+func (m HashMap[K, V]) Len() int { return len(m) }
 func (m HashMap[K, V]) Add(key K, val V) HashMap[K, V] {
 	m[key] = val
 	return m
@@ -20,9 +20,7 @@ func (m HashMap[K, V]) Remove(key K) HashMap[K, V] {
 	delete(m, key)
 	return m
 }
-func (m HashMap[K, V]) Get(key K) V {
-	return m[key]
-}
+func (m HashMap[K, V]) Get(key K) V { return m[key] }
 func (m HashMap[K, V]) GetOrDefault(key K, def V) V {
 	v, ok := m[key]
 	if !ok {
@@ -34,6 +32,26 @@ func (m HashMap[K, V]) Contains(key K) bool {
 	_, ok := m[key]
 	return ok
 }
+func (m HashMap[K, V]) IndexKey(key K) int {
+	index := 0
+	for k, _ := range m {
+		if k == key {
+			return index
+		}
+		index++
+	}
+	return -1
+}
+func (m HashMap[K, V]) IndexValue(value V) int {
+	index := 0
+	for _, v := range m {
+		if reflect.DeepEqual(value, v) {
+			return index
+		}
+		index++
+	}
+	return -1
+}
 func (m HashMap[K, V]) Keys() []K {
 	keys := make([]K, 0, m.Len())
 	for k := range m {
@@ -48,6 +66,16 @@ func (m HashMap[K, V]) Values() []V {
 	}
 	return values
 }
+func (m HashMap[K, V]) Items() ([]K, []V) {
+	keys := make(Slice[K], 0, m.Len())
+	values := make(Slice[V], 0, m.Len())
+	for k, v := range m {
+		keys = keys.Push(k)
+		values = values.Push(v)
+	}
+	return keys, values
+}
+
 func (m HashMap[K, V]) Filter(f func(K, V) bool) HashMap[K, V] {
 	out := make(HashMap[K, V])
 	for k, v := range m {
@@ -58,3 +86,15 @@ func (m HashMap[K, V]) Filter(f func(K, V) bool) HashMap[K, V] {
 	}
 	return out
 }
+func (m HashMap[K, V]) ForEach(f func(K, V)) HashMap[K, V] {
+	for k, v := range m {
+		f(k, v)
+	}
+	return m
+}
+func (m HashMap[K, V]) Map(f func(K, V) V) HashMap[K, V] {
+	for k, v := range m {
+		m[k] = f(k, v)
+	}
+	return m
+}

queue.go

@@ -10,62 +10,54 @@ var QueueFullErr = errors.New("queue full")
 type Queue[T any] struct {
 	size  uint64
 	mu    sync.RWMutex
-	queue []T
+	queue Slice[T]
 }
 
 func CreateQueue[T any](size uint64) *Queue[T] {
-	return &Queue[T]{
-		queue: make([]T, 0, size),
-		size:  size,
-	}
+	return &Queue[T]{queue: make(Slice[T], 0, size), size: size}
 }
 
+func (q *Queue[T]) Len() uint64 {
+	q.mu.RLock()
+	defer q.mu.RUnlock()
+	return uint64(q.queue.Len())
+}
+func (q *Queue[T]) IsEmpty() bool { return q.Len() == 0 }
+func (q *Queue[T]) IsFull() bool  { return q.Len() == q.size }
+func (q *Queue[T]) Size() uint64  { return q.size }
+
 func (q *Queue[T]) Enqueue(el T) error {
 	if q.IsFull() {
 		return QueueFullErr
 	}
-	q.queue = append(q.queue, el)
+	q.queue = q.queue.Push(el)
 	return nil
 }
 
-func (q *Queue[T]) Peak() T {
-	q.mu.RLock()
-	defer q.mu.RUnlock()
-	return q.queue[0]
-}
-
-func (q *Queue[T]) IsEmpty() bool {
-	return q.Length() == 0
-}
-
-func (q *Queue[T]) IsFull() bool {
-	return q.Length() == q.size
-}
-
-func (q *Queue[T]) Length() uint64 {
-	q.mu.RLock()
-	defer q.mu.RUnlock()
-	return uint64(len(q.queue))
-}
-
 func (q *Queue[T]) Dequeue() T {
 	q.mu.RLock()
-	el := q.queue[0]
+	el := q.queue.Get(0)
 	q.mu.RUnlock()
 
-	if q.Length() == 1 {
+	if q.Len() == 1 {
 		q.mu.Lock()
-		q.queue = make([]T, 0)
+		q.queue = make(Slice[T], 0)
 		q.mu.Unlock()
 		return el
 	}
 
 	q.mu.Lock()
-	q.queue = q.queue[1:]
+	q.queue = q.queue.Pop(0)
 	q.mu.Unlock()
 	return el
 }
 
-func (q *Queue[T]) Raw() []T {
+func (q *Queue[T]) Peak() T {
+	q.mu.RLock()
+	defer q.mu.RUnlock()
+	return q.queue.Get(0)
+}
+
+func (q *Queue[T]) Raw() Slice[T] {
 	return q.queue
 }

set.go

@@ -1,7 +1,9 @@
 package extypes
 
 import (
+	"fmt"
 	"reflect"
+	"strings"
 )
 
 type Set[T any] []T
@@ -16,18 +18,12 @@ func NewSetFrom[T any](slice []T) Set[T] {
 	}
 	return s
 }
-func (s Set[T]) Len() int {
-	return len(s)
-}
-func (s Set[T]) Cap() int {
-	return cap(s)
-}
-func (s Set[T]) Get(index int) T {
-	return s[index]
-}
-func (s Set[T]) Last() T {
-	return s[s.Len()-1]
-}
+func (s Set[T]) Len() int        { return len(s) }
+func (s Set[T]) Cap() int        { return cap(s) }
+func (s Set[T]) Get(index int) T { return s[index] }
+func (s Set[T]) Last() T         { return s.Get(s.Len() - 1) }
+func (s Set[T]) First() T        { return s.Get(0) }
+
 func (s Set[T]) Index(el T) int {
 	for i := range s {
 		if reflect.DeepEqual(s[i], el) {
@@ -62,33 +58,6 @@ func (s Set[T]) Remove(el T) Set[T] {
 	}
 	return s.Pop(index)
 }
-func (s Set[T]) Filter(f func(e T) bool) Set[T] {
-	out := make(Set[T], 0)
-	for _, v := range s {
-		if f(v) {
-			out = append(out, v)
-		}
-	}
-	return out
-}
-func (s Set[T]) ToSlice() Slice[T] {
-	out := make(Slice[T], s.Len())
-	copy(out, s)
-	return out
-}
-func (s Set[T]) ToArray() []T {
-	out := make([]T, len(s))
-	copy(out, s)
-	return out
-}
-func (s Set[T]) ToAnyArray() []any {
-	out := make([]any, len(s))
-	for i, v := range s {
-		out[i] = v
-	}
-	return out
-}
-
 func (s Set[T]) Equal(s2 Set[T]) bool {
 	if s.Len() != s2.Len() {
 		return false
@@ -103,3 +72,49 @@ func (s Set[T]) Equal(s2 Set[T]) bool {
 	}
 	return true
 }
+
+func (s Set[T]) ToSlice() Slice[T] { return NewSliceFrom(s) }
+func (s Set[T]) ToTuple() Tuple[T] { return NewTupleFrom(s) }
+func (s Set[T]) ToArray() []T {
+	out := make([]T, len(s))
+	copy(out, s)
+	return out
+}
+func (s Set[T]) ToAnyArray() []any {
+	out := make([]any, len(s))
+	for i, v := range s {
+		out[i] = v
+	}
+	return out
+}
+func (s Set[T]) Filter(f func(e T) bool) Set[T] {
+	out := make(Set[T], 0)
+	for _, v := range s {
+		if f(v) {
+			out = append(out, v)
+		}
+	}
+	return out
+}
+func (s Set[T]) ForEach(f func(int, T)) Set[T] {
+	for i, v := range s {
+		f(i, v)
+	}
+	return s
+}
+
+// Map is mutable func
+func (s Set[T]) Map(f func(e T) T) Set[T] {
+	for i, v := range s {
+		s[i] = f(v)
+	}
+	return s
+}
+
+func (s Set[T]) Join(sep string) string {
+	st := make([]string, len(s))
+	for i, v := range s {
+		st[i] = fmt.Sprintf("%v", v)
+	}
+	return strings.Join(st, sep)
+}

set_test.go

@@ -0,0 +1,19 @@
+package extypes
+
+import "testing"
+
+func TestSet(t *testing.T) {
+	s1 := make(Set[int], 0)
+	s2 := make(Set[int], 0)
+
+	s1 = s1.Add(1)
+	s1 = s1.Add(2)
+	s1 = s1.Add(2)
+
+	s2 = s2.Add(1)
+	s2 = s2.Add(2)
+
+	if !s1.Equal(s2) {
+		t.Error("s1 and s2 not equal")
+	}
+}

slice.go

@@ -1,7 +1,9 @@
 package extypes
 
 import (
+	"fmt"
 	"reflect"
+	"strings"
 )
 
 type Slice[T any] []T
@@ -11,18 +13,12 @@ func NewSliceFrom[T any](slice []T) Slice[T] {
 	copy(s, slice)
 	return s
 }
-func (s Slice[T]) Len() int {
-	return len(s)
-}
-func (s Slice[T]) Cap() int {
-	return cap(s)
-}
-func (s Slice[T]) Get(index int) T {
-	return s[index]
-}
-func (s Slice[T]) Last() T {
-	return s.Get(s.Len() - 1)
-}
+func (s Slice[T]) Len() int        { return len(s) }
+func (s Slice[T]) Cap() int        { return cap(s) }
+func (s Slice[T]) Get(index int) T { return s[index] }
+func (s Slice[T]) Last() T         { return s.Get(s.Len() - 1) }
+func (s Slice[T]) First() T        { return s.Get(0) }
+
 func (s Slice[T]) Index(el T) int {
 	for i := range s {
 		if reflect.DeepEqual(s[i], el) {
@@ -32,27 +28,7 @@ func (s Slice[T]) Index(el T) int {
 	return -1
 }
 
-// Swap is mutable func
-func (s Slice[T]) Swap(i, j int) Slice[T] {
-	s[i], s[j] = s[j], s[i]
-	return s
-}
-func (s Slice[T]) Filter(f func(e T) bool) Slice[T] {
-	out := make(Slice[T], 0)
-	for _, v := range s {
-		if f(v) {
-			out = append(out, v)
-		}
-	}
-	return out
-}
-func (s Slice[T]) Map(f func(e T) T) Slice[T] {
-	out := make(Slice[T], s.Len())
-	for i, v := range s {
-		out[i] = f(v)
-	}
-	return out
-}
+func (s Slice[T]) Push(e T) Slice[T] { return append(s, e) }
 func (s Slice[T]) Pop(index int) Slice[T] {
 	if index == 0 {
 		return s[1:]
@@ -69,31 +45,10 @@ func (s Slice[T]) Remove(el T) Slice[T] {
 	}
 	return s.Pop(index)
 }
-func (s Slice[T]) Push(e T) Slice[T] {
-	return append(s, e)
-}
 
-func (s Slice[T]) ToArray() []T {
-	out := make([]T, len(s))
-	copy(out, s)
-	return out
-}
-func (s Slice[T]) ToAnyArray() []any {
-	out := make([]any, len(s))
-	for i, v := range s {
-		out[i] = v
-	}
-	return out
-}
-func (s Slice[T]) ToSet() Set[T] {
-	return NewSetFrom(s)
-}
-
-// ForEach is mutable func
-func (s Slice[T]) ForEach(f func(e T) T) Slice[T] {
-	for i, v := range s {
-		s[i] = f(v)
-	}
+// Swap is mutable func
+func (s Slice[T]) Swap(i, j int) Slice[T] {
+	s[i], s[j] = s[j], s[i]
 	return s
 }
 
@@ -111,3 +66,50 @@ func (s Slice[T]) Equal(s2 Slice[T]) bool {
 	}
 	return true
 }
+
+func (s Slice[T]) ToArray() []T {
+	out := make([]T, len(s))
+	copy(out, s)
+	return out
+}
+func (s Slice[T]) ToAnyArray() []any {
+	out := make([]any, len(s))
+	for i, v := range s {
+		out[i] = v
+	}
+	return out
+}
+func (s Slice[T]) ToSet() Set[T]     { return NewSetFrom(s) }
+func (s Slice[T]) ToTuple() Tuple[T] { return NewTupleFrom(s) }
+
+func (s Slice[T]) Filter(f func(e T) bool) Slice[T] {
+	out := make(Slice[T], 0)
+	for _, v := range s {
+		if f(v) {
+			out = append(out, v)
+		}
+	}
+	return out
+}
+func (s Slice[T]) ForEach(f func(int, T)) Slice[T] {
+	for i, v := range s {
+		f(i, v)
+	}
+	return s
+}
+
+// Map is mutable func
+func (s Slice[T]) Map(f func(e T) T) Slice[T] {
+	for i, v := range s {
+		s[i] = f(v)
+	}
+	return s
+}
+
+func (s Slice[T]) Join(sep string) string {
+	st := make([]string, len(s))
+	for i, v := range s {
+		st[i] = fmt.Sprintf("%v", v)
+	}
+	return strings.Join(st, sep)
+}

tuple.go

@@ -0,0 +1,74 @@
+package extypes
+
+import (
+	"encoding/json"
+	"fmt"
+	"reflect"
+	"strings"
+)
+
+type Tuple[T any] struct {
+	s Slice[T]
+}
+
+func NewTupleFrom[T any](src []T) Tuple[T] { return Tuple[T]{src} }
+func NewTuple[T any]() Tuple[T]            { return Tuple[T]{make(Slice[T], 0)} }
+func (t Tuple[T]) Len() int                { return t.s.Len() }
+func (t Tuple[T]) Cap() int                { return t.s.Cap() }
+func (t Tuple[T]) Get(index int) T         { return t.s.Get(index) }
+
+func (t Tuple[T]) Equal(t2 Tuple[T]) bool {
+	if t.Len() != t2.Len() {
+		return false
+	}
+	for i := range t.s {
+		if !reflect.DeepEqual(t.s[i], t2.s[i]) {
+			return false
+		}
+	}
+	return true
+}
+
+func (t Tuple[T]) ToArray() []T {
+	out := make([]T, t.Len())
+	copy(out, t.s)
+	return out
+}
+func (t Tuple[T]) ToAnyArray() []any {
+	out := make([]any, t.Len())
+	for i, v := range t.s {
+		out[i] = v
+	}
+	return out
+}
+func (t Tuple[T]) ToSet() Set[T]     { return NewSetFrom(t.s) }
+func (t Tuple[T]) ToSlice() Slice[T] { return NewSliceFrom(t.s) }
+
+func (t Tuple[T]) Filter(f func(e T) bool) Tuple[T] {
+	out := make(Slice[T], 0)
+	for _, v := range t.s {
+		if f(v) {
+			out = append(out, v)
+		}
+	}
+	t.s = out
+	return t
+}
+func (t Tuple[T]) ForEach(f func(int, T)) Tuple[T] {
+	for i, v := range t.s {
+		f(i, v)
+	}
+	return t
+}
+
+func (t Tuple[T]) Join(sep string) string {
+	st := make([]string, len(t.s))
+	for i, v := range t.s {
+		st[i] = fmt.Sprintf("%v", v)
+	}
+	return strings.Join(st, sep)
+}
+
+func (t Tuple[T]) String() string                   { return fmt.Sprint(t.s) }
+func (t *Tuple[T]) UnmarshalJSON(data []byte) error { return json.Unmarshal(data, &t.s) }
+func (t Tuple[T]) MarshalJSON() ([]byte, error)     { return json.Marshal(t.s) }

tuple_test.go

@@ -0,0 +1,49 @@
+package extypes
+
+import (
+	"encoding/json"
+	"testing"
+)
+
+func TestTuple(t *testing.T) {
+	t1 := NewTupleFrom([]int{1, 2, 3})
+	t2 := NewTupleFrom([]int{1, 2, 3})
+
+	if !t1.Equal(t2) {
+		t.Error("t1 and t2 should be equal")
+	}
+}
+func TestTuple_MarshalJSON(t *testing.T) {
+	t1 := NewTupleFrom([]int{1, 2, 3})
+	data, err := json.Marshal(t1)
+	if err != nil {
+		t.Fatal(err)
+	}
+	s := "[1,2,3]"
+	if string(data) != s {
+		t.Error("t1 and t2 should be equal")
+	}
+}
+func TestTuple_UnmarshalJSON(t *testing.T) {
+	s := "[1,2,3]"
+	t1 := NewTuple[int]()
+	err := json.Unmarshal([]byte(s), &t1)
+	if err != nil {
+		t.Fatal(err)
+	}
+	t2 := NewTupleFrom([]int{1, 2, 3})
+	t.Log(t1, t2)
+	if !t1.Equal(t2) {
+		t.Error("t1 and t2 should be equal")
+	}
+
+	st1 := struct {
+		A int        `json:"a"`
+		B Tuple[int] `json:"b"`
+	}{}
+	s2 := `{"a":1,"b":[1,2,3]}`
+	if err := json.Unmarshal([]byte(s2), &st1); err != nil {
+		t.Fatal(err)
+	}
+	t.Log(st1)
+}