Paperclip/pkg/main.go

87 lines
2.3 KiB
Go
Raw Normal View History

2024-05-17 07:59:51 +00:00
package main
import (
"os"
"os/signal"
"syscall"
2024-06-22 04:18:54 +00:00
"git.solsynth.dev/hydrogen/paperclip/pkg/internal/database"
"git.solsynth.dev/hydrogen/paperclip/pkg/internal/gap"
"git.solsynth.dev/hydrogen/paperclip/pkg/internal/grpc"
2024-06-16 15:24:54 +00:00
"git.solsynth.dev/hydrogen/paperclip/pkg/internal/server"
"git.solsynth.dev/hydrogen/paperclip/pkg/internal/services"
2024-05-17 07:59:51 +00:00
"github.com/robfig/cron/v3"
2024-06-22 04:18:54 +00:00
pkg "git.solsynth.dev/hydrogen/paperclip/pkg/internal"
2024-05-17 07:59:51 +00:00
"github.com/rs/zerolog"
"github.com/rs/zerolog/log"
"github.com/spf13/viper"
)
func init() {
zerolog.TimeFieldFormat = zerolog.TimeFormatUnix
log.Logger = log.Output(zerolog.ConsoleWriter{Out: os.Stdout})
}
func main() {
// Configure settings
viper.AddConfigPath(".")
viper.AddConfigPath("..")
viper.SetConfigName("settings")
viper.SetConfigType("toml")
// Load settings
if err := viper.ReadInConfig(); err != nil {
log.Panic().Err(err).Msg("An error occurred when loading settings.")
}
// Connect to database
2024-06-16 15:32:52 +00:00
if err := database.NewSource(); err != nil {
2024-05-17 07:59:51 +00:00
log.Fatal().Err(err).Msg("An error occurred when connect to database.")
2024-06-16 15:32:52 +00:00
} else if err := database.RunMigration(database.C); err != nil {
2024-05-17 07:59:51 +00:00
log.Fatal().Err(err).Msg("An error occurred when running database auto migration.")
}
// Connect other services
2024-07-27 17:15:52 +00:00
if err := gap.RegisterService(); err != nil {
log.Error().Err(err).Msg("An error occurred when registering service to dealer...")
2024-06-16 15:32:52 +00:00
}
2024-05-17 07:59:51 +00:00
2024-07-28 16:53:40 +00:00
// Setup some workers
for idx := 0; idx < viper.GetInt("workers.files_deletion"); idx++ {
go services.StartConsumeDeletionTask()
}
for idx := 0; idx < viper.GetInt("workers.files_analyze"); idx++ {
go services.StartConsumeAnalyzeTask()
}
2024-05-17 07:59:51 +00:00
// Configure timed tasks
quartz := cron.New(cron.WithLogger(cron.VerbosePrintfLogger(&log.Logger)))
quartz.AddFunc("@every 60m", services.DoAutoDatabaseCleanup)
quartz.AddFunc("@midnight", services.RunScheduleDeletionTask)
2024-05-17 07:59:51 +00:00
quartz.Start()
// Server
server.NewServer()
go server.Listen()
// Grpc Server
2024-06-22 04:18:54 +00:00
grpc.NewGRPC()
go grpc.ListenGRPC()
2024-05-17 07:59:51 +00:00
// Messages
2024-06-16 15:24:54 +00:00
log.Info().Msgf("Paperclip v%s is started...", pkg.AppVersion)
2024-05-17 07:59:51 +00:00
services.ScanUnanalyzedFileFromDatabase()
services.RunScheduleDeletionTask()
2024-05-17 07:59:51 +00:00
quit := make(chan os.Signal, 1)
signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
<-quit
2024-06-16 15:24:54 +00:00
log.Info().Msgf("Paperclip v%s is quitting...", pkg.AppVersion)
2024-05-17 07:59:51 +00:00
quartz.Stop()
}