Commit 55900ee5 authored by shangtx's avatar shangtx

feat: 登录时进行推广统计

parent 8e95945e
...@@ -11,13 +11,16 @@ import com.onsiteservice.constant.enums.ServiceUserTypeEnum; ...@@ -11,13 +11,16 @@ import com.onsiteservice.constant.enums.ServiceUserTypeEnum;
import com.onsiteservice.core.exception.ServiceException; import com.onsiteservice.core.exception.ServiceException;
import com.onsiteservice.core.result.Result; import com.onsiteservice.core.result.Result;
import com.onsiteservice.core.security.jwt.JwtManager; import com.onsiteservice.core.security.jwt.JwtManager;
import com.onsiteservice.entity.service.ServicePromote;
import com.onsiteservice.entity.user.User; import com.onsiteservice.entity.user.User;
import com.onsiteservice.miniapp.controller.weixn.dto.LoginDTO; import com.onsiteservice.miniapp.controller.weixn.dto.LoginDTO;
import com.onsiteservice.miniapp.controller.weixn.dto.WxEncryptDTO; import com.onsiteservice.miniapp.controller.weixn.dto.WxEncryptDTO;
import com.onsiteservice.miniapp.service.user.ServicePromoteService;
import com.onsiteservice.miniapp.service.user.UserService; import com.onsiteservice.miniapp.service.user.UserService;
import io.micrometer.core.instrument.util.StringUtils; import io.micrometer.core.instrument.util.StringUtils;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import me.chanjar.weixin.common.error.WxErrorException; import me.chanjar.weixin.common.error.WxErrorException;
import org.apache.commons.lang3.ObjectUtils;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional; import org.springframework.transaction.annotation.Transactional;
...@@ -44,6 +47,8 @@ public class WeiXinService { ...@@ -44,6 +47,8 @@ public class WeiXinService {
private WxMaService wxMaService; private WxMaService wxMaService;
@Resource @Resource
private UserService userService; private UserService userService;
@Resource
private ServicePromoteService promoteService;
/** /**
...@@ -98,6 +103,9 @@ public class WeiXinService { ...@@ -98,6 +103,9 @@ public class WeiXinService {
.roleType(ServiceUserTypeEnum.USER.getId()) .roleType(ServiceUserTypeEnum.USER.getId())
.build(); .build();
userService.insertSelective(user); userService.insertSelective(user);
if(ObjectUtils.isNotEmpty(loginDTO.getShareUserId())) {
promoteService.promoteCount(loginDTO.getShareUserId());
}
return toLogin(user, user.getId(), "注册成功"); return toLogin(user, user.getId(), "注册成功");
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment