diff options
author | Nick White <git@njw.name> | 2019-12-16 16:05:47 +0000 |
---|---|---|
committer | Nick White <git@njw.name> | 2019-12-16 16:05:47 +0000 |
commit | d4c42f851cb2254a27940721de3e3ef2d73f2910 (patch) | |
tree | b886e8eed89a769a276cd0124103e97ff422f7f4 /cmd/addtoanalysequeue/main.go | |
parent | 70eecf79704452d600f3edca9dec2e9726dcfe6d (diff) |
Add a new tool, addtoqueue, which can be used to generically add any message to any queue
Diffstat (limited to 'cmd/addtoanalysequeue/main.go')
-rw-r--r-- | cmd/addtoanalysequeue/main.go | 72 |
1 files changed, 0 insertions, 72 deletions
diff --git a/cmd/addtoanalysequeue/main.go b/cmd/addtoanalysequeue/main.go deleted file mode 100644 index c9a34c7..0000000 --- a/cmd/addtoanalysequeue/main.go +++ /dev/null @@ -1,72 +0,0 @@ -package main - -import ( - "flag" - "fmt" - "log" - "os" - - "rescribe.xyz/bookpipeline" -) - -const usage = `Usage: addtoanalysequeue [-v] bookname - -addtoanalysequeue adds a book it to the Analyse queue. - -This should be done automatically by the bookpipeline tool once -the OCR job has completed, but sometimes it isn't, because of a -bug where if a file that is named like a preprocessed image -doesn't have a hOCR component. Once that bug is squashed, this -tool can be deleted. -` - -// null writer to enable non-verbose logging to be discarded -type NullWriter bool - -func (w NullWriter) Write(p []byte) (n int, err error) { - return len(p), nil -} - -type UnstickPipeliner interface { - Init() error - AddToQueue(url string, msg string) error - AnalyseQueueId() string -} - -func main() { - verbose := flag.Bool("v", false, "verbose") - flag.Usage = func() { - fmt.Fprintf(flag.CommandLine.Output(), usage) - flag.PrintDefaults() - } - flag.Parse() - - if flag.NArg() != 1 { - flag.Usage() - return - } - - var verboselog *log.Logger - if *verbose { - verboselog = log.New(os.Stdout, "", 0) - } else { - var n NullWriter - verboselog = log.New(n, "", 0) - } - - var conn UnstickPipeliner - conn = &bookpipeline.AwsConn{Region: "eu-west-2", Logger: verboselog} - - err := conn.Init() - if err != nil { - log.Fatalln("Error setting up cloud connection:", err) - } - - book := flag.Arg(0) - - err = conn.AddToQueue(conn.AnalyseQueueId(), book) - if err != nil { - log.Fatalln("Error adding message to Analyse queue:", err) - } - fmt.Println("Added message from to the Analyse queue.") -} |