Skip to content

Commit 6ad9dff

Browse files
added more logging. added tests for empty or bad json in meta file.
1 parent e12348b commit 6ad9dff

File tree

2 files changed

+26
-6
lines changed

2 files changed

+26
-6
lines changed

worker.go

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,7 @@ func parseUploadPath(dir string) (*UploadInfo, error) {
8181
func (w *Worker) Process(job Job) error {
8282
p, err := parseUploadPath(job.Dir)
8383
if err != nil {
84-
return err
84+
return fmt.Errorf("error parsing upload path: %s", err.Error())
8585
}
8686

8787
dataPath := filepath.Join(job.Root, job.Dir, "data")
@@ -91,7 +91,7 @@ func (w *Worker) Process(job Job) error {
9191
var meta MetaData
9292

9393
if err := readMetaFile(metaPath, &meta); err != nil {
94-
return err
94+
return fmt.Errorf("error reading meta file: %s", err.Error())
9595
}
9696

9797
// Add info extracted from path.
@@ -105,15 +105,15 @@ func (w *Worker) Process(job Job) error {
105105
s3path := fmt.Sprintf("node-data/%s/sage-%s-%s/%s", p.Namespace, p.Name, p.Version, p.NodeID)
106106

107107
if err := w.Uploader.UploadFile(dataPath, filepath.Join(s3path, targetNameData), &meta); err != nil {
108-
return err
108+
return fmt.Errorf("error uploading data file: %s", err.Error())
109109
}
110110

111111
if err := w.Uploader.UploadFile(metaPath, filepath.Join(s3path, targetNameMeta), nil); err != nil {
112-
return err
112+
return fmt.Errorf("error uploading meta file: %s", err.Error())
113113
}
114114

115115
if err := os.WriteFile(donePath, []byte{}, 0o644); err != nil {
116-
return fmt.Errorf("could not create flag file: %s", err.Error())
116+
return fmt.Errorf("error creating flag file: %s", err.Error())
117117
}
118118

119119
// TODO(sean) If we see the need to support various clean up strategies,
@@ -123,7 +123,7 @@ func (w *Worker) Process(job Job) error {
123123
// Clean up data, meta and done files.
124124
for _, name := range []string{dataPath, metaPath, donePath} {
125125
if err := os.Remove(name); err != nil {
126-
return fmt.Errorf("failed to clean up %s", name)
126+
return fmt.Errorf("error cleaning up %s: %s", name, err.Error())
127127
}
128128
}
129129

worker_test.go

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -156,6 +156,26 @@ func TestScanAndProcess(t *testing.T) {
156156
data: []byte(`{"ts":1638576647406523064,"labels":{"filename":"wow3.flac"}}`),
157157
},
158158
},
159+
"EmptyMeta": {
160+
{
161+
path: "node-000048b02d15bc7c/uploads/imagesampler-top/0.2.5/1638576647406523064-9801739daae44ec5293d4e1f53d3f4d2d426d91c/data",
162+
data: []byte(`we don't know this filename`),
163+
},
164+
{
165+
path: "node-000048b02d15bc7c/uploads/imagesampler-top/0.2.5/1638576647406523064-9801739daae44ec5293d4e1f53d3f4d2d426d91c/meta",
166+
data: []byte(``),
167+
},
168+
},
169+
"EmptyMetaBadJSON": {
170+
{
171+
path: "node-000048b02d15bc7c/uploads/imagesampler-top/0.2.5/1638576647406523064-9801739daae44ec5293d4e1f53d3f4d2d426d91c/data",
172+
data: []byte(`we don't know this filename`),
173+
},
174+
{
175+
path: "node-000048b02d15bc7c/uploads/imagesampler-top/0.2.5/1638576647406523064-9801739daae44ec5293d4e1f53d3f4d2d426d91c/meta",
176+
data: []byte(`{32-}`),
177+
},
178+
},
159179
"MissingMetaField": {
160180
{
161181
path: "node-000048b02d15bc7c/uploads/imagesampler-top/0.2.5/1638576647406523064-9801739daae44ec5293d4e1f53d3f4d2d426d91c/data",

0 commit comments

Comments
 (0)