[Feature] [Platform] Add Sharding to the Inventory Endpoint (#1848)

Author: ajanikow

CHANGELOG.md

@@ -9,6 +9,7 @@
 - (Maintenance) Allow GRPC Marshal Opts
 - (Bugfix) Propagate Gateway Idle Timeout
 - (Feature) Update Authz V1 Definitions
+- (Feature) (Platform) Add Sharding to the Inventory Endpoint
 
 ## [1.2.46](https://github.com/arangodb/kube-arangodb/tree/1.2.46) (2025-02-24)
 - (Bugfix) Clean Phase change properly during upgrade

integrations/inventory/v1/definition/inventory.go

@@ -21,8 +21,15 @@
 package definition
 
 import (
+	"strconv"
+
 	api "github.com/arangodb/kube-arangodb/pkg/apis/deployment/v1"
 	ugrpc "github.com/arangodb/kube-arangodb/pkg/util/grpc"
+	"github.com/arangodb/kube-arangodb/pkg/util/strings"
+)
+
+const (
+	forceOneShardFlag = "--cluster.force-one-shard"
 )
 
 func (s *Inventory) JSON() ([]byte, error) {
@@ -55,5 +62,29 @@ func NewArangoDBConfiguration(spec api.DeploymentSpec, status api.DeploymentStat
 		cfg.Version = string(i.ArangoDBVersion)
 	}
 
+	if spec.GetMode() != api.DeploymentModeCluster {
+		cfg.Sharding = ArangoDBSharding_Sharded
+	} else {
+		cfg.Sharding = getShardingFromArgs(spec.Coordinators.Args...)
+	}
+
 	return &cfg
 }
+
+func getShardingFromArgs(args ...string) ArangoDBSharding {
+	for _, arg := range args {
+		if arg == forceOneShardFlag {
+			return ArangoDBSharding_OneShardEnforced
+		}
+
+		if v := strings.SplitN(arg, "=", 2); v[0] == forceOneShardFlag && len(v) == 2 {
+			if q, err := strconv.ParseBool(v[1]); err != nil {
+				continue
+			} else if q {
+				return ArangoDBSharding_OneShardEnforced
+			}
+		}
+	}
+
+	return ArangoDBSharding_Sharded
+}

integrations/inventory/v1/definition/inventory.pb.go

@@ -56,6 +56,14 @@ enum ArangoDBEdition {
   Enterprise = 1;
 }
 
+// ArangoDB Sharding Mode
+enum ArangoDBSharding {
+  // Sharded Sharding Mode
+  Sharded = 0;
+  // OneShardEnforced Sharding Mode
+  OneShardEnforced = 1;
+}
+
 // ArangoDB Deployment Configuration
 message ArangoDBConfiguration {
   // ArangoDB Deployment Mode
@@ -64,4 +72,6 @@ message ArangoDBConfiguration {
   ArangoDBEdition edition = 2;
   // ArangoDB Deployment Version
   string version = 3;
+  // ArangoDB Sharding Mode
+  ArangoDBSharding sharding = 4;
 }

integrations/inventory/v1/definition/inventory.proto

@@ -21,6 +21,7 @@
 package definition
 
 import (
+	"fmt"
 	"testing"
 
 	"github.com/stretchr/testify/require"
@@ -52,3 +53,13 @@ func Test_State_Marshal(t *testing.T) {
 
 	require.EqualValues(t, &s, res)
 }
+
+func Test_getShardingFromArgs(t *testing.T) {
+	require.EqualValues(t, ArangoDBSharding_Sharded, getShardingFromArgs())
+	require.EqualValues(t, ArangoDBSharding_OneShardEnforced, getShardingFromArgs(forceOneShardFlag))
+	require.EqualValues(t, ArangoDBSharding_OneShardEnforced, getShardingFromArgs("--test=el", forceOneShardFlag))
+	require.EqualValues(t, ArangoDBSharding_OneShardEnforced, getShardingFromArgs(fmt.Sprintf("%s=true", forceOneShardFlag)))
+	require.EqualValues(t, ArangoDBSharding_Sharded, getShardingFromArgs(fmt.Sprintf("%s=false", forceOneShardFlag)))
+	require.EqualValues(t, ArangoDBSharding_OneShardEnforced, getShardingFromArgs(fmt.Sprintf("%s=True", forceOneShardFlag)))
+	require.EqualValues(t, ArangoDBSharding_Sharded, getShardingFromArgs(fmt.Sprintf("%s=Unknown", forceOneShardFlag)))
+}