apiVersion: apps/v1
kind: StatefulSet
metadata:
  name: pd-tso-server
spec:
  serviceName: tso-service
  selector:
    matchLabels:
      app.kubernetes.io/name: pd-tso-server
  template:
    metadata:
      annotations:
        prometheus.io/path: /metrics
        prometheus.io/port: "2379"
        prometheus.io/scrape: "true"
        cluster-autoscaler.kubernetes.io/safe-to-evict: "false"
      labels:
        app.kubernetes.io/name: pd-tso-server
    spec:
      affinity:
        nodeAffinity:
          requiredDuringSchedulingIgnoredDuringExecution:
            nodeSelectorTerms:
            - matchExpressions:
              - key: serverless.tidbcloud.com/node
                operator: In
                values:
                - critical-services
          preferredDuringSchedulingIgnoredDuringExecution:
          - weight: 100
            preference:
              matchExpressions:
              - key: serverless.tidbcloud.com/zone
                operator: In
                values:
                - primary
        podAntiAffinity:
          preferredDuringSchedulingIgnoredDuringExecution:
            - weight: 100
              podAffinityTerm:
                labelSelector:
                  matchExpressions:
                    - key: app.kubernetes.io/name
                      operator: In
                      values:
                        - pd-tso-server
                topologyKey: kubernetes.io/hostname
      containers:
      - name: pd-tso-server
        imagePullPolicy: IfNotPresent
        command:
          - /pd-server
          - services
          - tso
        args:
          - --backend-endpoints=https://serverless-cluster-pd.tidb-serverless.svc:2379
          - --listen-addr=https://$(POD_IP):2379
          - --advertise-listen-addr=https://$(POD_NAME).tso-service.tidb-serverless.svc:2379
        env:
          - name: POD_IP
            valueFrom:
              fieldRef:
                fieldPath: status.podIP
          - name: POD_NAME
            valueFrom:
              fieldRef:
                fieldPath: metadata.name
        resources:
          limits:
            memory: "8Gi"
            cpu: "4"
          requests:
            memory: "8Gi"
            cpu: "2"
        ports:
        - containerPort: 2379
      tolerations:
      - key: serverless.tidbcloud.com/node
        operator: Equal
        value: critical-services
        effect: NoSchedule