Add TLS disable option for ClickHouse and include Docker Compose file (#93)

Author: iuwqyir

README.md

@@ -343,6 +343,7 @@ storage:
       user: "admin"
       password: "admin"
       database: "base"
+      disableTLS: false
   staging:
     driver: "memory"
     memory:

configs/config.go

@@ -54,11 +54,12 @@ type StorageConnectionConfig struct {
 }
 
 type ClickhouseConfig struct {
-	Host     string `mapstructure:"host"`
-	Port     int    `mapstructure:"port"`
-	Username string `mapstructure:"username"`
-	Password string `mapstructure:"password"`
-	Database string `mapstructure:"database"`
+	Host       string `mapstructure:"host"`
+	Port       int    `mapstructure:"port"`
+	Username   string `mapstructure:"username"`
+	Password   string `mapstructure:"password"`
+	Database   string `mapstructure:"database"`
+	DisableTLS bool   `mapstructure:"disableTLS"`
 }
 
 type MemoryConfig struct {

docker-compose.yml

@@ -0,0 +1,65 @@
+services:
+  redis:
+    image: redis:latest
+    ports:
+      - "6379:6379"
+    volumes:
+      - redis_data:/data
+    command: redis-server --appendonly yes
+    profiles:
+      - redis
+
+  # http://localhost:8123/play <- web ui
+  clickhouse:
+    image: clickhouse/clickhouse-server:latest
+    ports:
+      - "8123:8123"
+      - "9000:9000"
+    volumes:
+      - clickhouse_data:/var/lib/clickhouse
+    ulimits:
+      nofile:
+        soft: 262144
+        hard: 262144
+
+  prometheus:
+    image: prom/prometheus:latest
+    ports:
+      - "9090:9090"
+    command:
+      - '--config.file=/etc/prometheus/prometheus.yml'
+    configs:
+      - source: prometheus_config
+        target: /etc/prometheus/prometheus.yml
+    extra_hosts:
+      - "host.docker.internal:host-gateway"
+    profiles:
+      - monitoring
+
+  grafana:
+    image: grafana/grafana:latest
+    environment:
+      - GF_SECURITY_ADMIN_USER=admin
+      - GF_SECURITY_ADMIN_PASSWORD=admin
+    ports:
+      - "4000:3000"
+    volumes:
+      - grafana_data:/var/lib/grafana
+    profiles:
+      - monitoring
+
+volumes:
+  redis_data:
+  clickhouse_data:
+  grafana_data:
+
+configs:
+  prometheus_config:
+    content: |
+      global:
+        scrape_interval: 15s
+
+      scrape_configs:
+        - job_name: 'local_service'
+          static_configs:
+            - targets: ['host.docker.internal:2112']

internal/storage/clickhouse.go

@@ -43,7 +43,12 @@ func connectDB(cfg *config.ClickhouseConfig) (clickhouse.Conn, error) {
 	conn, err := clickhouse.Open(&clickhouse.Options{
 		Addr:     []string{fmt.Sprintf("%s:%d", cfg.Host, port)},
 		Protocol: clickhouse.Native,
-		TLS:      &tls.Config{}, // enable secure TLS
+		TLS: func() *tls.Config {
+			if cfg.DisableTLS {
+				return nil
+			}
+			return &tls.Config{}
+		}(),
 		Auth: clickhouse.Auth{
 			Username: cfg.Username,
 			Password: cfg.Password,