com.gemstone.gemfire.internal.cache.EventStateHelper Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of gemfire-core Show documentation
SnappyData store based off Pivotal GemFireXD
The newest version!
/*
 * Copyright (c) 2010-2015 Pivotal Software, Inc. All rights reserved.
 *
 * Licensed 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. See accompanying
 * LICENSE file.
 */
package com.gemstone.gemfire.internal.cache;

import com.gemstone.gemfire.DataSerializer;
import com.gemstone.gemfire.distributed.internal.InternalDistributedSystem;
import com.gemstone.gemfire.distributed.internal.membership.InternalDistributedMember;
import com.gemstone.gemfire.internal.InternalDataSerializer;
import com.gemstone.gemfire.internal.cache.CreateRegionProcessor.CreateRegionReplyMessage;
import com.gemstone.gemfire.internal.cache.EventTracker.EventSeqnoHolder;
import com.gemstone.gemfire.internal.cache.InitialImageOperation.RegionStateMessage;
import com.gemstone.gemfire.internal.cache.ha.HARegionQueue.DispatchedAndCurrentEvents;
import com.gemstone.gemfire.internal.cache.ha.ThreadIdentifier;

import java.io.DataInput;
import java.io.DataOutput;
import java.io.IOException;
import java.util.Arrays;
import java.util.HashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;

/**
 * A helper class for serializing the event state map.
 * 
 * TODO - Store the event state map in DataSerializable object
 * that keeps the map in this compressed format in memory.
 * @author dsmith
 *
 */
public class EventStateHelper {
  /**
   * Post 7.1, if changes are made to this method make sure that it is backwards
   * compatible by creating toDataPreXX methods. Also make sure that the callers
   * to this method are backwards compatible by creating toDataPreXX methods for
   * them even if they are not changed. 

   * Callers for this method are: 

   * {@link CreateRegionReplyMessage#toData(DataOutput)} 

   * {@link RegionStateMessage#toData(DataOutput)} 

   */
  @SuppressWarnings("synthetic-access")
  public static void toData(DataOutput dop, Map eventState, boolean isHARegion) throws IOException {
    //For HARegionQueues, the event state map is uses different values
    //than a regular region :(
    InternalDistributedMember myId = InternalDistributedSystem.getAnyInstance().getDistributedMember();
    Map> groupedThreadIds = groupThreadIds(eventState);
    List orderedIds = new LinkedList();
    Map seenIds = new HashMap();
    
    myId.writeEssentialData(dop); // added in 7.0 for version tag processing in fromData
    
    for(MemberIdentifier memberId : groupedThreadIds.keySet()) {
      if (!seenIds.containsKey(memberId)) {
        orderedIds.add(memberId);
        seenIds.put(memberId, Integer.valueOf(seenIds.size()));
      }
    }

    dop.writeInt(seenIds.size());
    for (MemberIdentifier memberId: orderedIds) {
      DataSerializer.writeByteArray(memberId.bytes, dop);
    }

    dop.writeInt(groupedThreadIds.size());
    for(Map.Entry> memberIdEntry : groupedThreadIds.entrySet()) {
      MemberIdentifier memberId = memberIdEntry.getKey();
      dop.writeInt(seenIds.get(memberId).intValue());
      Map threadIdMap = memberIdEntry.getValue();
      dop.writeInt(threadIdMap.size());
      for(Object next : threadIdMap.entrySet()) {
        Map.Entry entry = (Map.Entry) next;
        ThreadIdentifier key = (ThreadIdentifier) entry.getKey();
        dop.writeLong(key.getThreadID());
        if(isHARegion) {
          DispatchedAndCurrentEvents value = (DispatchedAndCurrentEvents) entry.getValue();
          InternalDataSerializer.invokeToData(value, dop);
        } else {
          EventSeqnoHolder value = (EventSeqnoHolder) entry.getValue();
          InternalDataSerializer.invokeToData(value, dop);
        }
      }
    }
    
  }
  
  /**
   * Post 7.1, if changes are made to this method make sure that it is backwards
   * compatible by creating fromDataPreXX methods. Also make sure that the callers
   * to this method are backwards compatible by creating fromDataPreXX methods for
   * them even if they are not changed. 

   * Callers for this method are: 

   * {@link CreateRegionReplyMessage#fromData(DataInput)} 

   * {@link RegionStateMessage#fromData(DataInput)} 

   */
  public static Map fromData(DataInput dip, boolean isHARegion) throws IOException, ClassNotFoundException {
    
    InternalDistributedMember senderId = InternalDistributedMember.readEssentialData(dip);
    
    int numIds = dip.readInt();
    Map numberToMember = new HashMap();
    for (int i=0; i> groupThreadIds(
      Map eventState) {
    Map> results 
    = new HashMap>();
    for(Object next : eventState.entrySet()) {
      Map.Entry entry = (Map.Entry) next;
      ThreadIdentifier key = (ThreadIdentifier) entry.getKey();
      MemberIdentifier memberId  = new MemberIdentifier(key.getMembershipID());
      Object value = entry.getValue();
      Map subMap = results.get(memberId);
      if(subMap == null) {
        subMap = new HashMap();
        results.put(memberId, subMap);
      }
      subMap.put(key, value);
    }
    
    return results;
  }
  
  private static class MemberIdentifier {
    private final byte[] bytes;

    public MemberIdentifier(byte[] bytes) {
      this.bytes = bytes;
    }

    @Override
    public int hashCode() {
      final int prime = 31;
      int result = 1;
      result = prime * result + Arrays.hashCode(bytes);
      return result;
    }

    @Override
    public boolean equals(Object obj) {
      if (this == obj)
        return true;
      if (obj == null)
        return false;
      if (!(obj instanceof MemberIdentifier))
        return false;
      MemberIdentifier other = (MemberIdentifier) obj;
      if (!Arrays.equals(bytes, other.bytes))
        return false;
      return true;
    }
  }
}