From 7accbf7aeb3f1d4ddbb5a2bd43544f0f700cf069 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=8E=8B=E8=B6=85?= <cclcwangchao@hotmail.com>
Date: Sat, 11 Feb 2023 01:31:59 +0800
Subject: [PATCH] ttl: optimize bazel for TTL codes (#41278)

close pingcap/tidb#41276
---
 ttl/cache/BUILD.bazel      | 2 ++
 ttl/client/BUILD.bazel     | 2 ++
 ttl/metrics/BUILD.bazel    | 2 ++
 ttl/session/BUILD.bazel    | 2 ++
 ttl/sqlbuilder/BUILD.bazel | 2 ++
 ttl/ttlworker/BUILD.bazel  | 2 +-
 6 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/ttl/cache/BUILD.bazel b/ttl/cache/BUILD.bazel
index c926ee231cd93..1e448cbeccf3f 100644
--- a/ttl/cache/BUILD.bazel
+++ b/ttl/cache/BUILD.bazel
@@ -35,6 +35,7 @@ go_library(
 
 go_test(
     name = "cache_test",
+    timeout = "short",
     srcs = [
         "base_test.go",
         "infoschema_test.go",
@@ -46,6 +47,7 @@ go_test(
     ],
     embed = [":cache"],
     flaky = True,
+    shard_count = 50,
     deps = [
         "//infoschema",
         "//kv",
diff --git a/ttl/client/BUILD.bazel b/ttl/client/BUILD.bazel
index e842ad03a887b..31577c1161263 100644
--- a/ttl/client/BUILD.bazel
+++ b/ttl/client/BUILD.bazel
@@ -20,8 +20,10 @@ go_library(
 
 go_test(
     name = "client_test",
+    timeout = "short",
     srcs = ["command_test.go"],
     embed = [":client"],
+    shard_count = 5,
     deps = [
         "@com_github_pingcap_errors//:errors",
         "@com_github_stretchr_testify//require",
diff --git a/ttl/metrics/BUILD.bazel b/ttl/metrics/BUILD.bazel
index f0666b5c59530..f8fbe4ef8030b 100644
--- a/ttl/metrics/BUILD.bazel
+++ b/ttl/metrics/BUILD.bazel
@@ -13,7 +13,9 @@ go_library(
 
 go_test(
     name = "metrics_test",
+    timeout = "short",
     srcs = ["metrics_test.go"],
     embed = [":metrics"],
+    shard_count = 5,
     deps = ["@com_github_stretchr_testify//require"],
 )
diff --git a/ttl/session/BUILD.bazel b/ttl/session/BUILD.bazel
index 2c9dae3fc426f..a98067cff9854 100644
--- a/ttl/session/BUILD.bazel
+++ b/ttl/session/BUILD.bazel
@@ -21,12 +21,14 @@ go_library(
 
 go_test(
     name = "session_test",
+    timeout = "short",
     srcs = [
         "main_test.go",
         "session_test.go",
         "sysvar_test.go",
     ],
     flaky = True,
+    shard_count = 5,
     deps = [
         ":session",
         "//sessionctx/variable",
diff --git a/ttl/sqlbuilder/BUILD.bazel b/ttl/sqlbuilder/BUILD.bazel
index 505e9ffcb3576..5eafc4a7748d3 100644
--- a/ttl/sqlbuilder/BUILD.bazel
+++ b/ttl/sqlbuilder/BUILD.bazel
@@ -19,11 +19,13 @@ go_library(
 
 go_test(
     name = "sqlbuilder_test",
+    timeout = "short",
     srcs = [
         "main_test.go",
         "sql_test.go",
     ],
     flaky = True,
+    shard_count = 5,
     deps = [
         ":sqlbuilder",
         "//kv",
diff --git a/ttl/ttlworker/BUILD.bazel b/ttl/ttlworker/BUILD.bazel
index 74e050977463f..6e80e4f298145 100644
--- a/ttl/ttlworker/BUILD.bazel
+++ b/ttl/ttlworker/BUILD.bazel
@@ -44,7 +44,7 @@ go_library(
 
 go_test(
     name = "ttlworker_test",
-    timeout = "long",
+    timeout = "moderate",
     srcs = [
         "del_test.go",
         "job_manager_integration_test.go",