tgdrive/teldrive
1
1
Fork 0
mirror of https://github.com/tgdrive/teldrive.git synced 2024-09-20 16:25:54 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

remove extra db call in upload

Browse source
This commit is contained in:
divyam234 2023-11-06 03:56:37 +05:30
parent 193bcb104f
commit abb5029472
1 changed files with 3 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
services/upload.service.go
View file

@ -69,25 +69,12 @@ func (us *UploadService) UploadFile(c *gin.Context) (*schemas.UploadPartOut, *ty
uploadId := c.Param("id")
var uploadPart []models.Upload
us.Db.Model(&models.Upload{}).Where("upload_id = ?", uploadId).Where("part_no = ?", uploadQuery.PartNo).
Where("user_id = ?", userId).
Find(&uploadPart)
if len(uploadPart) == 1 {
out := mapper.MapUploadSchema(&uploadPart[0])
return out, nil
}
file := c.Request.Body
fileSize := c.Request.ContentLength
fileName := uploadQuery.Filename
var msgId int
tokens, err := GetBotsToken(c, userId)
if err != nil {
@ -156,16 +143,16 @@ func (us *UploadService) UploadFile(c *gin.Context) (*schemas.UploadPartOut, *ty
updates := res.(*tg.Updates)
msgId = updates.Updates[0].(*tg.UpdateMessageID).ID
message, ok := updates.Updates[1].(*tg.UpdateNewChannelMessage).Message.(*tg.Message)
if msgId == 0 {
if !ok {
return errors.New("failed to upload part")
}
partUpload := &models.Upload{
Name: fileName,
UploadId: uploadId,
PartId: msgId,
PartId: message.ID,
ChannelID: channelId,
Size: fileSize,
PartNo: uploadQuery.PartNo,