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 |   45 +++++++++++++++++++++++++++++++--------------
 1 files changed, 31 insertions(+), 14 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 7e0a34d..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() {
@@ -73,44 +75,46 @@
         props1.setProperty("algorithm-expression", "t_user_coupon$->{userId % 5 + 1}");
         result1.getShardingAlgorithms().put("t_user_coupon-inline", new AlgorithmConfiguration("INLINE", props1));
         result1.getKeyGenerators().put("t_user_coupon-snowflake", new AlgorithmConfiguration("SNOWFLAKE", new Properties()));
-        linkedList.add(result1);
 
         //分片规则配置
-        ShardingRuleConfiguration result2 = new ShardingRuleConfiguration();
-        result2.getTables().add(getUserPointsMerchandiseTableRuleConfiguration());
+        result1.getTables().add(getUserPointsMerchandiseTableRuleConfiguration());
         Properties props2 = new Properties();
         props2.setProperty("algorithm-expression", "t_user_points_merchandise$->{userId % 5 + 1}");
-        result2.getShardingAlgorithms().put("t_user_points_merchandise-inline", new AlgorithmConfiguration("INLINE", props2));
-        result2.getKeyGenerators().put("t_user_points_merchandise-snowflake", new AlgorithmConfiguration("SNOWFLAKE", new Properties()));
-        linkedList.add(result2);
+        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()));
 
         //分片规则配置
-        ShardingRuleConfiguration result3 = new ShardingRuleConfiguration();
-        result3.getTables().add(getUserBenefitsVideosTableRuleConfiguration());
+        result1.getTables().add(getUserBenefitsVideosTableRuleConfiguration());
         Properties props3 = new Properties();
         props3.setProperty("algorithm-expression", "t_user_benefits_videos$->{appUserId % 5 + 1}");
-        result3.getShardingAlgorithms().put("t_user_benefits_videos-inline", new AlgorithmConfiguration("INLINE", props3));
-        result3.getKeyGenerators().put("t_user_benefits_videos-snowflake", new AlgorithmConfiguration("SNOWFLAKE", new Properties()));
-        linkedList.add(result3);
+        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() {
@@ -123,6 +127,7 @@
 
     /**
      * 分片算法配置
+     *
      * @return
      */
     private ShardingTableRuleConfiguration getUserBenefitsVideosTableRuleConfiguration() {
@@ -132,4 +137,16 @@
         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