| // 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. |
| |
| #include "kudu/integration-tests/data_gen_util.h" |
| |
| #include <ostream> |
| |
| #include <glog/logging.h> |
| |
| #include "kudu/client/schema.h" |
| #include "kudu/common/partial_row.h" |
| #include "kudu/gutil/strings/numbers.h" |
| #include "kudu/util/random.h" // IWYU pragma: keep |
| #include "kudu/util/status.h" |
| |
| namespace kudu { |
| |
| void WriteValueToColumn(const client::KuduSchema& schema, |
| int col_idx, |
| uint64_t value, |
| KuduPartialRow* row) { |
| client::KuduColumnSchema::DataType type = schema.Column(col_idx).type(); |
| char buf[kFastToBufferSize]; |
| switch (type) { |
| case client::KuduColumnSchema::INT8: |
| CHECK_OK(row->SetInt8(col_idx, value)); |
| break; |
| case client::KuduColumnSchema::INT16: |
| CHECK_OK(row->SetInt16(col_idx, value)); |
| break; |
| case client::KuduColumnSchema::INT32: |
| CHECK_OK(row->SetInt32(col_idx, value)); |
| break; |
| case client::KuduColumnSchema::INT64: |
| CHECK_OK(row->SetInt64(col_idx, value)); |
| break; |
| case client::KuduColumnSchema::FLOAT: |
| CHECK_OK(row->SetFloat(col_idx, value / 123.0)); |
| break; |
| case client::KuduColumnSchema::DOUBLE: |
| CHECK_OK(row->SetDouble(col_idx, value / 123.0)); |
| break; |
| case client::KuduColumnSchema::STRING: |
| CHECK_OK(row->SetStringCopy(col_idx, FastHex64ToBuffer(value, buf))); |
| break; |
| case client::KuduColumnSchema::BOOL: |
| CHECK_OK(row->SetBool(col_idx, value)); |
| break; |
| case client::KuduColumnSchema::BINARY: |
| CHECK_OK(row->SetBinaryCopy(col_idx, FastHex64ToBuffer(value, buf))); |
| break; |
| case client::KuduColumnSchema::UNIXTIME_MICROS: |
| CHECK_OK(row->SetUnixTimeMicros(col_idx, value)); |
| break; |
| case client::KuduColumnSchema::DATE: |
| CHECK_OK(row->SetDate(col_idx, value)); |
| break; |
| case client::KuduColumnSchema::DECIMAL: |
| CHECK_OK(row->SetUnscaledDecimal(col_idx, value)); |
| break; |
| default: |
| LOG(FATAL) << "Unexpected data type: " << type; |
| } |
| } |
| |
| template <class RNG> |
| void GenerateDataForRow(const client::KuduSchema& schema, uint64_t record_id, |
| RNG* random, KuduPartialRow* row) { |
| for (int col_idx = 0; col_idx < schema.num_columns(); col_idx++) { |
| // We randomly generate the inserted data, except for the first column, |
| // which is always based on a monotonic "record id". |
| uint64_t value; |
| if (col_idx == 0) { |
| value = record_id; |
| } else { |
| value = random->Next64(); |
| } |
| WriteValueToColumn(schema, col_idx, value, row); |
| } |
| } |
| |
| template <class RNG> |
| void GenerateDataForRow(const client::KuduSchema& schema, |
| RNG* random, KuduPartialRow* row) { |
| GenerateDataForRow(schema, random->Next64(), random, row); |
| } |
| |
| // Explicit specialization for callers outside this compilation unit. |
| template |
| void GenerateDataForRow(const client::KuduSchema& schema, uint64_t record_id, |
| Random* random, KuduPartialRow* row); |
| template |
| void GenerateDataForRow(const client::KuduSchema& schema, uint64_t record_id, |
| ThreadSafeRandom* random, KuduPartialRow* row); |
| template |
| void GenerateDataForRow(const client::KuduSchema& schema, |
| Random* random, KuduPartialRow* row); |
| template |
| void GenerateDataForRow(const client::KuduSchema& schema, |
| ThreadSafeRandom* random, KuduPartialRow* row); |
| |
| } // namespace kudu |