Илия обнови решението на 17.11.2014 01:03 (преди над 3 години)
+package main
+
+import (
+ "strconv"
+ "strings"
+ "time"
+)
+
+type ExpireMap struct {
+ channel chan string
+ expires map[string]time.Time
+ data map[string]interface{}
+ m chan struct{}
+}
+
+type ExpireMapError struct {
+ cause string
+}
+
+func (e *ExpireMapError) Error() string {
+ return e.cause
+}
+
+// Връща указател към нов обект от тип ExpireMap, готов за използване.
+func NewExpireMap() *ExpireMap {
+ channel := make(chan string, 1)
+ expires := map[string]time.Time{}
+ data := map[string]interface{}{}
+ m := make(chan struct{}, 1)
+ return &ExpireMap{channel, expires, data, m}
+}
+
+func (em *ExpireMap) lock() {
+ em.m <- struct{}{}
+}
+func (em *ExpireMap) unlock() {
+ <-em.m
+}
+
+// Добавя в хранилището нов ключ key със стойност value за expire време. Duration е дефиниран в пакета time. expire време след добавянето на ключа той вече трябва да спре да бъде в хранилището. Стойността може да е от всякакъв тип, докато ключа е от тип string.
+func (em *ExpireMap) Set(key string, value interface{}, expire time.Duration) {
+ em.lock()
+ defer em.unlock()
+ em.data[key] = value
+ em.expires[key] = time.Now().Add(expire)
+ go func() {
+ time.Sleep(expire)
+ em.Delete(key)
+ select {
+ case em.channel <- key:
+ default:
+ }
+ }()
+}
+
+// Връща стойността на ключ и true, когато ключа е намерен. Когато ключа липсва или е изтекъл връща nil и false.
+func (em *ExpireMap) Get(key string) (v interface{}, ok bool) {
+ v, ok = em.data[key]
+ return
+}
+
+// Връща стойността, отговаряща на този ключ и true, когато ключа е намерен и стойността му е от тип int. Когато ключа липсва, изтекъл е или стойността му не е от този тип метода трябва да върне нула и false.
+func (em *ExpireMap) GetInt(key string) (v int, ok bool) {
+ v, ok = em.data[key].(int)
+ return
+}
+
+// Връща стойността на този ключ и true, когато ключа е намерен и стойността му е от тип float64. Когато ключа липсва, изтекъл е или стойността му не е от този тип метода трябва да върне нула и false.
+func (em *ExpireMap) GetFloat64(key string) (v float64, ok bool) {
+ v, ok = em.data[key].(float64)
+ return
+}
+
+// Връща стойността на ключ и true, когато ключа е намерен и стойността му е от тип string. Когато ключа липсва, изтекъл е или стойността не е от тип string метода трябва да върне празен стринг и false.
+func (em *ExpireMap) GetString(key string) (v string, ok bool) {
+ v, ok = em.data[key].(string)
+ return
+}
+
+// Връща стойността на ключ и true, когато ключа е намерен и стойността му е от тип bool. Когато ключа липсва, изтекъл е или стойността не е от тип bool метода трябва да върне false и false.
+func (em *ExpireMap) GetBool(key string) (v bool, ok bool) {
+ v, ok = em.data[key].(bool)
+ return
+}
+
+// За определен ключ тази функция трябва да върне времето, когато той ще изтече. Втората върната стойност е true когато ключа е намерен в хранилището. Когато не е намерен функцията трябва да върне нулевото време и false.
+func (em *ExpireMap) Expires(key string) (v time.Time, ok bool) {
+ v, ok = em.expires[key]
+ return
+}
+
+// Премахва ключа от хранилището. Когато няма такъв ключ метода не трябав да гърми.
+func (em *ExpireMap) Delete(key string) {
+ em.lock()
+ defer em.unlock()
+ delete(em.data, key)
+ delete(em.expires, key)
+}
+
+// Казва дали ключ е в хранилището или не. Отново, ключове са в хранилището само ако техния expire не е вече изтекъл.
+func (em *ExpireMap) Contains(key string) (ok bool) {
+ _, ok = em.Get(key)
+ return
+}
+
+// Връща големината на хранилището. Големина е бройката неизтекли ключове, които са в него.
+func (em *ExpireMap) Size() int {
+ return len(em.data)
+}
+
+func (em *ExpireMap) addValueToIntOrString(key string, delta int) error {
+ em.lock()
+ defer em.unlock()
+ if !em.Contains(key) {
+ return nil
+ }
+ if v, ok := em.GetInt(key); ok {
+ em.data[key] = v + delta
+ } else if v, ok := em.GetString(key); ok {
+ intV, ok := strconv.Atoi(v)
+ if ok != nil {
+ return ok
+ }
+ em.data[key] = strconv.Itoa(intV + delta)
+ } else {
+ return &ExpireMapError{""}
+ }
+ return nil
+}
+
+// За ключ увеличава стойността му числово с 1 когато тази стойност е от тип int или стринг, представящ цяло число. Когато стойността е от тип, различен от тези неща функцията трябва да върне грешка, различна от nil. float32 и float64 не са целочислен типове. Времето на изтичане на ключа не трябва да се променя.
+func (em *ExpireMap) Increment(key string) error {
+ return em.addValueToIntOrString(key, 1)
+}
+
+// Същото като Increment, но намалява стойността с 1 вместо да я увеличава.
+func (em *ExpireMap) Decrement(key string) error {
+ return em.addValueToIntOrString(key, -1)
+}
+
+// Ако стойността за ключа key е от тип string, то всички букви в този string трябва да станат главни. При успех метода трябва да върне nil, а при невъзможност да изпълни задачата - грешка.
+func (em *ExpireMap) ToUpper(key string) (ok error) {
+ em.lock()
+ defer em.unlock()
+ if v, ok := em.GetString(key); ok {
+ em.data[key] = strings.ToUpper(v)
+ }
+ return
+}
+
+func (em *ExpireMap) ToLower(key string) (ok error) {
+ em.lock()
+ defer em.unlock()
+ if v, ok := em.GetString(key); ok {
+ em.data[key] = strings.ToLower(v)
+ }
+ return
+}
+
+// Тази функция трябва да върне канал, по който да идват ключовете, които са изтекли. Те трябва да се пускат по канала в момента на изтичането си. Ключове, които са изтрити с Delete или Cleanup не трябва да се връщат по този канал. Единствено тези, които са достигнали до времето си на изтичане. Не е задължително през цялото време някой да чете от върнатия канал. Ключове добавени в инстанцията след извикването на ExpireChan също трябва да се пратят по канала, когато тяхното време дойде.
+func (em *ExpireMap) ExpiredChan() <-chan string {
+ return em.channel
+}
+
+// Изчиства хранилището. След извикване на функцията в ExpireMap-a не трябва да се съдържа нито един ключ.
+func (em *ExpireMap) Cleanup() {
+ em.lock()
+ defer em.unlock()
+ for k := range em.data {
+ delete(em.data, k)
+ }
+ for k := range em.expires {
+ delete(em.expires, k)
+ }
+}
+
+// Трябва да освободи всички ресурси, които е използвал този ExpireMap. Това означава изчистване на всички структури, затваряне на всички канали и спиране на всички горутини, свързани с нормалната му работа. След извикване на Destroy тази инстанция на ExpireMap няма да бъде повече използвана. Този метод би трябвало да се ползва по подобен начин:
+func (em *ExpireMap) Destroy() {
+ em.lock()
+ em.data = nil
+ em.expires = nil
+ close(em.channel)
+ close(em.m)
+ // TODO: close goroutines
+}
Браво че си се сетил че може и никой да не чете на expiredchan-а
Забележки:
Ако връщаме грешка че даден ключ не може да се увеличи защото стойността му не е от правилния тип, дали трябва да връщаме грешка ако го няма ?
Пусни си мапа да работи по дълго предлагам следния код:
em.Set("foo", 5, time.Millisecond)
em.Destroy()
time.Sleep(time.Second)
// panic here
Мога още но се надявам че след като оправиш тези и махнеш
// TODO: close goroutines няма да има нужда :)