package com.ruoyi.ai.context;
|
|
import com.ruoyi.framework.security.LoginUser;
|
import org.springframework.stereotype.Component;
|
import org.springframework.util.StringUtils;
|
|
import java.util.Map;
|
import java.util.concurrent.ConcurrentHashMap;
|
|
@Component
|
public class AiSessionUserContext {
|
|
private final Map<String, LoginUser> loginUserByMemoryId = new ConcurrentHashMap<>();
|
|
public void bind(String memoryId, LoginUser loginUser) {
|
if (!StringUtils.hasText(memoryId) || loginUser == null) {
|
return;
|
}
|
loginUserByMemoryId.put(memoryId, loginUser);
|
}
|
|
public LoginUser get(String memoryId) {
|
if (!StringUtils.hasText(memoryId)) {
|
return null;
|
}
|
return loginUserByMemoryId.get(memoryId);
|
}
|
|
public void remove(String memoryId) {
|
if (!StringUtils.hasText(memoryId)) {
|
return;
|
}
|
loginUserByMemoryId.remove(memoryId);
|
}
|
}
|
