Function fillTimestampValues

tools/src/CSVFileImport.cc:220–250

parse timestamp values in seconds

// parse timestamp values in seconds
void fillTimestampValues(const std::vector<std::string>& data, orc::ColumnVectorBatch* batch,
                         uint64_t numValues, uint64_t colIndex) {
  struct tm timeStruct;
  orc::TimestampVectorBatch* tsBatch = dynamic_cast<orc::TimestampVectorBatch*>(batch);
  bool hasNull = false;
  for (uint64_t i = 0; i < numValues; ++i) {
    std::string col = extractColumn(data[i], colIndex);
    if (col.empty()) {
      batch->notNull[i] = 0;
      hasNull = true;
    } else {
      memset(&timeStruct, 0, sizeof(timeStruct));
      char* left = strptime(col.c_str(), "%Y-%m-%d %H:%M:%S", &timeStruct);
      if (left == nullptr) {
        batch->notNull[i] = 0;
      } else {
        batch->notNull[i] = 1;
        tsBatch->data[i] = timegm(&timeStruct);
        char* tail;
        double d = strtod(left, &tail);
        if (tail != left) {
          tsBatch->nanoseconds[i] = static_cast<long>(d * 1000000000.0);
        } else {
          tsBatch->nanoseconds[i] = 0;
        }
      }
    }
  }
  tsBatch->hasNulls = hasNull;
  tsBatch->numElements = numValues;
}

void usage() {
  std::cout << "Usage: csv-import [-h] [--help]\n"