| /* |
| * 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 java.io.DataInput; |
| import java.io.DataOutput; |
| import java.io.IOException; |
| |
| import org.apache.geode.DataSerializable; |
| 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.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.configuration.domain.XmlEntity; |
| import org.apache.geode.management.internal.functions.CliFunctionResult; |
| import org.apache.geode.management.internal.i18n.CliStrings; |
| |
| /** |
| * 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 implements Function, DataSerializable { |
| |
| 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(); |
| } |
| |
| /** |
| * @since GemFire 8.1 |
| */ |
| public static Object[] toArgs(final String regionName, final String value) { |
| return new Object[] {regionName, value}; |
| } |
| |
| @Override |
| public void toData(DataOutput out) throws IOException { |
| |
| } |
| |
| @Override |
| public void fromData(DataInput in) throws IOException, ClassNotFoundException { |
| |
| } |
| } |