Merge pull request #2 from ObolNetwork/add-tls

Add tlsconfig to drivers
ObolNetwork · May 2, 2024 · 9a0f995 · 9a0f995
2 parents 5652bfb + 9d2031f
commit 9a0f995
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 6 deletions.
diff --git a/pkg/db/high_level_driver.go b/pkg/db/high_level_driver.go
@@ -2,7 +2,8 @@ package db
 
 import (
 	"context"
-
+
+	"crypto/tls"
 	"fmt"
 	"net"
 	"strings"
@@ -55,10 +56,8 @@ func ParseChUrlIntoOptionsHighLevel(url string) clickhouse.Options {
 			Username: user,
 			Password: password,
 		},
-		DialContext: func(ctx context.Context, addr string) (net.Conn, error) {
-			dialCount++
-			var d net.Dialer
-			return d.DialContext(ctx, "tcp", addr)
+		TLS: tls.Config{
+			InsecureSkipVerify: false,
 		},
 		Debug: false,
 		Debugf: func(format string, v ...any) {

diff --git a/pkg/db/low_level_driver.go b/pkg/db/low_level_driver.go
@@ -2,6 +2,7 @@ package db
 
 import (
 	"context"
+	"crypto/tls"
 	"strings"
 	"time"
 
@@ -51,7 +52,11 @@ func ParseChUrlIntoOptionsLowLevel(url string) ch.Options {
 		Address:  fqdn,
 		Database: database,
 		User:     user,
-		Password: password}
+		Password: password,
+		DialTimeout: time.Second * 30,
+		TLS: &tls.Config{
+			InsecureSkipVerify: false,
+		}}
 }
 
 func (p *DBService) Persist(