ljydlv
/
videoplayer
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge branch 'refs/heads/feat-file'

This commit is contained in:
junleea 2025-01-03 15:29:01 +08:00
parent 7607d29386 5dcd98ab3a
commit 191d466798
8 changed files with 521 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
dao/db.go
View File

@ -82,6 +82,18 @@ func Init() error {
return err
}
err = db.AutoMigrate(&ConfigFile{})
if err != nil {
fmt.Println("config file table:", err)
return err
}
err = db.AutoMigrate(&FileAuth{})
if err != nil {
fmt.Println("file auth table:", err)
return err
}
err = db.AutoMigrate(&Friend{})
if err != nil {
fmt.Println("friend table:", err)

81
dao/file.go
View File

@ -18,6 +18,23 @@ type File struct {
Md5 string `gorm:"column:md5;type:varchar(255);uniqueIndex:idx_file_name"`
}
type FileAuth struct {
gorm.Model
AuthID int `gorm:"column:auth_id"`
FileID int `gorm:"column:file_id"`
UserFileName string `gorm:"column:user_file_name;type:varchar(255);uniqueIndex:idx_file_name"` //对于同一个文件，不同用户有不同的文件名
UploadType string `gorm:"column:upload_type"` // 上传类型: im,avatar,file,config,config为系统文件
IsPrivate int `gorm:"column:is_private"` // 是否私有,私有文件只能自己下载或者通过分享链接下载,1为私有，0为公开
ShareCode string `gorm:"column:share_code"` // 分享码，用于分享时的验证，构建分享链接
}
type ConfigFile struct {
gorm.Model
AuthID int `gorm:"column:auth_id"`
FileName string `gorm:"column:file_name"`
FilePath string `gorm:"column:file_path"`
}
func CreateFile(fileStoreName, fileName, fileType, filePath, md5Str string, fileSize, authID int, NeedAuth bool) File {
file := File{FileStoreName: fileStoreName, FileName: fileName, FileType: fileType, FilePath: filePath, FileSize: fileSize, AuthID: authID, NeedAuth: NeedAuth, Md5: md5Str}
result := DB.Create(&file)
@ -96,3 +113,67 @@ func FindFileByMd5(md5 string) File {
}
return file
}
func CreateFileAuth(authID, fileID int, userFileName, uploadType string, isPrivate int, shareCode string) FileAuth {
fileAuth := FileAuth{AuthID: authID, FileID: fileID, UserFileName: userFileName, UploadType: uploadType, IsPrivate: isPrivate, ShareCode: shareCode}
var result *gorm.DB
if proto.Config.SERVER_SQL_LOG {
result = DB.Debug().Create(&fileAuth)
} else {
result = DB.Create(&fileAuth)
}
if result.Error != nil {
return FileAuth{}
}
return fileAuth
}
func CreateConfigFile(file ConfigFile) (id uint, err error) {
var result *gorm.DB
if proto.Config.SERVER_SQL_LOG {
result = DB.Debug().Create(&file)
} else {
result = DB.Create(&file)
}
return file.ID, result.Error
}
func FindConfigFileByID(id int, user_id int) ConfigFile {
var file ConfigFile
if proto.Config.SERVER_SQL_LOG {
DB.Debug().Where("id = ? and auth_id = ?", id, user_id).First(&file)
} else {
DB.Where("id = ? and auth_id = ?", id, user_id).First(&file)
}
return file
}
func DeleteConfigFileByID(id int) error {
var res *gorm.DB
if proto.Config.SERVER_SQL_LOG {
res = DB.Debug().Delete(&ConfigFile{}, id)
} else {
res = DB.Delete(&ConfigFile{}, id)
}
return res.Error
}
func UpdateConfigFileByID(id int, file ConfigFile) error {
var res *gorm.DB
if proto.Config.SERVER_SQL_LOG {
res = DB.Debug().Model(&ConfigFile{}).Where("id = ?", id).Updates(&file)
} else {
res = DB.Model(&ConfigFile{}).Where("id = ?", id).Updates(&file)
}
return res.Error
}
func FindConfigFileByAuthID(auth_id int) []ConfigFile {
var files []ConfigFile
if proto.Config.SERVER_SQL_LOG {
DB.Debug().Where("auth_id = ?", auth_id).Find(&files)
} else {
DB.Where("auth_id = ?", auth_id).Find(&files)
}
return files
}

194
handler/file.go Normal file
View File

@ -0,0 +1,194 @@
package handler
import (
"github.com/gin-gonic/gin"
"net/http"
"videoplayer/dao"
"videoplayer/proto"
"videoplayer/service"
)
func SetUpFileGroup(router *gin.Engine) {
fileGroup := router.Group("/file")
fileGroup.POST("/config/add", AddConfigFile)
fileGroup.POST("/config/delete", DeleteConfigFile)
fileGroup.POST("/config/update", UpdateConfigFile)
fileGroup.POST("/config/search", SearchConfigFile)
fileGroup.POST("/upload", UploadFileV2)
fileGroup.GET("/:filename", GetFile)
}
func AddConfigFile(c *gin.Context) {
id, _ := c.Get("id")
user_id := int(id.(float64))
var req proto.ConfigFileReq
if err := c.ShouldBind(&req); err == nil {
err2 := service.CreateConfigFile(&req, user_id)
if err2 != nil {
c.JSON(http.StatusOK, gin.H{"error": "add config file failed:" + err2.Error(), "code": proto.AddConfigFileFailed, "message": "failed"})
return
} else {
c.JSON(http.StatusOK, gin.H{"code": proto.SuccessCode, "message": "success"})
return
}
} else {
c.JSON(http.StatusOK, gin.H{"error": "upload form parameter decode error:" + err.Error(), "code": proto.ParameterError, "message": "failed"})
return
}
}
func DeleteConfigFile(c *gin.Context) {
id, _ := c.Get("id")
user_id := int(id.(float64))
var req proto.ConfigFileReq
if err := c.ShouldBind(&req); err == nil {
err2 := service.DeleteConfigFile(&req, user_id)
if err2 != nil {
c.JSON(http.StatusOK, gin.H{"error": "delete config file failed:" + err2.Error(), "code": proto.DeleteConfigFailed, "message": "failed"})
return
} else {
c.JSON(http.StatusOK, gin.H{"code": proto.SuccessCode, "message": "success"})
return
}
} else {
c.JSON(http.StatusOK, gin.H{"error": "upload form parameter decode error:" + err.Error(), "code": proto.ParameterError, "message": "failed"})
return
}
}
func UpdateConfigFile(c *gin.Context) {
id, _ := c.Get("id")
user_id := int(id.(float64))
var req proto.ConfigFileReq
if err := c.ShouldBind(&req); err == nil {
var configFileService service.ConfigFileService
err2 := configFileService.UpdateConfigFile(&req, user_id)
if err2 != nil {
c.JSON(http.StatusOK, gin.H{"error": "update config file failed:" + err2.Error(), "code": proto.UpdateConfigFailed, "message": "failed"})
return
} else {
c.JSON(http.StatusOK, gin.H{"code": proto.SuccessCode, "message": "success"})
return
}
} else {
c.JSON(http.StatusOK, gin.H{"error": "upload form parameter decode error:" + err.Error(), "code": proto.ParameterError, "message": "failed"})
return
}
}
func SearchConfigFile(c *gin.Context) {
id, _ := c.Get("id")
user_id := int(id.(float64))
var req proto.ConfigFileReq
if err := c.ShouldBind(&req); err == nil {
var configFileService service.ConfigFileService
if req.Type == "one" {
//有文件内容
configFile, err2 := configFileService.SearchOneConfigFile(&req, user_id)
if err2 == nil {
c.JSON(http.StatusOK, gin.H{"code": proto.SuccessCode, "msg": "success", "data": configFile})
} else {
c.JSON(http.StatusOK, gin.H{"code": proto.SearchConfigFileFailed, "msg": "info:" + err2.Error(), "data": configFile})
}
} else if req.Type == "all" {
configFileList, err3 := configFileService.SearchAllConfigFile(user_id)
if err3 == nil {
c.JSON(http.StatusOK, gin.H{"code": proto.SuccessCode, "msg": "success", "data": configFileList})
} else {
c.JSON(http.StatusOK, gin.H{"code": proto.SearchConfigFileFailed, "msg": "info:" + err3.Error(), "data": configFileList})
}
} else {
c.JSON(http.StatusOK, gin.H{"error": "search config file type error", "code": proto.ParameterError, "message": "failed"})
}
} else {
c.JSON(http.StatusOK, gin.H{"error": "upload form parameter decode error:" + err.Error(), "code": proto.ParameterError, "message": "failed"})
return
}
}
func UploadFileV2(c *gin.Context) {
//先查看是否有权限
id, _ := c.Get("id")
id1 := int(id.(float64))
var req proto.FileUploadReq
//获取post form参数
if err := c.ShouldBind(&req); err == nil {
//检查参数
if err2 := service.CheckUploadRequestParameters(&req); err2 != nil {
c.JSON(http.StatusOK, gin.H{"error": "upload form parameter check error:" + err2.Error(), "code": proto.ParameterError, "message": "failed"})
return
}
//
} else {
c.JSON(http.StatusOK, gin.H{"error": "upload form parameter decode error:" + err.Error(), "code": proto.ParameterError, "message": "failed"})
return
}
//从请求头获取upload_type
uploadType := c.PostForm("upload_type")
authType := c.PostForm("auth_type")
md5_ := c.PostForm("md5")
if uploadType == "" {
c.JSON(http.StatusOK, gin.H{"error": "upload_type is empty", "code": proto.ParameterError, "message": "failed"})
return
}
user := dao.FindUserByUserID(id1)
if user.Upload == false {
c.JSON(http.StatusOK, gin.H{"error": "no upload Permissions", "code": proto.NoUploadPermissions, "message": "failed"})
return
}
//上传文件
file, err := c.FormFile("file")
if err != nil {
c.JSON(http.StatusOK, gin.H{"error": "upload file failed", "code": proto.UploadFileFailed, "message": "failed"})
return
}
//计算文件md5值
if md5_ == "" {
file_, _ := file.Open()
md5_ = service.CalculateFileMd5(file_)
if md5_ == "" {
c.JSON(http.StatusOK, gin.H{"error": "计算文件MD5值失败", "code": proto.UploadFileFailed, "message": "failed"})
return
}
}
//查询文件是否已存在
fileExist := dao.FindFileByMd5(md5_)
if fileExist.ID != 0 {
fileExist.FilePath = ""
c.JSON(http.StatusOK, gin.H{"code": proto.SuccessCode, "message": "success", "data": fileExist})
return
}
//保存文件
filePath, fileStoreName, err := service.SaveFile(c, file, uploadType)
if err != nil {
c.JSON(http.StatusOK, gin.H{"error": "save file failed", "code": proto.SaveFileFailed, "message": "failed"})
return
}
//保存文件信息
fileSize := int(file.Size)
fileName := file.Filename
fileType := file.Header.Get("file_type")
var auth_type_ bool
if authType == "public" || authType == "" {
auth_type_ = false
} else if authType == "private" {
auth_type_ = true
}
file_record := dao.CreateFile(fileStoreName, fileName, fileType, filePath, md5_, fileSize, id1, auth_type_)
if file_record.ID == 0 {
c.JSON(http.StatusOK, gin.H{"error": "save file info failed", "code": proto.SaveFileInfoFailed, "message": "failed"})
return
}
file_record.FilePath = ""
c.JSON(http.StatusOK, gin.H{"code": proto.SuccessCode, "message": "success", "data": file_record})
}

2
proto/conf.go
View File

@ -11,7 +11,7 @@ var Config ConfigStruct
var SigningKey = []byte{}
var Url_map = map[string]bool{"/login": true, "/register": true, "/uuid": true, "/gqr": true, "/cid/callback": true, "/tool/monitor": true, "/user/sync": true, "/tool/file/": true} // 不需要token验证的url
var Per_menu_map = map[string]int{"/video/": 1, "/device/": 2, "/cid/": 3}
var File_Type = map[string]int{"im": 1, "avatar": 2, "file": 3, "config": 4} // 文件类型
const (
MYSQL_USER = "video_t2"
MYSQL_DB = "video_t2"

28
proto/file_req.go Normal file
View File

@ -0,0 +1,28 @@
package proto
import "time"
type FileUploadReq struct {
UploadType string `json:"upload_type" form:"upload_type" binding:"required"`
AuthType string `json:"auth_type" form:"auth_type"`
Md5 string `json:"md5" form:"md5"` //文件md5值
Type string `json:"type" form:"type"` //类型，im,avatar,file,config,config为系统文件
}
type ConfigFileReq struct {
ID int `json:"id" form:"id" binding:"required"`
Type string `json:"type" form:"type" binding:"required"` //查询类型，one,all
DelFile bool `json:"del_file" form:"del_file"` //删除文件
FileName string `json:"file_name" form:"file_name" binding:"required"`
FilePath string `json:"file_path" form:"file_path" binding:"required"`
Content string `json:"content" form:"content" binding:"required"`
}
type SearchOneConfigFileResp struct {
ID uint `json:"id" form:"id"`
CreatedAt time.Time `json:"created_at" form:"created_at"`
UpdatedAt time.Time `json:"updated_at" form:"updated_at"`
FileName string `json:"file_name" form:"file_name"`
FilePath string `json:"file_path" form:"file_path"`
Content string `json:"content" form:"content"`
}

5
proto/status.go
View File

@ -63,4 +63,9 @@ const (
DeleteFileInfoFailed = 78 // 删除文件信息失败
DataFormatError = 80 // 数据格式错误
AddConfigFileFailed = 90 // 添加配置文件失败
UpdateConfigFailed = 91 // 更新配置失败
DeleteConfigFailed = 92 // 删除配置失败
SearchConfigFileFailed = 93 // 获取配置失败
)

150
service/fileService.go
View File

@ -10,6 +10,7 @@ import (
"os"
"path"
"time"
"videoplayer/dao"
"videoplayer/proto"
"videoplayer/worker"
)
@ -55,3 +56,152 @@ func CalculateFileMd5(file io.Reader) string {
}
return fmt.Sprintf("%x", hash.Sum(nil))
}
func CheckUploadRequestParameters(req *proto.FileUploadReq) error {
var err error
if req.AuthType == "" {
err = fmt.Errorf("auth_type is empty")
}
if req.UploadType == "" {
req.UploadType = "1"
}
if req.UploadType != "1" {
req.UploadType = "2"
}
if proto.File_Type[req.Type] == 0 {
err = fmt.Errorf("file type is invalid")
}
return err
}
func CreateConfigFile(req *proto.ConfigFileReq, userId int) error {
var err error
user := GetUserByIDWithCache(userId)
if user.ID == 0 || user.Role != "admin" {
err = fmt.Errorf("user not found or no permission")
return err
}
//查看系统中是否存在文件,不存在则创建
file := req.FilePath + "/" + req.FileName
_, err = os.Stat(file)
if err != nil {
//创建文件
f, err2 := os.Create(file)
if err2 != nil {
err = err2
return err
}
defer func(f *os.File) {
err := f.Close()
if err != nil {
fmt.Println("Error closing file")
}
}(f)
}
//创建
configFile := dao.ConfigFile{FilePath: req.FilePath, FileName: req.FileName, AuthID: userId}
_, err3 := dao.CreateConfigFile(configFile)
if err3 != nil {
err = err3
return err
}
return err
}
func DeleteConfigFile(req *proto.ConfigFileReq, userId int) error {
var err error
user := GetUserByIDWithCache(userId)
if user.ID == 0 || user.Role != "admin" {
err = fmt.Errorf("user not found or no permission")
return err
}
//删除文件
config_file := dao.FindConfigFileByID(req.ID, userId)
if config_file.ID == 0 {
err = fmt.Errorf("config file not found")
return err
}
err = dao.DeleteConfigFileByID(req.ID)
if req.DelFile {
file := config_file.FilePath + "/" + config_file.FileName
err = os.Remove(file)
if err != nil {
return err
}
}
//删除数据库记录
return err
}
type ConfigFileService struct {
}
func (c *ConfigFileService) UpdateConfigFile(req *proto.ConfigFileReq, userId int) error {
var err error
user := GetUserByIDWithCache(userId)
if user.ID == 0 || user.Role != "admin" {
err = fmt.Errorf("user not found or no permission")
return err
}
config_file := dao.FindConfigFileByID(req.ID, userId)
if config_file.ID == 0 {
err = fmt.Errorf("config file not found")
return err
}
//修改文件名
if req.FileName != "" {
file := config_file.FilePath + "/" + config_file.FileName
new_file := config_file.FilePath + "/" + req.FileName
err = os.Rename(file, new_file)
if err != nil {
return err
}
err = dao.UpdateConfigFileByID(req.ID, dao.ConfigFile{FileName: req.FileName})
}
if req.Content != "" {
file := config_file.FilePath + "/" + config_file.FileName
f, err2 := os.OpenFile(file, os.O_WRONLY|os.O_TRUNC, 0644) //打开文件，清空文件内容，写入新内容，不存在则创建
if err2 != nil {
err = err2
return err
}
defer func(f *os.File) {
err3 := f.Close()
if err3 != nil {
fmt.Println("Error closing file")
}
}(f)
_, err = f.WriteString(req.Content)
if err != nil {
return err
}
}
return err
}
func (c *ConfigFileService) SearchOneConfigFile(req *proto.ConfigFileReq, userId int) ([]proto.SearchOneConfigFileResp, error) {
user := GetUserByIDWithCache(userId)
if user.ID == 0 || user.Role != "admin" {
return []proto.SearchOneConfigFileResp{}, fmt.Errorf("user not found or no permission")
}
config_file := dao.FindConfigFileByID(req.ID, userId)
if config_file.ID == 0 {
return []proto.SearchOneConfigFileResp{}, fmt.Errorf("config file not found")
}
file := config_file.FilePath + "/" + config_file.FileName
content, err2 := os.ReadFile(file)
if err2 != nil {
return []proto.SearchOneConfigFileResp{}, err2
}
resp := []proto.SearchOneConfigFileResp{{ID: config_file.ID, FileName: config_file.FileName, Content: string(content), CreatedAt: config_file.CreatedAt, UpdatedAt: config_file.UpdatedAt}}
return resp, nil
}
func (c *ConfigFileService) SearchAllConfigFile(userId int) ([]dao.ConfigFile, error) {
user := GetUserByIDWithCache(userId)
if user.ID == 0 || user.Role != "admin" {
return []dao.ConfigFile{}, fmt.Errorf("user not found or no permission")
}
config_files := dao.FindConfigFileByAuthID(userId)
return config_files, nil
}

50
service/userService.go
View File

@ -1,6 +1,7 @@
package service
import (
"encoding/json"
"errors"
"fmt"
"regexp"
@ -51,6 +52,38 @@ func GetUserByID(id int) []proto.User {
return dao.FindUserByID(id)
}
// 获取用户信息，有redis缓存
func GetUserByIDWithCache(id int) dao.User {
if id <= 0 {
return dao.User{}
}
var user dao.User
//先从redis获取
key := "user_info_" + strconv.Itoa(id)
user_str := worker.GetRedis(key)
if user_str != "" {
err := json.Unmarshal([]byte(user_str), &user)
if err != nil {
fmt.Println("get user info , json unmarshal error:", err, "\tuser_str:", user_str)
return dao.User{}
}
} else {
user = dao.FindUserByID2(id)
if user.ID != 0 {
userJson, err := json.Marshal(user)
if err != nil {
fmt.Println("get user info , json marshal error:", err)
return dao.User{}
}
success := worker.SetRedis(key, string(userJson))
if !success {
fmt.Println("set redis error,user json:", string(userJson))
}
}
}
return user
}
func GetUserByNameLike(name string) []proto.User {
return dao.FindUserByNameLike(name)
}
@ -63,6 +96,7 @@ func UpdateUser(user_id int, req proto.UpdateUserInfoReq) (int, error) {
//添加修改用户信息到同步列表
if err == nil {
err2 := setSyncUserDataSet("update", user_id)
UpdateUserCache(user_id)
if err2 != nil {
fmt.Println("set sync user data set error:", err2)
return user_id, nil
@ -74,6 +108,7 @@ func UpdateUser(user_id int, req proto.UpdateUserInfoReq) (int, error) {
if err == nil {
//添加修改用户信息到同步列表
err2 := setSyncUserDataSet("update", req.ID)
UpdateUserCache(req.ID)
if err2 != nil {
fmt.Println("set sync user data set error:", err2)
return req.ID, nil
@ -85,6 +120,21 @@ func UpdateUser(user_id int, req proto.UpdateUserInfoReq) (int, error) {
}
}
func UpdateUserCache(id int) {
key := "user_info_" + strconv.Itoa(id)
if worker.IsContainKey(key) {
user := GetUserByID(id)
userJson, err := json.Marshal(user)
if err != nil {
fmt.Println("get user info , json marshal error:", err)
}
success := worker.SetRedis(key, string(userJson))
if !success {
fmt.Println("set redis error,user json:", string(userJson))
}
}
}
func DeleteUserService(id, user_id int) int {
res := 0
if user_id == id {