diff --git a/src/main/java/com/muyu/system/domain/ConnectorUser.java b/src/main/java/com/muyu/system/domain/ConnectorUser.java index f8d1937..241713b 100644 --- a/src/main/java/com/muyu/system/domain/ConnectorUser.java +++ b/src/main/java/com/muyu/system/domain/ConnectorUser.java @@ -2,6 +2,8 @@ package com.muyu.system.domain; import lombok.Data; +import java.math.BigDecimal; + /** * @Author:chaiyapeng * @Package:com.muyu.system.domain @@ -35,6 +37,10 @@ public class ConnectorUser { * 接口名称 */ private String connectorName; + /** + * 价钱 + */ + private Double connectorPrice; /** * 用户名称 */ diff --git a/src/main/java/com/muyu/system/mapper/BuyMapper.java b/src/main/java/com/muyu/system/mapper/BuyMapper.java index 50fa01e..f7803c9 100644 --- a/src/main/java/com/muyu/system/mapper/BuyMapper.java +++ b/src/main/java/com/muyu/system/mapper/BuyMapper.java @@ -1,8 +1,10 @@ package com.muyu.system.mapper; import com.baomidou.mybatisplus.core.mapper.BaseMapper; +import com.muyu.common.system.domain.SysUser; import com.muyu.system.domain.Connector; import com.muyu.system.domain.ConnectorUser; +import org.apache.catalina.User; /** * @Author:chaiyapeng @@ -18,4 +20,8 @@ public interface BuyMapper extends BaseMapper { ConnectorUser selectConnectorUser(ConnectorUser connectorUserr); Integer addConnectorUser(ConnectorUser connectorUser); + + SysUser selectUser(Long userId); + + void buyPrice(SysUser user); } diff --git a/src/main/java/com/muyu/system/service/impl/BuyServiceImpl.java b/src/main/java/com/muyu/system/service/impl/BuyServiceImpl.java index b822ab6..9d47e1e 100644 --- a/src/main/java/com/muyu/system/service/impl/BuyServiceImpl.java +++ b/src/main/java/com/muyu/system/service/impl/BuyServiceImpl.java @@ -4,10 +4,12 @@ import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import com.muyu.common.core.domain.Result; import com.muyu.common.security.utils.SecurityUtils; import com.muyu.common.system.domain.LoginUser; +import com.muyu.common.system.domain.SysUser; import com.muyu.system.domain.Connector; import com.muyu.system.domain.ConnectorUser; import com.muyu.system.mapper.BuyMapper; import com.muyu.system.service.BuyService; +import org.apache.catalina.User; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; @@ -36,6 +38,9 @@ public class BuyServiceImpl extends ServiceImpl implements connectorUser.setUserId(userId); Integer i = buyMapper.doBuyInterface(connectorUser); if (i>0){ + SysUser user = buyMapper.selectUser(userId); + user.setUserBalance(user.getUserBalance()-connectorUser.getConnectorFrequency()*connectorUser.getConnectorPrice()); + buyMapper.buyPrice(user); return Result.success(i,"购买成功"); } ConnectorUser connectorUser1 = buyMapper.selectConnectorUser(connectorUser); @@ -43,6 +48,9 @@ public class BuyServiceImpl extends ServiceImpl implements Integer i1 = buyMapper.addConnectorUser(connectorUser); if (i1>0){ buyMapper.doBuyInterface(connectorUser); + SysUser user = buyMapper.selectUser(userId); + user.setUserBalance(user.getUserBalance()-connectorUser.getConnectorFrequency()*connectorUser.getConnectorPrice()); + buyMapper.buyPrice(user); return Result.success(i,"购买成功"); } } diff --git a/src/main/resources/mapper/system/BuyMapper.xml b/src/main/resources/mapper/system/BuyMapper.xml index ab0adf8..4f0123c 100644 --- a/src/main/resources/mapper/system/BuyMapper.xml +++ b/src/main/resources/mapper/system/BuyMapper.xml @@ -14,13 +14,17 @@ update connector_user set connector_residue_degree=connector_residue_degree+#{connectorFrequency},connector_frequency=connector_frequency+#{connectorFrequency} where connector_id=#{connectorId} and user_id=#{userId} + + update sys_user set user_balance=#{userBalance} where user_id=#{userId} + + +