/* * 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.extension.mock; import org.apache.geode.cache.Cache; import org.apache.geode.cache.CacheFactory; import org.apache.geode.cache.Region; import org.apache.geode.cache.execute.Function; import org.apache.geode.cache.execute.FunctionAdapter; import org.apache.geode.cache.execute.FunctionContext; import org.apache.geode.cache.execute.FunctionException; import org.apache.geode.cache.execute.ResultSender; import org.apache.geode.internal.cache.extension.Extensible; import org.apache.geode.internal.cache.extension.Extension; import org.apache.geode.internal.cache.xmlcache.CacheXml; import org.apache.geode.management.internal.cli.CliUtil; import org.apache.geode.management.internal.cli.functions.CliFunctionResult; import org.apache.geode.management.internal.cli.i18n.CliStrings; import org.apache.geode.management.internal.configuration.domain.XmlEntity; /** * Function to alter {@link MockRegionExtension} on a {@link Region}. * * <dl> * <dt>Arguments:</dt> * <dd> * <dl> * <dt>{@link String} regionName</dt> * <dd>Name of region on which to create {@link MockCacheExtension}.</dd> * <dt>{@link String} value</dt> * <dd>Value to set. See {@link MockCacheExtension#setValue(String)}.</dd> * </dl> * </dt> * </dl> * * * @since GemFire 8.1 */ public class AlterMockRegionExtensionFunction extends FunctionAdapter { private static final long serialVersionUID = 1L; public static final Function INSTANCE = new AlterMockRegionExtensionFunction(); @Override public void execute(FunctionContext context) { final Cache cache = CacheFactory.getAnyInstance(); final Region<?, ?> region = cache.getRegion((String) ((Object[]) context.getArguments())[0]); if (!(region instanceof Extensible)) { throw new FunctionException("Not extensible region."); } final String value = (String) ((Object[]) context.getArguments())[1]; @SuppressWarnings("unchecked") final Extensible<Region<?, ?>> extensible = (Extensible<Region<?, ?>>) region; for (Extension<Region<?, ?>> extension : extensible.getExtensionPoint().getExtensions()) { if (extension instanceof MockRegionExtension) { ((MockRegionExtension) extension).setValue(value); } } XmlEntity xmlEntity = new XmlEntity(CacheXml.REGION, "name", region.getName()); final ResultSender<Object> resultSender = context.getResultSender(); final String memberNameOrId = CliUtil.getMemberNameOrId(cache.getDistributedSystem().getDistributedMember()); resultSender.lastResult(new CliFunctionResult(memberNameOrId, xmlEntity, CliStrings.format("Mock region extension \"{0}\" altered on \"{1}\"", new Object[] {region.getFullPath(), memberNameOrId}))); } @Override public String getId() { return AlterMockRegionExtensionFunction.class.getName(); } /** * @param regionName * @param value * @return * @since GemFire 8.1 */ public static Object[] toArgs(final String regionName, final String value) { return new Object[] {regionName, value}; } }