YugabyteDB (2.13.0.0-b42, bfc6a6643e7399ac8a0e81d06a3ee6d6571b33ab)

// Copyright (c) YugaByte, Inc.

//

// Licensed 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 "yb/client/client.h"

#include "yb/client/ql-dml-test-base.h"

#include "yb/client/schema.h"

#include "yb/client/session.h"

#include "yb/client/table_handle.h"

#include "yb/client/yb_op.h"

#include "yb/common/ql_value.h"

#include "yb/util/status_log.h"

#include "yb/yql/cql/ql/util/statement_result.h"

namespace yb {

namespace client {

using yb::ql::RowsResult;

namespace {

const std::vector<std::string> kAllColumns = {"k", "c1", "c2", "c3", "c4"};

}

class QLDmlTTLTest : public QLDmlTestBase<MiniCluster> {

 public:

  void SetUp() override {

    QLDmlTestBase::SetUp();

    YBSchemaBuilder b;

    b.AddColumn("k")->Type(INT32)->HashPrimaryKey()->NotNull();

    b.AddColumn("c1")->Type(INT32);

    b.AddColumn("c2")->Type(STRING);

    b.AddColumn("c3")->Type(INT32);

    b.AddColumn("c4")->Type(STRING);

    ASSERT_OK(table_.Create(kTableName, CalcNumTablets(3), client_.get(), &b));

  }

  TableHandle table_;

};

TEST_F(QLDmlTTLTest, TestInsertWithTTL) {

  const YBSessionPtr session(NewSession());

  {

    // insert into t (k, c1, c2) values (1, 1, "yuga-hello") using ttl 2;

    const YBqlWriteOpPtr op = table_.NewWriteOp(QLWriteRequestPB::QL_STMT_INSERT);

    auto* const req = op->mutable_request();

    QLAddInt32HashValue(req, 1);

    table_.AddInt32ColumnValue(req, "c1", 1);

    table_.AddStringColumnValue(req, "c2", "yuga-hello");

    req->set_ttl(2 * 1000);

    CHECK_OK(session->ApplyAndFlush(op));

    EXPECT_EQ(op->response().status(), QLResponsePB::YQL_STATUS_OK);

  }

  {

    // insert into t (k, c3, c4) values (1, 2, "yuga-hi") using ttl 4;

    const YBqlWriteOpPtr op = table_.NewWriteOp(QLWriteRequestPB::QL_STMT_INSERT);

    auto* const req = op->mutable_request();

    QLAddInt32HashValue(req, 1);

    table_.AddInt32ColumnValue(req, "c3", 2);

    table_.AddStringColumnValue(req, "c4", "yuga-hi");

    req->set_ttl(4 * 1000);

    CHECK_OK(session->ApplyAndFlush(op));

    EXPECT_EQ(op->response().status(), QLResponsePB::YQL_STATUS_OK);

  }

  {

    // select * from t where k = 1;

    const YBqlReadOpPtr op = table_.NewReadOp();

    auto* const req = op->mutable_request();

    QLAddInt32HashValue(req, 1);

    table_.AddColumns(kAllColumns, req);

    CHECK_OK(session->ApplyAndFlush(op));

    // Expect all 4 columns (c1, c2, c3, c4) to be valid right now.

    EXPECT_EQ(op->response().status(), QLResponsePB::YQL_STATUS_OK);

    auto rowblock = RowsResult(op.get()).GetRowBlock();

    EXPECT_EQ(rowblock->row_count(), 1);

    const auto& row = rowblock->row(0);

    EXPECT_EQ(row.column(0).int32_value(), 1);

    EXPECT_EQ(row.column(1).int32_value(), 1);

    EXPECT_EQ(row.column(2).string_value(), "yuga-hello");

    EXPECT_EQ(row.column(3).int32_value(), 2);

    EXPECT_EQ(row.column(4).string_value(), "yuga-hi");

  }

  LOG(INFO) << "Sleep for 2.5 seconds..";

  SleepFor(MonoDelta::FromMilliseconds(2500));

  {

    // select * from t where k = 1;

    const YBqlReadOpPtr op = table_.NewReadOp();

    auto* const req = op->mutable_request();

    QLAddInt32HashValue(req, 1);

    table_.AddColumns(kAllColumns, req);

    CHECK_OK(session->ApplyAndFlush(op));

    // Expect columns (c1, c2) to be null and (c3, c4) to be valid right now.

    EXPECT_EQ(op->response().status(), QLResponsePB::YQL_STATUS_OK);

    auto rowblock = RowsResult(op.get()).GetRowBlock();

    EXPECT_EQ(rowblock->row_count(), 1);

    const auto& row = rowblock->row(0);

    EXPECT_EQ(row.column(0).int32_value(), 1);

    EXPECT_TRUE(row.column(1).IsNull());

    EXPECT_TRUE(row.column(2).IsNull());

    EXPECT_EQ(row.column(3).int32_value(), 2);

    EXPECT_EQ(row.column(4).string_value(), "yuga-hi");

  }

  LOG(INFO) << "Sleep for 2.5 seconds..";

  SleepFor(MonoDelta::FromMilliseconds(2500));

  {

    // select * from t where k = 1;

    const YBqlReadOpPtr op = table_.NewReadOp();

    auto* const req = op->mutable_request();

    QLAddInt32HashValue(req, 1);

    table_.AddColumns(kAllColumns, req);

    CHECK_OK(session->ApplyAndFlush(op));

    // Expect all 4 columns (c1, c2, c3, c4) to be null.

    EXPECT_EQ(op->response().status(), QLResponsePB::YQL_STATUS_OK);

    auto rowblock = RowsResult(op.get()).GetRowBlock();

    EXPECT_EQ(rowblock->row_count(), 0);

  }

}

}  // namespace client

}  // namespace yb