generated from michaelpeterswa/go-start
-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
53 lines (45 loc) · 1.44 KB
/
main.go
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
package main
import (
"log"
"net/http"
"sync"
"github.com/gorilla/mux"
"github.com/michaelpeterswa/yt-playlist-ripper/internal/handlers"
"github.com/michaelpeterswa/yt-playlist-ripper/internal/logging"
"github.com/michaelpeterswa/yt-playlist-ripper/internal/settings"
"github.com/michaelpeterswa/yt-playlist-ripper/internal/ytdl"
"github.com/prometheus/client_golang/prometheus/promhttp"
"github.com/robfig/cron/v3"
"go.uber.org/zap"
)
func main() {
logger, err := logging.InitZap()
if err != nil {
log.Fatalf("could not acquire zap logger: %s", err.Error())
}
logger.Info("yt-playlist-ripper init...")
settings, err := settings.InitSettings()
if err != nil {
logger.Fatal("could not init settings", zap.Error(err))
}
lockMap := make(map[string]*sync.Mutex, len(settings.Playlists))
ytdlClient := ytdl.New(logger, lockMap)
c := cron.New()
for _, playlist := range settings.Playlists {
ytdlClient.LockMap[playlist] = &sync.Mutex{}
logger.Info("adding playlist to cron", zap.String("playlist", playlist))
_, err := c.AddFunc(settings.CronString, ytdlClient.Run(playlist))
if err != nil {
logger.Error("could not add cron job", zap.Error(err))
}
}
c.Start()
r := mux.NewRouter()
r.HandleFunc("/healthcheck", handlers.HealthcheckHandler)
r.Handle("/metrics", promhttp.Handler())
http.Handle("/", r)
err = http.ListenAndServe(":8080", nil)
if err != nil {
logger.Fatal("could not start http server", zap.Error(err))
}
}