fix(snowflake): support per-replica WorkerId via env for multi-instance StatefulSet

This commit is contained in:
zetaloop
2026-05-05 12:29:47 +08:00
parent cba510c675
commit c456f3e296
3 changed files with 34 additions and 3 deletions
+2 -2
View File
@@ -2,5 +2,5 @@ Name: snowflake.rpc
ListenOn: 0.0.0.0:8080
Snowflake:
DatacenterId: 1
WorkerId: 0
DatacenterId: ${SNOWFLAKE_DATACENTER_ID}
WorkerId: ${SNOWFLAKE_WORKER_ID}
+3
View File
@@ -210,6 +210,9 @@ services:
image: juwan/snowflake-rpc:dev
container_name: juwan-snowflake
restart: unless-stopped
environment:
SNOWFLAKE_DATACENTER_ID: 1
SNOWFLAKE_WORKER_ID: 0
authz-adapter:
image: juwan/authz-adapter:dev
+29 -1
View File
@@ -1,11 +1,12 @@
apiVersion: apps/v1
kind: Deployment
kind: StatefulSet
metadata:
name: snowflake
namespace: juwan
labels:
app: snowflake
spec:
serviceName: snowflake-headless
replicas: 1
selector:
matchLabels:
@@ -20,6 +21,18 @@ spec:
containers:
- name: snowflake
image: registry.juwan.xhttp.zip/juwan/snowflake-rpc:latest
command: ["/bin/sh", "-c"]
args:
- |
export SNOWFLAKE_WORKER_ID="${POD_NAME##*-}"
exec /app/main -f /app/etc/snowflake.yaml
env:
- name: POD_NAME
valueFrom:
fieldRef:
fieldPath: metadata.name
- name: SNOWFLAKE_DATACENTER_ID
value: "1"
ports:
- name: grpc
containerPort: 8080
@@ -28,6 +41,21 @@ spec:
cpu: 20m
memory: 32Mi
---
apiVersion: v1
kind: Service
metadata:
name: snowflake-headless
namespace: juwan
spec:
clusterIP: None
selector:
app: snowflake
ports:
- name: grpc
port: 8080
targetPort: 8080
---
apiVersion: v1
kind: Service