Make Leveldb save data into HDFS files. You have to set USE_HDFS in your environment variable to compile leveldb with HDFS support.
Test Plan: Run benchmark. Differential Revision: https://reviews.facebook.net/D3549main
Dhruba Borthakur
13 years ago
parent
338939e5c1
commit
a35e574344
@ -0,0 +1,26 @@ |
|||||||
|
This directory contains the hdfs extensions needed to make leveldb store |
||||||
|
files in HDFS. |
||||||
|
|
||||||
|
The hdfs.h file is copied from the Apache Hadoop 1.0 source code. |
||||||
|
It defines the libhdfs library |
||||||
|
(http://hadoop.apache.org/common/docs/r0.20.2/libhdfs.html) to access |
||||||
|
data in HDFS. The libhdfs.a is copied from the Apache Hadoop 1.0 build. |
||||||
|
It implements the API defined in hdfs.h. If your hadoop cluster is running |
||||||
|
a different hadoop release, then install these two files manually from your |
||||||
|
hadoop distribution and then recompile leveldb. |
||||||
|
|
||||||
|
The env_hdfs.h file defines the leveldb objects that are needed to talk to an |
||||||
|
underlying filesystem. |
||||||
|
|
||||||
|
If you want to compile leveldb with hdfs support, please set the following |
||||||
|
enviroment variables appropriately: |
||||||
|
USE_HDFS=1 |
||||||
|
JAVA_HOME=/usr/local/jdk-6u22-64 |
||||||
|
LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$JAVA_HOME/jre/lib/amd64/server:$JAVA_HOME/jre/lib/amd64/ |
||||||
|
make clean all db_bench |
||||||
|
|
||||||
|
To run dbbench, |
||||||
|
set CLASSPATH to include your hadoop distribution |
||||||
|
db_bench --hdfs="hdfs://hbaseudbperf001.snc1.facebook.com:9000" |
||||||
|
|
||||||
|
|
||||||
@ -0,0 +1,253 @@ |
|||||||
|
/**
|
||||||
|
* 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. |
||||||
|
*/ |
||||||
|
// Copyright (c) 2012 Facebook. All rights reserved.
|
||||||
|
|
||||||
|
#ifndef LEVELDB_HDFS_FILE_H |
||||||
|
#define LEVELDB_HDFS_FILE_H |
||||||
|
|
||||||
|
#include <algorithm> |
||||||
|
#include <stdio.h> |
||||||
|
#include <sys/time.h> |
||||||
|
#include <time.h> |
||||||
|
#include <iostream> |
||||||
|
#include "leveldb/env.h" |
||||||
|
#include "leveldb/status.h" |
||||||
|
|
||||||
|
#ifdef USE_HDFS |
||||||
|
#include "hdfs/hdfs.h" |
||||||
|
|
||||||
|
namespace leveldb { |
||||||
|
|
||||||
|
static const std::string kProto = "hdfs://"; |
||||||
|
static const std::string pathsep = "/"; |
||||||
|
|
||||||
|
// Thrown during execution when there is an issue with the supplied
|
||||||
|
// arguments.
|
||||||
|
class HdfsUsageException : public std::exception { }; |
||||||
|
|
||||||
|
// A simple exception that indicates something went wrong that is not
|
||||||
|
// recoverable. The intention is for the message to be printed (with
|
||||||
|
// nothing else) and the process terminate.
|
||||||
|
class HdfsFatalException : public std::exception { |
||||||
|
public: |
||||||
|
explicit HdfsFatalException(const std::string& s) : what_(s) { } |
||||||
|
virtual ~HdfsFatalException() throw() { } |
||||||
|
virtual const char* what() const throw() { |
||||||
|
return what_.c_str(); |
||||||
|
} |
||||||
|
private: |
||||||
|
const std::string what_; |
||||||
|
}; |
||||||
|
|
||||||
|
//
|
||||||
|
// The HDFS environment for leveldb. This class overrides all the
|
||||||
|
// file/dir access methods and delegates the thread-mgmt methods to the
|
||||||
|
// default posix environment.
|
||||||
|
//
|
||||||
|
class HdfsEnv : public Env { |
||||||
|
|
||||||
|
public: |
||||||
|
HdfsEnv(const std::string& fsname) : fsname_(fsname) { |
||||||
|
posixEnv = Env::Default(); |
||||||
|
fileSys_ = connectToPath(fsname_); |
||||||
|
} |
||||||
|
|
||||||
|
virtual ~HdfsEnv() { |
||||||
|
fprintf(stderr, "Destroying HdfsEnv::Default()\n"); |
||||||
|
hdfsDisconnect(fileSys_); |
||||||
|
} |
||||||
|
|
||||||
|
virtual Status NewSequentialFile(const std::string& fname, |
||||||
|
SequentialFile** result); |
||||||
|
|
||||||
|
virtual Status NewRandomAccessFile(const std::string& fname, |
||||||
|
RandomAccessFile** result); |
||||||
|
|
||||||
|
virtual Status NewWritableFile(const std::string& fname, |
||||||
|
WritableFile** result); |
||||||
|
|
||||||
|
virtual bool FileExists(const std::string& fname); |
||||||
|
|
||||||
|
virtual Status GetChildren(const std::string& path, |
||||||
|
std::vector<std::string>* result); |
||||||
|
|
||||||
|
virtual Status DeleteFile(const std::string& fname); |
||||||
|
|
||||||
|
virtual Status CreateDir(const std::string& name); |
||||||
|
|
||||||
|
virtual Status DeleteDir(const std::string& name); |
||||||
|
|
||||||
|
virtual Status GetFileSize(const std::string& fname, uint64_t* size); |
||||||
|
|
||||||
|
virtual Status RenameFile(const std::string& src, const std::string& target); |
||||||
|
|
||||||
|
virtual Status LockFile(const std::string& fname, FileLock** lock); |
||||||
|
|
||||||
|
virtual Status UnlockFile(FileLock* lock); |
||||||
|
|
||||||
|
virtual Status NewLogger(const std::string& fname, Logger** result); |
||||||
|
|
||||||
|
virtual void Schedule( void (*function)(void* arg), void* arg) { |
||||||
|
posixEnv->Schedule(function, arg); |
||||||
|
} |
||||||
|
|
||||||
|
virtual void StartThread(void (*function)(void* arg), void* arg) { |
||||||
|
posixEnv->StartThread(function, arg); |
||||||
|
} |
||||||
|
|
||||||
|
virtual Status GetTestDirectory(std::string* path) { |
||||||
|
return posixEnv->GetTestDirectory(path); |
||||||
|
} |
||||||
|
|
||||||
|
virtual uint64_t NowMicros() { |
||||||
|
return posixEnv->NowMicros(); |
||||||
|
} |
||||||
|
|
||||||
|
virtual void SleepForMicroseconds(int micros) { |
||||||
|
posixEnv->SleepForMicroseconds(micros); |
||||||
|
} |
||||||
|
|
||||||
|
static uint64_t gettid() { |
||||||
|
assert(sizeof(pthread_t) <= sizeof(uint64_t)); |
||||||
|
return (uint64_t)pthread_self(); |
||||||
|
} |
||||||
|
|
||||||
|
private: |
||||||
|
std::string fsname_; // string of the form "hdfs://hostname:port/"
|
||||||
|
hdfsFS fileSys_; // a single FileSystem object for all files
|
||||||
|
Env* posixEnv; // This object is derived from Env, but not from
|
||||||
|
// posixEnv. We have posixnv as an encapsulated
|
||||||
|
// object here so that we can use posix timers,
|
||||||
|
// posix threads, etc.
|
||||||
|
|
||||||
|
/**
|
||||||
|
* If the URI is specified of the form hdfs://server:port/path,
|
||||||
|
* then connect to the specified cluster |
||||||
|
* else connect to default. |
||||||
|
*/ |
||||||
|
hdfsFS connectToPath(const std::string& uri) { |
||||||
|
if (uri.empty()) { |
||||||
|
return NULL; |
||||||
|
} |
||||||
|
if (uri.find(kProto) != 0) { |
||||||
|
// uri doesn't start with hdfs:// -> use default:0, which is special
|
||||||
|
// to libhdfs.
|
||||||
|
return hdfsConnectNewInstance("default", 0); |
||||||
|
} |
||||||
|
const std::string hostport = uri.substr(kProto.length()); |
||||||
|
|
||||||
|
std::vector <std::string> parts; |
||||||
|
split(hostport, ':', parts); |
||||||
|
if (parts.size() != 2) { |
||||||
|
throw HdfsFatalException("Bad uri for hdfs " + uri); |
||||||
|
} |
||||||
|
// parts[0] = hosts, parts[1] = port/xxx/yyy
|
||||||
|
std::string host(parts[0]); |
||||||
|
std::string remaining(parts[1]); |
||||||
|
|
||||||
|
int rem = remaining.find(pathsep); |
||||||
|
std::string portStr = (rem == 0 ? remaining : |
||||||
|
remaining.substr(0, rem)); |
||||||
|
|
||||||
|
tPort port; |
||||||
|
port = atoi(portStr.c_str()); |
||||||
|
if (port == 0) { |
||||||
|
throw HdfsFatalException("Bad host-port for hdfs " + uri); |
||||||
|
} |
||||||
|
hdfsFS fs = hdfsConnectNewInstance(host.c_str(), port); |
||||||
|
return fs; |
||||||
|
} |
||||||
|
|
||||||
|
void split(const std::string &s, char delim,
|
||||||
|
std::vector<std::string> &elems) { |
||||||
|
elems.clear(); |
||||||
|
size_t prev = 0; |
||||||
|
size_t pos = s.find(delim); |
||||||
|
while (pos != std::string::npos) { |
||||||
|
elems.push_back(s.substr(prev, pos)); |
||||||
|
prev = pos + 1; |
||||||
|
pos = s.find(delim, prev); |
||||||
|
} |
||||||
|
elems.push_back(s.substr(prev, s.size())); |
||||||
|
} |
||||||
|
}; |
||||||
|
|
||||||
|
} // namespace leveldb
|
||||||
|
|
||||||
|
#else // USE_HDFS
|
||||||
|
|
||||||
|
|
||||||
|
namespace leveldb { |
||||||
|
|
||||||
|
class HdfsEnv : public Env { |
||||||
|
|
||||||
|
public: |
||||||
|
HdfsEnv(const std::string& fsname) { |
||||||
|
fprintf(stderr, "You have not build leveldb with HDFS support\n"); |
||||||
|
fprintf(stderr, "Please see hdfs/README for details\n"); |
||||||
|
throw new std::exception(); |
||||||
|
} |
||||||
|
|
||||||
|
virtual ~HdfsEnv() { |
||||||
|
} |
||||||
|
|
||||||
|
virtual Status NewSequentialFile(const std::string& fname, |
||||||
|
SequentialFile** result); |
||||||
|
|
||||||
|
virtual Status NewRandomAccessFile(const std::string& fname, |
||||||
|
RandomAccessFile** result) {} |
||||||
|
|
||||||
|
virtual Status NewWritableFile(const std::string& fname, |
||||||
|
WritableFile** result){} |
||||||
|
|
||||||
|
virtual bool FileExists(const std::string& fname){} |
||||||
|
|
||||||
|
virtual Status GetChildren(const std::string& path, |
||||||
|
std::vector<std::string>* result){} |
||||||
|
|
||||||
|
virtual Status DeleteFile(const std::string& fname){} |
||||||
|
|
||||||
|
virtual Status CreateDir(const std::string& name){} |
||||||
|
|
||||||
|
virtual Status DeleteDir(const std::string& name){} |
||||||
|
|
||||||
|
virtual Status GetFileSize(const std::string& fname, uint64_t* size){} |
||||||
|
|
||||||
|
virtual Status RenameFile(const std::string& src, const std::string& target){} |
||||||
|
|
||||||
|
virtual Status LockFile(const std::string& fname, FileLock** lock){} |
||||||
|
|
||||||
|
virtual Status UnlockFile(FileLock* lock){} |
||||||
|
|
||||||
|
virtual Status NewLogger(const std::string& fname, Logger** result){} |
||||||
|
|
||||||
|
virtual void Schedule( void (*function)(void* arg), void* arg) {} |
||||||
|
|
||||||
|
virtual void StartThread(void (*function)(void* arg), void* arg) {} |
||||||
|
|
||||||
|
virtual Status GetTestDirectory(std::string* path) {} |
||||||
|
|
||||||
|
virtual uint64_t NowMicros() {} |
||||||
|
|
||||||
|
virtual void SleepForMicroseconds(int micros) {} |
||||||
|
}; |
||||||
|
} |
||||||
|
|
||||||
|
#endif // USE_HDFS
|
||||||
|
|
||||||
|
#endif // LEVELDB_HDFS_FILE_H
|
||||||
@ -0,0 +1,490 @@ |
|||||||
|
/**
|
||||||
|
* 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. |
||||||
|
*/ |
||||||
|
|
||||||
|
#ifndef LIBHDFS_HDFS_H |
||||||
|
#define LIBHDFS_HDFS_H |
||||||
|
|
||||||
|
#include <sys/types.h> |
||||||
|
#include <sys/stat.h> |
||||||
|
|
||||||
|
#include <fcntl.h> |
||||||
|
#include <stdio.h> |
||||||
|
#include <stdint.h> |
||||||
|
#include <string.h> |
||||||
|
#include <stdlib.h> |
||||||
|
#include <time.h> |
||||||
|
#include <errno.h> |
||||||
|
|
||||||
|
#include <jni.h> |
||||||
|
|
||||||
|
#ifndef O_RDONLY |
||||||
|
#define O_RDONLY 1 |
||||||
|
#endif |
||||||
|
|
||||||
|
#ifndef O_WRONLY |
||||||
|
#define O_WRONLY 2 |
||||||
|
#endif |
||||||
|
|
||||||
|
#ifndef EINTERNAL |
||||||
|
#define EINTERNAL 255 |
||||||
|
#endif |
||||||
|
|
||||||
|
|
||||||
|
/** All APIs set errno to meaningful values */ |
||||||
|
#ifdef __cplusplus |
||||||
|
extern "C" { |
||||||
|
#endif |
||||||
|
|
||||||
|
/**
|
||||||
|
* Some utility decls used in libhdfs. |
||||||
|
*/ |
||||||
|
|
||||||
|
typedef int32_t tSize; /// size of data for read/write io ops
|
||||||
|
typedef time_t tTime; /// time type in seconds
|
||||||
|
typedef int64_t tOffset;/// offset within the file
|
||||||
|
typedef uint16_t tPort; /// port
|
||||||
|
typedef enum tObjectKind { |
||||||
|
kObjectKindFile = 'F', |
||||||
|
kObjectKindDirectory = 'D', |
||||||
|
} tObjectKind; |
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* The C reflection of org.apache.org.hadoop.FileSystem . |
||||||
|
*/ |
||||||
|
typedef void* hdfsFS; |
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* The C equivalent of org.apache.org.hadoop.FSData(Input|Output)Stream . |
||||||
|
*/ |
||||||
|
enum hdfsStreamType |
||||||
|
{ |
||||||
|
UNINITIALIZED = 0, |
||||||
|
INPUT = 1, |
||||||
|
OUTPUT = 2, |
||||||
|
}; |
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* The 'file-handle' to a file in hdfs. |
||||||
|
*/ |
||||||
|
struct hdfsFile_internal { |
||||||
|
void* file; |
||||||
|
enum hdfsStreamType type; |
||||||
|
}; |
||||||
|
typedef struct hdfsFile_internal* hdfsFile; |
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* hdfsConnectAsUser - Connect to a hdfs file system as a specific user |
||||||
|
* Connect to the hdfs. |
||||||
|
* @param host A string containing either a host name, or an ip address |
||||||
|
* of the namenode of a hdfs cluster. 'host' should be passed as NULL if |
||||||
|
* you want to connect to local filesystem. 'host' should be passed as |
||||||
|
* 'default' (and port as 0) to used the 'configured' filesystem |
||||||
|
* (core-site/core-default.xml). |
||||||
|
* @param port The port on which the server is listening. |
||||||
|
* @param user the user name (this is hadoop domain user). Or NULL is equivelant to hhdfsConnect(host, port) |
||||||
|
* @param groups the groups (these are hadoop domain groups) |
||||||
|
* @return Returns a handle to the filesystem or NULL on error. |
||||||
|
*/ |
||||||
|
hdfsFS hdfsConnectAsUser(const char* host, tPort port, const char *user , const char *groups[], int groups_size ); |
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* hdfsConnect - Connect to a hdfs file system. |
||||||
|
* Connect to the hdfs. |
||||||
|
* @param host A string containing either a host name, or an ip address |
||||||
|
* of the namenode of a hdfs cluster. 'host' should be passed as NULL if |
||||||
|
* you want to connect to local filesystem. 'host' should be passed as |
||||||
|
* 'default' (and port as 0) to used the 'configured' filesystem |
||||||
|
* (core-site/core-default.xml). |
||||||
|
* @param port The port on which the server is listening. |
||||||
|
* @return Returns a handle to the filesystem or NULL on error. |
||||||
|
*/ |
||||||
|
hdfsFS hdfsConnect(const char* host, tPort port); |
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* This are the same as hdfsConnectAsUser except that every invocation returns a new FileSystem handle. |
||||||
|
* Applications should call a hdfsDisconnect for every call to hdfsConnectAsUserNewInstance. |
||||||
|
*/ |
||||||
|
hdfsFS hdfsConnectAsUserNewInstance(const char* host, tPort port, const char *user , const char *groups[], int groups_size ); |
||||||
|
hdfsFS hdfsConnectNewInstance(const char* host, tPort port); |
||||||
|
hdfsFS hdfsConnectPath(const char* uri); |
||||||
|
|
||||||
|
/**
|
||||||
|
* hdfsDisconnect - Disconnect from the hdfs file system. |
||||||
|
* Disconnect from hdfs. |
||||||
|
* @param fs The configured filesystem handle. |
||||||
|
* @return Returns 0 on success, -1 on error.
|
||||||
|
*/ |
||||||
|
int hdfsDisconnect(hdfsFS fs); |
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* hdfsOpenFile - Open a hdfs file in given mode. |
||||||
|
* @param fs The configured filesystem handle. |
||||||
|
* @param path The full path to the file. |
||||||
|
* @param flags - an | of bits/fcntl.h file flags - supported flags are O_RDONLY, O_WRONLY (meaning create or overwrite i.e., implies O_TRUNCAT),
|
||||||
|
* O_WRONLY|O_APPEND. Other flags are generally ignored other than (O_RDWR || (O_EXCL & O_CREAT)) which return NULL and set errno equal ENOTSUP. |
||||||
|
* @param bufferSize Size of buffer for read/write - pass 0 if you want |
||||||
|
* to use the default configured values. |
||||||
|
* @param replication Block replication - pass 0 if you want to use |
||||||
|
* the default configured values. |
||||||
|
* @param blocksize Size of block - pass 0 if you want to use the |
||||||
|
* default configured values. |
||||||
|
* @return Returns the handle to the open file or NULL on error. |
||||||
|
*/ |
||||||
|
hdfsFile hdfsOpenFile(hdfsFS fs, const char* path, int flags, |
||||||
|
int bufferSize, short replication, tSize blocksize); |
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* hdfsCloseFile - Close an open file.
|
||||||
|
* @param fs The configured filesystem handle. |
||||||
|
* @param file The file handle. |
||||||
|
* @return Returns 0 on success, -1 on error.
|
||||||
|
*/ |
||||||
|
int hdfsCloseFile(hdfsFS fs, hdfsFile file); |
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* hdfsExists - Checks if a given path exsits on the filesystem
|
||||||
|
* @param fs The configured filesystem handle. |
||||||
|
* @param path The path to look for |
||||||
|
* @return Returns 0 on exists, 1 on non-exists, -1/-2 on error.
|
||||||
|
*/ |
||||||
|
int hdfsExists(hdfsFS fs, const char *path); |
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* hdfsSeek - Seek to given offset in file.
|
||||||
|
* This works only for files opened in read-only mode.
|
||||||
|
* @param fs The configured filesystem handle. |
||||||
|
* @param file The file handle. |
||||||
|
* @param desiredPos Offset into the file to seek into. |
||||||
|
* @return Returns 0 on success, -1 on error.
|
||||||
|
*/ |
||||||
|
int hdfsSeek(hdfsFS fs, hdfsFile file, tOffset desiredPos);
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* hdfsTell - Get the current offset in the file, in bytes. |
||||||
|
* @param fs The configured filesystem handle. |
||||||
|
* @param file The file handle. |
||||||
|
* @return Current offset, -1 on error. |
||||||
|
*/ |
||||||
|
tOffset hdfsTell(hdfsFS fs, hdfsFile file); |
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* hdfsRead - Read data from an open file. |
||||||
|
* @param fs The configured filesystem handle. |
||||||
|
* @param file The file handle. |
||||||
|
* @param buffer The buffer to copy read bytes into. |
||||||
|
* @param length The length of the buffer. |
||||||
|
* @return Returns the number of bytes actually read, possibly less |
||||||
|
* than than length;-1 on error. |
||||||
|
*/ |
||||||
|
tSize hdfsRead(hdfsFS fs, hdfsFile file, void* buffer, tSize length); |
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* hdfsPread - Positional read of data from an open file. |
||||||
|
* @param fs The configured filesystem handle. |
||||||
|
* @param file The file handle. |
||||||
|
* @param position Position from which to read |
||||||
|
* @param buffer The buffer to copy read bytes into. |
||||||
|
* @param length The length of the buffer. |
||||||
|
* @return Returns the number of bytes actually read, possibly less than |
||||||
|
* than length;-1 on error. |
||||||
|
*/ |
||||||
|
tSize hdfsPread(hdfsFS fs, hdfsFile file, tOffset position, |
||||||
|
void* buffer, tSize length); |
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* hdfsWrite - Write data into an open file. |
||||||
|
* @param fs The configured filesystem handle. |
||||||
|
* @param file The file handle. |
||||||
|
* @param buffer The data. |
||||||
|
* @param length The no. of bytes to write.
|
||||||
|
* @return Returns the number of bytes written, -1 on error. |
||||||
|
*/ |
||||||
|
tSize hdfsWrite(hdfsFS fs, hdfsFile file, const void* buffer, |
||||||
|
tSize length); |
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* hdfsWrite - Flush the data.
|
||||||
|
* @param fs The configured filesystem handle. |
||||||
|
* @param file The file handle. |
||||||
|
* @return Returns 0 on success, -1 on error.
|
||||||
|
*/ |
||||||
|
int hdfsFlush(hdfsFS fs, hdfsFile file); |
||||||
|
|
||||||
|
/**
|
||||||
|
* hdfsSync - Sync the data to persistent store. |
||||||
|
* @param fs The configured filesystem handle. |
||||||
|
* @param file The file handle. |
||||||
|
* @return Returns 0 on success, -1 on error. |
||||||
|
*/ |
||||||
|
int hdfsSync(hdfsFS fs, hdfsFile file); |
||||||
|
|
||||||
|
/**
|
||||||
|
* hdfsGetNumReplicasInPipeline - get number of remaining replicas in
|
||||||
|
* pipeline |
||||||
|
* @param fs The configured filesystem handle |
||||||
|
* @param file the file handle |
||||||
|
* @return returns the # of datanodes in the write pipeline; -1 on error |
||||||
|
*/ |
||||||
|
int hdfsGetNumCurrentReplicas(hdfsFS, hdfsFile file); |
||||||
|
|
||||||
|
/**
|
||||||
|
* hdfsAvailable - Number of bytes that can be read from this |
||||||
|
* input stream without blocking. |
||||||
|
* @param fs The configured filesystem handle. |
||||||
|
* @param file The file handle. |
||||||
|
* @return Returns available bytes; -1 on error.
|
||||||
|
*/ |
||||||
|
int hdfsAvailable(hdfsFS fs, hdfsFile file); |
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* hdfsCopy - Copy file from one filesystem to another. |
||||||
|
* @param srcFS The handle to source filesystem. |
||||||
|
* @param src The path of source file.
|
||||||
|
* @param dstFS The handle to destination filesystem. |
||||||
|
* @param dst The path of destination file.
|
||||||
|
* @return Returns 0 on success, -1 on error.
|
||||||
|
*/ |
||||||
|
int hdfsCopy(hdfsFS srcFS, const char* src, hdfsFS dstFS, const char* dst); |
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* hdfsMove - Move file from one filesystem to another. |
||||||
|
* @param srcFS The handle to source filesystem. |
||||||
|
* @param src The path of source file.
|
||||||
|
* @param dstFS The handle to destination filesystem. |
||||||
|
* @param dst The path of destination file.
|
||||||
|
* @return Returns 0 on success, -1 on error.
|
||||||
|
*/ |
||||||
|
int hdfsMove(hdfsFS srcFS, const char* src, hdfsFS dstFS, const char* dst); |
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* hdfsDelete - Delete file.
|
||||||
|
* @param fs The configured filesystem handle. |
||||||
|
* @param path The path of the file.
|
||||||
|
* @return Returns 0 on success, -1 on error.
|
||||||
|
*/ |
||||||
|
int hdfsDelete(hdfsFS fs, const char* path); |
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* hdfsRename - Rename file.
|
||||||
|
* @param fs The configured filesystem handle. |
||||||
|
* @param oldPath The path of the source file.
|
||||||
|
* @param newPath The path of the destination file.
|
||||||
|
* @return Returns 0 on success, -1 on error.
|
||||||
|
*/ |
||||||
|
int hdfsRename(hdfsFS fs, const char* oldPath, const char* newPath); |
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* hdfsGetWorkingDirectory - Get the current working directory for |
||||||
|
* the given filesystem. |
||||||
|
* @param fs The configured filesystem handle. |
||||||
|
* @param buffer The user-buffer to copy path of cwd into.
|
||||||
|
* @param bufferSize The length of user-buffer. |
||||||
|
* @return Returns buffer, NULL on error. |
||||||
|
*/ |
||||||
|
char* hdfsGetWorkingDirectory(hdfsFS fs, char *buffer, size_t bufferSize); |
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* hdfsSetWorkingDirectory - Set the working directory. All relative |
||||||
|
* paths will be resolved relative to it. |
||||||
|
* @param fs The configured filesystem handle. |
||||||
|
* @param path The path of the new 'cwd'.
|
||||||
|
* @return Returns 0 on success, -1 on error.
|
||||||
|
*/ |
||||||
|
int hdfsSetWorkingDirectory(hdfsFS fs, const char* path); |
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* hdfsCreateDirectory - Make the given file and all non-existent |
||||||
|
* parents into directories. |
||||||
|
* @param fs The configured filesystem handle. |
||||||
|
* @param path The path of the directory.
|
||||||
|
* @return Returns 0 on success, -1 on error.
|
||||||
|
*/ |
||||||
|
int hdfsCreateDirectory(hdfsFS fs, const char* path); |
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* hdfsSetReplication - Set the replication of the specified |
||||||
|
* file to the supplied value |
||||||
|
* @param fs The configured filesystem handle. |
||||||
|
* @param path The path of the file.
|
||||||
|
* @return Returns 0 on success, -1 on error.
|
||||||
|
*/ |
||||||
|
int hdfsSetReplication(hdfsFS fs, const char* path, int16_t replication); |
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* hdfsFileInfo - Information about a file/directory. |
||||||
|
*/ |
||||||
|
typedef struct { |
||||||
|
tObjectKind mKind; /* file or directory */ |
||||||
|
char *mName; /* the name of the file */ |
||||||
|
tTime mLastMod; /* the last modification time for the file in seconds */ |
||||||
|
tOffset mSize; /* the size of the file in bytes */ |
||||||
|
short mReplication; /* the count of replicas */ |
||||||
|
tOffset mBlockSize; /* the block size for the file */ |
||||||
|
char *mOwner; /* the owner of the file */ |
||||||
|
char *mGroup; /* the group associated with the file */ |
||||||
|
short mPermissions; /* the permissions associated with the file */ |
||||||
|
tTime mLastAccess; /* the last access time for the file in seconds */ |
||||||
|
} hdfsFileInfo; |
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* hdfsListDirectory - Get list of files/directories for a given |
||||||
|
* directory-path. hdfsFreeFileInfo should be called to deallocate memory if |
||||||
|
* the function returns non-NULL value. |
||||||
|
* @param fs The configured filesystem handle. |
||||||
|
* @param path The path of the directory.
|
||||||
|
* @param numEntries Set to the number of files/directories in path. |
||||||
|
* @return Returns a dynamically-allocated array of hdfsFileInfo |
||||||
|
* objects; NULL if empty or on error. |
||||||
|
* on error, numEntries will be -1. |
||||||
|
*/ |
||||||
|
hdfsFileInfo *hdfsListDirectory(hdfsFS fs, const char* path, |
||||||
|
int *numEntries); |
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* hdfsGetPathInfo - Get information about a path as a (dynamically |
||||||
|
* allocated) single hdfsFileInfo struct. hdfsFreeFileInfo should be |
||||||
|
* called when the pointer is no longer needed. |
||||||
|
* @param fs The configured filesystem handle. |
||||||
|
* @param path The path of the file.
|
||||||
|
* @return Returns a dynamically-allocated hdfsFileInfo object; |
||||||
|
* NULL on error. |
||||||
|
*/ |
||||||
|
hdfsFileInfo *hdfsGetPathInfo(hdfsFS fs, const char* path); |
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* hdfsFreeFileInfo - Free up the hdfsFileInfo array (including fields)
|
||||||
|
* @param hdfsFileInfo The array of dynamically-allocated hdfsFileInfo |
||||||
|
* objects. |
||||||
|
* @param numEntries The size of the array. |
||||||
|
*/ |
||||||
|
void hdfsFreeFileInfo(hdfsFileInfo *hdfsFileInfo, int numEntries); |
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* hdfsGetHosts - Get hostnames where a particular block (determined by |
||||||
|
* pos & blocksize) of a file is stored. The last element in the array |
||||||
|
* is NULL. Due to replication, a single block could be present on |
||||||
|
* multiple hosts. |
||||||
|
* @param fs The configured filesystem handle. |
||||||
|
* @param path The path of the file.
|
||||||
|
* @param start The start of the block. |
||||||
|
* @param length The length of the block. |
||||||
|
* @return Returns a dynamically-allocated 2-d array of blocks-hosts; |
||||||
|
* NULL on error. |
||||||
|
*/ |
||||||
|
char*** hdfsGetHosts(hdfsFS fs, const char* path,
|
||||||
|
tOffset start, tOffset length); |
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* hdfsFreeHosts - Free up the structure returned by hdfsGetHosts |
||||||
|
* @param hdfsFileInfo The array of dynamically-allocated hdfsFileInfo |
||||||
|
* objects. |
||||||
|
* @param numEntries The size of the array. |
||||||
|
*/ |
||||||
|
void hdfsFreeHosts(char ***blockHosts); |
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* hdfsGetDefaultBlockSize - Get the optimum blocksize. |
||||||
|
* @param fs The configured filesystem handle. |
||||||
|
* @return Returns the blocksize; -1 on error.
|
||||||
|
*/ |
||||||
|
tOffset hdfsGetDefaultBlockSize(hdfsFS fs); |
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* hdfsGetCapacity - Return the raw capacity of the filesystem.
|
||||||
|
* @param fs The configured filesystem handle. |
||||||
|
* @return Returns the raw-capacity; -1 on error.
|
||||||
|
*/ |
||||||
|
tOffset hdfsGetCapacity(hdfsFS fs); |
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* hdfsGetUsed - Return the total raw size of all files in the filesystem. |
||||||
|
* @param fs The configured filesystem handle. |
||||||
|
* @return Returns the total-size; -1 on error.
|
||||||
|
*/ |
||||||
|
tOffset hdfsGetUsed(hdfsFS fs); |
||||||
|
|
||||||
|
/**
|
||||||
|
* hdfsChown
|
||||||
|
* @param fs The configured filesystem handle. |
||||||
|
* @param path the path to the file or directory |
||||||
|
* @param owner this is a string in Hadoop land. Set to null or "" if only setting group |
||||||
|
* @param group this is a string in Hadoop land. Set to null or "" if only setting user |
||||||
|
* @return 0 on success else -1 |
||||||
|
*/ |
||||||
|
int hdfsChown(hdfsFS fs, const char* path, const char *owner, const char *group); |
||||||
|
|
||||||
|
/**
|
||||||
|
* hdfsChmod |
||||||
|
* @param fs The configured filesystem handle. |
||||||
|
* @param path the path to the file or directory |
||||||
|
* @param mode the bitmask to set it to |
||||||
|
* @return 0 on success else -1 |
||||||
|
*/ |
||||||
|
int hdfsChmod(hdfsFS fs, const char* path, short mode); |
||||||
|
|
||||||
|
/**
|
||||||
|
* hdfsUtime |
||||||
|
* @param fs The configured filesystem handle. |
||||||
|
* @param path the path to the file or directory |
||||||
|
* @param mtime new modification time or 0 for only set access time in seconds |
||||||
|
* @param atime new access time or 0 for only set modification time in seconds |
||||||
|
* @return 0 on success else -1 |
||||||
|
*/ |
||||||
|
int hdfsUtime(hdfsFS fs, const char* path, tTime mtime, tTime atime); |
||||||
|
|
||||||
|
#ifdef __cplusplus |
||||||
|
} |
||||||
|
#endif |
||||||
|
|
||||||
|
#endif /*LIBHDFS_HDFS_H*/ |
||||||
|
|
||||||
|
/**
|
||||||
|
* vim: ts=4: sw=4: et |
||||||
|
*/ |
||||||
@ -0,0 +1,499 @@ |
|||||||
|
/**
|
||||||
|
* 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. |
||||||
|
*/ |
||||||
|
// Copyright (c) 2012 Facebook. All rights reserved.
|
||||||
|
|
||||||
|
#ifdef USE_HDFS |
||||||
|
#ifndef LEVELDB_HDFS_FILE_C |
||||||
|
#define LEVELDB_HDFS_FILE_C |
||||||
|
|
||||||
|
#include <algorithm> |
||||||
|
#include <stdio.h> |
||||||
|
#include <sys/time.h> |
||||||
|
#include <time.h> |
||||||
|
#include <iostream> |
||||||
|
#include <sstream> |
||||||
|
#include "leveldb/env.h" |
||||||
|
#include "leveldb/status.h" |
||||||
|
#include "hdfs/hdfs.h" |
||||||
|
#include "hdfs/env_hdfs.h" |
||||||
|
|
||||||
|
//
|
||||||
|
// This file defines an HDFS environment for leveldb. It uses the libhdfs
|
||||||
|
// api to access HDFS. All HDFS files created by one instance of leveldb
|
||||||
|
// will reside on the same HDFS cluster.
|
||||||
|
//
|
||||||
|
|
||||||
|
namespace leveldb { |
||||||
|
|
||||||
|
namespace { |
||||||
|
|
||||||
|
// Log error message
|
||||||
|
static Status IOError(const std::string& context, int err_number) { |
||||||
|
return Status::IOError(context, strerror(err_number)); |
||||||
|
} |
||||||
|
|
||||||
|
// assume that there is one global logger for now. It is not thread-safe,
|
||||||
|
// but need not be because the logger is initialized at db-open time.
|
||||||
|
static Logger* mylog = NULL; |
||||||
|
|
||||||
|
// Used for reading a file from HDFS. It implements both sequential-read
|
||||||
|
// access methods as well as random read access methods.
|
||||||
|
class HdfsReadableFile: virtual public SequentialFile, virtual public RandomAccessFile { |
||||||
|
private: |
||||||
|
hdfsFS fileSys_; |
||||||
|
std::string filename_; |
||||||
|
hdfsFile hfile_; |
||||||
|
|
||||||
|
public: |
||||||
|
HdfsReadableFile(hdfsFS fileSys, const std::string& fname) |
||||||
|
: fileSys_(fileSys), filename_(fname), hfile_(NULL) { |
||||||
|
Log(mylog, "[hdfs] HdfsReadableFile opening file %s\n", |
||||||
|
filename_.c_str()); |
||||||
|
hfile_ = hdfsOpenFile(fileSys_, filename_.c_str(), O_RDONLY, 0, 0, 0); |
||||||
|
Log(mylog, "[hdfs] HdfsReadableFile opened file %s hfile_=0x%p\n",
|
||||||
|
filename_.c_str(), hfile_); |
||||||
|
} |
||||||
|
|
||||||
|
virtual ~HdfsReadableFile() { |
||||||
|
Log(mylog, "[hdfs] HdfsReadableFile closing file %s\n",
|
||||||
|
filename_.c_str()); |
||||||
|
hdfsCloseFile(fileSys_, hfile_); |
||||||
|
Log(mylog, "[hdfs] HdfsReadableFile closed file %s\n",
|
||||||
|
filename_.c_str()); |
||||||
|
hfile_ = NULL; |
||||||
|
} |
||||||
|
|
||||||
|
bool isValid() { |
||||||
|
return hfile_ != NULL; |
||||||
|
} |
||||||
|
|
||||||
|
// sequential access, read data at current offset in file
|
||||||
|
virtual Status Read(size_t n, Slice* result, char* scratch) { |
||||||
|
Status s; |
||||||
|
Log(mylog, "[hdfs] HdfsReadableFile reading %s %ld\n",
|
||||||
|
filename_.c_str(), n); |
||||||
|
size_t bytes_read = hdfsRead(fileSys_, hfile_, scratch, (tSize)n); |
||||||
|
Log(mylog, "[hdfs] HdfsReadableFile read %s\n", filename_.c_str()); |
||||||
|
*result = Slice(scratch, bytes_read); |
||||||
|
if (bytes_read < n) { |
||||||
|
if (feof()) { |
||||||
|
// We leave status as ok if we hit the end of the file
|
||||||
|
} else { |
||||||
|
// A partial read with an error: return a non-ok status
|
||||||
|
s = IOError(filename_, errno); |
||||||
|
} |
||||||
|
} |
||||||
|
return s; |
||||||
|
} |
||||||
|
|
||||||
|
// random access, read data from specified offset in file
|
||||||
|
virtual Status Read(uint64_t offset, size_t n, Slice* result, |
||||||
|
char* scratch) const { |
||||||
|
Status s; |
||||||
|
Log(mylog, "[hdfs] HdfsReadableFile preading %s\n", filename_.c_str()); |
||||||
|
ssize_t bytes_read = hdfsPread(fileSys_, hfile_, offset,
|
||||||
|
(void*)scratch, (tSize)n); |
||||||
|
Log(mylog, "[hdfs] HdfsReadableFile pread %s\n", filename_.c_str()); |
||||||
|
*result = Slice(scratch, (bytes_read < 0) ? 0 : bytes_read); |
||||||
|
if (bytes_read < 0) { |
||||||
|
// An error: return a non-ok status
|
||||||
|
s = IOError(filename_, errno); |
||||||
|
} |
||||||
|
return s; |
||||||
|
} |
||||||
|
|
||||||
|
virtual Status Skip(uint64_t n) { |
||||||
|
Log(mylog, "[hdfs] HdfsReadableFile skip %s\n", filename_.c_str()); |
||||||
|
// get current offset from file
|
||||||
|
tOffset current = hdfsTell(fileSys_, hfile_); |
||||||
|
if (current < 0) { |
||||||
|
return IOError(filename_, errno); |
||||||
|
} |
||||||
|
// seek to new offset in file
|
||||||
|
tOffset newoffset = current + n; |
||||||
|
int val = hdfsSeek(fileSys_, hfile_, newoffset); |
||||||
|
if (val < 0) { |
||||||
|
return IOError(filename_, errno); |
||||||
|
} |
||||||
|
return Status::OK(); |
||||||
|
} |
||||||
|
|
||||||
|
private: |
||||||
|
|
||||||
|
// returns true if we are at the end of file, false otherwise
|
||||||
|
bool feof() { |
||||||
|
Log(mylog, "[hdfs] HdfsReadableFile feof %s\n", filename_.c_str()); |
||||||
|
if (hdfsTell(fileSys_, hfile_) == fileSize()) { |
||||||
|
return true; |
||||||
|
} |
||||||
|
return false; |
||||||
|
} |
||||||
|
|
||||||
|
// the current size of the file
|
||||||
|
tOffset fileSize() { |
||||||
|
Log(mylog, "[hdfs] HdfsReadableFile fileSize %s\n", filename_.c_str()); |
||||||
|
hdfsFileInfo* pFileInfo = hdfsGetPathInfo(fileSys_, filename_.c_str()); |
||||||
|
tOffset size = 0L; |
||||||
|
if (pFileInfo != NULL) { |
||||||
|
size = pFileInfo->mSize; |
||||||
|
hdfsFreeFileInfo(pFileInfo, 1); |
||||||
|
} else { |
||||||
|
throw new leveldb::HdfsFatalException("fileSize on unknown file " +
|
||||||
|
filename_); |
||||||
|
} |
||||||
|
return size; |
||||||
|
} |
||||||
|
}; |
||||||
|
|
||||||
|
// Appends to an existing file in HDFS.
|
||||||
|
class HdfsWritableFile: public WritableFile { |
||||||
|
private: |
||||||
|
hdfsFS fileSys_; |
||||||
|
std::string filename_; |
||||||
|
hdfsFile hfile_; |
||||||
|
|
||||||
|
public: |
||||||
|
HdfsWritableFile(hdfsFS fileSys, const std::string& fname) |
||||||
|
: fileSys_(fileSys), filename_(fname) , hfile_(NULL) { |
||||||
|
Log(mylog, "[hdfs] HdfsWritableFile opening %s\n", filename_.c_str()); |
||||||
|
hfile_ = hdfsOpenFile(fileSys_, filename_.c_str(), O_WRONLY, 0, 0, 0); |
||||||
|
Log(mylog, "[hdfs] HdfsWritableFile opened %s\n", filename_.c_str()); |
||||||
|
assert(hfile_ != NULL); |
||||||
|
} |
||||||
|
virtual ~HdfsWritableFile() { |
||||||
|
if (hfile_ != NULL) { |
||||||
|
Log(mylog, "[hdfs] HdfsWritableFile closing %s\n", filename_.c_str()); |
||||||
|
hdfsCloseFile(fileSys_, hfile_); |
||||||
|
Log(mylog, "[hdfs] HdfsWritableFile closed %s\n", filename_.c_str()); |
||||||
|
hfile_ = NULL; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// If the file was successfully created, then this returns true.
|
||||||
|
// Otherwise returns false.
|
||||||
|
bool isValid() { |
||||||
|
return hfile_ != NULL; |
||||||
|
} |
||||||
|
|
||||||
|
// The name of the file, mostly needed for debug logging.
|
||||||
|
const std::string& getName() { |
||||||
|
return filename_; |
||||||
|
} |
||||||
|
|
||||||
|
virtual Status Append(const Slice& data) { |
||||||
|
Log(mylog, "[hdfs] HdfsWritableFile Append %s\n", filename_.c_str()); |
||||||
|
const char* src = data.data(); |
||||||
|
size_t left = data.size(); |
||||||
|
size_t ret = hdfsWrite(fileSys_, hfile_, src, left); |
||||||
|
Log(mylog, "[hdfs] HdfsWritableFile Appended %s\n", filename_.c_str()); |
||||||
|
if (ret != left) { |
||||||
|
return IOError(filename_, errno); |
||||||
|
} |
||||||
|
return Status::OK(); |
||||||
|
} |
||||||
|
|
||||||
|
virtual Status Flush() { |
||||||
|
return Status::OK(); |
||||||
|
} |
||||||
|
|
||||||
|
virtual Status Sync() { |
||||||
|
Status s; |
||||||
|
Log(mylog, "[hdfs] HdfsWritableFile Sync %s\n", filename_.c_str()); |
||||||
|
if (hdfsFlush(fileSys_, hfile_) == -1) { |
||||||
|
return IOError(filename_, errno); |
||||||
|
} |
||||||
|
if (hdfsSync(fileSys_, hfile_) == -1) { |
||||||
|
return IOError(filename_, errno); |
||||||
|
} |
||||||
|
Log(mylog, "[hdfs] HdfsWritableFile Synced %s\n", filename_.c_str()); |
||||||
|
return Status::OK(); |
||||||
|
} |
||||||
|
|
||||||
|
// This is used by HdfsLogger to write data to the debug log file
|
||||||
|
virtual Status Append(const char* src, size_t size) { |
||||||
|
if (hdfsWrite(fileSys_, hfile_, src, size) != (tSize)size) { |
||||||
|
return IOError(filename_, errno); |
||||||
|
} |
||||||
|
return Status::OK(); |
||||||
|
} |
||||||
|
|
||||||
|
virtual Status Close() { |
||||||
|
Log(mylog, "[hdfs] HdfsWritableFile closing %s\n", filename_.c_str()); |
||||||
|
if (hdfsCloseFile(fileSys_, hfile_) != 0) { |
||||||
|
return IOError(filename_, errno); |
||||||
|
} |
||||||
|
Log(mylog, "[hdfs] HdfsWritableFile closed %s\n", filename_.c_str()); |
||||||
|
hfile_ = NULL; |
||||||
|
return Status::OK(); |
||||||
|
} |
||||||
|
}; |
||||||
|
|
||||||
|
// The object that implements the debug logs to reside in HDFS.
|
||||||
|
class HdfsLogger : public Logger { |
||||||
|
private: |
||||||
|
HdfsWritableFile* file_; |
||||||
|
uint64_t (*gettid_)(); // Return the thread id for the current thread
|
||||||
|
|
||||||
|
public: |
||||||
|
HdfsLogger(HdfsWritableFile* f, uint64_t (*gettid)())
|
||||||
|
: file_(f), gettid_(gettid) { |
||||||
|
Log(mylog, "[hdfs] HdfsLogger opened %s\n",
|
||||||
|
file_->getName().c_str()); |
||||||
|
} |
||||||
|
|
||||||
|
virtual ~HdfsLogger() { |
||||||
|
Log(mylog, "[hdfs] HdfsLogger closed %s\n",
|
||||||
|
file_->getName().c_str()); |
||||||
|
delete file_; |
||||||
|
if (mylog != NULL && mylog == this) { |
||||||
|
mylog = NULL; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
virtual void Logv(const char* format, va_list ap) { |
||||||
|
const uint64_t thread_id = (*gettid_)(); |
||||||
|
|
||||||
|
// We try twice: the first time with a fixed-size stack allocated buffer,
|
||||||
|
// and the second time with a much larger dynamically allocated buffer.
|
||||||
|
char buffer[500]; |
||||||
|
for (int iter = 0; iter < 2; iter++) { |
||||||
|
char* base; |
||||||
|
int bufsize; |
||||||
|
if (iter == 0) { |
||||||
|
bufsize = sizeof(buffer); |
||||||
|
base = buffer; |
||||||
|
} else { |
||||||
|
bufsize = 30000; |
||||||
|
base = new char[bufsize]; |
||||||
|
} |
||||||
|
char* p = base; |
||||||
|
char* limit = base + bufsize; |
||||||
|
|
||||||
|
struct timeval now_tv; |
||||||
|
gettimeofday(&now_tv, NULL); |
||||||
|
const time_t seconds = now_tv.tv_sec; |
||||||
|
struct tm t; |
||||||
|
localtime_r(&seconds, &t); |
||||||
|
p += snprintf(p, limit - p, |
||||||
|
"%04d/%02d/%02d-%02d:%02d:%02d.%06d %llx ", |
||||||
|
t.tm_year + 1900, |
||||||
|
t.tm_mon + 1, |
||||||
|
t.tm_mday, |
||||||
|
t.tm_hour, |
||||||
|
t.tm_min, |
||||||
|
t.tm_sec, |
||||||
|
static_cast<int>(now_tv.tv_usec), |
||||||
|
static_cast<long long unsigned int>(thread_id)); |
||||||
|
|
||||||
|
// Print the message
|
||||||
|
if (p < limit) { |
||||||
|
va_list backup_ap; |
||||||
|
va_copy(backup_ap, ap); |
||||||
|
p += vsnprintf(p, limit - p, format, backup_ap); |
||||||
|
va_end(backup_ap); |
||||||
|
} |
||||||
|
|
||||||
|
// Truncate to available space if necessary
|
||||||
|
if (p >= limit) { |
||||||
|
if (iter == 0) { |
||||||
|
continue; // Try again with larger buffer
|
||||||
|
} else { |
||||||
|
p = limit - 1; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// Add newline if necessary
|
||||||
|
if (p == base || p[-1] != '\n') { |
||||||
|
*p++ = '\n'; |
||||||
|
} |
||||||
|
|
||||||
|
assert(p <= limit); |
||||||
|
file_->Append(base, p-base); |
||||||
|
file_->Flush(); |
||||||
|
if (base != buffer) { |
||||||
|
delete[] base; |
||||||
|
} |
||||||
|
break; |
||||||
|
} |
||||||
|
} |
||||||
|
}; |
||||||
|
|
||||||
|
} // namespace
|
||||||
|
|
||||||
|
// Finally, the hdfs environment
|
||||||
|
|
||||||
|
// open a file for sequential reading
|
||||||
|
Status HdfsEnv::NewSequentialFile(const std::string& fname, |
||||||
|
SequentialFile** result) { |
||||||
|
HdfsReadableFile* f = new HdfsReadableFile(fileSys_, fname); |
||||||
|
if (f == NULL) { |
||||||
|
*result = NULL; |
||||||
|
return IOError(fname, errno); |
||||||
|
} |
||||||
|
*result = dynamic_cast<SequentialFile*>(f); |
||||||
|
return Status::OK(); |
||||||
|
} |
||||||
|
|
||||||
|
// open a file for random reading
|
||||||
|
Status HdfsEnv::NewRandomAccessFile(const std::string& fname, |
||||||
|
RandomAccessFile** result) { |
||||||
|
HdfsReadableFile* f = new HdfsReadableFile(fileSys_, fname); |
||||||
|
if (f == NULL) { |
||||||
|
*result = NULL; |
||||||
|
return IOError(fname, errno); |
||||||
|
} |
||||||
|
*result = dynamic_cast<RandomAccessFile*>(f); |
||||||
|
return Status::OK(); |
||||||
|
} |
||||||
|
|
||||||
|
// create a new file for writing
|
||||||
|
Status HdfsEnv::NewWritableFile(const std::string& fname, |
||||||
|
WritableFile** result) { |
||||||
|
Status s; |
||||||
|
HdfsWritableFile* f = new HdfsWritableFile(fileSys_, fname); |
||||||
|
if (f == NULL || !f->isValid()) { |
||||||
|
*result = NULL; |
||||||
|
return IOError(fname, errno); |
||||||
|
} |
||||||
|
*result = dynamic_cast<WritableFile*>(f); |
||||||
|
return Status::OK(); |
||||||
|
} |
||||||
|
|
||||||
|
bool HdfsEnv::FileExists(const std::string& fname) { |
||||||
|
int value = hdfsExists(fileSys_, fname.c_str()); |
||||||
|
if (value == 0) { |
||||||
|
return true; |
||||||
|
} |
||||||
|
return false; |
||||||
|
} |
||||||
|
|
||||||
|
Status HdfsEnv::GetChildren(const std::string& path, |
||||||
|
std::vector<std::string>* result) { |
||||||
|
int value = hdfsExists(fileSys_, path.c_str()); |
||||||
|
switch (value) { |
||||||
|
case 0: { |
||||||
|
int numEntries = 0; |
||||||
|
hdfsFileInfo* pHdfsFileInfo = 0; |
||||||
|
pHdfsFileInfo = hdfsListDirectory(fileSys_, path.c_str(), &numEntries); |
||||||
|
if (numEntries >= 0) { |
||||||
|
for(int i = 0; i < numEntries; i++) { |
||||||
|
char* pathname = pHdfsFileInfo[i].mName; |
||||||
|
char* filename = rindex(pathname, '/'); |
||||||
|
if (filename != NULL) { |
||||||
|
result->push_back(filename+1); |
||||||
|
} |
||||||
|
} |
||||||
|
if (pHdfsFileInfo != NULL) { |
||||||
|
hdfsFreeFileInfo(pHdfsFileInfo, numEntries); |
||||||
|
} |
||||||
|
} else { |
||||||
|
// numEntries < 0 indicates error
|
||||||
|
Log(mylog, "hdfsListDirectory call failed with error "); |
||||||
|
throw HdfsFatalException("hdfsListDirectory call failed negative error.\n"); |
||||||
|
} |
||||||
|
break; |
||||||
|
} |
||||||
|
case 1: // directory does not exist, exit
|
||||||
|
break; |
||||||
|
default: // anything else should be an error
|
||||||
|
Log(mylog, "hdfsListDirectory call failed with error "); |
||||||
|
throw HdfsFatalException("hdfsListDirectory call failed with error.\n"); |
||||||
|
} |
||||||
|
return Status::OK(); |
||||||
|
} |
||||||
|
|
||||||
|
Status HdfsEnv::DeleteFile(const std::string& fname) { |
||||||
|
if (hdfsDelete(fileSys_, fname.c_str()) == 0) { |
||||||
|
return Status::OK(); |
||||||
|
} |
||||||
|
return IOError(fname, errno); |
||||||
|
}; |
||||||
|
|
||||||
|
Status HdfsEnv::CreateDir(const std::string& name) { |
||||||
|
if (hdfsCreateDirectory(fileSys_, name.c_str()) == 0) { |
||||||
|
return Status::OK(); |
||||||
|
} |
||||||
|
return IOError(name, errno); |
||||||
|
}; |
||||||
|
|
||||||
|
Status HdfsEnv::DeleteDir(const std::string& name) { |
||||||
|
return DeleteFile(name); |
||||||
|
}; |
||||||
|
|
||||||
|
Status HdfsEnv::GetFileSize(const std::string& fname, uint64_t* size) { |
||||||
|
*size = 0L; |
||||||
|
hdfsFileInfo* pFileInfo = hdfsGetPathInfo(fileSys_, fname.c_str()); |
||||||
|
if (pFileInfo != NULL) { |
||||||
|
*size = pFileInfo->mSize; |
||||||
|
hdfsFreeFileInfo(pFileInfo, 1); |
||||||
|
return Status::OK(); |
||||||
|
} |
||||||
|
return IOError(fname, errno); |
||||||
|
} |
||||||
|
|
||||||
|
// The rename is not atomic. HDFS does not allow a renaming if the
|
||||||
|
// target already exists. So, we delete the target before attemting the
|
||||||
|
// rename.
|
||||||
|
Status HdfsEnv::RenameFile(const std::string& src, const std::string& target) { |
||||||
|
hdfsDelete(fileSys_, target.c_str()); |
||||||
|
if (hdfsRename(fileSys_, src.c_str(), target.c_str()) == 0) { |
||||||
|
return Status::OK(); |
||||||
|
} |
||||||
|
return IOError(src, errno); |
||||||
|
} |
||||||
|
|
||||||
|
Status HdfsEnv::LockFile(const std::string& fname, FileLock** lock) { |
||||||
|
// there isn's a very good way to atomically check and create
|
||||||
|
// a file via libhdfs
|
||||||
|
*lock = NULL; |
||||||
|
return Status::OK(); |
||||||
|
} |
||||||
|
|
||||||
|
Status HdfsEnv::UnlockFile(FileLock* lock) { |
||||||
|
return Status::OK(); |
||||||
|
} |
||||||
|
|
||||||
|
Status HdfsEnv::NewLogger(const std::string& fname, Logger** result) { |
||||||
|
HdfsWritableFile* f = new HdfsWritableFile(fileSys_, fname); |
||||||
|
if (f == NULL || !f->isValid()) { |
||||||
|
*result = NULL; |
||||||
|
return IOError(fname, errno); |
||||||
|
} |
||||||
|
HdfsLogger* h = new HdfsLogger(f, &HdfsEnv::gettid); |
||||||
|
*result = h; |
||||||
|
if (mylog == NULL) { |
||||||
|
// mylog = h; // uncomment this for detailed logging
|
||||||
|
} |
||||||
|
return Status::OK(); |
||||||
|
} |
||||||
|
|
||||||
|
} // namespace leveldb
|
||||||
|
|
||||||
|
#endif // LEVELDB_HDFS_FILE_C
|
||||||
|
|
||||||
|
#else // USE_HDFS
|
||||||
|
|
||||||
|
// dummy placeholders used when HDFS is not available
|
||||||
|
#include "leveldb/env.h" |
||||||
|
#include "hdfs/env_hdfs.h" |
||||||
|
namespace leveldb { |
||||||
|
Status HdfsEnv::NewSequentialFile(const std::string& fname, |
||||||
|
SequentialFile** result) {} |
||||||
|
} |
||||||
|
|
||||||
|
#endif |
||||||
Loading…
Reference in new issue