diff --git a/src/main/java/com/muyu/system/mapper/BuyMapper.java b/src/main/java/com/muyu/system/mapper/BuyMapper.java index 6b08fb9..50fa01e 100644 --- a/src/main/java/com/muyu/system/mapper/BuyMapper.java +++ b/src/main/java/com/muyu/system/mapper/BuyMapper.java @@ -17,4 +17,5 @@ public interface BuyMapper extends BaseMapper { ConnectorUser selectConnectorUser(ConnectorUser connectorUserr); + Integer addConnectorUser(ConnectorUser connectorUser); } 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 627c07c..3929bc2 100644 --- a/src/main/java/com/muyu/system/service/impl/BuyServiceImpl.java +++ b/src/main/java/com/muyu/system/service/impl/BuyServiceImpl.java @@ -35,7 +35,10 @@ public class BuyServiceImpl extends ServiceImpl implements // LoginUser loginUser = SecurityUtils.getLoginUser(); ConnectorUser connectorUser1 = buyMapper.selectConnectorUser(connectorUser); if (connectorUser1==null){ - return Result.error("购买失败,该API不存在,请先申请API"); + Integer i = buyMapper.addConnectorUser(connectorUser); + if (i>0){ + buyMapper.doBuyInterface(connectorUser); + } } connectorUser.setUserId(userId); Integer i = buyMapper.doBuyInterface(connectorUser); diff --git a/src/main/resources/mapper/system/BuyMapper.xml b/src/main/resources/mapper/system/BuyMapper.xml index 269ce7a..d77bc27 100644 --- a/src/main/resources/mapper/system/BuyMapper.xml +++ b/src/main/resources/mapper/system/BuyMapper.xml @@ -3,6 +3,12 @@ PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd"> + + INSERT INTO `three-groups`.`connector_user` + (`connector_user_id`, `user_id`, `connector_id`, `connector_residue_degree`, `connector_frequency`) + VALUES (0,#{userId}, #{connectorId}, 0, 0) + +