| /** |
| * 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.hadoop.yarn.server.timelineservice.storage.apptoflow; |
| |
| |
| import org.apache.hadoop.conf.Configuration; |
| import org.apache.hadoop.hbase.HColumnDescriptor; |
| import org.apache.hadoop.hbase.HTableDescriptor; |
| import org.apache.hadoop.hbase.TableName; |
| import org.apache.hadoop.hbase.client.Admin; |
| import org.apache.hadoop.hbase.regionserver.BloomType; |
| import org.apache.hadoop.yarn.conf.YarnConfiguration; |
| import org.apache.hadoop.yarn.server.timelineservice.storage.common.BaseTableRW; |
| import org.apache.hadoop.yarn.server.timelineservice.storage.common.TimelineHBaseSchemaConstants; |
| import org.slf4j.Logger; |
| import org.slf4j.LoggerFactory; |
| |
| import java.io.IOException; |
| |
| /** |
| * Create, read and write to the AppToFlow Table. |
| */ |
| public class AppToFlowTableRW extends BaseTableRW<AppToFlowTable> { |
| /** app_flow prefix. */ |
| private static final String PREFIX = |
| YarnConfiguration.TIMELINE_SERVICE_PREFIX + "app-flow"; |
| |
| /** config param name that specifies the app_flow table name. */ |
| public static final String TABLE_NAME_CONF_NAME = PREFIX + ".table.name"; |
| |
| /** default value for app_flow table name. */ |
| private static final String DEFAULT_TABLE_NAME = "timelineservice.app_flow"; |
| |
| private static final Logger LOG = |
| LoggerFactory.getLogger(AppToFlowTableRW.class); |
| |
| public AppToFlowTableRW() { |
| super(TABLE_NAME_CONF_NAME, DEFAULT_TABLE_NAME); |
| } |
| |
| /* |
| * (non-Javadoc) |
| * |
| * @see |
| * org.apache.hadoop.yarn.server.timelineservice.storage.BaseTableRW# |
| * createTable(org.apache.hadoop.hbase.client.Admin, |
| * org.apache.hadoop.conf.Configuration) |
| */ |
| public void createTable(Admin admin, Configuration hbaseConf) |
| throws IOException { |
| |
| TableName table = getTableName(hbaseConf); |
| if (admin.tableExists(table)) { |
| // do not disable / delete existing table |
| // similar to the approach taken by map-reduce jobs when |
| // output directory exists |
| throw new IOException("Table " + table.getNameAsString() |
| + " already exists."); |
| } |
| |
| HTableDescriptor appToFlowTableDescp = new HTableDescriptor(table); |
| HColumnDescriptor mappCF = |
| new HColumnDescriptor(AppToFlowColumnFamily.MAPPING.getBytes()); |
| mappCF.setBloomFilterType(BloomType.ROWCOL); |
| appToFlowTableDescp.addFamily(mappCF); |
| |
| appToFlowTableDescp |
| .setRegionSplitPolicyClassName( |
| "org.apache.hadoop.hbase.regionserver.KeyPrefixRegionSplitPolicy"); |
| appToFlowTableDescp.setValue("KeyPrefixRegionSplitPolicy.prefix_length", |
| TimelineHBaseSchemaConstants.USERNAME_SPLIT_KEY_PREFIX_LENGTH); |
| admin.createTable(appToFlowTableDescp, |
| TimelineHBaseSchemaConstants.getUsernameSplits()); |
| LOG.info("Status of table creation for " + table.getNameAsString() + "=" |
| + admin.tableExists(table)); |
| } |
| } |