60 行
2.3 KiB
Java
60 行
2.3 KiB
Java
|
package com.xuqm.im.service;
|
||
|
|
|
||
|
|
import com.xuqm.common.exception.BusinessException;
|
||
|
|
import com.xuqm.common.security.JwtUtil;
|
||
|
|
import com.xuqm.im.entity.ImAccountEntity;
|
||
|
|
import com.xuqm.im.repository.ImAccountRepository;
|
||
|
|
import org.springframework.stereotype.Service;
|
||
|
|
|
||
|
|
import java.time.LocalDateTime;
|
||
|
|
import java.util.Map;
|
||
|
|
import java.util.UUID;
|
||
|
|
|
||
|
|
@Service
|
||
|
|
public class ImAccountService {
|
||
|
|
|
||
|
|
private final ImAccountRepository accountRepository;
|
||
|
|
private final JwtUtil jwtUtil;
|
||
|
|
|
||
|
|
public ImAccountService(ImAccountRepository accountRepository, JwtUtil jwtUtil) {
|
||
|
|
this.accountRepository = accountRepository;
|
||
|
|
this.jwtUtil = jwtUtil;
|
||
|
|
}
|
||
|
|
|
||
|
|
public String loginOrRegister(String appId, String userId, String nickname, String avatar) {
|
||
|
|
ImAccountEntity account = accountRepository.findByAppIdAndUserId(appId, userId)
|
||
|
|
.orElseGet(() -> {
|
||
|
|
ImAccountEntity e = new ImAccountEntity();
|
||
|
|
e.setId(UUID.randomUUID().toString());
|
||
|
|
e.setAppId(appId);
|
||
|
|
e.setUserId(userId);
|
||
|
|
e.setNickname(nickname);
|
||
|
|
e.setAvatar(avatar);
|
||
|
|
e.setGender(ImAccountEntity.Gender.UNKNOWN);
|
||
|
|
e.setStatus(ImAccountEntity.Status.ACTIVE);
|
||
|
|
e.setCreatedAt(LocalDateTime.now());
|
||
|
|
return accountRepository.save(e);
|
||
|
|
});
|
||
|
|
|
||
|
|
if (account.getStatus() == ImAccountEntity.Status.BANNED) {
|
||
|
|
throw new BusinessException(403, "账号已被封禁");
|
||
|
|
}
|
||
|
|
|
||
|
|
return jwtUtil.generate(userId, Map.of("appId", appId, "role", "USER"));
|
||
|
|
}
|
||
|
|
|
||
|
|
public ImAccountEntity getAccount(String appId, String userId) {
|
||
|
|
return accountRepository.findByAppIdAndUserId(appId, userId)
|
||
|
|
.orElseThrow(() -> new BusinessException(404, "账号不存在"));
|
||
|
|
}
|
||
|
|
|
||
|
|
public ImAccountEntity updateAccount(String appId, String userId, String nickname,
|
||
|
|
String avatar, ImAccountEntity.Gender gender) {
|
||
|
|
ImAccountEntity account = getAccount(appId, userId);
|
||
|
|
if (nickname != null) account.setNickname(nickname);
|
||
|
|
if (avatar != null) account.setAvatar(avatar);
|
||
|
|
if (gender != null) account.setGender(gender);
|
||
|
|
return accountRepository.save(account);
|
||
|
|
}
|
||
|
|
}
|
