| /* |
| * 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.internal.cache.tier.sockets.command; |
| |
| import java.io.IOException; |
| import java.nio.ByteBuffer; |
| |
| import org.apache.geode.annotations.Immutable; |
| import org.apache.geode.cache.EntryNotFoundException; |
| import org.apache.geode.cache.RegionDestroyedException; |
| import org.apache.geode.cache.client.internal.DestroyOp; |
| import org.apache.geode.cache.client.internal.InvalidateOp; |
| import org.apache.geode.cache.operations.InvalidateOperationContext; |
| import org.apache.geode.distributed.internal.DistributionStats; |
| import org.apache.geode.internal.cache.EventID; |
| import org.apache.geode.internal.cache.EventIDHolder; |
| import org.apache.geode.internal.cache.LocalRegion; |
| import org.apache.geode.internal.cache.PartitionedRegion; |
| import org.apache.geode.internal.cache.tier.Command; |
| import org.apache.geode.internal.cache.tier.MessageType; |
| import org.apache.geode.internal.cache.tier.sockets.BaseCommand; |
| import org.apache.geode.internal.cache.tier.sockets.CacheServerStats; |
| import org.apache.geode.internal.cache.tier.sockets.Message; |
| import org.apache.geode.internal.cache.tier.sockets.Part; |
| import org.apache.geode.internal.cache.tier.sockets.ServerConnection; |
| import org.apache.geode.internal.cache.versions.VersionTag; |
| import org.apache.geode.internal.security.AuthorizeRequest; |
| import org.apache.geode.internal.security.SecurityService; |
| import org.apache.geode.internal.util.Breadcrumbs; |
| import org.apache.geode.security.GemFireSecurityException; |
| import org.apache.geode.security.ResourcePermission; |
| |
| public class Invalidate70 extends BaseCommand { |
| |
| @Immutable |
| private static final Invalidate70 singleton = new Invalidate70(); |
| |
| public static Command getCommand() { |
| return singleton; |
| } |
| |
| private Invalidate70() {} |
| |
| @Override |
| public void cmdExecute(final Message clientMessage, final ServerConnection serverConnection, |
| final SecurityService securityService, long start) throws IOException, InterruptedException { |
| Part regionNamePart = null, keyPart = null, callbackArgPart = null; |
| String regionName = null; |
| Object callbackArg = null, key = null; |
| Part eventPart = null; |
| StringBuilder errMessage = new StringBuilder(); |
| CacheServerStats stats = serverConnection.getCacheServerStats(); |
| serverConnection.setAsTrue(REQUIRES_RESPONSE); |
| |
| { |
| long oldStart = start; |
| start = DistributionStats.getStatTime(); |
| stats.incReadInvalidateRequestTime(start - oldStart); |
| } |
| // Retrieve the data from the message parts |
| regionNamePart = clientMessage.getPart(0); |
| keyPart = clientMessage.getPart(1); |
| eventPart = clientMessage.getPart(2); |
| // callbackArgPart = null; (redundant assignment) |
| if (clientMessage.getNumberOfParts() > 3) { |
| callbackArgPart = clientMessage.getPart(3); |
| try { |
| callbackArg = callbackArgPart.getObject(); |
| } catch (Exception e) { |
| writeException(clientMessage, e, false, serverConnection); |
| serverConnection.setAsTrue(RESPONDED); |
| return; |
| } |
| } |
| regionName = regionNamePart.getCachedString(); |
| try { |
| key = keyPart.getStringOrObject(); |
| } catch (Exception e) { |
| writeException(clientMessage, e, false, serverConnection); |
| serverConnection.setAsTrue(RESPONDED); |
| return; |
| } |
| if (logger.isDebugEnabled()) { |
| logger.debug(serverConnection.getName() + ": Received invalidate request (" |
| + clientMessage.getPayloadLength() + " bytes) from " + serverConnection.getSocketString() |
| + " for region " + regionName + " key " + key); |
| } |
| |
| // Process the invalidate request |
| if (key == null || regionName == null) { |
| if (key == null) { |
| logger.warn("The input key for the invalidate request is null"); |
| errMessage.append("The input key for the invalidate request is null"); |
| } |
| if (regionName == null) { |
| logger.warn("The input region name for the invalidate request is null"); |
| errMessage |
| .append("The input region name for the invalidate request is null"); |
| } |
| writeErrorResponse(clientMessage, MessageType.DESTROY_DATA_ERROR, errMessage.toString(), |
| serverConnection); |
| serverConnection.setAsTrue(RESPONDED); |
| return; |
| } |
| LocalRegion region = (LocalRegion) serverConnection.getCache().getRegion(regionName); |
| if (region == null) { |
| String reason = " was not found during invalidate request"; |
| writeRegionDestroyedEx(clientMessage, regionName, reason, serverConnection); |
| serverConnection.setAsTrue(RESPONDED); |
| return; |
| } |
| // Invalidate the entry |
| ByteBuffer eventIdPartsBuffer = ByteBuffer.wrap(eventPart.getSerializedForm()); |
| long threadId = EventID.readEventIdPartsFromOptmizedByteArray(eventIdPartsBuffer); |
| long sequenceId = EventID.readEventIdPartsFromOptmizedByteArray(eventIdPartsBuffer); |
| EventID eventId = |
| new EventID(serverConnection.getEventMemberIDByteArray(), threadId, sequenceId); |
| |
| Breadcrumbs.setEventId(eventId); |
| |
| VersionTag tag = null; |
| |
| try { |
| // for integrated security |
| securityService.authorize(ResourcePermission.Resource.DATA, |
| ResourcePermission.Operation.WRITE, regionName, key); |
| |
| AuthorizeRequest authzRequest = serverConnection.getAuthzRequest(); |
| if (authzRequest != null) { |
| InvalidateOperationContext invalidateContext = |
| authzRequest.invalidateAuthorize(regionName, key, callbackArg); |
| callbackArg = invalidateContext.getCallbackArg(); |
| } |
| EventIDHolder clientEvent = new EventIDHolder(eventId); |
| |
| // msg.isRetry might be set by v7.0 and later clients |
| if (clientMessage.isRetry()) { |
| // if (logger.isDebugEnabled()) { |
| // logger.debug("DEBUG: encountered isRetry in Invalidate"); |
| // } |
| clientEvent.setPossibleDuplicate(true); |
| if (region.getAttributes().getConcurrencyChecksEnabled()) { |
| // recover the version tag from other servers |
| clientEvent.setRegion(region); |
| if (!recoverVersionTagForRetriedOperation(clientEvent)) { |
| clientEvent.setPossibleDuplicate(false); // no-one has seen this event |
| } |
| } |
| } |
| |
| region.basicBridgeInvalidate(key, callbackArg, serverConnection.getProxyID(), true, |
| clientEvent); |
| tag = clientEvent.getVersionTag(); |
| serverConnection.setModificationInfo(true, regionName, key); |
| } catch (EntryNotFoundException e) { |
| // Don't send an exception back to the client if this |
| // exception happens. Just log it and continue. |
| logger.info("During {} no entry was found for key {}", |
| new Object[] {"invalidate", key}); |
| } catch (RegionDestroyedException rde) { |
| writeException(clientMessage, rde, false, serverConnection); |
| serverConnection.setAsTrue(RESPONDED); |
| return; |
| } catch (Exception e) { |
| // If an interrupted exception is thrown , rethrow it |
| checkForInterrupt(serverConnection, e); |
| |
| // If an exception occurs during the destroy, preserve the connection |
| writeException(clientMessage, e, false, serverConnection); |
| serverConnection.setAsTrue(RESPONDED); |
| if (e instanceof GemFireSecurityException) { |
| // Fine logging for security exceptions since these are already |
| // logged by the security logger |
| if (logger.isDebugEnabled()) { |
| logger.debug("{}: Unexpected Security exception", serverConnection.getName(), e); |
| } |
| } else { |
| logger.warn(String.format("%s: Unexpected Exception", |
| serverConnection.getName()), e); |
| } |
| return; |
| } |
| |
| // Update the statistics and write the reply |
| { |
| long oldStart = start; |
| start = DistributionStats.getStatTime(); |
| stats.incProcessInvalidateTime(start - oldStart); |
| } |
| if (region instanceof PartitionedRegion) { |
| PartitionedRegion pr = (PartitionedRegion) region; |
| if (pr.getNetworkHopType() != PartitionedRegion.NETWORK_HOP_NONE) { |
| writeReplyWithRefreshMetadata(clientMessage, serverConnection, pr, pr.getNetworkHopType(), |
| tag); |
| pr.clearNetworkHopData(); |
| } else { |
| writeReply(clientMessage, serverConnection, tag); |
| } |
| } else { |
| writeReply(clientMessage, serverConnection, tag); |
| } |
| serverConnection.setAsTrue(RESPONDED); |
| if (logger.isDebugEnabled()) { |
| logger.debug("{}: Sent invalidate response for region {} key {}", serverConnection.getName(), |
| regionName, key); |
| } |
| stats.incWriteInvalidateResponseTime(DistributionStats.getStatTime() - start); |
| } |
| |
| protected void writeReplyWithRefreshMetadata(Message origMsg, ServerConnection servConn, |
| PartitionedRegion pr, byte nwHop, VersionTag versionTag) throws IOException { |
| Message replyMsg = servConn.getReplyMessage(); |
| servConn.getCache().getCancelCriterion().checkCancelInProgress(null); |
| replyMsg.setMessageType(MessageType.REPLY); |
| int flags = 0; |
| int numParts = 2; |
| if (versionTag != null) { |
| flags |= InvalidateOp.HAS_VERSION_TAG; |
| numParts++; |
| } |
| replyMsg.setNumberOfParts(numParts); |
| replyMsg.setTransactionId(origMsg.getTransactionId()); |
| replyMsg.addIntPart(flags); |
| if (versionTag != null) { |
| replyMsg.addObjPart(versionTag); |
| } |
| replyMsg.addBytesPart(new byte[] {pr.getMetadataVersion(), nwHop}); |
| pr.getPrStats().incPRMetaDataSentCount(); |
| replyMsg.send(servConn); |
| if (logger.isTraceEnabled()) { |
| logger.trace("{}: rpl with REFRESH_METADATA tx: {}", servConn.getName(), |
| origMsg.getTransactionId()); |
| } |
| } |
| |
| protected void writeReply(Message origMsg, ServerConnection servConn, VersionTag versionTag) |
| throws IOException { |
| Message replyMsg = servConn.getReplyMessage(); |
| servConn.getCache().getCancelCriterion().checkCancelInProgress(null); |
| replyMsg.setMessageType(MessageType.REPLY); |
| int flags = 0; |
| int numParts = 2; |
| if (versionTag != null) { |
| flags |= DestroyOp.HAS_VERSION_TAG; |
| numParts++; |
| } |
| replyMsg.setNumberOfParts(numParts); |
| replyMsg.setTransactionId(origMsg.getTransactionId()); |
| replyMsg.addIntPart(flags); |
| if (versionTag != null) { |
| if (logger.isDebugEnabled()) { |
| logger.debug("wrote version tag in response: {}", versionTag); |
| } |
| replyMsg.addObjPart(versionTag); |
| } else { |
| if (logger.isDebugEnabled()) { |
| logger.debug("response has no version tag"); |
| } |
| } |
| replyMsg.addBytesPart(okBytes()); // make old single-hop code happy by putting byte[]{0} here |
| replyMsg.send(servConn); |
| if (logger.isTraceEnabled()) { |
| logger.trace("{}: rpl tx: {} parts={}", servConn.getName(), origMsg.getTransactionId(), |
| replyMsg.getNumberOfParts()); |
| } |
| } |
| } |
