60 lines
886 B
Go
60 lines
886 B
Go
package main
|
|
|
|
import (
|
|
"database/sql"
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/golang/glog"
|
|
_ "github.com/lib/pq"
|
|
)
|
|
|
|
func fetchMessage(db *sql.DB, hash string) error {
|
|
rows, err := db.Query(`
|
|
SELECT
|
|
blob
|
|
FROM
|
|
original
|
|
WHERE
|
|
hash = $1
|
|
`, hash)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
var b []byte
|
|
|
|
for rows.Next() {
|
|
if err := rows.Scan(&b); err != nil {
|
|
return err
|
|
}
|
|
fmt.Println(string(b))
|
|
}
|
|
if err := rows.Err(); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func main() {
|
|
defer glog.Flush()
|
|
flag.Parse()
|
|
|
|
// TODO(wathiede): make a set of flags.
|
|
db, err := sql.Open("postgres", "user=gomail dbname=gomail sslmode=disable")
|
|
if err != nil {
|
|
glog.Fatal(err)
|
|
}
|
|
|
|
if flag.NArg() == 0 {
|
|
fmt.Println("Must specify message hashes to print")
|
|
os.Exit(1)
|
|
}
|
|
|
|
for _, hash := range flag.Args() {
|
|
if err := fetchMessage(db, hash); err != nil {
|
|
glog.Fatal(err)
|
|
}
|
|
}
|
|
}
|