func SetupDatabase(wf *web.WebFace, db *database.StatTrackerDB) {
webBuf := bytes.NewBufferString("Starting Server")
fileCounter := 0
numFiles := 0
outBuf := io.MultiWriter(webBuf, os.Stdout)
fmt.Fprintf(outBuf, "Hosting on %s \n", *addr)
wf.RedirectHandler = func(rw http.ResponseWriter, req *http.Request) {
fmt.Fprintf(rw, "Files Processed: %4d/%d \n", fileCounter, numFiles)
fmt.Fprint(rw, webBuf)
}
fmt.Fprintln(outBuf, "Fetching File List")
cPage := make(chan int)
var driveFilelist []*drive.File
go func() {
var errDrv error
driveFilelist, errDrv = google.AllFiles("mimeType = 'application/vnd.google-apps.document'", cPage)
if errDrv != nil {
log.Fatalln("File List Error:", errDrv)
}
}()
for i := <-cPage; i != -1; i = <-cPage {
fmt.Fprintf(outBuf, "Getting Page: %d \n", i)
}
// Handle per file
docStatList := []*stat.DocStat{}
numFiles = len(driveFilelist)
for ifile, file := range driveFilelist {
fileCounter = ifile
db.WriteFile(file)
dStat := FilePullCalc(file, db)
db.WriteFileStats(dStat)
fmt.Fprintf(outBuf, "Stats File Generated: %s... %s %s\n", file.Id[:6], dStat.LastMod[:10], file.Title)
docStatList = append(docStatList, dStat)
}
// Generate Daily Stat
dates := stat.CreateDailyUserStat(docStatList)
// Slower but good test (and get sorting from DB)
for _, v := range dates {
db.WriteDailyUserStats(&v)
}
wf.RedirectHandler = nil
}
