feat: add timeouts to clickhouse

PeerDB-io · May 20, 2024 · 4f0afac · 4f0afac
1 parent f4f055c
commit 4f0afac
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/flow/connectors/clickhouse/clickhouse.go b/flow/connectors/clickhouse/clickhouse.go
@@ -8,6 +8,7 @@ import (
 	"fmt"
 	"net/url"
 	"strings"
+	"time"
 
 	"github.com/ClickHouse/clickhouse-go/v2"
 	_ "github.com/ClickHouse/clickhouse-go/v2/lib/driver"
@@ -184,6 +185,13 @@ func connect(ctx context.Context, config *protos.ClickhouseConfig) (*sql.DB, err
 				{Name: "peerdb"},
 			},
 		},
+		Settings: map[string]any{
+			"idle_timeout":    10000,
+			"connect_timeout": 10000,
+			"send_timeout":    10000,
+			"receive_timeout": 10000,
+		},
+		ReadTimeout: 10000 * time.Second,
 	})
 
 	if err := conn.PingContext(ctx); err != nil {