Просмотр исходного кода

trade: 分销业务后台功能:修改推广资格

owen 1 год назад
Родитель
Сommit
4e9be7a181

+ 6 - 0
yudao-module-mall/yudao-module-trade-biz/src/main/java/cn/iocoder/yudao/module/trade/dal/mysql/brokerage/user/BrokerageUserMapper.java

@@ -105,4 +105,10 @@ public interface BrokerageUserMapper extends BaseMapperX<BrokerageUserDO> {
                 .eq(BrokerageUserDO::getId, id)
                 .set(BrokerageUserDO::getBindUserId, null).set(BrokerageUserDO::getBindUserTime, null));
     }
+
+    default void updateEnabledFalseAndBrokerageTimeToNull(Long id) {
+        update(null, new LambdaUpdateWrapper<BrokerageUserDO>()
+                .eq(BrokerageUserDO::getId, id)
+                .set(BrokerageUserDO::getBrokerageEnabled, false).set(BrokerageUserDO::getBrokerageTime, null));
+    }
 }

+ 8 - 1
yudao-module-mall/yudao-module-trade-biz/src/main/java/cn/iocoder/yudao/module/trade/service/brokerage/user/BrokerageUserServiceImpl.java

@@ -71,7 +71,14 @@ public class BrokerageUserServiceImpl implements BrokerageUserService {
     public void updateBrokerageEnabled(Long id, Boolean enabled) {
         // 校验存在
         validateBrokerageUserExists(id);
-        // TODO @疯狂:貌似没实现完
+        if (BooleanUtil.isTrue(enabled)) {
+            // 开通推广资格
+            brokerageUserMapper.updateById(new BrokerageUserDO().setId(id)
+                    .setBrokerageEnabled(true).setBrokerageTime(LocalDateTime.now()));
+        } else {
+            // 取消推广资格
+            brokerageUserMapper.updateEnabledFalseAndBrokerageTimeToNull(id);
+        }
     }
 
     private void validateBrokerageUserExists(Long id) {