package main

import (
	"log"
	"time"

	mgo "gopkg.in/mgo.v2"
	"gopkg.in/mgo.v2/bson"
)

var DB, _ = conn("127.0.0.1:27017", "agent")

func cleanThread() {
	client := DB.C("client")
	for {
		var offlineIPList []string
		log.Println(bson.M{"uptime": bson.M{"$lte": time.Now().Add(time.Hour * time.Duration(-72))}})
		err := client.Find(
			bson.M{"uptime": bson.M{"$lte": time.Now().Add(time.Hour * time.Duration(-72))}}).Distinct("ip", &offlineIPList)
		if err != nil {
			log.Println(err.Error())
		}
		if len(offlineIPList) >= 100 {
			time.Sleep(time.Second * 60)
			continue
		}
		for _, ip := range offlineIPList {
			log.Println(bson.M{"uptime": bson.M{"$lte": time.Now().Add(time.Hour * time.Duration(-72))}})
			err = DB.C("client").Remove(bson.M{"ip": ip})
			if err != nil {
				log.Println(err.Error())
			}
		}
	}
}

func conn(netloc string, dbname string) (*mgo.Database, error) {
	session, err := mgo.Dial(netloc)
	if err != nil {
		return nil, err
	}
	session.SetMode(mgo.Monotonic, true)
	db := session.DB(dbname)
	return db, nil
}

func main() {
	cleanThread()
}