sftppoller.go

package main

import (
	"context"
	"os"
	"os/signal"

	log "github.com/Sirupsen/logrus"
	"github.com/da4nik/sftppoller/config"
	"github.com/da4nik/sftppoller/merger"
	"github.com/da4nik/sftppoller/poller"
	_ "github.com/joho/godotenv/autoload"
)

var logFile *os.File
var version string
var buildTime string

func initLogger() {
	// Setting up logger
	log.SetFormatter(&log.TextFormatter{
		FullTimestamp: true,
	})
	log.SetLevel(log.DebugLevel)

	log.SetOutput(os.Stdout)
	if len(config.LogFileName) > 0 {
		var err error
		logFile, err = os.OpenFile(config.LogFileName, os.O_WRONLY|os.O_CREATE, 0664)
		if err != nil {
			log.Warningf("File %s, can't be opened, using STDOUT for logging.", config.LogFileName)
		} else {
			log.SetOutput(logFile)
		}
	}
}

func main() {
	config.Load()
	initLogger()

	if version != "" && buildTime != "" {
		log.Infof("Starting sftppoller v%s build at %s", version, buildTime)
	}

	log.Infof("SFTP will be polled every %d seconds.", config.SFTPPollIntervalSeconds)

	pollerCtx, pollerCancel := context.WithCancel(context.Background())
	poller.Start(pollerCtx)
	defer pollerCancel()

	mergerCtx, mergerCancel := context.WithCancel(context.Background())
	merger.Init(mergerCtx)
	defer mergerCancel()

	c := make(chan os.Signal, 1)
	signal.Notify(c, os.Interrupt)
	<-c
}