poll.go

package main

import (
	"os"
	"runtime"
	"time"

	"google.golang.org/grpc"
)

func DumpPoll(s *grpc.Server, done chan bool, sigs chan os.Signal, url, token, dir string, d time.Duration) {
	runtime.GC()
	Info.Printf("Complete GC\n")
	DumpRefresh(url, token, dir)
	for {
		timer := time.NewTimer(d * time.Second)
		select {
		case <-timer.C:
			DumpRefresh(url, token, dir)
		case <-sigs:
			s.Stop()
			done <- true
		}
	}
}

func DumpRefresh(url, token, dir string) {
	ts := time.Now().Unix()
	lastDump, err := GetLastDumpId(ts, url, token)
	if err != nil {
		Error.Printf("Can't get last dump id: %s\n", err.Error())
		return
	}
	if lastDump.Id == "" {
		Error.Println("Last dump Id is empty...")
		return
	}
	Info.Printf("Last dump id: %s\n", lastDump.Id)
	cachedDump, err := ReadCurrentDumpId(dir + "/current")
	if err != nil {
		Error.Printf("Can't read cached dump id: %s\n", err.Error())
		return
	}
	if cachedDump.Id == "" {
		Warning.Println("Cashed dump Id is empty...")
	}
	// two states...
	if lastDump.CRC != cachedDump.CRC {
		Info.Printf("Getting new dump..")
		err := FetchDump(lastDump.Id, dir+"/dump.xml", url, token)
		if err != nil {
			Error.Printf("Can't fetch last dump: %s\n", err.Error())
			return
		}
		Info.Println("Last dump fetched")
		// parse xml
		if dumpFile, err := os.Open(dir + "/dump.xml"); err != nil {
			Error.Printf("Can't open dump file: %s\n", err.Error())
			return
		} else {
			defer dumpFile.Close()
			err = Parse(dumpFile)
			if err != nil {
				Error.Printf("Parse error: %s\n", err.Error())
				return
			} else {
				Info.Printf("Dump parsed")
				dumpFile.Close()
				runtime.GC()
				Info.Printf("Complete GC\n")
			}
			err = WriteCurrentDumpId(dir+"/current", lastDump)
			if err != nil {
				Error.Printf("Can't write currentdump file: %s\n", err.Error())
				return
			}
			Info.Println("Last dump metainfo saved")
		}
	} else if lastDump.Id != cachedDump.Id {
		Info.Printf("Not changed, but new dump metainfo")
		Parse2(lastDump.UpdateTime)
		runtime.GC()
		Info.Printf("Complete GC\n")
	} else {
		Info.Printf("No new dump")
	}
}