BJ_ZTEITS-Developers / parking-cloud-yudao

Commit 4e9be7a1815f8c96567294d7600870ee88c7c028

Authored by owen
1 parent 25973496

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

Inline Side-by-side
Showing 2 changed files with 14 additions and 1 deletions
yudao-module-mall/yudao-module-trade-biz/src/main/java/cn/iocoder/yudao/module/trade/dal/mysql/brokerage/user/BrokerageUserMapper.java
  View file @4e9be7a
@@ -105,4 +105,10 @@ public interface BrokerageUserMapper extends BaseMapperX<BrokerageUserDO> { @@ -105,4 +105,10 @@ public interface BrokerageUserMapper extends BaseMapperX<BrokerageUserDO> {
105 .eq(BrokerageUserDO::getId, id) 105 .eq(BrokerageUserDO::getId, id)
106 .set(BrokerageUserDO::getBindUserId, null).set(BrokerageUserDO::getBindUserTime, null)); 106 .set(BrokerageUserDO::getBindUserId, null).set(BrokerageUserDO::getBindUserTime, null));
107 } 107 }
  108 +
  109 + default void updateEnabledFalseAndBrokerageTimeToNull(Long id) {
  110 + update(null, new LambdaUpdateWrapper<BrokerageUserDO>()
  111 + .eq(BrokerageUserDO::getId, id)
  112 + .set(BrokerageUserDO::getBrokerageEnabled, false).set(BrokerageUserDO::getBrokerageTime, null));
  113 + }
108 } 114 }
yudao-module-mall/yudao-module-trade-biz/src/main/java/cn/iocoder/yudao/module/trade/service/brokerage/user/BrokerageUserServiceImpl.java
  View file @4e9be7a
@@ -71,7 +71,14 @@ public class BrokerageUserServiceImpl implements BrokerageUserService { @@ -71,7 +71,14 @@ public class BrokerageUserServiceImpl implements BrokerageUserService {
71 public void updateBrokerageEnabled(Long id, Boolean enabled) { 71 public void updateBrokerageEnabled(Long id, Boolean enabled) {
72 // 校验存在 72 // 校验存在
73 validateBrokerageUserExists(id); 73 validateBrokerageUserExists(id);
74 - // TODO @疯狂：貌似没实现完 74 + if (BooleanUtil.isTrue(enabled)) {
  75 + // 开通推广资格
  76 + brokerageUserMapper.updateById(new BrokerageUserDO().setId(id)
  77 + .setBrokerageEnabled(true).setBrokerageTime(LocalDateTime.now()));
  78 + } else {
  79 + // 取消推广资格
  80 + brokerageUserMapper.updateEnabledFalseAndBrokerageTimeToNull(id);
  81 + }
75 } 82 }
76 83
77 private void validateBrokerageUserExists(Long id) { 84 private void validateBrokerageUserExists(Long id) {