fix(edgegroups): convert the related endpoint IDs to roaring bitmaps to increase performance BE-12053 (#903)

api/internal/edge/edgegroup_benchmark_test.go

@@ -0,0 +1,104 @@
+package edge
+
+import (
+	"testing"
+
+	portainer "github.com/portainer/portainer/api"
+	"github.com/portainer/portainer/api/datastore"
+	"github.com/portainer/portainer/api/roar"
+
+	"github.com/rs/zerolog"
+	"github.com/stretchr/testify/require"
+)
+
+const n = 1_000_000
+
+func BenchmarkWriteEdgeGroupOld(b *testing.B) {
+	zerolog.SetGlobalLevel(zerolog.ErrorLevel)
+
+	_, store := datastore.MustNewTestStore(b, false, false)
+
+	var endpointIDs []portainer.EndpointID
+
+	for i := range n {
+		endpointIDs = append(endpointIDs, portainer.EndpointID(i+1))
+	}
+
+	for b.Loop() {
+		err := store.EdgeGroup().Create(&portainer.EdgeGroup{
+			Name:      "Test Edge Group",
+			Endpoints: endpointIDs,
+		})
+		require.NoError(b, err)
+	}
+}
+
+func BenchmarkWriteEdgeGroupNew(b *testing.B) {
+	zerolog.SetGlobalLevel(zerolog.ErrorLevel)
+
+	_, store := datastore.MustNewTestStore(b, false, false)
+
+	var ts []portainer.EndpointID
+
+	for i := range n {
+		ts = append(ts, portainer.EndpointID(i+1))
+	}
+
+	endpointIDs := roar.FromSlice(ts)
+
+	for b.Loop() {
+		err := store.EdgeGroup().Create(&portainer.EdgeGroup{
+			Name:        "Test Edge Group",
+			EndpointIDs: endpointIDs,
+		})
+		require.NoError(b, err)
+	}
+}
+
+func BenchmarkReadEdgeGroupOld(b *testing.B) {
+	zerolog.SetGlobalLevel(zerolog.ErrorLevel)
+
+	_, store := datastore.MustNewTestStore(b, false, false)
+
+	var endpointIDs []portainer.EndpointID
+
+	for i := range n {
+		endpointIDs = append(endpointIDs, portainer.EndpointID(i+1))
+	}
+
+	err := store.EdgeGroup().Create(&portainer.EdgeGroup{
+		Name:      "Test Edge Group",
+		Endpoints: endpointIDs,
+	})
+	require.NoError(b, err)
+
+	for b.Loop() {
+		_, err := store.EdgeGroup().ReadAll()
+		require.NoError(b, err)
+	}
+}
+
+func BenchmarkReadEdgeGroupNew(b *testing.B) {
+	zerolog.SetGlobalLevel(zerolog.ErrorLevel)
+
+	_, store := datastore.MustNewTestStore(b, false, false)
+
+	var ts []portainer.EndpointID
+
+	for i := range n {
+		ts = append(ts, portainer.EndpointID(i+1))
+	}
+
+	endpointIDs := roar.FromSlice(ts)
+
+	err := store.EdgeGroup().Create(&portainer.EdgeGroup{
+		Name:        "Test Edge Group",
+		EndpointIDs: endpointIDs,
+	})
+	require.NoError(b, err)
+
+	for b.Loop() {
+		_, err := store.EdgeGroup().ReadAll()
+		require.NoError(b, err)
+	}
+}