fix(db_connector): add timeout for db connection check

internal/connectors/db/connector.go

@@ -60,13 +60,17 @@ type YdbConnector struct {
 	driver *ydb.Driver
 }
 
-func select1(ctx context.Context, db *ydb.Driver) error {
+func select1(baseCtx context.Context, db *ydb.Driver, timeout time.Duration) error {
 	readTx := table.TxControl(
 		table.BeginTx(
 			table.WithOnlineReadOnly(),
 		),
 		table.CommitTx(),
 	)
+
+	ctx, cancel := context.WithTimeout(baseCtx, timeout)
+	defer cancel()
+
 	return db.Table().Do(
 		ctx, func(ctx context.Context, s table.Session) error {
 			_, res, err := s.Execute(
@@ -93,8 +97,9 @@ func select1(ctx context.Context, db *ydb.Driver) error {
 }
 
 func NewYdbConnector(ctx context.Context, config config.YDBConnectionConfig) (*YdbConnector, error) {
+	dialTimeout := time.Second * time.Duration(config.DialTimeoutSeconds)
 	opts := []ydb.Option{
-		ydb.WithDialTimeout(time.Second * time.Duration(config.DialTimeoutSeconds)),
+		ydb.WithDialTimeout(dialTimeout),
 	}
 	if config.Insecure {
 		opts = append(opts, ydb.WithTLSSInsecureSkipVerify())
@@ -113,7 +118,7 @@ func NewYdbConnector(ctx context.Context, config config.YDBConnectionConfig) (*Y
 	if err != nil {
 		return nil, fmt.Errorf("can't connect to YDB, dsn %s: %w", config.ConnectionString, err)
 	}
-	err = select1(ctx, driver)
+	err = select1(ctx, driver, dialTimeout)
 	if err != nil {
 		return nil, fmt.Errorf("can't connect to YDB, dsn %s: %w", config.ConnectionString, err)
 	}

internal/handlers/take_backup_retry.go

@@ -186,8 +186,9 @@ func TBWROperationHandler(
 			if err != nil {
 				tbwr.State = types.OperationStateError
 				tbwr.Message = err.Error()
-				tbwr.UpdatedAt = timestamppb.New(clock.Now())
-				tbwr.Audit.CompletedAt = timestamppb.New(clock.Now())
+				now := clock.Now()
+				tbwr.UpdatedAt = timestamppb.New(now)
+				tbwr.Audit.CompletedAt = timestamppb.New(now)
 
 				errup := db.ExecuteUpsert(ctx, queryBuilderFactory().WithUpdateOperation(tbwr))
 				if errup != nil {
@@ -207,8 +208,9 @@ func TBWROperationHandler(
 				{
 					tbwr.State = types.OperationStateDone
 					tbwr.Message = "Success"
-					tbwr.UpdatedAt = timestamppb.New(clock.Now())
-					tbwr.Audit.CompletedAt = timestamppb.New(clock.Now())
+					now := clock.Now()
+					tbwr.UpdatedAt = timestamppb.New(now)
+					tbwr.Audit.CompletedAt = timestamppb.New(now)
 					return db.ExecuteUpsert(ctx, queryBuilderFactory().WithUpdateOperation(tbwr))
 				}
 			case Skip:
@@ -223,8 +225,9 @@ func TBWROperationHandler(
 						return ids
 					}(), ", ")
 					tbwr.State = types.OperationStateError
-					tbwr.UpdatedAt = timestamppb.New(clock.Now())
-					tbwr.Audit.CompletedAt = timestamppb.New(clock.Now())
+					now := clock.Now()
+					tbwr.UpdatedAt = timestamppb.New(now)
+					tbwr.Audit.CompletedAt = timestamppb.New(now)
 
 					tbwr.Message = fmt.Sprintf("retry attempts exceeded limit: %d.", len(ops))
 					fields := []zap.Field{
@@ -258,8 +261,9 @@ func TBWROperationHandler(
 			default:
 				tbwr.State = types.OperationStateError
 				tbwr.Message = "unexpected operation state"
-				tbwr.UpdatedAt = timestamppb.New(clock.Now())
-				tbwr.Audit.CompletedAt = timestamppb.New(clock.Now())
+				now := clock.Now()
+				tbwr.UpdatedAt = timestamppb.New(now)
+				tbwr.Audit.CompletedAt = timestamppb.New(now)
 
 				_ = db.ExecuteUpsert(ctx, queryBuilderFactory().WithUpdateOperation(tbwr))
 				return errors.New(tbwr.Message)
@@ -278,8 +282,9 @@ func TBWROperationHandler(
 			if last == nil || !types.IsActive(last) {
 				tbwr.State = types.OperationStateCancelled
 				tbwr.Message = "Success"
-				tbwr.UpdatedAt = timestamppb.New(clock.Now())
-				tbwr.Audit.CompletedAt = timestamppb.New(clock.Now())
+				now := clock.Now()
+				tbwr.UpdatedAt = timestamppb.New(now)
+				tbwr.Audit.CompletedAt = timestamppb.New(now)
 				return db.ExecuteUpsert(ctx, queryBuilderFactory().WithUpdateOperation(tbwr))
 			} else {
 				if last.State == types.OperationStatePending || last.State == types.OperationStateRunning {
@@ -295,8 +300,9 @@ func TBWROperationHandler(
 		{
 			tbwr.State = types.OperationStateError
 			tbwr.Message = "unexpected operation state"
-			tbwr.UpdatedAt = timestamppb.New(clock.Now())
-			tbwr.Audit.CompletedAt = timestamppb.New(clock.Now())
+			now := clock.Now()
+			tbwr.UpdatedAt = timestamppb.New(now)
+			tbwr.Audit.CompletedAt = timestamppb.New(now)
 			_ = db.ExecuteUpsert(ctx, queryBuilderFactory().WithUpdateOperation(tbwr))
 			return errors.New(tbwr.Message)
 		}