| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735 |
- package controllers
- import (
- "bufio"
- "fmt"
- "github.com/kgiannakakis/mp3duration/src/mp3duration"
- "hongze/hz_crm_api/models"
- "hongze/hz_crm_api/services"
- "hongze/hz_crm_api/utils"
- "io"
- "os"
- "path"
- "regexp"
- "strings"
- "time"
- )
- type ResourceController struct {
-
- BaseCommonController
- }
- func (this *ResourceController) Upload() {
- br := new(models.BaseResponse).Init()
- defer func() {
- this.Data["json"] = br
- this.ServeJSON()
- }()
- f, h, err := this.GetFile("file")
- if err != nil {
- br.Msg = "获取资源信息失败"
- br.ErrMsg = "获取资源信息失败,Err:" + err.Error()
- return
- }
- uploadFileName := h.Filename
- ext := path.Ext(h.Filename)
- dateDir := time.Now().Format("20060102")
- uploadDir := utils.STATIC_DIR + "hongze/" + dateDir
- err = os.MkdirAll(uploadDir, 766)
- if err != nil {
- br.Msg = "存储目录创建失败"
- br.ErrMsg = "存储目录创建失败,Err:" + err.Error()
- return
- }
- randStr := utils.GetRandStringNoSpecialChar(28)
- fileName := randStr + ext
- fpath := uploadDir + "/" + fileName
- defer f.Close()
- err = this.SaveToFile("file", fpath)
- if err != nil {
- br.Msg = "文件上传失败"
- br.ErrMsg = "文件上传失败，Err:" + err.Error()
- return
- }
-
- resourceUrl, err := services.UploadAliyunV2(fileName, fpath)
- if err != nil {
- br.Msg = "文件上传失败"
- br.ErrMsg = "文件上传失败，Err:" + err.Error()
- return
- }
- defer func() {
- os.Remove(fpath)
- }()
- item := new(models.Resource)
- item.ResourceUrl = resourceUrl
- item.ResourceType = 1
- item.CreateTime = time.Now()
- newId, err := models.AddResource(item)
- if err != nil {
- br.Msg = "资源上传失败"
- br.ErrMsg = "资源上传失败,Err:" + err.Error()
- return
- }
- resp := models.ResourceResp{
- Id: newId,
- ResourceUrl: resourceUrl,
- ResourceName: uploadFileName,
- }
- br.Msg = "上传成功"
- br.Ret = 200
- br.Success = true
- br.Data = resp
- return
- }
- func (this *ResourceController) VideoUpload() {
- utils.FileLog.Info("start VideoUpload ")
- utils.FileLog.Info("%s:", time.Now().Format(utils.FormatDateTime))
- br := new(models.BaseResponse).Init()
- defer func() {
- this.Data["json"] = br
- this.ServeJSON()
- }()
- f, h, err := this.GetFile("file")
- if err != nil {
- br.Msg = "获取资源信息失败"
- br.ErrMsg = "获取资源信息失败,Err:" + err.Error()
- return
- }
- ext := path.Ext(h.Filename)
- dateDir := time.Now().Format("20060102")
- uploadDir := utils.STATIC_DIR + "hongze/" + dateDir
-
- fmt.Println(uploadDir)
- err = os.MkdirAll(uploadDir, 777)
- if err != nil {
- br.Msg = "存储目录创建失败"
- br.ErrMsg = "存储目录创建失败,Err:" + err.Error()
- return
- }
- randStr := utils.GetRandStringNoSpecialChar(28)
- fileName := randStr + ext
- fpath := uploadDir + "/" + fileName
- defer f.Close()
- utils.FileLog.Info("start save")
- utils.FileLog.Info("%s:", time.Now().Format(utils.FormatDateTime))
- utils.FileLog.Info("fpath %s:", fpath)
- fileHandle, err := os.OpenFile(fpath, os.O_RDWR|os.O_CREATE|os.O_APPEND, os.ModeAppend)
- if err != nil {
- br.Msg = "创建文件失败"
- br.ErrMsg = "创建文件失败,Err:" + err.Error()
- return
- }
- defer fileHandle.Close()
-
- for {
- buffer := make([]byte, 4096)
- _, err := f.Read(buffer)
-
- if err != nil && err != io.EOF {
- br.Msg = "上传失败"
- br.ErrMsg = "读取文件内容失败,Err:" + err.Error()
- return
- }
-
- if err == io.EOF {
- break
- }
-
-
- buf := bufio.NewWriter(fileHandle)
-
-
- buf.WriteString(string(buffer))
-
- err = buf.Flush()
- if err != nil {
- br.Msg = "上传失败"
- br.ErrMsg = "写入文件内容失败,Err:" + err.Error()
- return
- }
- }
- utils.FileLog.Info("%s:", time.Now().Format(utils.FormatDateTime))
- utils.FileLog.Info("end save")
-
-
-
-
-
-
-
- utils.FileLog.Info("start update oss ")
- utils.FileLog.Info("%s:", time.Now().Format(utils.FormatDateTime))
- savePath := utils.Upload_Audio_Dir + time.Now().Format("200601/20060102/")
- savePath += fileName
-
- err = services.UploadVideoAliyun(fileName, fpath, savePath)
- if err != nil {
- br.Msg = "文件上传失败"
- br.ErrMsg = "文件上传失败，Err:" + err.Error()
- return
- }
- utils.FileLog.Info("%s:", time.Now().Format(utils.FormatDateTime))
- utils.FileLog.Info("end update oss ")
- resourceUrl := utils.Imghost + savePath
- item := new(models.Resource)
- item.ResourceUrl = resourceUrl
- item.ResourceType = 3
- item.CreateTime = time.Now()
- newId, err := models.AddResource(item)
- if err != nil {
- br.Msg = "资源上传失败"
- br.ErrMsg = "资源上传失败,Err:" + err.Error()
- return
- }
- resp := new(models.ResourceResp)
- resp.Id = newId
- resp.ResourceUrl = resourceUrl
- utils.FileLog.Info("start GetMP4Duration ")
- utils.FileLog.Info("%s:", time.Now().Format(utils.FormatDateTime))
- duration, err := services.GetMP4Duration(f)
- if err != nil {
- panic(err)
- }
- utils.FileLog.Info("end GetMP4Duration ")
- utils.FileLog.Info("%s:", time.Now().Format(utils.FormatDateTime))
- resp.PlaySeconds = duration
- br.Msg = "上传成功"
- br.Ret = 200
- br.Success = true
- br.Data = resp
- utils.FileLog.Info("end VideoUpload ")
- utils.FileLog.Info("%s:", time.Now().Format(utils.FormatDateTime))
- return
- }
- func (this *ResourceController) VoiceUpload() {
- utils.FileLog.Info("start VoiceUpload ")
- utils.FileLog.Info(fmt.Sprintf("%s:", time.Now().Format(utils.FormatDateTime)))
- br := new(models.BaseResponse).Init()
- defer func() {
- this.Data["json"] = br
- this.ServeJSON()
- }()
- f, h, err := this.GetFile("file")
- if err != nil {
- br.Msg = "获取资源信息失败"
- br.ErrMsg = "获取资源信息失败,Err:" + err.Error()
- return
- }
- uploadFileName := h.Filename
- ext := path.Ext(h.Filename)
- dateDir := time.Now().Format("20060102")
- uploadDir := utils.STATIC_DIR + "hongze/" + dateDir
-
- fmt.Println(uploadDir)
- err = os.MkdirAll(uploadDir, 766)
- if err != nil {
- br.Msg = "存储目录创建失败"
- br.ErrMsg = "存储目录创建失败,Err:" + err.Error()
- return
- }
- randStr := utils.GetRandStringNoSpecialChar(28)
- fileName := randStr + ext
- fpath := uploadDir + "/" + fileName
- defer f.Close()
- utils.FileLog.Info("start save")
- utils.FileLog.Info(fmt.Sprintf("%s:", time.Now().Format(utils.FormatDateTime)))
- utils.FileLog.Info(fmt.Sprintf("fpath %s:", fpath))
- fileHandle, err := os.OpenFile(fpath, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0755)
- if err != nil {
- br.Msg = "创建文件失败"
- br.ErrMsg = "创建文件失败,Err:" + err.Error()
- return
- }
- defer fileHandle.Close()
- localFileBuffer := make([]byte, 0)
-
- for {
- buffer := make([]byte, 4096)
- maxLen, err := f.Read(buffer)
-
- if err != nil && err != io.EOF {
- br.Msg = "上传失败"
- br.ErrMsg = "读取文件内容失败,Err:" + err.Error()
- return
- }
-
- if err == io.EOF {
- break
- }
-
-
- buf := bufio.NewWriter(fileHandle)
-
-
- buf.WriteString(string(buffer))
- localFileBuffer = append(localFileBuffer, buffer[0:maxLen]...)
-
- err = buf.Flush()
- if err != nil {
- br.Msg = "上传失败"
- br.ErrMsg = "写入文件内容失败,Err:" + err.Error()
- return
- }
- }
- utils.FileLog.Info(fmt.Sprintf("%s:", time.Now().Format(utils.FormatDateTime)))
- utils.FileLog.Info("end save")
- defer func() {
- if utils.FileIsExist(fpath) {
- os.Remove(fpath)
- }
- }()
- var playSeconds float64
- if ext == ".m4a" {
- utils.FileLog.Info("start calculate m4a duration ")
- tmpPlaySeconds, err := services.GetMP4Duration(f)
- if err != nil {
- br.Msg = "音频资源时长计算失败"
- br.ErrMsg = "音频资源时长计算失败,Err:" + err.Error()
- return
- }
- playSeconds = float64(tmpPlaySeconds)
- utils.FileLog.Info("end calculate m4a duration ")
- } else {
- utils.FileLog.Info("start calculate mp3duration ")
- playSeconds, err = mp3duration.Calculate(fpath)
- if playSeconds <= 0 {
- playSeconds, err = utils.GetVideoPlaySeconds(fpath)
- if err != nil {
- br.Msg = "音频资源时长计算失败"
- br.ErrMsg = "音频资源时长计算失败,Err:" + err.Error()
- return
- }
- }
- utils.FileLog.Info("end calculate mp3duration ")
- }
- utils.FileLog.Info("start update oss ")
- utils.FileLog.Info(fmt.Sprintf("%s:", time.Now().Format(utils.FormatDateTime)))
- savePath := utils.Upload_Audio_Dir + time.Now().Format("200601/20060102/")
- savePath += fileName
-
- err = services.UploadVideoAliyun(fileName, fpath, savePath)
- if err != nil {
- br.Msg = "文件上传失败"
- br.ErrMsg = "文件上传失败，Err:" + err.Error()
- return
- }
- utils.FileLog.Info(fmt.Sprintf("%s:", time.Now().Format(utils.FormatDateTime)))
- utils.FileLog.Info("end update oss ")
- resourceUrl := utils.Imghost + savePath
- item := new(models.Resource)
- item.ResourceUrl = resourceUrl
- item.ResourceType = 3
- item.CreateTime = time.Now()
- newId, err := models.AddResource(item)
- if err != nil {
- br.Msg = "音频资源上传失败"
- br.ErrMsg = "音频资源上传失败,Err:" + err.Error()
- return
- }
- resp := models.ResourceResp{
- Id: newId,
- ResourceUrl: resourceUrl,
- ResourceName: uploadFileName,
- PlaySeconds: uint32(playSeconds),
- }
- br.Msg = "上传成功"
- br.Ret = 200
- br.Success = true
- br.Data = resp
- utils.FileLog.Info("end VoiceUpload ")
- utils.FileLog.Info(fmt.Sprintf("%s:", time.Now().Format(utils.FormatDateTime)))
- return
- }
- func init123() {
- fmt.Println("start")
- fmt.Println(time.Now())
- tofile := "./20201123.mp4"
- fromFile := "./第五期下.mp4"
- f, err := os.Open(fromFile)
- defer f.Close()
- fileHandle, err := os.OpenFile(tofile, os.O_RDONLY|os.O_CREATE|os.O_APPEND, 0666)
- if err != nil {
- fmt.Println("open file error :", err)
- return
- }
- defer fileHandle.Close()
- if err != nil {
- panic(err)
- } else {
-
- for {
- buffer := make([]byte, 4096)
- _, err := f.Read(buffer)
-
- if err != nil && err != io.EOF {
- panic(err)
- }
-
- if err == io.EOF {
- fmt.Println("读取完毕")
- break
- }
-
-
- buf := bufio.NewWriter(fileHandle)
-
-
- buf.WriteString(string(buffer))
-
- err = buf.Flush()
- if err != nil {
- fmt.Println("flush error :" + err.Error())
- return
- }
- }
- }
- fmt.Println(time.Now())
- fmt.Println("end")
- }
- func (this *ResourceController) UploadImageBase64() {
- br := new(models.BaseResponse).Init()
- defer func() {
- this.Data["json"] = br
- this.ServeJSON()
- }()
- var err error
- ext := ".png"
- uploadDir := "./static"
- randStr := utils.GetRandStringNoSpecialChar(28)
- fileName := randStr + ext
- fpath := uploadDir + "/" + fileName
- image := this.Ctx.Request.FormValue("Image")
- if image == "" {
- br.Msg = "参数错误"
- br.ErrMsg = "图片内容不能为空"
- return
- }
- b, _ := regexp.MatchString(`^data:\s*image\/(\w+);base64,`, image)
- if !b {
- br.Msg = "图片格式不正确"
- br.ErrMsg = "图片格式不正确"
- return
- }
- re, _ := regexp.Compile(`^data:\s*image\/(\w+);base64,`)
- base64Str := re.ReplaceAllString(image, "")
- base64Str = strings.Replace(base64Str, " ", "", -1)
- err = utils.SaveBase64ToFile(base64Str, fpath)
- if err != nil {
- br.Msg = "图片保存失败"
- br.ErrMsg = "图片保存失败,Err:" + err.Error()
- return
- }
- fmt.Println("end")
- defer os.Remove(fpath)
- hzUploadDir := "static/images/"
- savePath := hzUploadDir + time.Now().Format("200601/20060102/")
- savePath += fileName
-
- err = services.UploadFileToAliyun(fileName, fpath, savePath)
- if err != nil {
- br.Msg = "文件上传失败"
- br.ErrMsg = "文件上传失败，Err:" + err.Error()
- return
- }
- utils.FileLog.Info("%s:", time.Now().Format(utils.FormatDateTime))
- utils.FileLog.Info("end update oss ")
- fileHost := "https://hzstatic.hzinsights.com/"
- resourceUrl := fileHost + savePath
- item := new(models.Resource)
- item.ResourceUrl = resourceUrl
- item.ResourceType = 1
- item.CreateTime = time.Now()
- newId, err := models.AddResource(item)
- if err != nil {
- br.Msg = "资源上传失败"
- br.ErrMsg = "资源上传失败,Err:" + err.Error()
- return
- }
- resp := new(models.ResourceResp)
- resp.Id = newId
- resp.ResourceUrl = resourceUrl
- br.Msg = "上传成功"
- br.Ret = 200
- br.Success = true
- br.Data = resp
- utils.FileLog.Info("end VideoUpload ")
- utils.FileLog.Info("%s:", time.Now().Format(utils.FormatDateTime))
- }
- func IsFileExist(filename string) bool {
- _, err := os.Stat(filename)
- if os.IsNotExist(err) {
- return false
- }
- return true
- }
- func (this *ResourceController) UploadV2() {
- br := new(models.BaseResponse).Init()
- defer func() {
- this.Data["json"] = br
- this.ServeJSON()
- }()
- businessType := this.Ctx.Request.Form.Get("business_type")
-
- fmt.Println("businessType:", businessType)
- fmt.Println(this.Ctx.Request.Form)
- fmt.Println("===========")
- br.Data = businessType
- f, h, err := this.GetFile("file")
- if err != nil {
- br.Msg = "获取资源信息失败"
- br.ErrMsg = "获取资源信息失败,Err:" + err.Error()
- return
- }
- ext := path.Ext(h.Filename)
- dateDir := time.Now().Format("20060102")
- uploadDir := utils.STATIC_DIR + "hongze/" + dateDir
- err = os.MkdirAll(uploadDir, 766)
- if err != nil {
- br.Msg = "存储目录创建失败"
- br.ErrMsg = "存储目录创建失败,Err:" + err.Error()
- return
- }
- randStr := utils.GetRandStringNoSpecialChar(28)
- fileName := randStr + ext
- fpath := uploadDir + "/" + fileName
- defer f.Close()
- err = this.SaveToFile("file", fpath)
- if err != nil {
- br.Msg = "文件上传失败"
- br.ErrMsg = "文件上传失败，Err:" + err.Error()
- return
- }
-
- resourceUrl, err := services.UploadAliyunV2(fileName, fpath)
- if err != nil {
- br.Msg = "文件上传失败"
- br.ErrMsg = "文件上传失败，Err:" + err.Error()
- return
- }
- defer func() {
- os.Remove(fpath)
- }()
- item := new(models.Resource)
- item.ResourceUrl = resourceUrl
- item.ResourceType = 1
- item.CreateTime = time.Now()
- newId, err := models.AddResource(item)
- if err != nil {
- br.Msg = "资源上传失败"
- br.ErrMsg = "资源上传失败,Err:" + err.Error()
- return
- }
- resp := new(models.ResourceResp)
- resp.Id = newId
- resp.ResourceUrl = resourceUrl
- br.Msg = "上传成功"
- br.Ret = 200
- br.Success = true
-
- return
- }
- func (this *ResourceController) OssSTSToken() {
- br := new(models.BaseResponse).Init()
- defer func() {
- this.Data["json"] = br
- this.ServeJSON()
- }()
- resp, err := services.GetOssSTSToken()
- if err != nil {
- br.Msg = "获取失败"
- br.ErrMsg = "获取STSToken失败, Err: " + err.Error()
- return
- }
- br.Msg = "获取成功"
- br.Ret = 200
- br.Success = true
- br.Data = resp
- }
|
