深入解析CustomerInfoServiceImpl和CustomerServiceImpl：微信小程序登录功能的实现

Takumilovexu

于 2024-08-10 21:57:33 发布

阅读量590

点赞数 20

文章标签：微信小程序小程序 java spring cloud

本文链接：https://blog.csdn.net/Takumilove/article/details/141096838

版权

文章目录

在现代移动应用开发中，微信小程序已经成为了许多企业的重要选择。为了支持小程序的用户登录功能，我们需要与微信的认证服务进行集成。这篇文章将带大家深入解析两个关键的服务实现类： CustomerInfoServiceImpl 和 CustomerServiceImpl，它们是如何共同工作来实现微信小程序登录功能的。

一、CustomerInfoServiceImpl类详解

CustomerInfoServiceImpl类主要负责处理用户信息的管理和登录日志的记录。它继承了ServiceImpl，并实现了CustomerInfoService接口。下面我们来详细了解它的核心逻辑。

package com.atguigu.daijia.customer.service.impl;

import cn.binarywang.wx.miniapp.api.WxMaService;
import cn.binarywang.wx.miniapp.bean.WxMaJscode2SessionResult;
import com.atguigu.daijia.customer.mapper.CustomerInfoMapper;
import com.atguigu.daijia.customer.mapper.CustomerLoginLogMapper;
import com.atguigu.daijia.customer.service.CustomerInfoService;
import com.atguigu.daijia.model.entity.customer.CustomerInfo;
import com.atguigu.daijia.model.entity.customer.CustomerLoginLog;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import me.chanjar.weixin.common.error.WxErrorException;
import org.springframework.stereotype.Service;

@Slf4j
@Service
@SuppressWarnings({"unchecked", "rawtypes"})
@RequiredArgsConstructor
public class CustomerInfoServiceImpl extends ServiceImpl<CustomerInfoMapper, CustomerInfo> implements
        CustomerInfoService {

    private final WxMaService wxMaService;
    private final CustomerInfoMapper customerInfoMapper;
    private final CustomerLoginLogMapper customerLoginLogMapper;

    // 微信小程序登录接口
    @Override
    public Long login(String code) {
        // 1.获取code值，使用微信工具包对象，获取微信唯一标识openid
        String openid = null;
        try {
            WxMaJscode2SessionResult sessionInfo = wxMaService.getUserService().getSessionInfo(code);
            openid = sessionInfo.getOpenid();
        } catch (WxErrorException e) {
            throw new RuntimeException(e);
        }
        // 2.根据openid查询数据库表，判断是否第一次登录
        // 如果不存返回null，如果存在返回一条记录
        LambdaQueryWrapper<CustomerInfo> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(CustomerInfo::getWxOpenId, openid);
        CustomerInfo customerInfo = customerInfoMapper.selectOne(wrapper);
        // 3.如果是第一次登录，将用户信息插入数据库表
        if (customerInfo == null) {
            customerInfo = new CustomerInfo();
            customerInfo.setNickname(String.valueOf(System.currentTimeMillis()));
            customerInfo.setAvatarUrl("https://oss.aliyuncs.com/aliyun_id_photo_bucket/default_handsome.jpg");
            customerInfo.setWxOpenId(openid);
            customerInfoMapper.insert(customerInfo);
        }
        // 4.记录登录日志
        CustomerLoginLog customerLoginLog = new CustomerLoginLog();
        customerLoginLog.setCustomerId(customerInfo.getId());
        customerLoginLog.setMsg("小程序登录");
        customerLoginLogMapper.insert(customerLoginLog);

        // 5.返回用户id
        return customerInfo.getId();
    }
}

获取微信用户的openid

在微信小程序中，code是通过调用wx.login接口获取的，作为登录的凭证。我们通过wxMaService来获取openid，这是每个微信用户的唯一标识。
```
WxMaJscode2SessionResult sessionInfo = wxMaService.getUserService().getSessionInfo(code);
String openid = sessionInfo.getOpenid();
```

检查用户是否是第一次登录

通过openid在数据库中查询用户信息。如果没有找到对应的用户信息，则说明这是用户的首次登录。在这种情况下，我们会创建一条新的用户记录。

LambdaQueryWrapper<CustomerInfo> wrapper = new LambdaQueryWrapper<>();
wrapper.eq(CustomerInfo::getWxOpenId, openid);
CustomerInfo customerInfo = customerInfoMapper.selectOne(wrapper);

if (customerInfo == null) {
    customerInfo = new CustomerInfo();
    customerInfo.setNickname(String.valueOf(System.currentTimeMillis()));
    customerInfo.setAvatarUrl("https://oss.aliyuncs.com/aliyun_id_photo_bucket/default_handsome.jpg");
    customerInfo.setWxOpenId(openid);
    customerInfoMapper.insert(customerInfo);
}

记录登录日志

每次用户登录，我们都会记录一条登录日志，以便后续分析和追踪。

CustomerLoginLog customerLoginLog = new CustomerLoginLog();
customerLoginLog.setCustomerId(customerInfo.getId());
customerLoginLog.setMsg("小程序登录");
customerLoginLogMapper.insert(customerLoginLog);

返回用户ID

最后，我们将用户的ID返回，供后续使用。
```
return customerInfo.getId();
```

二、CustomerServiceImpl类详解

CustomerServiceImpl类则主要负责与其他微服务的交互以及处理登录后的Token管理。它通过调用远程的用户服务，获取用户的登录信息，并将登录状态存储在Redis中。

package com.atguigu.daijia.customer.service.impl;

import com.atguigu.daijia.common.constant.RedisConstant;
import com.atguigu.daijia.common.execption.GuiguException;
import com.atguigu.daijia.common.result.Result;
import com.atguigu.daijia.common.result.ResultCodeEnum;
import com.atguigu.daijia.customer.client.CustomerInfoFeignClient;
import com.atguigu.daijia.customer.service.CustomerService;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;

import java.util.UUID;
import java.util.concurrent.TimeUnit;

@Slf4j
@Service
@SuppressWarnings({"unchecked", "rawtypes"})
@RequiredArgsConstructor
public class CustomerServiceImpl implements CustomerService {

    // 注入远程调用接口
    private final CustomerInfoFeignClient client;
    private final RedisTemplate redisTemplate;

    @Override
    public String login(String code) {
        // 1.远程调用，返回用户id
        Result<Long> loginResult = client.login(code);

        // 2.判断如果返回失败，返回错误提示
        Integer codeResult = loginResult.getCode();
        if (codeResult != 200) {
            throw new GuiguException(ResultCodeEnum.DATA_ERROR);
        }
        // 3.获取远程调用返回的用户id
        Long customerId = loginResult.getData();
        // 4.判断返回用户id是否为空，如果为空，返回错误提示
        if (customerId == null) {
            throw new GuiguException(ResultCodeEnum.DATA_ERROR);
        }
        // 5.生成token字符串
        String token = UUID.randomUUID().toString().replaceAll("-", "");
        // 6.把用户id放到Redis，设置过期时间
        // key:token value:customerId
        redisTemplate.opsForValue().set(RedisConstant.USER_LOGIN_KEY_PREFIX + token,
                                        customerId.toString(),
                                        RedisConstant.USER_LOGIN_KEY_TIMEOUT,
                                        TimeUnit.SECONDS);
        // 7.返回token字符串
        return token;
    }
}

远程调用获取用户ID

CustomerServiceImpl通过Feign客户端远程调用CustomerInfoService的login方法，获取用户ID。
```
Result<Long> loginResult = client.login(code);
```
错误处理

在调用远程服务后，首先要检查返回的结果状态码，如果状态码不为200，说明请求失败，抛出自定义异常。
```
if (loginResult.getCode() != 200) {
    throw new GuiguException(ResultCodeEnum.DATA_ERROR);
}
```

生成Token并存储到Redis

为了管理用户的会话状态，我们生成了一个唯一的Token，并将其与用户ID绑定存储到Redis中，这样可以在后续的请求中验证用户身份。

String token = UUID.randomUUID().toString().replaceAll("-", "");
redisTemplate.opsForValue().set(RedisConstant.USER_LOGIN_KEY_PREFIX + token,
                                customerId.toString(),
                                RedisConstant.USER_LOGIN_KEY_TIMEOUT,
                                TimeUnit.SECONDS);

返回Token

最终，将生成的Token返回给前端，前端会在后续的请求中使用这个Token来进行身份验证。
```
return token;
```

三、总结

CustomerInfoServiceImpl和CustomerServiceImpl这两个类的协作，完整实现了微信小程序的用户登录功能。前者负责处理微信用户信息的管理，后者则负责会话状态的管理与安全。这种设计使得代码清晰且职责明确，便于维护和扩展。

Takumilovexu

关注

20
点赞
踩
16

收藏

觉得还不错? 一键收藏
0
评论
深入解析CustomerInfoServiceImpl和CustomerServiceImpl：微信小程序登录功能的实现

和这两个类的协作，完整实现了微信小程序的用户登录功能。前者负责处理微信用户信息的管理，后者则负责会话状态的管理与安全。这种设计使得代码清晰且职责明确，便于维护和扩展。
复制链接

扫一扫