|
|
@@ -501,12 +501,20 @@ public class User implements Serializable {
|
|
|
*/
|
|
|
public void setCurrentEnterpriseRoles() {
|
|
|
if(!CollectionUtils.isEmpty(this.roles) && this.enterprise != null) {
|
|
|
- Iterator<Role> iterator = this.roles.iterator();
|
|
|
+// Iterator<Role> iterator = this.roles.iterator();
|
|
|
+// Long enuu = this.enterprise.getUu();
|
|
|
+// while (iterator.hasNext()) {
|
|
|
+// Role role = iterator.next();
|
|
|
+// if(!role.getEnUU().equals(enuu)) {
|
|
|
+// iterator.remove();
|
|
|
+// }
|
|
|
+// }
|
|
|
+ Set<Role> roles = this.roles;
|
|
|
+ this.roles = new HashSet<>();
|
|
|
Long enuu = this.enterprise.getUu();
|
|
|
- while (iterator.hasNext()) {
|
|
|
- Role role = iterator.next();
|
|
|
- if(!role.getEnUU().equals(enuu)) {
|
|
|
- iterator.remove();
|
|
|
+ for (Role role : roles) {
|
|
|
+ if (role.getEnUU().equals(enuu)) {
|
|
|
+ this.roles.add(role);
|
|
|
}
|
|
|
}
|
|
|
}
|