package main

import (
	"fmt"
	"strconv"
	"sync"
	"time"

	"github.com/wgliang/timewheel"
)

type Wheel struct {
	mux   *sync.Mutex
	Cache map[string]string
}

func NewWheel() *Wheel {
	w := &Wheel{
		mux: new(sync.Mutex),
	}
	w.Cache = make(map[string]string, 0)
	return w
}

func (w *Wheel) Get(key string) string {
	w.mux.Lock()
	defer w.mux.Unlock()
	return w.Cache[key]
}

func (w *Wheel) Add(key, value string) {
	w.mux.Lock()
	defer w.mux.Unlock()
	w.Cache[key] = value
}

func (w *Wheel) Remove(key string) {
	w.mux.Lock()
	defer w.mux.Unlock()
	delete(w.Cache, key)
}

func goValue(tw *timewheel.TimeWheel, w *Wheel) {
	ti := time.Tick(1 * time.Second)
	key := 0
	for {
		select {
		case <-ti:
			if key == 20 {
				return
			}
			key++
			value := time.Now().Format("2016-11-14 22:22:22")
			// 业务中对资源的管理
			w.Add(strconv.Itoa(key), value)
			fmt.Println("add to Wheel...", value)
			// 不要忘记同时在时间轮里也要做改变，原则就是业务中的改变记得通知时间轮，但时间轮做的工作我们无需关心
			tw.Add(strconv.Itoa(key))
			fmt.Println("add to TimeWheel...", key)
		}
	}
}

func PrintWheel(w *Wheel) {
	ti := time.Tick(1 * time.Second)
	key := 0
	for {
		select {
		case <-ti:
			if key == 30 {
				return
			}
			key++
			w.mux.Lock()
			fmt.Println(w.Cache)
			w.mux.Unlock()
		}
	}
}

func main() {
    // 初始化你的资源和接口
	w := NewWheel()
	// 传入你要对你的资源要做的操作，以及传入回调函数
	wheel := timewheel.NewTimeWheel(time.Second*1, 10, func(w interface{}, key interface{}) {
		w.(*Wheel).Remove(key.(string))
	}, w)
	// 开启我们的时间轮
	wheel.Start()
	// 在这里代表你的项目中对资源做的改变，增加等等，然后剩下的就交给时间轮管理吧
	go goValue(wheel, w)
	go PrintWheel(w)
	time.Sleep(30 * time.Second)
}