slices: add mapping for pointer slices and pointer values in slices f…

…or source and dest Signed-off-by: Dmitrii Aleksandrov <goodmobiledevices@gmail.com>
Insei · Apr 4, 2024 · 5f59c26 · 5f59c26
1 parent 73fc47f
commit 5f59c26
Show file tree

Hide file tree

Showing 2 changed files with 88 additions and 0 deletions.
diff --git a/mapper_test.go b/mapper_test.go
@@ -57,6 +57,44 @@ func TestMapTo(t *testing.T) {
 			assert.Equal(t, source[i].Name, dest[i].Name)
 		}
 	})
+	t.Run("Source is a slice with pointer elements", func(t *testing.T) {
+		source := []*TestingStructSource{
+			{
+				Name: "ArrayTest1",
+			},
+			{
+				Name: "ArrayTest2",
+			},
+			{
+				Name: "ArrayTest3",
+			},
+		}
+		dest, err := MapTo[[]TestingStructDest](source)
+		assert.NoError(t, err)
+		assert.Equal(t, len(source), len(dest))
+		for i, _ := range source {
+			assert.Equal(t, source[i].Name, dest[i].Name)
+		}
+	})
+	t.Run("Source is a slice with pointer elements, dest is a slice with pointer elements", func(t *testing.T) {
+		source := []*TestingStructSource{
+			{
+				Name: "ArrayTest1",
+			},
+			{
+				Name: "ArrayTest2",
+			},
+			{
+				Name: "ArrayTest3",
+			},
+		}
+		dest, err := MapTo[[]*TestingStructDest](source)
+		assert.NoError(t, err)
+		assert.Equal(t, len(source), len(dest))
+		for i, _ := range source {
+			assert.Equal(t, source[i].Name, dest[i].Name)
+		}
+	})
 	t.Run("Source is a slice pointer", func(t *testing.T) {
 		source := &[]TestingStructSource{
 			{

diff --git a/routes.go b/routes.go
@@ -56,6 +56,56 @@ func addSliceRoutes[TSource, TDest any]() {
 		}
 		return nil
 	})
+	addSliceRoute[*[]TSource, *[]*TDest](func(sourceSlice *[]TSource, pointerDestSlice *[]*TDest) error {
+		for _, source := range *sourceSlice {
+			dest, err := MapTo[TDest](source)
+			if err != nil {
+				return err
+			}
+			*pointerDestSlice = append(*pointerDestSlice, &dest)
+		}
+		return nil
+	})
+	addSliceRoute[[]*TSource, *[]TDest](func(sourceSlice []*TSource, pointerDestSlice *[]TDest) error {
+		for _, source := range sourceSlice {
+			dest, err := MapTo[TDest](source)
+			if err != nil {
+				return err
+			}
+			*pointerDestSlice = append(*pointerDestSlice, dest)
+		}
+		return nil
+	})
+	addSliceRoute[[]*TSource, *[]*TDest](func(sourceSlice []*TSource, pointerDestSlice *[]*TDest) error {
+		for _, source := range sourceSlice {
+			dest, err := MapTo[TDest](source)
+			if err != nil {
+				return err
+			}
+			*pointerDestSlice = append(*pointerDestSlice, &dest)
+		}
+		return nil
+	})
+	addSliceRoute[*[]*TSource, *[]TDest](func(sourceSlice *[]*TSource, pointerDestSlice *[]TDest) error {
+		for _, source := range *sourceSlice {
+			dest, err := MapTo[TDest](source)
+			if err != nil {
+				return err
+			}
+			*pointerDestSlice = append(*pointerDestSlice, dest)
+		}
+		return nil
+	})
+	addSliceRoute[*[]*TSource, *[]*TDest](func(sourceSlice *[]*TSource, pointerDestSlice *[]*TDest) error {
+		for _, source := range *sourceSlice {
+			dest, err := MapTo[TDest](source)
+			if err != nil {
+				return err
+			}
+			*pointerDestSlice = append(*pointerDestSlice, &dest)
+		}
+		return nil
+	})
 }
 
 func AddRoute[TSource, TDest any | []any](mapFunc func(source TSource, dest *TDest) error) error {