视频素材 时长修复

Makefile

@@ -1,11 +1,15 @@
 BINARY="build/iot_service"
 VERSION=1.0.0
 BASH_PATH=D:\Programs\Git\bin\bash.exe
-export GOARCH=amd64 #环境变量设置
+
+export GOARCH=amd64
 export GOOS=linux
+clear:
+	$(RM) ${BINARY}
+
 build:
 	@go build -o ${BINARY} ./
 
-push-test:
+push-test:build
 	${BASH_PATH} ./push.sh
 .PHONY: build push-test

README.md

@@ -1 +1,5 @@
-go 平江智慧路灯大数据平台
+## go 平江智慧路灯大数据平台
+
+### 常见问题
+* 1. 2022-12-15 `github.com/tosone/minimp`报gcc不存在
+    参考这个方法在本地安装gcc ![https://www.cnblogs.com/lonecloud/p/15468154.html](https://www.cnblogs.com/lonecloud/p/15468154.html)

app/multimedia/service/libraryService.go

@@ -131,8 +131,14 @@ func (s *libraryService) UploadFile(tenantId int, fileHeader *multipart.FileHead
 		fileType = 3
 	}
 	duration := 0
-	if strings.Contains(objectName, "mp4") || strings.Contains(objectName, "mp3") {
-
+	if fileType == 2 {
+		duration = 3
+	} else {
+		duration, err = common.GetMediaDuration(fileContent, contentType)
+		if err != nil {
+			logger.Logger.Errorf("UploadFile duration fail, err = %s", err.Error())
+			duration = 0
+		}
 	}
 	return &model.RspUploadFile{
 		Link:         config.Instance().Minio.Link + "/" + bucket + "/" + objectName,

go.mod

@@ -13,10 +13,13 @@ require (
 	github.com/robfig/cron v1.2.0
 	github.com/satori/go.uuid v1.2.0
 	github.com/sirupsen/logrus v1.9.0
+	github.com/tosone/minimp3 v1.0.1
 	gopkg.in/yaml.v2 v2.4.0
 	gorm.io/driver/mysql v1.4.4
 	gorm.io/gorm v1.24.1
+)
 
+require (
 	github.com/dustin/go-humanize v1.0.0 // indirect
 	github.com/fastly/go-utils v0.0.0-20180712184237-d95a45783239 // indirect
 	github.com/fogleman/gg v1.3.0 // indirect

main.go

@@ -89,6 +89,7 @@ func initDB() {
 // 		[windows] CMD下执行 setx "DEV_ENV" "1"
 //		[linux] export DEV_ENV="1"
 func isDevEnv() bool {
+	//return true
 	if os.Getenv("DEV_ENV") == "1" {
 		return true
 	}

util/common/mp3mp4_duration.go

@@ -0,0 +1,99 @@
+package common
+
+import (
+	"bytes"
+	"encoding/binary"
+	"mime/multipart"
+)
+
+// GetMediaDuration 得到媒体文件时长
+func GetMediaDuration(reader multipart.File, contentType string) (int, error) {
+	var duration int
+	var err error
+	if contentType == "video/mp4" {
+		duration, err = GetMP4Duration(reader)
+	}
+	if contentType == "audio/mpeg" {
+		duration, err = GetMP3PlayDuration(reader)
+	}
+	if err != nil {
+		return 0, err
+	}
+	return duration * 1000, err
+}
+
+// GetMP4Duration 获取视频时长，以秒计
+func GetMP4Duration(reader multipart.File) (lengthOfTime int, err error) {
+	var info = make([]byte, 0x10)
+	var boxHeader BoxHeader
+	var offset int64 = 0
+	// 获取moov结构偏移
+	for {
+		_, err = reader.ReadAt(info, offset)
+		if err != nil {
+			return
+		}
+		boxHeader = getHeaderBoxInfo(info)
+		fourccType := getFourccType(boxHeader)
+		if fourccType == "moov" {
+			break
+		}
+		// 有一部分mp4 mdat尺寸过大需要特殊处理
+		if fourccType == "mdat" {
+			if boxHeader.Size == 1 {
+				offset += int64(boxHeader.Size64)
+				continue
+			}
+		}
+		offset += int64(boxHeader.Size)
+	}
+	// 获取moov结构开头一部分
+	moovStartBytes := make([]byte, 0x100)
+	_, err = reader.ReadAt(moovStartBytes, offset)
+	if err != nil {
+		return
+	}
+	// 定义timeScale与Duration偏移
+	timeScaleOffset := 0x1C
+	durationOffest := 0x20
+	timeScale := binary.BigEndian.Uint32(moovStartBytes[timeScaleOffset : timeScaleOffset+4])
+	Duration := binary.BigEndian.Uint32(moovStartBytes[durationOffest : durationOffest+4])
+	lengthOfTime = int(Duration / timeScale)
+	return
+}
+
+// GetMP3PlayDuration 获取mp3时长，以秒计
+func GetMP3PlayDuration(reader multipart.File) (seconds int, err error) {
+	return 0, nil
+	//buf := bytes.NewBuffer(nil)
+	//if _, err := io.Copy(buf, reader); err != nil {
+	//	return 0, err
+	//}
+	//mp3Data := buf.Bytes()
+	//dec, _, err := minimp3.DecodeFull(mp3Data)
+	//if err != nil {
+	//	return 0, err
+	//}
+	//// 音乐时长 = (文件大小(byte) - 128(ID3信息)) * 8(to bit) / (码率(kbps b:bit) * 1000)(kilo bit to bit)
+	//seconds = (len(mp3Data) - 128) * 8 / (dec.Kbps * 1000)
+	//return seconds, nil
+}
+
+// BoxHeader 信息头
+type BoxHeader struct {
+	Size       uint32
+	FourccType [4]byte
+	Size64     uint64
+}
+
+func getHeaderBoxInfo(data []byte) (boxHeader BoxHeader) {
+	buf := bytes.NewBuffer(data)
+	binary.Read(buf, binary.BigEndian, &boxHeader)
+	return
+}
+
+// getFourccType 获取信息头类型
+func getFourccType(boxHeader BoxHeader) (fourccType string) {
+	fourccType = string(boxHeader.FourccType[:])
+	return
+}