54 lines
1021 B
Go
54 lines
1021 B
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"log"
|
||
|
|
"os"
|
||
|
|
"os/signal"
|
||
|
|
"syscall"
|
||
|
|
_cron "twcron/cron"
|
||
|
|
"twcron/mode"
|
||
|
|
|
||
|
|
"github.com/go-pg/pg/v10"
|
||
|
|
"github.com/joho/godotenv"
|
||
|
|
"github.com/robfig/cron/v3"
|
||
|
|
)
|
||
|
|
|
||
|
|
func init() {
|
||
|
|
os.Setenv("TZ", "UTC")
|
||
|
|
|
||
|
|
if mode.Get() == mode.DevelopmentMode {
|
||
|
|
godotenv.Load(".env.development")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
db := pg.Connect(&pg.Options{
|
||
|
|
User: os.Getenv("DB_USER"),
|
||
|
|
Password: os.Getenv("DB_PASSWORD"),
|
||
|
|
Database: os.Getenv("DB_NAME"),
|
||
|
|
Addr: os.Getenv("DB_HOST") + ":" + os.Getenv("DB_PORT"),
|
||
|
|
})
|
||
|
|
defer func() {
|
||
|
|
if err := db.Close(); err != nil {
|
||
|
|
log.Fatal(err)
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
|
||
|
|
c := cron.New(cron.WithChain(
|
||
|
|
cron.SkipIfStillRunning(cron.VerbosePrintfLogger(log.New(os.Stdout, "cron: ", log.LstdFlags))),
|
||
|
|
))
|
||
|
|
if err := _cron.Attach(c, db); err != nil {
|
||
|
|
log.Fatal(err)
|
||
|
|
}
|
||
|
|
c.Start()
|
||
|
|
defer c.Stop()
|
||
|
|
|
||
|
|
log.Print("Cron is running!")
|
||
|
|
|
||
|
|
channel := make(chan os.Signal, 1)
|
||
|
|
signal.Notify(channel, os.Interrupt, os.Kill, syscall.SIGTERM, syscall.SIGINT)
|
||
|
|
<-channel
|
||
|
|
|
||
|
|
log.Print("shutting down")
|
||
|
|
}
|