明星电力充电桩
blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
Pu Zhibing
2024-09-30 f541eaa5c92cb112af6d3f656275105606d31b39 
 ruoyi-service/ruoyi-account/src/main/java/com/ruoyi/account/controller/WxLoginController.java


@@ -1,6 +1,7 @@


package com.ruoyi.account.controller;




import com.alibaba.fastjson.JSONObject;


import com.alipay.api.internal.util.codec.Base64;


import com.baomidou.mybatisplus.core.conditions.Wrapper;


import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;


import com.baomidou.mybatisplus.core.toolkit.ObjectUtils;


@@ -15,6 +16,7 @@


import com.ruoyi.account.wx.tools.WxAppletTools;


import com.ruoyi.account.wx.tools.WxUtils;


import com.ruoyi.common.core.exception.ServiceException;


import com.ruoyi.common.core.utils.FileUploadUtils;


import com.ruoyi.common.core.utils.StringUtils;


import com.ruoyi.common.core.web.domain.AjaxResult;


import com.ruoyi.common.redis.service.RedisService;


@@ -24,12 +26,25 @@


import io.swagger.annotations.ApiParam;


import lombok.extern.slf4j.Slf4j;


import org.springframework.beans.factory.annotation.Autowired;


import org.springframework.beans.factory.annotation.Value;


import org.springframework.http.HttpEntity;


import org.springframework.http.HttpMethod;


import org.springframework.http.ResponseEntity;


import org.springframework.mock.web.MockMultipartFile;


import org.springframework.util.LinkedMultiValueMap;


import org.springframework.util.MultiValueMap;


import org.springframework.web.bind.annotation.*;


import org.springframework.web.client.RestTemplate;


import org.springframework.web.multipart.MultipartFile;




import java.util.HashMap;


import java.util.Map;


import java.util.Objects;


import java.io.ByteArrayInputStream;


import java.io.IOException;


import java.io.InputStream;


import java.io.OutputStream;


import java.security.SecureRandom;


import java.text.SimpleDateFormat;


import java.util.*;




/**


 * <p>


 *  微信小程序登录 前端控制器


@@ -48,6 +63,11 @@


    private WeixinProperties wxConfig;


    @Autowired


    private RestTemplate wxRestTemplate;


    /**


     * 上传文件存储在本地的根路径


     */


    @Value("${file.path}")


    private String localFilePath;


    @ApiOperation(value = "通过code获得openid,获取用户信息",tags = {"微信小程序登录"})


    @PostMapping("/openIdByJsCode")


    public AjaxResult<Map<String, Object>> openIdByJsCode(@RequestBody AppletUserEncrypteData data) {


@@ -79,4 +99,62 @@


    public AjaxResult<Map<String, Object>> openIdByJsCode(@RequestBody AppletUserDecodeData appletUserDecodeData) {


        return AjaxResult.success(appUserService.wxLogin(appletUserDecodeData, null));


    }


    public static MultipartFile convertInputStreamToMultipartFile(InputStream inputStream, String fileName, String contentType) throws IOException {


        return new MockMultipartFile(fileName, fileName, contentType, inputStream);


    }


    @ApiOperation(value = "获取微信小程序二维码",tags = {"获取微信小程序二维码"})


    @PostMapping("/getQRCode")


    public AjaxResult getQRCode() {


        InputStream inputStream = null;


        OutputStream outputStream = null;


        WxAppletTools appletTools = new WxAppletTools(wxRestTemplate, wxConfig);


        String accessToken = appletTools.getAccessToken("");


        try {


            String url = "https://api.weixin.qq.com/wxa/getwxacodeunlimit?access_token=" + accessToken;


            Map<String, Object> param = new HashMap<>();


//            param.put("page", "pageA/houseDetail");


            param.put("check_path", false);


            param.put("env_version", "trial");


            param.put("width", 200); //二维码尺寸


            param.put("is_hyaline", true); // 是否需要透明底色， is_hyaline 为true时，生成透明底色的小程序码 参数仅对小程序码生效


            param.put("auto_color", true); // 自动配置线条颜色，如果颜色依然是黑色，则说明不建议配置主色调 参数仅对小程序码生效


            Map<String, Object> line_color = new HashMap<>();


            line_color.put("r", 0);


            line_color.put("g", 0);


            line_color.put("b", 0);


            param.put("line_color", line_color);


            System.err.println("调用生成微信URL接口传参:" + param);


            MultiValueMap<String, String> headers = new LinkedMultiValueMap<>();


            HttpEntity requestEntity = new HttpEntity(param, headers);


            ResponseEntity<byte[]> entity = wxRestTemplate.exchange(url, HttpMethod.POST, requestEntity, byte[].class, new Object[0]);


            System.err.println("调用小程序生成微信永久小程序码URL接口返回结果:" + entity.getBody());


            byte[] result = entity.getBody();


            System.err.println(Base64.encodeBase64String(result));


            inputStream = new ByteArrayInputStream(result);


            String finalFileName = System.currentTimeMillis() + "" + new SecureRandom().nextInt(0x0400) + ".jpeg";


            MultipartFile multipartFile = convertInputStreamToMultipartFile(inputStream, finalFileName, "image/jpeg");


            String name = FileUploadUtils.upload(localFilePath, multipartFile);


            System.err.println(name);


            return AjaxResult.success(name);


        } catch (Exception e) {


            System.err.println("调用小程序生成微信永久小程序码URL接口异常" + e);


        } finally {


            if (inputStream != null) {


                try {


                    inputStream.close();


                } catch (IOException e) {


                    e.printStackTrace();


                }


            }


            if (outputStream != null) {


                try {


                    outputStream.close();


                } catch (IOException e) {


                    e.printStackTrace();


                }


            }


        }


        return AjaxResult.success();


    }




}