mirror of https://github.com/grafana/grafana
Cloud migrations: create snapshot files (#89693)
* Cloud migrations: create snapshot and store it on disk * fix merge conflicts * implement StartSnapshot for gms client * pass snapshot directory as argument to snapshot builder * ensure snapshot folder is set * make swagger-gen * remove Test_ExecuteAsyncWorkflow * pass signed in user to buildSnapshot method / use github.com/grafana/grafana-cloud-migration-snapshot to create snapshot files * fix FakeServiceImpl.CreateSnapshot * remove new linepull/89999/head
Bruno
1 year ago
committed by
GitHub
parent
7b29242600
commit
d1952bb681
@ -0,0 +1,33 @@ |
||||
package slicesext |
||||
|
||||
import "math" |
||||
|
||||
// Partitions the input into slices where the length is <= chunkSize.
|
||||
//
|
||||
// Example:
|
||||
//
|
||||
// Chunks(2, []int{1, 2, 3, 4})
|
||||
// => [][]int{{1, 2}, {3, 4}}
|
||||
func Chunks[T any](chunkSize int, xs []T) [][]T { |
||||
if chunkSize < 0 { |
||||
panic("chunk size must be greater than or equal to 0") |
||||
} |
||||
if chunkSize == 0 { |
||||
return [][]T{} |
||||
} |
||||
|
||||
out := make([][]T, 0, int(math.Ceil(float64(len(xs))/float64(chunkSize)))) |
||||
|
||||
for i := 0; i < len(xs); i += chunkSize { |
||||
var chunk []T |
||||
if i+chunkSize < len(xs) { |
||||
chunk = xs[i : i+chunkSize] |
||||
} else { |
||||
chunk = xs[i:] |
||||
} |
||||
|
||||
out = append(out, chunk) |
||||
} |
||||
|
||||
return out |
||||
} |
||||
@ -0,0 +1,80 @@ |
||||
package slicesext |
||||
|
||||
import ( |
||||
"testing" |
||||
|
||||
"github.com/stretchr/testify/assert" |
||||
) |
||||
|
||||
func TestChunks(t *testing.T) { |
||||
t.Parallel() |
||||
|
||||
t.Run("chunkSize must be greater than 0", func(t *testing.T) { |
||||
t.Parallel() |
||||
|
||||
assert.PanicsWithValue(t, "chunk size must be greater than or equal to 0", func() { |
||||
Chunks(-1, []string{}) |
||||
}) |
||||
}) |
||||
|
||||
t.Run("basic", func(t *testing.T) { |
||||
t.Parallel() |
||||
|
||||
cases := []struct { |
||||
description string |
||||
chunkSize int |
||||
input []int |
||||
expected [][]int |
||||
}{ |
||||
{ |
||||
description: "empty slice", |
||||
chunkSize: 2, |
||||
input: []int{}, |
||||
expected: [][]int{}, |
||||
}, |
||||
{ |
||||
description: "nil slice", |
||||
chunkSize: 2, |
||||
input: nil, |
||||
expected: [][]int{}, |
||||
}, |
||||
{ |
||||
description: "chunk size is 0", |
||||
chunkSize: 0, |
||||
input: []int{1, 2, 3}, |
||||
expected: [][]int{}, |
||||
}, |
||||
{ |
||||
description: "chunk size is greater than slice length", |
||||
chunkSize: 3, |
||||
input: []int{1}, |
||||
expected: [][]int{{1}}, |
||||
}, |
||||
{ |
||||
description: "chunk size is 1", |
||||
chunkSize: 1, |
||||
input: []int{1, 2, 3}, |
||||
expected: [][]int{{1}, {2}, {3}}, |
||||
}, |
||||
{ |
||||
description: "chunk size is 2 and slice length is 3", |
||||
chunkSize: 2, |
||||
input: []int{1, 2, 3}, |
||||
expected: [][]int{{1, 2}, {3}}, |
||||
}, |
||||
{ |
||||
description: "chunk size is 2 and slice length is 6", |
||||
chunkSize: 2, |
||||
input: []int{1, 2, 3, 4, 5, 6}, |
||||
expected: [][]int{{1, 2}, {3, 4}, {5, 6}}, |
||||
}, |
||||
} |
||||
|
||||
for _, tt := range cases { |
||||
t.Run(tt.description, func(t *testing.T) { |
||||
result := Chunks(tt.chunkSize, tt.input) |
||||
assert.Equal(t, tt.expected, result) |
||||
}) |
||||
} |
||||
}) |
||||
} |
||||
Loading…
Reference in new issue