temp-witstalk/lib-user/src/main/java/top/xinsin/controller/UserController.java

package top.xinsin.controller;

import com.alibaba.fastjson2.JSON;
import com.alibaba.fastjson2.JSONObject;
import lombok.SneakyThrows;
import org.apache.commons.httpclient.HttpClient;
import org.apache.commons.httpclient.methods.GetMethod;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.web.bind.annotation.*;
import top.xinsin.pojo.User;
import top.xinsin.services.UserService;
import top.xinsin.utils.IpUtils;
import top.xinsin.utils.RData;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.util.List;

/**
 * @author xinxin
 * Created On 2021/12/11 18:06
 * @version 1.0
 */
@RestController
public class UserController {
    private final UserService userService;
    @Autowired
    public UserController(UserService userService){
        this.userService = userService;
    }

    /**
     * 登录接口
     * @param user 用户
     * @param response 请求体
     * @return 是否可以登陆
     */
    @PostMapping("/user/api/login")
    public RData<JSONObject> login(@RequestBody User user, HttpServletResponse response) {
        return userService.canLogin(user,response);
    }

    /**
     * 添加用户接口
     * @param user 需要添加的用户
     * @return 受影响行数
     */
    @PostMapping("/user/api/adduser")
    public RData<JSONObject> addUser(@RequestBody User user) {return userService.addUser(user);}

    /**
     * @param user 需要修改密码的用户，此用户的password字段为需要为修改后的密码
     * @return 受影响行数
     */
    @PostMapping("/user/api/changepassword")
    public RData<JSONObject> changePassword(@RequestBody User user, HttpServletRequest request) {
        return userService.changePassword(user, request);
    }

    /**
     * 设置用户头像接口
     *
     * @param user 需要修改头像的用户，此用户的head
     * @return 受影响行数
     */
    @PostMapping("/user/api/setHeadPortrait")
    public RData<JSONObject> setHeadPortrait(@RequestBody User user, HttpServletRequest request) {
        return userService.setHeadPortrait(user, request);
    }

    /**
     * 获取用户头像接口
     * @param username 用户名
     * @return 用户头像的base64
     */
    @GetMapping("/user/api/getUserHeadPortrait/{username}")
    public String getUserHeadPortrait(@PathVariable String username){
        return userService.getUserHeadPortrait(username).toString();
    }

    /**
     * 获取所有在线用户接口
     *
     * @return 在线用户
     */
    @GetMapping("/user/api/getOnlineUser")
    public RData<List<User>> getOnlineUser() {
        return userService.getOnlineUser();
    }

    /**
     * 修改用户名
     *
     * @param user    新用户
     * @param request 请求
     * @return 是否成功
     */
    @PostMapping("/user/api/changeUsername")
    public RData<Boolean> changeUsername(@RequestBody User user, HttpServletRequest request, HttpServletResponse response) {
        return userService.changeUsername(user, request, response);
    }
    @GetMapping("/user/api/setUserExclusiveColor")
    public RData<Boolean> setUserExclusiveColor(@RequestHeader("color") String color,@RequestHeader("username")String username){return userService.setColorById(color,username);}
    @GetMapping("/user/api/autoLogin")
    public RData<JSONObject> autoLogin(HttpServletRequest request){return userService.autoLogin(request);}
    @Value("${map-key}")
    private String key;
    @SneakyThrows
    @GetMapping("/user/api/getWeather")
    public RData<JSONObject> getWeather(HttpServletRequest httpServletRequest){
        String ipAddr = IpUtils.getIpAddr(httpServletRequest);
//        检测为本地ip则进行默认地点查询
        if ("127.0.0.1".equals(ipAddr)){
            HttpClient httpClient = new HttpClient();
            httpClient.getHttpConnectionManager().getParams().setConnectionTimeout(15000);
            GetMethod getMethod = new GetMethod("https://restapi.amap.com/v3/ip?key=" + key);
            httpClient.executeMethod(getMethod);
            String responseBodyAsString = getMethod.getResponseBodyAsString();
            JSONObject jsonObject = JSON.parseObject(responseBodyAsString);
            getMethod.releaseConnection();
            GetMethod getWeather = new GetMethod("https://restapi.amap.com/v3/weather/weatherInfo?key=" + key + "&city="+jsonObject.getString("adcode")+"&awextensions=base");
            httpClient.executeMethod(getWeather);
            String getWeatherResult = getWeather.getResponseBodyAsString();
            getWeather.releaseConnection();
            return RData.success(JSON.parseObject(getWeatherResult));
        }else{
            HttpClient httpClient = new HttpClient();
            httpClient.getHttpConnectionManager().getParams().setConnectionTimeout(15000);
            GetMethod getMethod = new GetMethod("https://restapi.amap.com/v3/ip?key=" + key + "&ip=" + ipAddr);
            httpClient.executeMethod(getMethod);
            String result = getMethod.getResponseBodyAsString();
            getMethod.releaseConnection();
            String adcode = JSON.parseObject(result).getString("adcode");
            if (adcode != null){
                GetMethod getWeather = new GetMethod("https://restapi.amap.com/v3/weather/weatherInfo?key=" + key + "&city=" + adcode + "&extensions=base");
                httpClient.executeMethod(getWeather);
                String getWeatherResult = getWeather.getResponseBodyAsString();
                getWeather.releaseConnection();
                return RData.success(JSON.parseObject(getWeatherResult));
            }else{
                return RData.success(new JSONObject().fluentPut("status","未查询到相关天气信息"));
            }
        }
    }
}