package org.ruoyi.mcp.service;

import lombok.RequiredArgsConstructor;
import org.ruoyi.system.domain.vo.SysUserVo;
import org.ruoyi.system.mapper.SysUserMapper;
import org.springframework.ai.tool.annotation.Tool;
import org.springframework.stereotype.Service;

/**
 * @author ageer
 */
@Service
@RequiredArgsConstructor
public class McpCustomService {

	private final SysUserMapper userMapper;

	public record User(String userName, Double userBalance) {
	}

	@Tool(description = "根据用户名称查询用户信息")
	public User getUserBalance(String username) {
		SysUserVo sysUserVo = userMapper.selectUserByUserName(username);
		return new User(sysUserVo.getUserName(),sysUserVo.getUserBalance());
	}

}