001/** 002 * Licensed to the Apache Software Foundation (ASF) under one 003 * or more contributor license agreements. See the NOTICE file 004 * distributed with this work for additional information 005 * regarding copyright ownership. The ASF licenses this file 006 * to you under the Apache License, Version 2.0 (the 007 * "License"); you may not use this file except in compliance 008 * with the License. You may obtain a copy of the License at 009 * 010 * http://www.apache.org/licenses/LICENSE-2.0 011 * 012 * Unless required by applicable law or agreed to in writing, software 013 * distributed under the License is distributed on an "AS IS" BASIS, 014 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 015 * See the License for the specific language governing permissions and 016 * limitations under the License. 017 */ 018package org.apache.hadoop.hbase.rsgroup; 019 020import static org.junit.Assert.assertTrue; 021 022import java.io.IOException; 023import java.util.HashSet; 024import java.util.Set; 025import org.apache.hadoop.hbase.net.Address; 026import org.apache.yetus.audience.InterfaceAudience; 027import org.apache.yetus.audience.InterfaceStability; 028import org.slf4j.Logger; 029import org.slf4j.LoggerFactory; 030 031@InterfaceAudience.Private 032@InterfaceStability.Evolving 033public abstract class RSGroupTestingUtil { 034 private static final Logger LOG = LoggerFactory.getLogger(RSGroupTestingUtil.class); 035 036 private RSGroupTestingUtil() { 037 } 038 039 public static RSGroupInfo addRSGroup(final RSGroupAdmin rsGroupAdmin, String groupName, 040 int groupRSCount) throws IOException { 041 RSGroupInfo defaultInfo = rsGroupAdmin.getRSGroupInfo(RSGroupInfo.DEFAULT_GROUP); 042 assertTrue(defaultInfo != null); 043 assertTrue(defaultInfo.getServers().size() >= groupRSCount); 044 rsGroupAdmin.addRSGroup(groupName); 045 046 Set<Address> set = new HashSet<>(); 047 for(Address server: defaultInfo.getServers()) { 048 if(set.size() == groupRSCount) { 049 break; 050 } 051 set.add(server); 052 } 053 rsGroupAdmin.moveServers(set, groupName); 054 RSGroupInfo result = rsGroupAdmin.getRSGroupInfo(groupName); 055 assertTrue(result.getServers().size() >= groupRSCount); 056 return result; 057 } 058 059}