speechAnalysis/service/process.go

package service

import (
	"bytes"
	"encoding/json"
	"errors"
	"gorm.io/gorm"
	"io"
	"mime/multipart"
	"net/http"
	"os"
	"speechAnalysis/conf"
	"speechAnalysis/constvar"
	"speechAnalysis/models"
	"speechAnalysis/pkg/logx"
)

// Response 结构体用于存储响应体的内容
type Response struct {
	Code   int     `json:"code"`
	Msg    string  `json:"msg"`
	Result string  `json:"result"`
	Score  float64 `json:"score"`
}

func AnalysisAudio(filename string, targetURL string) (resp Response, err error) {
	file, err := os.Open(filename)
	if err != nil {
		return
	}
	defer file.Close()

	// 创建一个缓冲区来存储表单数据
	var requestBody bytes.Buffer
	writer := multipart.NewWriter(&requestBody)

	// 创建一个表单字段，用于存储文件
	fileWriter, err := writer.CreateFormFile("audio", filename)
	if err != nil {
		return
	}

	// 将文件内容复制到表单字段中
	_, err = io.Copy(fileWriter, file)
	if err != nil {
		return
	}

	// 关闭表单写入器，以便写入末尾的边界
	writer.Close()

	// 创建POST请求，指定URL和请求体
	request, err := http.NewRequest("POST", targetURL, &requestBody)
	if err != nil {
		return
	}

	// 设置请求头，指定Content-Type为multipart/form-data
	request.Header.Set("Content-Type", writer.FormDataContentType())

	// 发送请求
	client := &http.Client{}
	response, err := client.Do(request)
	if err != nil {
		return
	}
	defer response.Body.Close()

	// 读取响应
	body := &bytes.Buffer{}
	_, err = io.Copy(body, response.Body)
	if err != nil {
		return
	}

	err = json.NewDecoder(body).Decode(&resp)
	if err != nil {
		return
	}

	return
}

func Process(audioId uint) (err error) {
	audio, err := models.NewAudioSearch().SetID(audioId).First()

	if err != nil {
		return errors.New("查找音频失败")
	}

	if audio.AudioStatus != constvar.AudioStatusUploadOk && audio.AudioStatus != constvar.AudioStatusFailed {
		return errors.New("状态不正确")
	}

	err = models.NewAudioSearch().SetID(audioId).UpdateByMap(map[string]interface{}{"audio_status": constvar.AudioStatusProcessing})
	if err != nil {
		return errors.New("DB错误")
	}

	go func() {
		resp, err := AnalysisAudio(audio.FilePath, conf.AanlysisConf.Url)
		if err != nil {
			logx.Errorf("err when AnalysisAudio:%v", err)
			_ = models.NewAudioSearch().SetID(audioId).UpdateByMap(map[string]interface{}{"audio_status": constvar.AudioStatusFailed})
			return
		}
		if resp.Code != 0 {
			logx.Errorf("AnalysisAudio error return:%v", resp)
			_ = models.NewAudioSearch().SetID(audioId).UpdateByMap(map[string]interface{}{"audio_status": constvar.AudioStatusFailed})
			return
		}
		logx.Infof("AnalysisAudio result: %v", resp)
		err = models.WithTransaction(func(db *gorm.DB) error {
			err = models.NewAudioSearch().SetID(audioId).UpdateByMap(map[string]interface{}{
				"audio_status": constvar.AudioStatusFinish,
				"score":        resp.Score,
			})
			if err != nil {
				return err
			}
			err = models.NewAudioTextSearch().Save(&models.AudioText{
				AudioID:   audio.ID,
				AudioText: resp.Result,
			})
			return err
		})
		if err != nil {
			logx.Infof("AnalysisAudio success but update record failed: %v", err)
			return
		}
	}()

	return nil
}