feat: add no-op database client

probe-lab · Jan 8, 2025 · 820fd74 · 820fd74
1 parent 96f9435
commit 820fd74
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 12 deletions.
diff --git a/cmd/ants/main.go b/cmd/ants/main.go
@@ -266,17 +266,24 @@ func runQueenCommand(c *cli.Context) error {
 	logger.Debugln("Starting metrics server", "host", queenConfig.MetricsHost, "port", queenConfig.MetricsPort)
 	go metrics.ListenAndServe(queenConfig.MetricsHost, queenConfig.MetricsPort)
 
-	// initializing a new clickhouse client
-	client, err := db.NewClient(
-		queenConfig.ClickhouseAddress,
-		queenConfig.ClickhouseDatabase,
-		queenConfig.ClickhouseUsername,
-		queenConfig.ClickhousePassword,
-		queenConfig.ClickhouseSSL,
-		telemetry,
-	)
-	if err != nil {
-		return fmt.Errorf("init database client: %w", err)
+	var client db.Client
+	if queenConfig.ClickhouseAddress == "" {
+		logger.Warn("No clickhouse address provided, using no-op client.")
+		client = db.NewNoopClient()
+	} else {
+		// initializing a new clickhouse client
+		client, err = db.NewClickhouseClient(
+			queenConfig.ClickhouseAddress,
+			queenConfig.ClickhouseDatabase,
+			queenConfig.ClickhouseUsername,
+			queenConfig.ClickhousePassword,
+			queenConfig.ClickhouseSSL,
+			telemetry,
+		)
+		if err != nil {
+			return fmt.Errorf("init database client: %w", err)
+		}
+
 	}
 
 	// pinging database to check availability

diff --git a/db/client.go b/db/client.go
@@ -25,14 +25,30 @@ type Client interface {
 	BulkInsertRequests(ctx context.Context, requests []*Request) error
 }
 
+type NoopClient struct{}
+
+var _ Client = (*NoopClient)(nil)
+
+func NewNoopClient() *NoopClient {
+	return &NoopClient{}
+}
+
+func (n NoopClient) Ping(ctx context.Context) error {
+	return nil
+}
+
+func (n NoopClient) BulkInsertRequests(ctx context.Context, requests []*Request) error {
+	return nil
+}
+
 type ClickhouseClient struct {
 	driver.Conn
 	telemetry *metrics.Telemetry
 }
 
 var _ Client = (*ClickhouseClient)(nil)
 
-func NewClient(address, database, username, password string, ssl bool, telemetry *metrics.Telemetry) (*ClickhouseClient, error) {
+func NewClickhouseClient(address, database, username, password string, ssl bool, telemetry *metrics.Telemetry) (*ClickhouseClient, error) {
 	logger.Infoln("Creating new clickhouse client...")
 
 	conn, err := clickhouse.Open(&clickhouse.Options{