-
Notifications
You must be signed in to change notification settings - Fork 258
/
upload-datasource.go
92 lines (82 loc) · 2.95 KB
/
upload-datasource.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
92
package importer
import (
"io"
"net/url"
"path/filepath"
"k8s.io/klog"
"kubevirt.io/containerized-data-importer/pkg/util"
)
// UploadDataSource contains all the information need to upload data into a data volume.
// Sequence of phases:
// 1a. ProcessingPhaseInfo -> ProcessingPhaseTransferScratch (In Info phase the format readers are configured) In case the readers don't contain a raw file.
// 1b. ProcessingPhaseInfo -> ProcessingPhaseTransferDataFile, in the case the readers contain a raw file.
// 2a. ProcessingPhaseTransferScratch -> ProcessingPhaseProcess
// 2b. ProcessingPhaseTransferDataFile -> ProcessingPhaseResize
// 3. ProcessingPhaseProcess -> ProcessingPhaseConvert
type UploadDataSource struct {
// Data strean
stream io.ReadCloser
// stack of readers
readers *FormatReaders
// url to a file in scratch space.
url *url.URL
}
// NewUploadDataSource creates a new instance of an UploadDataSource
func NewUploadDataSource(stream io.ReadCloser) *UploadDataSource {
return &UploadDataSource{
stream: stream,
}
}
// Info is called to get initial information about the data.
func (ud *UploadDataSource) Info() (ProcessingPhase, error) {
var err error
// Hardcoded to only accept kubevirt content type.
ud.readers, err = NewFormatReaders(ud.stream, uint64(0))
if err != nil {
klog.Errorf("Error creating readers: %v", err)
return ProcessingPhaseError, err
}
if !ud.readers.Convert {
// Uploading a raw file, we can write that directly to the target.
return ProcessingPhaseTransferDataFile, nil
}
return ProcessingPhaseTransferScratch, nil
}
// Transfer is called to transfer the data from the source to the passed in path.
func (ud *UploadDataSource) Transfer(path string) (ProcessingPhase, error) {
if util.GetAvailableSpace(path) <= int64(0) {
//Path provided is invalid.
return ProcessingPhaseError, ErrInvalidPath
}
file := filepath.Join(path, tempFile)
err := util.StreamDataToFile(ud.readers.TopReader(), file)
if err != nil {
return ProcessingPhaseError, err
}
// If we successfully wrote to the file, then the parse will succeed.
ud.url, _ = url.Parse(file)
return ProcessingPhaseProcess, nil
}
// TransferFile is called to transfer the data from the source to the passed in file.
func (ud *UploadDataSource) TransferFile(fileName string) (ProcessingPhase, error) {
err := util.StreamDataToFile(ud.readers.TopReader(), fileName)
if err != nil {
return ProcessingPhaseError, err
}
return ProcessingPhaseResize, nil
}
// Process is called to do any special processing before giving the url to the data back to the processor
func (ud *UploadDataSource) Process() (ProcessingPhase, error) {
return ProcessingPhaseConvert, nil
}
// GetURL returns the url that the data processor can use when converting the data.
func (ud *UploadDataSource) GetURL() *url.URL {
return ud.url
}
// Close closes any readers or other open resources.
func (ud *UploadDataSource) Close() error {
if ud.stream != nil {
return ud.stream.Close()
}
return nil
}