StackSaga SQL 数据库分区支持 (SQL Database Partition Support)

# 启用/禁用 starter (默认: true)
stacksaga.partitioning.sql.enabled=true

# 启动时校验数据库连通性 (默认: true)
stacksaga.partitioning.sql.validate-on-startup=true

# 启动时校验用户分区变更权限 (默认: true)
stacksaga.partitioning.sql.validate-privileges=true

# 启动时立即执行一次分区创建 (默认: true)
stacksaga.partitioning.sql.run-on-startup=true

# 定时调度 cron (默认: 每天中午 12:00)
stacksaga.partitioning.sql.cron=0 0 12 * * *

# 跨数据源的并发度限制 (默认: 4)
stacksaga.partitioning.sql.concurrency=4

# 建立 R2DBC 连接的超时时间 (默认: 10s)
stacksaga.partitioning.sql.connect-timeout=10s

# --- 数据源 1: 订单服务数据库 (MySQL) - 按天分区 ---
stacksaga.partitioning.sql.datasources.orderdb.url=r2dbc:mysql://localhost:3306/order_db
stacksaga.partitioning.sql.datasources.orderdb.username=partition_admin
stacksaga.partitioning.sql.datasources.orderdb.password=secret
stacksaga.partitioning.sql.datasources.orderdb.unit=DAYS
stacksaga.partitioning.sql.datasources.orderdb.ahead-count=2

# --- 数据源 2: 支付服务数据库 (PostgreSQL) - 按周分区 ---
stacksaga.partitioning.sql.datasources.paymentdb.url=r2dbc:postgresql://localhost:5432/payment_db
stacksaga.partitioning.sql.datasources.paymentdb.username=partition_admin
stacksaga.partitioning.sql.datasources.paymentdb.password=secret
stacksaga.partitioning.sql.datasources.paymentdb.unit=WEEK
stacksaga.partitioning.sql.datasources.paymentdb.ahead-count=3

# --- 数据源 3: 库存服务数据库 (MySQL) - 按月分区 ---
stacksaga.partitioning.sql.datasources.inventorydb.url=r2dbc:mysql://localhost:3306/inventory_db
stacksaga.partitioning.sql.datasources.inventorydb.username=partition_admin
stacksaga.partitioning.sql.datasources.inventorydb.password=secret
stacksaga.partitioning.sql.datasources.inventorydb.unit=MONTH
stacksaga.partitioning.sql.datasources.inventorydb.ahead-count=1

apiVersion: batch/v1
kind: CronJob
metadata:
  name: stacksaga-partition-runner
  namespace: stacksaga
spec:
  schedule: "0 12 * * *"  # 每天中午 12:00 准时触发
  concurrencyPolicy: Forbid
  successfulJobsHistoryLimit: 3
  failedJobsHistoryLimit: 3
  jobTemplate:
    spec:
      template:
        spec:
          restartPolicy: OnFailure
          containers:
            - name: partition-runner
              image: your-docker-registry.internal/stacksaga/stacksaga-partition-runner:1.0.0
              env:
                # Pod 启动后立即执行分区创建
                - name: STACKSAGA_PARTITIONING_SQL_RUN_ON_STARTUP
                  value: "true"
                # 启动时校验连通性与权限
                - name: STACKSAGA_PARTITIONING_SQL_VALIDATE_ON_STARTUP
                  value: "true"
                - name: STACKSAGA_PARTITIONING_SQL_VALIDATE_PRIVILEGES
                  value: "true"
                # 禁用内部 cron，调度权交给 Kubernetes
                - name: STACKSAGA_PARTITIONING_SQL_CRON
                  value: "none"
                # 执行完毕后自动以退出码 0 (或错误时为 1) 退出
                - name: STACKSAGA_PARTITIONING_SQL_EXIT_ON_COMPLETION
                  value: "true"
              resources:
                requests:
                  memory: "256Mi"
                  cpu: "100m"
                limits:
                  memory: "512Mi"
                  cpu: "500m"

import lombok.extern.slf4j.Slf4j;
import org.springframework.context.event.EventListener;
import org.springframework.stereotype.Component;
import org.stacksaga.partition.event.PartitionExecutionCompletedEvent;

@Component
@Slf4j
public class PartitionMonitoringListener {

    @EventListener
    public void onPartitionCompleted(PartitionExecutionCompletedEvent event) {
        if (!event.isAllSuccessful()) {
            log.error("ALERT: 分区创建失败！共评估 {} 个数据源，失败 {} 个，耗时 {} ms！",
                    event.getTotalDatasources(), event.getFailureCount(), event.getDuration().toMillis());

            event.getReports().forEach((dsName, report) -> {
                if (!report.isSuccess()) {
                    log.error("数据源 '{}' 分区失败: {}", dsName, report.getErrorMessage(), report.getError());
                    // 示例：向钉钉、企微、Slack 发送告警通知
                    // alertService.sendAlert("数据库分区失败，数据源: " + dsName);
                }
            });
        } else {
            log.info("SUCCESS: 所有 {} 个数据源分区全部成功完成，总耗时 {} ms。",
                    event.getTotalDatasources(), event.getDuration().toMillis());
        }
    }
}