/
dfs.writeFilePartData_handler.go
91 lines (79 loc) · 2.34 KB
/
dfs.writeFilePartData_handler.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
/*
* Created from 'scheme.tl' by 'mtprotoc'
*
* Copyright (c) 2021-present, Teamgram Studio (https://teamgram.io).
* All rights reserved.
*
* Author: teamgramio (teamgram.io@gmail.com)
*/
package core
import (
"time"
"github.com/teamgram/proto/mtproto"
"github.com/teamgram/teamgram-server/app/service/dfs/dfs"
"github.com/teamgram/teamgram-server/app/service/dfs/internal/model"
)
// DfsWriteFilePartData
// dfs.writeFilePartData flags:# creator:long file_id:long bytes:bytes big:flags.0?true file_total_parts:flags.1?int = Bool;
func (c *DfsCore) DfsWriteFilePartData(in *dfs.TLDfsWriteFilePartData) (*mtproto.Bool, error) {
var (
err error
)
if len(in.GetBytes()) == 0 {
err = mtproto.ErrFilePartEmpty
c.Logger.Errorf("dfs.writeFilePartData - error: %v", err)
return nil, err
}
err = c.svcCtx.Dao.WriteFilePartData(c.ctx, in.Creator, in.FileId, in.FilePart, in.Bytes)
if err != nil {
c.Logger.Errorf("dfs.writeFilePartData - error: %v", err)
return nil, err
}
if in.FilePart == 0 {
err = c.svcCtx.Dao.SetFileInfo(c.ctx, &model.DfsFileInfo{
Creator: in.Creator,
FileId: in.FileId,
Big: in.Big,
FileName: "",
FileTotalParts: int(in.GetFileTotalParts().GetValue()),
FirstFilePartSize: len(in.Bytes),
FilePartSize: 0,
LastFilePartSize: 0,
MimeType: "",
Mtime: time.Now().Unix(),
})
// TODO(@benqi): error
if err != nil {
c.Logger.Errorf("dfs.writeFilePartData - error: %v", err)
return nil, mtproto.ErrFilePartInvalid
}
} else if in.FilePart == 1 {
err = c.svcCtx.Dao.SetFileInfo(c.ctx, &model.DfsFileInfo{
Creator: in.Creator,
FileId: in.FileId,
FilePartSize: len(in.Bytes),
})
// TODO(@benqi): error
if err != nil {
c.Logger.Errorf("dfs.writeFilePartData - error: %v", err)
return nil, mtproto.ErrFilePartInvalid
}
}
if in.GetFileTotalParts() != nil {
if in.GetFileTotalParts().GetValue() == in.FilePart+1 {
err = c.svcCtx.Dao.SetFileInfo(
c.ctx,
&model.DfsFileInfo{
Creator: in.Creator,
FileId: in.FileId,
LastFilePartSize: len(in.Bytes),
})
// TODO(@benqi): error
if err != nil {
c.Logger.Errorf("dfs.writeFilePartData - error: %v", err)
return nil, mtproto.ErrFilePartInvalid
}
}
}
return mtproto.BoolTrue, nil
}