| /* |
| * 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 org.apache.geode.annotations.Immutable; |
| import org.apache.geode.distributed.internal.DistributionStats; |
| import org.apache.geode.internal.cache.LocalRegion; |
| 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.security.AuthorizeRequest; |
| import org.apache.geode.internal.security.SecurityService; |
| import org.apache.geode.security.NotAuthorizedException; |
| import org.apache.geode.security.ResourcePermission.Operation; |
| import org.apache.geode.security.ResourcePermission.Resource; |
| |
| public class ContainsKey extends BaseCommand { |
| |
| @Immutable |
| private static final ContainsKey singleton = new ContainsKey(); |
| |
| public static Command getCommand() { |
| return singleton; |
| } |
| |
| private static void writeContainsKeyResponse(boolean containsKey, Message origMsg, |
| ServerConnection servConn) throws IOException { |
| Message responseMsg = servConn.getResponseMessage(); |
| responseMsg.setMessageType(MessageType.RESPONSE); |
| responseMsg.setNumberOfParts(1); |
| responseMsg.setTransactionId(origMsg.getTransactionId()); |
| responseMsg.addObjPart(containsKey ? Boolean.TRUE : Boolean.FALSE); |
| responseMsg.send(servConn); |
| } |
| |
| @Override |
| public void cmdExecute(final Message clientMessage, final ServerConnection serverConnection, |
| final SecurityService securityService, long start) throws IOException { |
| Part regionNamePart = null; |
| Part keyPart = null; |
| String regionName = null; |
| Object key = null; |
| |
| CacheServerStats stats = serverConnection.getCacheServerStats(); |
| |
| serverConnection.setAsTrue(REQUIRES_RESPONSE); |
| { |
| long oldStart = start; |
| start = DistributionStats.getStatTime(); |
| stats.incReadContainsKeyRequestTime(start - oldStart); |
| } |
| // Retrieve the data from the message parts |
| regionNamePart = clientMessage.getPart(0); |
| keyPart = clientMessage.getPart(1); |
| regionName = regionNamePart.getCachedString(); |
| try { |
| key = keyPart.getStringOrObject(); |
| } catch (Exception e) { |
| writeException(clientMessage, e, false, serverConnection); |
| serverConnection.setAsTrue(RESPONDED); |
| return; |
| } |
| if (logger.isDebugEnabled()) { |
| logger.debug("{}: Received containsKey request ({} bytes) from {} for region {} key {}", |
| serverConnection.getName(), clientMessage.getPayloadLength(), |
| serverConnection.getSocketString(), regionName, key); |
| } |
| |
| // Process the containsKey request |
| if (key == null || regionName == null) { |
| String errMessage = ""; |
| if (key == null) { |
| logger.warn("{}: The input key for the containsKey request is null", |
| serverConnection.getName()); |
| errMessage = "The input key for the containsKey request is null"; |
| } |
| if (regionName == null) { |
| logger.warn("{}: The input region name for the containsKey request is null", |
| serverConnection.getName()); |
| errMessage = |
| "The input region name for the containsKey request is null"; |
| } |
| writeErrorResponse(clientMessage, MessageType.CONTAINS_KEY_DATA_ERROR, errMessage, |
| serverConnection); |
| serverConnection.setAsTrue(RESPONDED); |
| return; |
| } |
| |
| LocalRegion region = (LocalRegion) serverConnection.getCache().getRegion(regionName); |
| if (region == null) { |
| String reason = |
| "was not found during containsKey request"; |
| writeRegionDestroyedEx(clientMessage, regionName, reason, serverConnection); |
| serverConnection.setAsTrue(RESPONDED); |
| return; |
| } |
| |
| try { |
| securityService.authorize(Resource.DATA, Operation.READ, regionName, key); |
| } catch (NotAuthorizedException ex) { |
| writeException(clientMessage, ex, false, serverConnection); |
| serverConnection.setAsTrue(RESPONDED); |
| return; |
| } |
| |
| AuthorizeRequest authzRequest = serverConnection.getAuthzRequest(); |
| if (authzRequest != null) { |
| try { |
| authzRequest.containsKeyAuthorize(regionName, key); |
| } catch (NotAuthorizedException ex) { |
| writeException(clientMessage, ex, false, serverConnection); |
| serverConnection.setAsTrue(RESPONDED); |
| return; |
| } |
| } |
| // Execute the containsKey |
| boolean containsKey = region.containsKey(key); |
| |
| // Update the statistics and write the reply |
| { |
| long oldStart = start; |
| start = DistributionStats.getStatTime(); |
| stats.incProcessContainsKeyTime(start - oldStart); |
| } |
| writeContainsKeyResponse(containsKey, clientMessage, serverConnection); |
| serverConnection.setAsTrue(RESPONDED); |
| if (logger.isDebugEnabled()) { |
| logger.debug("{}: Sent containsKey response for region {} key {}", serverConnection.getName(), |
| regionName, key); |
| } |
| stats.incWriteContainsKeyResponseTime(DistributionStats.getStatTime() - start); |
| } |
| |
| } |