1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
// Copyright 2019 Nick White.
// Use of this source code is governed by the GPLv3
// license that can be found in the LICENSE file.
package main
import (
"flag"
"fmt"
"log"
"os"
"time"
"rescribe.xyz/bookpipeline"
)
const usage = `Usage: unstickocr [-v] bookname
unstickocr deletes a book from the OCR queue and adds 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
nasty bug. 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
CheckQueue(url string, timeout int64) (bookpipeline.Qmsg, error)
AddToQueue(url string, msg string) error
DelFromQueue(url string, handle string) error
OCRQueueId() string
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)
done := false
for a := 0; a < 5; a++ {
for i := 0; i < 10; i++ {
verboselog.Println("Checking OCR queue for", book)
msg, err := conn.CheckQueue(conn.OCRQueueId(), 10)
if err != nil {
log.Fatalln("Error checking OCR queue:", err)
continue
}
if msg.Handle == "" {
verboselog.Println("No message received on OCR queue")
continue
}
if msg.Body != book {
verboselog.Println("Message received on OCR queue is not the one we're",
"looking for, so will try again - found", msg.Body)
continue
}
err = conn.DelFromQueue(conn.OCRQueueId(), msg.Handle)
if err != nil {
log.Fatalln("Error deleting message from OCR queue:", err)
}
err = conn.AddToQueue(conn.AnalyseQueueId(), book)
if err != nil {
log.Fatalln("Error adding message to Analyse queue:", err)
}
done = true
break
}
if done == true {
break
}
log.Println("No message found yet, sleeping for 30 seconds to try again")
time.Sleep(30 * time.Minute)
}
if done == true {
fmt.Println("Succeeded moving message from OCR queue to Analyse queue.")
} else {
log.Fatalln("Failed to find message", book, "on OCR queue; is it still being processed?",
"It can only be discovered and processed by this tool when it is available.",
"Try shutting down any instance that is using it, waiting a few minutes,",
"and rerunning this tool.")
}
}
|
