|
| 1 | +{{- if .Values.clickhouse.enabled }} |
| 2 | +{{- if .Values.clickhouse.clusterName }} |
| 3 | +apiVersion: batch/v1 |
| 4 | +kind: Job |
| 5 | +metadata: |
| 6 | + name: {{ include "alphatrion.fullname" . }}-clickhouse-migration |
| 7 | + labels: |
| 8 | + {{- include "alphatrion.labels" . | nindent 4 }} |
| 9 | + app.kubernetes.io/component: clickhouse-migration |
| 10 | + annotations: |
| 11 | + "helm.sh/hook": pre-install,pre-upgrade |
| 12 | + "helm.sh/hook-weight": "-4" |
| 13 | + "helm.sh/hook-delete-policy": before-hook-creation |
| 14 | +spec: |
| 15 | + backoffLimit: 3 |
| 16 | + template: |
| 17 | + metadata: |
| 18 | + labels: |
| 19 | + {{- include "alphatrion.selectorLabels" . | nindent 8 }} |
| 20 | + app.kubernetes.io/component: clickhouse-migration |
| 21 | + spec: |
| 22 | + restartPolicy: Never |
| 23 | + containers: |
| 24 | + - name: clickhouse-migration |
| 25 | + image: clickhouse/clickhouse-server:24.3-alpine |
| 26 | + command: |
| 27 | + - sh |
| 28 | + - -c |
| 29 | + - | |
| 30 | + set -e |
| 31 | + echo "==========================================" |
| 32 | + echo "ClickHouse Cluster Table Migration" |
| 33 | + echo "==========================================" |
| 34 | + echo "" |
| 35 | + echo "Cluster: {{ .Values.clickhouse.clusterName }}" |
| 36 | + echo "Database: {{ .Values.clickhouse.database }}" |
| 37 | + echo "" |
| 38 | +
|
| 39 | + # Wait for ClickHouse to be ready |
| 40 | + echo "Waiting for ClickHouse to be ready..." |
| 41 | + until clickhouse-client --host {{ .Values.clickhouse.host }} --port 9000 \ |
| 42 | + --user {{ .Values.clickhouse.username }} --password "$CLICKHOUSE_PASSWORD" \ |
| 43 | + --query "SELECT 1" > /dev/null 2>&1; do |
| 44 | + echo " Waiting for ClickHouse..." |
| 45 | + sleep 2 |
| 46 | + done |
| 47 | + echo "✓ ClickHouse is ready" |
| 48 | + echo "" |
| 49 | +
|
| 50 | + # Check if cluster exists |
| 51 | + echo "Checking if cluster '{{ .Values.clickhouse.clusterName }}' exists..." |
| 52 | + CLUSTER_EXISTS=$(clickhouse-client --host {{ .Values.clickhouse.host }} --port 9000 \ |
| 53 | + --user {{ .Values.clickhouse.username }} --password "$CLICKHOUSE_PASSWORD" \ |
| 54 | + --query "SELECT count() FROM system.clusters WHERE cluster='{{ .Values.clickhouse.clusterName }}'" 2>/dev/null || echo "0") |
| 55 | +
|
| 56 | + if [ "$CLUSTER_EXISTS" -eq "0" ]; then |
| 57 | + echo "✗ Cluster '{{ .Values.clickhouse.clusterName }}' not found!" |
| 58 | + echo " Please configure ClickHouse cluster first." |
| 59 | + exit 1 |
| 60 | + fi |
| 61 | + echo "✓ Cluster exists" |
| 62 | + echo "" |
| 63 | +
|
| 64 | + # Drop old non-replicated table if exists |
| 65 | + echo "Dropping old non-replicated tables (if any)..." |
| 66 | + clickhouse-client --host {{ .Values.clickhouse.host }} --port 9000 \ |
| 67 | + --user {{ .Values.clickhouse.username }} --password "$CLICKHOUSE_PASSWORD" \ |
| 68 | + --query "DROP TABLE IF EXISTS {{ .Values.clickhouse.database }}.otel_spans ON CLUSTER {{ .Values.clickhouse.clusterName }} SYNC" \ |
| 69 | + 2>/dev/null || echo " No existing table to drop" |
| 70 | + echo "✓ Old tables dropped" |
| 71 | + echo "" |
| 72 | +
|
| 73 | + # Create replicated table on cluster |
| 74 | + echo "Creating replicated table on cluster..." |
| 75 | + clickhouse-client --host {{ .Values.clickhouse.host }} --port 9000 \ |
| 76 | + --user {{ .Values.clickhouse.username }} --password "$CLICKHOUSE_PASSWORD" \ |
| 77 | + --multiquery <<'EOF' |
| 78 | + CREATE TABLE IF NOT EXISTS {{ .Values.clickhouse.database }}.otel_spans ON CLUSTER {{ .Values.clickhouse.clusterName }} ( |
| 79 | + Timestamp DateTime64(9) CODEC(Delta, ZSTD(1)), |
| 80 | + TraceId String CODEC(ZSTD(1)), |
| 81 | + SpanId String CODEC(ZSTD(1)), |
| 82 | + ParentSpanId String CODEC(ZSTD(1)), |
| 83 | + SpanName LowCardinality(String) CODEC(ZSTD(1)), |
| 84 | + SpanKind LowCardinality(String) CODEC(ZSTD(1)), |
| 85 | + SemanticKind LowCardinality(String) CODEC(ZSTD(1)), |
| 86 | + ServiceName LowCardinality(String) CODEC(ZSTD(1)), |
| 87 | + Duration UInt64 CODEC(ZSTD(1)), |
| 88 | + StatusCode LowCardinality(String) CODEC(ZSTD(1)), |
| 89 | + StatusMessage String CODEC(ZSTD(1)), |
| 90 | + TeamId String CODEC(ZSTD(1)), |
| 91 | + RunId String CODEC(ZSTD(1)), |
| 92 | + ExperimentId String CODEC(ZSTD(1)), |
| 93 | + SpanAttributes Map(LowCardinality(String), String) CODEC(ZSTD(1)), |
| 94 | + ResourceAttributes Map(LowCardinality(String), String) CODEC(ZSTD(1)), |
| 95 | + Events Nested( |
| 96 | + Timestamp DateTime64(9), |
| 97 | + Name LowCardinality(String), |
| 98 | + Attributes Map(LowCardinality(String), String) |
| 99 | + ) CODEC(ZSTD(1)), |
| 100 | + Links Nested( |
| 101 | + TraceId String, |
| 102 | + SpanId String, |
| 103 | + Attributes Map(LowCardinality(String), String) |
| 104 | + ) CODEC(ZSTD(1)), |
| 105 | + INDEX idx_trace_id TraceId TYPE bloom_filter(0.001) GRANULARITY 1, |
| 106 | + INDEX idx_span_id SpanId TYPE bloom_filter(0.001) GRANULARITY 1, |
| 107 | + INDEX idx_run_id RunId TYPE bloom_filter(0.001) GRANULARITY 1, |
| 108 | + INDEX idx_team_id TeamId TYPE bloom_filter(0.001) GRANULARITY 1, |
| 109 | + INDEX idx_semantic_kind SemanticKind TYPE set(0) GRANULARITY 1, |
| 110 | + INDEX idx_attr_keys mapKeys(SpanAttributes) TYPE bloom_filter(0.01) GRANULARITY 1 |
| 111 | + ) ENGINE = ReplicatedMergeTree('/clickhouse/tables/{shard}/otel_spans', '{replica}') |
| 112 | + PARTITION BY toDate(Timestamp) |
| 113 | + ORDER BY (ServiceName, toUnixTimestamp(Timestamp)) |
| 114 | + SETTINGS index_granularity = 8192; |
| 115 | + EOF |
| 116 | +
|
| 117 | + echo "✓ Replicated table created" |
| 118 | + echo "" |
| 119 | +
|
| 120 | + # Verify table exists locally |
| 121 | + echo "Verifying table creation..." |
| 122 | + TABLE_COUNT=$(clickhouse-client --host {{ .Values.clickhouse.host }} --port 9000 \ |
| 123 | + --user {{ .Values.clickhouse.username }} --password "$CLICKHOUSE_PASSWORD" \ |
| 124 | + --query "SELECT count() FROM system.tables WHERE database = '{{ .Values.clickhouse.database }}' AND name = 'otel_spans'") |
| 125 | +
|
| 126 | + if [ "$TABLE_COUNT" -eq "1" ]; then |
| 127 | + echo "✓ Table 'otel_spans' verified" |
| 128 | + else |
| 129 | + echo "✗ Table verification failed!" |
| 130 | + exit 1 |
| 131 | + fi |
| 132 | +
|
| 133 | + echo "" |
| 134 | + echo "==========================================" |
| 135 | + echo "✓ Migration completed successfully!" |
| 136 | + echo "==========================================" |
| 137 | + env: |
| 138 | + - name: CLICKHOUSE_PASSWORD |
| 139 | + valueFrom: |
| 140 | + secretKeyRef: |
| 141 | + name: {{ include "alphatrion.clickhouse.secretName" . }} |
| 142 | + key: clickhouse-password |
| 143 | +{{- end }} |
| 144 | +{{- end }} |
0 commit comments