80 lines
2.0 KiB
Go
80 lines
2.0 KiB
Go
|
package service
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"encoding/base64"
|
|||
|
|
"encoding/json"
|
|||
|
|
"fmt"
|
|||
|
|
"github.com/tidwall/gjson"
|
|||
|
|
"io/ioutil"
|
|||
|
|
"net/http"
|
|||
|
|
"strings"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type BaiduOCR struct {
|
|||
|
|
API_KEY, SECRET_KEY string
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (this *BaiduOCR) OCR(imgBase64 string) (string, error) {
|
|||
|
|
url := "https://aip.baidubce.com/rest/2.0/ocr/v1/general_basic?access_token=" + this.GetAccessToken()
|
|||
|
|
|
|||
|
|
// image 可以通过 GetFileContentAsBase64("C:\fakepath\微信图片_20240430171848.jpg") 方法获取，如需转码请使用 url.QueryEscape()
|
|||
|
|
payload := strings.NewReader("image=" + imgBase64)
|
|||
|
|
client := &http.Client{}
|
|||
|
|
req, err := http.NewRequest("POST", url, payload)
|
|||
|
|
|
|||
|
|
if err != nil {
|
|||
|
|
return "", err
|
|||
|
|
}
|
|||
|
|
req.Header.Add("Content-Type", "application/x-www-form-urlencoded")
|
|||
|
|
req.Header.Add("Accept", "application/json")
|
|||
|
|
|
|||
|
|
res, err := client.Do(req)
|
|||
|
|
if err != nil {
|
|||
|
|
return "", err
|
|||
|
|
}
|
|||
|
|
defer res.Body.Close()
|
|||
|
|
|
|||
|
|
body, err := ioutil.ReadAll(res.Body)
|
|||
|
|
if err != nil {
|
|||
|
|
return "", err
|
|||
|
|
}
|
|||
|
|
words := gjson.Get(string(body), "words_result").Array()
|
|||
|
|
result := ""
|
|||
|
|
for _, word := range words {
|
|||
|
|
result += word.Get("words").String()
|
|||
|
|
}
|
|||
|
|
return result, nil
|
|||
|
|
}
|
|||
|
|
func (this *BaiduOCR) GetAccessToken() string {
|
|||
|
|
url := "https://aip.baidubce.com/oauth/2.0/token"
|
|||
|
|
postData := fmt.Sprintf("grant_type=client_credentials&client_id=%s&client_secret=%s", this.API_KEY, this.SECRET_KEY)
|
|||
|
|
resp, err := http.Post(url, "application/x-www-form-urlencoded", strings.NewReader(postData))
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Println(err)
|
|||
|
|
return ""
|
|||
|
|
}
|
|||
|
|
defer resp.Body.Close()
|
|||
|
|
body, err := ioutil.ReadAll(resp.Body)
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Println(err)
|
|||
|
|
return ""
|
|||
|
|
}
|
|||
|
|
accessTokenObj := map[string]any{}
|
|||
|
|
_ = json.Unmarshal([]byte(body), &accessTokenObj)
|
|||
|
|
return accessTokenObj["access_token"].(string)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 获取文件base64编码
|
|||
|
|
* @param string path 文件路径
|
|||
|
|
* @return string base64编码信息，不带文件头
|
|||
|
|
*/
|
|||
|
|
func GetFileContentAsBase64(path string) string {
|
|||
|
|
srcByte, err := ioutil.ReadFile(path)
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Println(err)
|
|||
|
|
return ""
|
|||
|
|
}
|
|||
|
|
return base64.StdEncoding.EncodeToString(srcByte)
|
|||
|
|
}
|