Skip to content

Commit 2f6fc1a

Browse files
MattShirleyBenGalewsky
authored andcommitted
add proper s3 object sourcing
1 parent 1d747d1 commit 2f6fc1a

File tree

4 files changed

+21
-18
lines changed

4 files changed

+21
-18
lines changed

servicex_app/servicex_app/resources/internal/transformer_file_complete.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -89,8 +89,9 @@ def put(self, request_id):
8989
info = request.get_json()
9090
logger = current_app.logger
9191
log_extra = {
92-
'requestId': request_id,
93-
'file-id': info['file-id']
92+
'requestId': request_id,
93+
'file-id': info['file-id'],
94+
's3-object-name': info['s3-object-name'],
9495
}
9596

9697
logger.info("FileComplete", extra={**log_extra, 'metric': info})

servicex_app/servicex_app/resources/transformation/results.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,3 @@
1-
import logging
21
import datetime
32

43
from flask_restful import reqparse
@@ -7,8 +6,6 @@
76
from servicex_app.models import TransformationResult
87
from servicex_app.resources.servicex_resource import ServiceXResource
98

10-
logger = logging.getLogger(__name__)
11-
129

1310
class TransformationResults(ServiceXResource):
1411
@auth_required

transformer_sidecar/src/transformer_sidecar/servicex_adapter.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -104,10 +104,13 @@ def put_file_complete(self, rec: FileCompleteRecord):
104104
fkwargs={"json": rec.to_json(), "timeout": (0.5, None)},
105105
tries=MAX_RETRIES,
106106
delay=RETRY_DELAY)
107-
self.logger.info("Put file complete.", extra={'requestId': rec.request_id,
108-
"file-id": rec.file_id,
109-
"place": PLACE,
110-
"file_path": rec.file_path})
107+
self.logger.info("Put file complete.", extra={
108+
'requestId': rec.request_id,
109+
"file-id": rec.file_id,
110+
"place": PLACE,
111+
"file_path": rec.file_path,
112+
"s3-object-name": rec.s3_object_name,
113+
})
111114
except requests.exceptions.ConnectionError:
112115
self.logger.exception("Connection Error in put_file_complete",
113116
extra={'requestId': rec.request_id,

transformer_sidecar/src/transformer_sidecar/transformer.py

Lines changed: 11 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -163,7 +163,7 @@ def transform_file(
163163
transformer_stats = TransformerStats()
164164
try:
165165
# Loop through the replicas
166-
for path, _file_path in zip(paths, _file_paths):
166+
for _file_path in _file_paths:
167167
logger.info(
168168
"trying to transform file",
169169
extra={
@@ -215,19 +215,19 @@ def transform_file(
215215
rec = FileCompleteRecord(
216216
request_id=request_id,
217217
file_path=_file_path,
218-
s3_object_name=path,
218+
s3_object_name="none",
219219
file_id=file_id,
220220
status="success",
221221
total_time=time.time() - total_time,
222222
total_events=transformer_stats.total_events,
223223
total_bytes=os.path.getsize(output_path),
224224
)
225-
226225
if object_store:
227-
upload_file(Path(transform_request["safeOutputFileName"]),
228-
servicex,
229-
rec
230-
)
226+
upload_file(
227+
Path(transform_request["safeOutputFileName"]),
228+
servicex,
229+
rec
230+
)
231231
else:
232232
servicex.put_file_complete(rec)
233233

@@ -257,8 +257,8 @@ def transform_file(
257257
rec = FileCompleteRecord(
258258
request_id=request_id,
259259
file_path=_file_paths[0],
260-
s3_object_name=paths[0],
261260
file_id=file_id,
261+
s3_object_name="none",
262262
status="failure",
263263
total_time=time.time() - total_time,
264264
total_events=0,
@@ -294,7 +294,7 @@ def transform_file(
294294
rec = FileCompleteRecord(
295295
request_id=request_id,
296296
file_path=_file_paths[0],
297-
s3_object_name=paths[0],
297+
s3_object_name="none",
298298
file_id=file_id,
299299
status="failure",
300300
total_time=time.time() - total_time,
@@ -354,6 +354,8 @@ def upload_file(source_path: Path,
354354
file_to_upload = source_path
355355
object_name = source_path.name
356356

357+
rec.s3_object_name = object_name
358+
357359
logger.info("Uploading file to object store.",
358360
extra={'requestId': request_id,
359361
"file-id": rec.file_id,

0 commit comments

Comments
 (0)