From 9486766c806fe1d9e082b2fd02ea1cc558f1b443 Mon Sep 17 00:00:00 2001
From: 无关风月 <443237572@qq.com>
Date: 星期四, 08 五月 2025 09:21:57 +0800
Subject: [PATCH] bug修改

---
 cloud-server-activity/src/main/java/com/dsh/config/Sharding_jdbc/ShardingConfig.java |   47 +++++++++++++++++++++++++++++++++++++++++++----
 1 files changed, 43 insertions(+), 4 deletions(-)

diff --git a/cloud-server-activity/src/main/java/com/dsh/config/Sharding_jdbc/ShardingConfig.java b/cloud-server-activity/src/main/java/com/dsh/config/Sharding_jdbc/ShardingConfig.java
index 2c53536..4e66630 100644
--- a/cloud-server-activity/src/main/java/com/dsh/config/Sharding_jdbc/ShardingConfig.java
+++ b/cloud-server-activity/src/main/java/com/dsh/config/Sharding_jdbc/ShardingConfig.java
@@ -24,7 +24,7 @@
 
 
     @Bean
-    public DataSource getDataSource(){
+    public DataSource getDataSource() {
         DataSource dataSource = null;
         try {
             Properties properties = new Properties();
@@ -40,6 +40,7 @@
 
     /**
      * 配置多数据源
+     *
      * @return
      */
     private Map<String, DataSource> createDataSourceMap() {
@@ -61,6 +62,7 @@
 
     /**
      * 分片配置
+     *
      * @return
      */
     private Collection<RuleConfiguration> createShardingRuleConfiguration() {
@@ -81,26 +83,38 @@
         result1.getShardingAlgorithms().put("t_user_points_merchandise-inline", new AlgorithmConfiguration("INLINE", props2));
         result1.getKeyGenerators().put("t_user_points_merchandise-snowflake", new AlgorithmConfiguration("SNOWFLAKE", new Properties()));
 
-        linkedList.add(result1);
+        //分片规则配置
+        result1.getTables().add(getUserBenefitsVideosTableRuleConfiguration());
+        Properties props3 = new Properties();
+        props3.setProperty("algorithm-expression", "t_user_benefits_videos$->{appUserId % 5 + 1}");
+        result1.getShardingAlgorithms().put("t_user_benefits_videos-inline", new AlgorithmConfiguration("INLINE", props3));
+        result1.getKeyGenerators().put("t_user_benefits_videos-snowflake", new AlgorithmConfiguration("SNOWFLAKE", new Properties()));
 
+        result1.getTables().add(getUserCouponTableRulePaymentConfiguration());
+        Properties props4 = new Properties();
+        props4.setProperty("algorithm-expression", "t_user_coupon_payment$->{userId % 5 + 1}");
+        result1.getShardingAlgorithms().put("t_user_coupon_payment-inline", new AlgorithmConfiguration("INLINE", props4));
+        result1.getKeyGenerators().put("t_user_coupon_payment-snowflake", new AlgorithmConfiguration("SNOWFLAKE", new Properties()));
+        linkedList.add(result1);
         return linkedList;
     }
 
     /**
      * 分片算法配置
+     *
      * @return
      */
     private ShardingTableRuleConfiguration getUserCouponTableRuleConfiguration() {
-        ShardingTableRuleConfiguration result = new ShardingTableRuleConfiguration("t_user_coupon", "m_$->{0}.t_user_coupon->{1..5}");//50
+        ShardingTableRuleConfiguration result = new ShardingTableRuleConfiguration("t_user_coupon", "m_$->{0}.t_user_coupon$->{1..5}");//50
         result.setTableShardingStrategy(new StandardShardingStrategyConfiguration("userId", "t_user_coupon-inline"));
         result.setKeyGenerateStrategy(new KeyGenerateStrategyConfiguration("id", "t_user_coupon-snowflake"));
         return result;
     }
 
 
-
     /**
      * 分片算法配置
+     *
      * @return
      */
     private ShardingTableRuleConfiguration getUserPointsMerchandiseTableRuleConfiguration() {
@@ -110,4 +124,29 @@
         return result;
     }
 
+
+    /**
+     * 分片算法配置
+     *
+     * @return
+     */
+    private ShardingTableRuleConfiguration getUserBenefitsVideosTableRuleConfiguration() {
+        ShardingTableRuleConfiguration result = new ShardingTableRuleConfiguration("t_user_benefits_videos", "m_$->{0}.t_user_benefits_videos$->{1..5}");//30
+        result.setTableShardingStrategy(new StandardShardingStrategyConfiguration("appUserId", "t_user_benefits_videos-inline"));
+        result.setKeyGenerateStrategy(new KeyGenerateStrategyConfiguration("id", "t_user_benefits_videos-snowflake"));
+        return result;
+    }
+
+    /**
+     * 分片算法配置
+     *
+     * @return
+     */
+    private ShardingTableRuleConfiguration getUserCouponTableRulePaymentConfiguration() {
+        ShardingTableRuleConfiguration result = new ShardingTableRuleConfiguration("t_user_coupon_payment", "m_$->{0}.t_user_coupon_payment$->{1..5}");//50
+        result.setTableShardingStrategy(new StandardShardingStrategyConfiguration("userId", "t_user_coupon_payment-inline"));
+        result.setKeyGenerateStrategy(new KeyGenerateStrategyConfiguration("id", "t_user_coupon_payment-snowflake"));
+        return result;
+    }
+
 }

--
Gitblit v1.7.1