BugFix: Updating read and write quotas in cluster.

aerospike · Aug 26, 2024 · 10b385f · 10b385f
1 parent 09c04bb
commit 10b385f
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/accesscontrol/access_control.go b/accesscontrol/access_control.go
@@ -314,6 +314,16 @@ func (roleCreate AerospikeRoleCreateUpdate) UpdateRole(
 		}
 	}
 
+	if role.ReadQuota != roleCreate.ReadQuota || role.WriteQuota != roleCreate.WriteQuota {
+		if err := client.SetQuotas(
+			adminPolicy, roleCreate.Name, roleCreate.ReadQuota, roleCreate.WriteQuota,
+		); err != nil {
+			return fmt.Errorf(
+				"error setting quotas for role %s: %v", roleCreate.Name, err,
+			)
+		}
+	}
+
 	logger.Info("Updated role", "role name", roleCreate.Name)
 
 	return nil