log-server/service/levelMonitor/image_record.go

package levelMonitor

import (
	"context"
	"encoding/json"
	"errors"
	"fmt"
	"go.uber.org/zap"
	"gorm.io/gorm"
	"io/ioutil"
	"log-server/global"
	"log-server/model/common/request"
	"log-server/model/levelMonitor"
	levelRequest "log-server/model/levelMonitor/request"
	"log-server/model/log"
	"log-server/service/cache"
	"net/http"
	"net/url"
	"strconv"
	"strings"
	"time"
)

type ImageRecordService struct {
	cache cache.Cache
}

//创建图片记录
func (s *ImageRecordService) CreateImageRecord(record levelMonitor.ImageRecord) (err error) {
	//获取今日日期
	date := time.Now().Format("2006-01-02")
	key := fmt.Sprintf(cache.UploadImageNum, date, record.TaskId)
	ctx := context.Background()
	num, _ := s.cache.GetCacheNum(ctx, key)
	if num > 50 {
		return errors.New("数据上传已到上限")
	}
	var entity levelMonitor.ImageRecord
	//创建日期赋值
	record.CreateDate = date
	record.Status = 2
	//查找数据库是否存在此账号此游戏今日记录
	err = global.GVA_DB.Model(&levelMonitor.ImageRecord{}).Where("task_id = ? and account = ? and create_date = ?", record.TaskId, record.Account, date).First(&entity).Error
	if !errors.Is(err, gorm.ErrRecordNotFound) {
		return errors.New("此记录已存在，请勿重复添加")
	}
	err = global.GVA_DB.Model(&levelMonitor.ImageRecord{}).Omit("create_time", "update_time").Create(&record).Error
	if err != nil {
		return err
	}
	go s.ImageIdentify(record, key)
	return
}

func (s *ImageRecordService) UploadOrNot(record levelMonitor.ImageRecord) (err error) {
	date := time.Now().Format("2006-01-02")
	key := fmt.Sprintf(cache.UploadImageNum, date, record.TaskId)
	ctx := context.Background()
	num, _ := s.cache.GetCacheNum(ctx, key)
	if num > 50 {
		return errors.New("数据上传已到上限")
	}
	return nil
}

//删除三日前的图片记录
func (s *ImageRecordService) DeleteExpireImageRecord() {
	markTime := time.Now().Add(-time.Hour * 48).Format("2006-01-02")
	err := global.GVA_DB.Where("create_date < ?", markTime).Delete(&levelMonitor.ImageRecord{}).Error
	if err != nil {
		global.GVA_LOG.Info("删除图片信息失败:" + err.Error() + time.Now().Format("2006-01-02 15:04:05"))
	}
	return
}

//获取图片记录列表
func (s *ImageRecordService) GetImageRecordList(record levelRequest.ImageRecordRequest, info request.PageInfo, order string, desc bool) (recordList []levelMonitor.ImageRecord, total int64, err error) {
	limit := info.PageSize
	offset := info.PageSize * (info.Page - 1)
	db := global.GVA_DB.Model(&levelMonitor.ImageRecord{})
	startDate := time.Now().Format("2006-01-02")
	endDate := time.Now().Format("2006-01-02")
	if len(record.Date) == 2 {
		startDate = record.Date[0]
		endDate = record.Date[1]
	}
	//筛选日期
	db = db.Where("create_date >= ? and create_date <= ?", startDate, endDate)
	if record.TaskId != 0 {
		db = db.Where("task_id = ?", record.TaskId)
	}
	err = db.Count(&total).Error
	if err != nil {
		return recordList, total, err
	} else {
		db = db.Limit(limit).Offset(offset)
		if order != "" {
			var OrderStr string
			// 设置有效排序key 防止sql注入
			// 感谢 Tom4t0 提交漏洞信息
			orderMap := make(map[string]bool, 4)
			orderMap["task_id"] = true
			orderMap["create_date"] = true
			orderMap["create_time"] = true
			orderMap["update_time"] = true
			if orderMap[order] {
				if desc {
					OrderStr = order + " desc"
				} else {
					OrderStr = order
				}
			} else { // didn't matched any order key in `orderMap`
				global.GVA_LOG.Error("获取失败!", zap.Error(err))
				return recordList, total, err
			}
			err = db.Order(OrderStr).Find(&recordList).Error
		} else {
			err = db.Order("id desc").Find(&recordList).Error
		}
	}
	//遍历更改日期格式
	for i, _ := range recordList {
		recordList[i].CreateDate = recordList[i].CreateDate[:10]
	}
	return recordList, total, err
}

const API_KEY = "z9GNcyrC7VeV3g1xXEj3YL1s"
const SECRET_KEY = "VASRBsEzeVsyKduSkkflfL87r5yqoqvj"
const BAIDU_IDENTIFY = "baiduIdentifyToken"

func (s *ImageRecordService) ImageIdentify(record levelMonitor.ImageRecord, key string) {
	token, err := GetAccessToken()
	if err != nil {
		global.GVA_LOG.Error("get token fail", zap.Error(err))
		s.UpdateImageRecordStatus(record.Id, "get token fail", -1, 0, 0)
		return
	}
	body, err := Request(token, record.ImageBase64)
	if err != nil {
		global.GVA_LOG.Error("Read Body Fail", zap.Error(err))
		s.UpdateImageRecordStatus(record.Id, err.Error(), -1, 0, 0)
		return
	}
	type Data struct {
		WordsResult []struct {
			Words string `json:"words"`
		} `json:"words_result"`
		WordsResultNum int `json:"words_result_num"`
		LogId          int `json:"log_id"`
	}
	var data Data
	err = json.Unmarshal(body, &data)
	if err != nil {
		global.GVA_LOG.Error("Read Body Fail", zap.Error(err))
		s.UpdateImageRecordStatus(record.Id, "Read Body Fail", -1, 0, 0)
		return
	}

	identify := 0
	if data.WordsResultNum == 0 {
		s.UpdateImageRecordStatus(record.Id, string(body), -1, 0, 0)
		return
	}
	if len(data.WordsResult) > 1 {
		for _, words := range data.WordsResult {
			i, err := strconv.Atoi(words.Words)
			if err == nil {
				identify = i
				break
			}
		}
	} else {
		i, _ := strconv.Atoi(data.WordsResult[0].Words)
		identify = i
	}
	if identify == 0 {
		s.UpdateImageRecordStatus(record.Id, string(body), -1, 0, 0)
		return
	}
	var gameAccount log.GameAccount
	global.GVA_DB.Where("game_id", record.TaskId).Where("account", record.Account).First(&gameAccount)
	s.UpdateImageRecordStatus(record.Id, string(body), 1, gameAccount.UseNum, identify)
	UpdateGameAccountIdentify(record.Account, record.TaskId, identify)
	ctx := context.Background()
	s.cache.SetCacheNum(ctx, key)
	return
}

func (s *ImageRecordService) UpdateImageRecordStatus(id int, result string, status int, useNum, identify int) {
	update := map[string]interface{}{
		"result":   result,
		"status":   status,
		"use_num":  useNum,
		"identify": identify,
	}
	global.GVA_DB.Model(&levelMonitor.ImageRecord{}).Where("id", id).Updates(update)
}

func Request(baiduToken, content string) (body []byte, err error) {
	reqUrl := "https://aip.baidubce.com/rest/2.0/ocr/v1/general_basic?access_token=" + baiduToken
	var (
		params = url.Values{}
	)
	params.Set("image", content)
	requestData := params.Encode()
	payload := strings.NewReader(requestData)
	client := &http.Client{}
	req, err := http.NewRequest("POST", reqUrl, payload)

	if err != nil {
		fmt.Println(err)
		return
	}
	req.Header.Add("Content-Type", "application/x-www-form-urlencoded")
	req.Header.Add("Accept", "application/json")

	res, err := client.Do(req)
	if err != nil {
		global.GVA_LOG.Error("request Identify fail", zap.Error(err))
		return
	}
	defer res.Body.Close()

	body, err = ioutil.ReadAll(res.Body)
	if err != nil {
		global.GVA_LOG.Error("Read Body Fail", zap.Error(err))
		return
	}
	fmt.Println(string(body))
	return
}

// 更新game_acount
func UpdateGameAccountIdentify(account string, gameId int, identify int) {
	global.GVA_DB.Table("game_account").Where("game_id", gameId).Where("account", account).Update("identify", identify)
}

/**
 * 使用 AK，SK 生成鉴权签名（Access Token）
 * @return string 鉴权签名信息（Access Token）
 */
func GetAccessToken() (string, error) {
	ctx := context.Background()
	token, err := global.GVA_REDIS.Get(ctx, BAIDU_IDENTIFY).Result()
	if err != nil || token == "" {
		reqUrl := "https://aip.baidubce.com/oauth/2.0/token"
		postData := fmt.Sprintf("grant_type=client_credentials&client_id=%s&client_secret=%s", API_KEY, SECRET_KEY)
		resp, err := http.Post(reqUrl, "application/x-www-form-urlencoded", strings.NewReader(postData))
		if err != nil {
			fmt.Println(err)
			return "", err
		}
		defer resp.Body.Close()
		body, err := ioutil.ReadAll(resp.Body)
		if err != nil {
			fmt.Println(err)
			return "", err
		}
		accessTokenObj := map[string]string{}
		json.Unmarshal(body, &accessTokenObj)
		global.GVA_REDIS.Set(ctx, BAIDU_IDENTIFY, accessTokenObj["access_token"], time.Hour*24*29)
		return accessTokenObj["access_token"], err
	}
	return token, err
}