/*
* Copyright 2010 The Apache Software Foundation
*
* 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.hbase.rest.model;
import java.io.IOException;
import java.io.Serializable;
import java.util.ArrayList;
import java.util.List;
import javax.xml.bind.annotation.XmlAttribute;
import javax.xml.bind.annotation.XmlElement;
import javax.xml.bind.annotation.XmlRootElement;
import org.apache.hadoop.hbase.rest.ProtobufMessageHandler;
import org.apache.hadoop.hbase.rest.protobuf.generated.TableInfoMessage.TableInfo;
import com.google.protobuf.ByteString;
/**
* Representation of a list of table regions.
*
* <pre>
* <complexType name="TableInfo">
* <sequence>
* <element name="region" type="tns:TableRegion"
* maxOccurs="unbounded" minOccurs="1"></element>
* </sequence>
* <attribute name="name" type="string"></attribute>
* </complexType>
* </pre>
*/
@XmlRootElement(name="TableInfo")
public class TableInfoModel implements Serializable, ProtobufMessageHandler {
private static final long serialVersionUID = 1L;
private String name;
private List<TableRegionModel> regions = new ArrayList<TableRegionModel>();
/**
* Default constructor
*/
public TableInfoModel() {}
/**
* Constructor
* @param name
*/
public TableInfoModel(String name) {
this.name = name;
}
/**
* Add a region model to the list
* @param region the region
*/
public void add(TableRegionModel region) {
regions.add(region);
}
/**
* @param index the index
* @return the region model
*/
public TableRegionModel get(int index) {
return regions.get(index);
}
/**
* @return the table name
*/
@XmlAttribute
public String getName() {
return name;
}
/**
* @return the regions
*/
@XmlElement(name="Region")
public List<TableRegionModel> getRegions() {
return regions;
}
/**
* @param name the table name
*/
public void setName(String name) {
this.name = name;
}
/**
* @param regions the regions to set
*/
public void setRegions(List<TableRegionModel> regions) {
this.regions = regions;
}
/* (non-Javadoc)
* @see java.lang.Object#toString()
*/
@Override
public String toString() {
StringBuilder sb = new StringBuilder();
for(TableRegionModel aRegion : regions) {
sb.append(aRegion.toString());
sb.append('\n');
}
return sb.toString();
}
@Override
public byte[] createProtobufOutput() {
TableInfo.Builder builder = TableInfo.newBuilder();
builder.setName(name);
for (TableRegionModel aRegion: regions) {
TableInfo.Region.Builder regionBuilder = TableInfo.Region.newBuilder();
regionBuilder.setName(aRegion.getName());
regionBuilder.setId(aRegion.getId());
regionBuilder.setStartKey(ByteString.copyFrom(aRegion.getStartKey()));
regionBuilder.setEndKey(ByteString.copyFrom(aRegion.getEndKey()));
regionBuilder.setLocation(aRegion.getLocation());
builder.addRegions(regionBuilder);
}
return builder.build().toByteArray();
}
@Override
public ProtobufMessageHandler getObjectFromMessage(byte[] message)
throws IOException {
TableInfo.Builder builder = TableInfo.newBuilder();
builder.mergeFrom(message);
setName(builder.getName());
for (TableInfo.Region region: builder.getRegionsList()) {
add(new TableRegionModel(builder.getName(), region.getId(),
region.getStartKey().toByteArray(),
region.getEndKey().toByteArray(),
region.getLocation()));
}
return this;
}
}