/**
* 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.hive.ql.metadata.formatting;
import java.io.DataOutputStream;
import java.io.OutputStream;
import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.net.URLDecoder;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Set;
import com.facebook.presto.hive.shaded.org.apache.commons.lang.StringUtils;
import com.facebook.presto.hive.shaded.org.apache.commons.logging.Log;
import com.facebook.presto.hive.shaded.org.apache.commons.logging.LogFactory;
import org.apache.hadoop.fs.FileStatus;
import org.apache.hadoop.fs.FileSystem;
import org.apache.hadoop.fs.Path;
import org.apache.hadoop.hive.conf.HiveConf;
import org.apache.hadoop.hive.metastore.api.FieldSchema;
import org.apache.hadoop.hive.ql.metadata.Hive;
import org.apache.hadoop.hive.ql.metadata.HiveException;
import org.apache.hadoop.hive.ql.metadata.Partition;
import org.apache.hadoop.hive.ql.metadata.Table;
import org.apache.hadoop.hive.ql.session.SessionState.LogHelper;
import org.apache.hadoop.hive.shims.ShimLoader;
import com.facebook.presto.hive.shaded.org.codehaus.jackson.map.ObjectMapper;
/**
* Format table and index information for machine readability using
* json.
*/
public class JsonMetaDataFormatter implements MetaDataFormatter {
private static final Log LOG = LogFactory.getLog(JsonMetaDataFormatter.class);
/**
* Convert the map to a JSON string.
*/
private void asJson(OutputStream out, Map<String, Object> data)
throws HiveException
{
try {
new ObjectMapper().writeValue(out, data);
} catch (IOException e) {
throw new HiveException("Unable to convert to json", e);
}
}
/**
* Write an error message.
*/
@Override
public void error(OutputStream out, String msg, int errorCode, String sqlState)
throws HiveException
{
error(out, msg, errorCode, sqlState, null);
}
@Override
public void error(OutputStream out, String errorMessage, int errorCode, String sqlState, String errorDetail) throws HiveException {
MapBuilder mb = MapBuilder.create().put("error", errorMessage);
if(errorDetail != null) {
mb.put("errorDetail", errorDetail);
}
mb.put("errorCode", errorCode);
if(sqlState != null) {
mb.put("sqlState", sqlState);
}
asJson(out,mb.build());
}
/**
* Show a list of tables.
*/
@Override
public void showTables(DataOutputStream out, Set<String> tables)
throws HiveException
{
asJson(out,
MapBuilder.create()
.put("tables", tables)
.build());
}
/**
* Describe table.
*/
@Override
public void describeTable(DataOutputStream out,
String colPath, String tableName,
Table tbl, Partition part, List<FieldSchema> cols,
boolean isFormatted, boolean isExt,
boolean isPretty)
throws HiveException
{
MapBuilder builder = MapBuilder.create();
builder.put("columns", makeColsUnformatted(cols));
if (isExt) {
if (part != null)
builder.put("partitionInfo", part.getTPartition());
else
builder.put("tableInfo", tbl.getTTable());
}
asJson(out, builder.build());
}
private List<Map<String, Object>> makeColsUnformatted(List<FieldSchema> cols) {
ArrayList<Map<String, Object>> res = new ArrayList<Map<String, Object>>();
for (FieldSchema col : cols)
res.add(makeOneColUnformatted(col));
return res;
}
private Map<String, Object> makeOneColUnformatted(FieldSchema col) {
return MapBuilder.create()
.put("name", col.getName())
.put("type", col.getType())
.put("comment", col.getComment())
.build();
}
@Override
public void showTableStatus(DataOutputStream out,
Hive db,
HiveConf conf,
List<Table> tbls,
Map<String, String> part,
Partition par)
throws HiveException
{
asJson(out, MapBuilder
.create()
.put("tables", makeAllTableStatus(db, conf,
tbls, part, par))
.build());
}
private List<Map<String, Object>> makeAllTableStatus(Hive db,
HiveConf conf,
List<Table> tbls,
Map<String, String> part,
Partition par)
throws HiveException
{
try {
ArrayList<Map<String, Object>> res = new ArrayList<Map<String, Object>>();
for (Table tbl : tbls)
res.add(makeOneTableStatus(tbl, db, conf, part, par));
return res;
} catch(IOException e) {
throw new HiveException(e);
}
}
private Map<String, Object> makeOneTableStatus(Table tbl,
Hive db,
HiveConf conf,
Map<String, String> part,
Partition par)
throws HiveException, IOException
{
String tblLoc = null;
String inputFormattCls = null;
String outputFormattCls = null;
if (part != null) {
if (par != null) {
if (par.getLocation() != null) {
tblLoc = par.getDataLocation().toString();
}
inputFormattCls = par.getInputFormatClass().getName();
outputFormattCls = par.getOutputFormatClass().getName();
}
} else {
if (tbl.getPath() != null) {
tblLoc = tbl.getDataLocation().toString();
}
inputFormattCls = tbl.getInputFormatClass().getName();
outputFormattCls = tbl.getOutputFormatClass().getName();
}
MapBuilder builder = MapBuilder.create();
builder.put("tableName", tbl.getTableName());
builder.put("owner", tbl.getOwner());
builder.put("location", tblLoc);
builder.put("inputFormat", inputFormattCls);
builder.put("outputFormat", outputFormattCls);
builder.put("columns", makeColsUnformatted(tbl.getCols()));
builder.put("partitioned", tbl.isPartitioned());
if (tbl.isPartitioned())
builder.put("partitionColumns", makeColsUnformatted(tbl.getPartCols()));
putFileSystemsStats(builder, makeTableStatusLocations(tbl, db, par),
conf, tbl.getPath());
return builder.build();
}
private List<Path> makeTableStatusLocations(Table tbl, Hive db, Partition par)
throws HiveException
{
// output file system information
Path tblPath = tbl.getPath();
List<Path> locations = new ArrayList<Path>();
if (tbl.isPartitioned()) {
if (par == null) {
for (Partition curPart : db.getPartitions(tbl)) {
if (curPart.getLocation() != null) {
locations.add(new Path(curPart.getLocation()));
}
}
} else {
if (par.getLocation() != null) {
locations.add(new Path(par.getLocation()));
}
}
} else {
if (tblPath != null) {
locations.add(tblPath);
}
}
return locations;
}
// Duplicates logic in TextMetaDataFormatter
private void putFileSystemsStats(MapBuilder builder, List<Path> locations,
HiveConf conf, Path tblPath)
throws IOException
{
long totalFileSize = 0;
long maxFileSize = 0;
long minFileSize = Long.MAX_VALUE;
long lastAccessTime = 0;
long lastUpdateTime = 0;
int numOfFiles = 0;
boolean unknown = false;
FileSystem fs = tblPath.getFileSystem(conf);
// in case all files in locations do not exist
try {
FileStatus tmpStatus = fs.getFileStatus(tblPath);
lastAccessTime = ShimLoader.getHadoopShims().getAccessTime(tmpStatus);
lastUpdateTime = tmpStatus.getModificationTime();
} catch (IOException e) {
LOG.warn(
"Cannot access File System. File System status will be unknown: ", e);
unknown = true;
}
if (!unknown) {
for (Path loc : locations) {
try {
FileStatus status = fs.getFileStatus(tblPath);
FileStatus[] files = fs.listStatus(loc);
long accessTime = ShimLoader.getHadoopShims().getAccessTime(status);
long updateTime = status.getModificationTime();
// no matter loc is the table location or part location, it must be a
// directory.
if (!status.isDir()) {
continue;
}
if (accessTime > lastAccessTime) {
lastAccessTime = accessTime;
}
if (updateTime > lastUpdateTime) {
lastUpdateTime = updateTime;
}
for (FileStatus currentStatus : files) {
if (currentStatus.isDir()) {
continue;
}
numOfFiles++;
long fileLen = currentStatus.getLen();
totalFileSize += fileLen;
if (fileLen > maxFileSize) {
maxFileSize = fileLen;
}
if (fileLen < minFileSize) {
minFileSize = fileLen;
}
accessTime = ShimLoader.getHadoopShims().getAccessTime(
currentStatus);
updateTime = currentStatus.getModificationTime();
if (accessTime > lastAccessTime) {
lastAccessTime = accessTime;
}
if (updateTime > lastUpdateTime) {
lastUpdateTime = updateTime;
}
}
} catch (IOException e) {
// ignore
}
}
}
builder
.put("totalNumberFiles", numOfFiles, ! unknown)
.put("totalFileSize", totalFileSize, ! unknown)
.put("maxFileSize", maxFileSize, ! unknown)
.put("minFileSize", numOfFiles > 0 ? minFileSize : 0, ! unknown)
.put("lastAccessTime", lastAccessTime, ! (unknown || lastAccessTime < 0))
.put("lastUpdateTime", lastUpdateTime, ! unknown);
}
/**
* Show the table partitions.
*/
@Override
public void showTablePartitons(DataOutputStream out, List<String> parts)
throws HiveException
{
asJson(out,
MapBuilder.create()
.put("partitions", makeTablePartions(parts))
.build());
}
private List<Map<String, Object>> makeTablePartions(List<String> parts)
throws HiveException
{
try {
ArrayList<Map<String, Object>> res = new ArrayList<Map<String, Object>>();
for (String part : parts)
res.add(makeOneTablePartition(part));
return res;
} catch (UnsupportedEncodingException e) {
throw new HiveException(e);
}
}
// This seems like a very wrong implementation.
private Map<String, Object> makeOneTablePartition(String partIdent)
throws UnsupportedEncodingException
{
ArrayList<Map<String, Object>> res = new ArrayList<Map<String, Object>>();
ArrayList<String> names = new ArrayList<String>();
for (String part : StringUtils.split(partIdent, "/")) {
String name = part;
String val = null;
String[] kv = StringUtils.split(part, "=", 2);
if (kv != null) {
name = kv[0];
if (kv.length > 1)
val = URLDecoder.decode(kv[1], "UTF-8");
}
if (val != null)
names.add(name + "='" + val + "'");
else
names.add(name);
res.add(MapBuilder.create()
.put("columnName", name)
.put("columnValue", val)
.build());
}
return MapBuilder.create()
.put("name", StringUtils.join(names, ","))
.put("values", res)
.build();
}
/**
* Show a list of databases
*/
@Override
public void showDatabases(DataOutputStream out, List<String> databases)
throws HiveException
{
asJson(out,
MapBuilder.create()
.put("databases", databases)
.build());
}
/**
* Show the description of a database
*/
@Override
public void showDatabaseDescription(DataOutputStream out,
String database,
String comment,
String location,
Map<String, String> params)
throws HiveException
{
if (params == null || params.isEmpty()) {
asJson(out, MapBuilder
.create()
.put("database", database)
.put("comment", comment)
.put("location", location)
.build());
} else {
asJson(out, MapBuilder
.create()
.put("database", database)
.put("comment", comment)
.put("location", location)
.put("params", params)
.build());
}
}
}