Fix logging.

This commit is contained in:
Auke Kok 2018-01-08 16:57:58 -08:00
parent c82f794f77
commit ffffafd347

15
main.go
View File

@ -12,6 +12,7 @@ import (
"gopkg.in/gomail.v2" "gopkg.in/gomail.v2"
"io/ioutil" "io/ioutil"
"log" "log"
"fmt"
"net" "net"
"net/http" "net/http"
"net/http/fcgi" "net/http/fcgi"
@ -19,6 +20,14 @@ import (
"time" "time"
) )
// Change our logging to be journalctl friendly
type logWriter struct {
}
func (writer logWriter) Write(bytes []byte) (int, error) {
return fmt.Print(string(bytes))
}
// DB related stuff // DB related stuff
var db *sql.DB var db *sql.DB
@ -736,7 +745,9 @@ send_response:
} }
func main() { func main() {
// Custom log format
log.SetFlags(0) log.SetFlags(0)
log.SetOutput(new(logWriter))
// config stuffs // config stuffs
viper.SetConfigName("mt2fa") viper.SetConfigName("mt2fa")
@ -766,7 +777,7 @@ func main() {
listener, err := net.Listen("unix", s) listener, err := net.Listen("unix", s)
if err != nil { if err != nil {
log.Fatal("mt2fa: net.Listen: ", err) log.Fatal("net.Listen: ", err)
} }
os.Chmod(s, 0666) os.Chmod(s, 0666)
@ -821,7 +832,7 @@ func main() {
// serve requests. // serve requests.
h := new(FastCGIServer) h := new(FastCGIServer)
log.Print("mt2fa: started") log.Print("started")
err = fcgi.Serve(listener, h) err = fcgi.Serve(listener, h)
} }