/hazelcast/src/main/java/com/hazelcast/impl/partition/BackupSafeMemberGroupFactory.java

https://bitbucket.org/gabral6_gmailcom/hazelcast · Java · 60 lines · 34 code · 9 blank · 17 comment · 7 complexity · ff3043086f55a32fecd37e52cdcf141e MD5 · raw file

  1. /*
  2. * Copyright (c) 2008-2013, Hazelcast, Inc. All Rights Reserved.
  3. *
  4. * Licensed under the Apache License, Version 2.0 (the "License");
  5. * you may not use this file except in compliance with the License.
  6. * You may obtain a copy of the License at
  7. *
  8. * http://www.apache.org/licenses/LICENSE-2.0
  9. *
  10. * Unless required by applicable law or agreed to in writing, software
  11. * distributed under the License is distributed on an "AS IS" BASIS,
  12. * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13. * See the License for the specific language governing permissions and
  14. * limitations under the License.
  15. */
  16. package com.hazelcast.impl.partition;
  17. import com.hazelcast.impl.MemberImpl;
  18. import java.util.Collection;
  19. import java.util.Iterator;
  20. import java.util.LinkedList;
  21. import java.util.Set;
  22. abstract class BackupSafeMemberGroupFactory implements MemberGroupFactory {
  23. public final Collection<MemberGroup> createMemberGroups(final Collection<MemberImpl> allMembers) {
  24. final Collection<MemberImpl> members = removeLiteMembers(allMembers);
  25. final Collection<MemberGroup> groups = createInternalMemberGroups(members);
  26. if (groups.size() == 1 && members.size() > 1) {
  27. // If there are more than one members and just one group
  28. // then split members into two groups to guarantee at least the first backup.
  29. MemberGroup group1 = groups.iterator().next();
  30. MemberGroup group2 = new DefaultMemberGroup();
  31. final int sizePerGroup = group1.size() / 2;
  32. final Iterator<MemberImpl> iter = group1.iterator();
  33. while (group2.size() < sizePerGroup && iter.hasNext()) {
  34. group2.addMember(iter.next());
  35. iter.remove();
  36. }
  37. groups.add(group2);
  38. }
  39. return groups;
  40. }
  41. protected abstract Set<MemberGroup> createInternalMemberGroups(final Collection<MemberImpl> allMembers) ;
  42. private Collection<MemberImpl> removeLiteMembers(Collection<MemberImpl> members) {
  43. final Collection<MemberImpl> result = new LinkedList<MemberImpl>();
  44. for (MemberImpl member : members) {
  45. if (!member.isLiteMember()) {
  46. result.add(member);
  47. }
  48. }
  49. return result;
  50. }
  51. }