implement file streaming in agent

crossfire/agent.rkt

@@ -64,9 +64,10 @@
     (with-handlers ([exn:fail? (lambda (ex)
                                  (logging-report-error ex)
                                  (cleanup))])
-      ;; TODO this should be updated with the streaming interface
+      (call-with-output-file
-      (call-with-output-file tgz-file (lambda (out) (write-bytes (get-project-file tid) out))
+        tgz-file
-                             #:exists 'truncate)
+        (lambda (out) (define ft (get-project-file tid)) (file-transfer-connect ft out))
+        #:exists 'truncate)
       (log-agent-info "extracting task data for ~a" tid)
       (untgz tgz-file #:dest extract-dir)
       (set-box! success-box #t)))

crossfire/client.rkt

@@ -302,10 +302,11 @@
     ;; TODO : xterm/vt100/etc-specific
     ;; this should use a library maybe
     (printf "\r\x1b[Ktransferred ~a% [~a/~a]" pct a b)
-    (when (= a b)
-      (printf "\n"))
     (flush-output))
 
+  (define (finish-progress)
+    (printf "\n"))
+
   (define (make-random-filename)
     (string-append (bytes->hex-string (crypto-random-bytes 8)) ".agent"))
 
@@ -358,6 +359,7 @@
       ;; do submit
       (report-status "submitting project...!")
       (cmd-submit (current-directory) print-progress)
+      (finish-progress)
       (report-status "project submitted!! time for crab"))
 
     (subcommand (delete "Delete an executed or completed project")
@@ -422,6 +424,7 @@
          (define out-name (make-random-filename))
          (call-with-output-file out-name
            (lambda (o) (cmd-get-deployment aid o print-progress)))
+         (finish-progress)
          (do-final out-name)]
         [_ (error "you must provide -l, -c, -d, or -g for this command")]))
 

crossfire/comms.rkt

@@ -560,7 +560,9 @@
                 (loop (+ offs n))])]
           [(== thread-evt)
            (match (thread-receive)
-             [(msg:file-request _ _ #f) (log-tm-info "file transfer ~a complete" trans-id)]
+             [(msg:file-request _ _ #f)
+              (close-input-port port)
+              (log-tm-info "file transfer ~a complete" trans-id)]
              [(msg:file-request _ _ new-offs) (loop new-offs)]
              [x (log-tm-warning "invalid data during file transfer ~a" x) (loop offs)])]))
       (void)))

crossfire/server.rkt

@@ -172,7 +172,7 @@
 
 (define (server-get-file taskid)
   (define path (get-project-file-path taskid))
-  (file->bytes path))
+  (and (file-exists? path) (open-input-file path)))
 
 
 ;; rpc helpers
@@ -348,10 +348,11 @@
     (project-info id name mf progress matches (hash-ref agent-states id '()))))
 
 (define/contract (get-project-file taskid)
-  (-> integer? bytes?)
+  (-> integer? file-transfer?)
   ;; TODO : streaming interface
-  (with-handlers ([exn:fail? (lambda (ex) (error "unable to fetch the requested file"))])
+  (match (server-get-file taskid)
-    (server-get-file taskid)))
+    [#f (error "no such project file")]
+    [port (make-file-transfer port)]))
 
 (define/contract (get-project-matches taskid)
   (-> integer? (listof (listof integer?)))