mitao
2025-01-21 66f7e9045df3aa0808870ae2a8522f1cf1da0ef4
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
package com.ruoyi.system.controller;
 
import com.alibaba.fastjson2.JSONObject;
import com.ruoyi.common.core.domain.R;
import com.ruoyi.common.core.utils.UrlDownloader;
import com.ruoyi.common.core.utils.orc.OcrUtils;
import com.ruoyi.system.api.model.IdCard;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.Parameter;
import io.swagger.v3.oas.annotations.tags.Tag;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
 
import java.io.InputStream;
import java.net.URLEncoder;
 
@RestController
@RequestMapping("/common")
@Tag(name = "通用接口")
public class CommonController {
 
    /**
     * ocr提取身份证信息
     */
    @GetMapping("/idCard")
    @Operation(summary = "ocr提取身份证信息", description = "ocr提取身份证信息")
    public R<IdCard> idCard(@Parameter(name = "url", description = "图片地址") String url) {
        InputStream inputStream;
        try {
            // 找到最后一个斜杠的位置
            int lastSlashIndex = url.lastIndexOf("/");
            // 从斜杠后提取文件名
            String baseUrl = url.substring(0, lastSlashIndex);
            String fileName = url.substring(lastSlashIndex + 1);
            String encode = URLEncoder.encode(fileName, "UTF-8");
            url = baseUrl + "/" + encode;
            inputStream = UrlDownloader.downloadAsStream(url);
            JSONObject jsonObject = OcrUtils.idCard(inputStream);
            IdCard idCard = JSONObject.parseObject(jsonObject.toJSONString(), IdCard.class);
            return R.ok(idCard);
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }
}