LoadDataSourceFactory.cc

Go to the documentation of this file.

#include "Common/Compat.h"
#include "HqlInterpreter.h"
#include "LoadDataSourceFactory.h"
#include "LoadDataSourceFileLocal.h"
#include "LoadDataSourceFileDfs.h"
#include "LoadDataSourceStdin.h"

using namespace Hypertable;
using namespace std;

LoadDataSource *
LoadDataSourceFactory::create(FsBroker::Lib::ClientPtr &dfs_client,
    const string &input_fname, const int src,
    const string &header_fname, const int header_src,
    const std::vector<String> &key_columns, const string &timestamp_column,
    char field_separator, int row_uniquify_chars, int load_flags) {

  LoadDataSource *lds;

  switch (src) {
    case LOCAL_FILE:
      lds = new LoadDataSourceFileLocal(input_fname, header_fname,
                                        row_uniquify_chars, load_flags);
      break;
    case DFS_FILE:
      lds = new LoadDataSourceFileDfs(dfs_client, input_fname, header_fname,
                                      row_uniquify_chars, load_flags );
      break;
    case STDIN:
      lds = new LoadDataSourceStdin(header_fname, row_uniquify_chars, load_flags);
      break;
    default:
      HT_THROW(Error::HQL_PARSE_ERROR, "LOAD DATA - bad filename");
  }

  lds->init(key_columns, timestamp_column, field_separator);
  return lds;
}