Collection: Add Compact endpoint in V2

Author: bluepal-prasanthi-moparthi

v2/arangodb/collection.go

@@ -79,8 +79,14 @@ type Collection interface {
 	// Renaming collections is only supported in single server deployments.
 	Rename(ctx context.Context, req RenameCollectionRequest) (CollectionInfo, error)
 
+	// RecalculateCount recalculates the count of documents in the collection.
 	RecalculateCount(ctx context.Context) (bool, int64, error)
 
+	//Compacts the data of a collection in order to reclaim disk space.
+	// This operation is only supported in single server deployments.
+	// In cluster deployments, the compaction is done automatically by the server.
+	Compact(ctx context.Context) (CollectionInfo, error)
+
 	CollectionDocuments
 	CollectionIndexes
 }

v2/arangodb/collection_impl.go

@@ -405,6 +405,27 @@ func (c collection) RecalculateCount(ctx context.Context) (bool, int64, error) {
 	}
 }
 
+func (c collection) Compact(ctx context.Context) (CollectionInfo, error) {
+	urlEndpoint := c.url("collection", "compact")
+
+	var response struct {
+		shared.ResponseStruct `json:",inline"`
+		CollectionInfo        `json:",inline"`
+	}
+
+	resp, err := connection.CallPut(ctx, c.connection(), urlEndpoint, &response, nil, c.withModifiers()...)
+	if err != nil {
+		return CollectionInfo{}, errors.WithStack(err)
+	}
+
+	switch code := resp.Code(); code {
+	case http.StatusOK:
+		return response.CollectionInfo, nil
+	default:
+		return CollectionInfo{}, response.AsArangoErrorWithCode(code)
+	}
+}
+
 type RemoveCollectionOptions struct {
 	// IsSystem when set to true allows to remove system collections.
 	// Use on your own risk!

v2/tests/database_collection_operations_test.go

@@ -745,3 +745,33 @@ func Test_CollectionRecalculateCount(t *testing.T) {
 		})
 	})
 }
+
+func Test_CollectionCompact(t *testing.T) {
+	Wrap(t, func(t *testing.T, client arangodb.Client) {
+		WithDatabase(t, client, nil, func(db arangodb.Database) {
+			WithCollectionV2(t, db, nil, func(col arangodb.Collection) {
+				withContextT(t, defaultTestTimeout, func(ctx context.Context, tb testing.TB) {
+					role, err := client.ServerRole(ctx)
+					require.NoError(t, err)
+
+					if role == arangodb.ServerRoleSingle {
+						// Create dummy data
+						for i := 0; i < 10; i++ {
+							_, err := col.CreateDocument(ctx, map[string]interface{}{
+								"_key": fmt.Sprintf("key%d", i),
+								"val":  i,
+							})
+							require.NoError(t, err)
+						}
+
+						result, err := col.Compact(ctx)
+						require.NoError(t, err)
+						fmt.Printf("Compacted Collection: %s, ID: %s, Status: %d\n", result.Name, result.ID, result.Status)
+					} else {
+						t.Skip("Compaction is not supported in cluster mode")
+					}
+				})
+			})
+		})
+	})
+}