From 27f38e66145666ee64ee08624a1e959d1d383931 Mon Sep 17 00:00:00 2001 From: Bill Thiede Date: Sun, 30 Mar 2014 21:24:44 -0700 Subject: [PATCH] Add dryrun mode to contacts. --- cmd/contacts/contacts.go | 17 ++++++++++++++--- 1 file changed, 14 insertions(+), 3 deletions(-) diff --git a/cmd/contacts/contacts.go b/cmd/contacts/contacts.go index ef0c363..3042ea8 100644 --- a/cmd/contacts/contacts.go +++ b/cmd/contacts/contacts.go @@ -5,17 +5,22 @@ import ( "database/sql" "flag" "fmt" - "net/mail" "time" "unicode/utf8" "xinu.tv/types" "github.com/golang/glog" _ "github.com/lib/pq" + + "xinu.tv/std/net/mail" ) -var all = flag.Bool("all", false, - "extract addresses from all messages. Default is to only extract from unprocessed messages.") +var ( + all = flag.Bool("all", false, + "extract addresses from all messages. Default is to only extract from unprocessed messages.") + + dryrun = flag.Bool("dryrun", false, "don't write to the DB.") +) type contact struct { hash string @@ -24,6 +29,12 @@ type contact struct { } func insertAddresses(db *sql.DB, batch int, contactCh chan *contact, errc chan error) { + if *dryrun { + for _ = range contactCh { + } + errc <- nil + return + } var ( txn *sql.Tx stmt *sql.Stmt