From 2089289d7d34c16d9530699f4635ff0f69d7d556 Mon Sep 17 00:00:00 2001
From: Bryan Newbold <bnewbold@robocracy.org>
Date: Wed, 26 Dec 2018 18:21:42 -0800
Subject: fileset/web support for exports

---
 rust/src/bin/fatcat-export.rs | 10 ++++++++++
 1 file changed, 10 insertions(+)

(limited to 'rust/src')

diff --git a/rust/src/bin/fatcat-export.rs b/rust/src/bin/fatcat-export.rs
index dd3f4da4..8b7c4442 100644
--- a/rust/src/bin/fatcat-export.rs
+++ b/rust/src/bin/fatcat-export.rs
@@ -46,6 +46,8 @@ arg_enum! {
         Creator,
         Container,
         File,
+        Fileset,
+        Webcapture,
         Release,
         Work
     }
@@ -107,6 +109,8 @@ macro_rules! generic_loop_work {
 generic_loop_work!(loop_work_container, ContainerEntity);
 generic_loop_work!(loop_work_creator, CreatorEntity);
 generic_loop_work!(loop_work_file, FileEntity);
+generic_loop_work!(loop_work_fileset, FilesetEntity);
+generic_loop_work!(loop_work_webcapture, WebcaptureEntity);
 generic_loop_work!(loop_work_release, ReleaseEntity);
 generic_loop_work!(loop_work_work, WorkEntity);
 
@@ -203,6 +207,12 @@ pub fn do_export(
             ExportEntityType::File => {
                 thread::spawn(move || loop_work_file(row_receiver, output_sender, &db_conn, expand))
             }
+            ExportEntityType::Fileset => {
+                thread::spawn(move || loop_work_fileset(row_receiver, output_sender, &db_conn, expand))
+            }
+            ExportEntityType::Webcapture=> {
+                thread::spawn(move || loop_work_webcapture(row_receiver, output_sender, &db_conn, expand))
+            }
             ExportEntityType::Release => thread::spawn(move || {
                 loop_work_release(row_receiver, output_sender, &db_conn, expand)
             }),
-- 
cgit v1.2.3