| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more contributor license |
| * agreements. See the NOTICE file distributed with this work for additional information regarding |
| * copyright ownership. The ASF licenses this file to You under the Apache License, Version 2.0 (the |
| * "License"); you may not use this file except in compliance with the License. You may obtain a |
| * copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software distributed under the License |
| * is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express |
| * or implied. See the License for the specific language governing permissions and limitations under |
| * the License. |
| */ |
| package org.apache.geode.management.internal.cli.functions; |
| |
| import java.util.HashSet; |
| import java.util.List; |
| import java.util.Map; |
| import java.util.Set; |
| |
| import org.apache.geode.cache.execute.FunctionContext; |
| import org.apache.geode.distributed.DistributedMember; |
| import org.apache.geode.internal.cache.InternalCache; |
| import org.apache.geode.internal.cache.PartitionedRegion; |
| import org.apache.geode.internal.cache.execute.InternalFunction; |
| import org.apache.geode.internal.cache.partitioned.ColocatedRegionDetails; |
| import org.apache.geode.internal.cache.persistence.PersistentMemberID; |
| import org.apache.geode.internal.cache.persistence.PersistentMemberManager; |
| import org.apache.geode.internal.cache.persistence.PersistentMemberPattern; |
| |
| public class ShowMissingDiskStoresFunction implements InternalFunction { |
| |
| @Override |
| public void execute(FunctionContext context) { |
| final Set<PersistentMemberPattern> memberMissingIDs = new HashSet<PersistentMemberPattern>(); |
| Set<ColocatedRegionDetails> missingColocatedRegions = new HashSet<ColocatedRegionDetails>(); |
| |
| if (context == null) { |
| throw new RuntimeException(); |
| } |
| try { |
| final InternalCache cache = (InternalCache) context.getCache(); |
| |
| if (cache != null && !cache.isClosed()) { |
| final DistributedMember member = cache.getMyId(); |
| |
| // Missing DiskStores |
| PersistentMemberManager mm = cache.getPersistentMemberManager(); |
| Map<String, Set<PersistentMemberID>> waitingRegions = mm.getWaitingRegions(); |
| for (Set<PersistentMemberID> entry : waitingRegions.values()) { |
| for (PersistentMemberID id : entry) { |
| memberMissingIDs.add(new PersistentMemberPattern(id)); |
| } |
| } |
| // Missing colocated regions |
| Set<PartitionedRegion> prs = cache.getPartitionedRegions(); |
| for (PartitionedRegion pr : prs) { |
| List<String> missingChildRegions = pr.getMissingColocatedChildren(); |
| for (String child : missingChildRegions) { |
| missingColocatedRegions.add(new ColocatedRegionDetails(member.getHost(), |
| member.getName(), pr.getFullPath(), child)); |
| } |
| } |
| } |
| } catch (Exception e) { |
| context.getResultSender().sendException(e); |
| } |
| |
| if (memberMissingIDs.isEmpty() && missingColocatedRegions.isEmpty()) { |
| context.getResultSender().lastResult(null); |
| } else { |
| if (!memberMissingIDs.isEmpty()) { |
| if (missingColocatedRegions.isEmpty()) { |
| context.getResultSender().lastResult(memberMissingIDs); |
| } else { |
| context.getResultSender().sendResult(memberMissingIDs); |
| } |
| } |
| if (!missingColocatedRegions.isEmpty()) { |
| context.getResultSender().lastResult(missingColocatedRegions); |
| } |
| } |
| } |
| |
| @Override |
| public String getId() { |
| return getClass().getName(); |
| } |
| } |