episodecache.go

   1 package anidb
   2
   3 import (
   4         "encoding/gob"
   5         "strconv"
   6         "strings"
   7         "time"
   8 )
   9
  10 func init() {
  11         gob.RegisterName("*github.com/Kovensky/go-anidb.Episode", &Episode{})
  12 }
  13
  14 func (e *Episode) Touch() {
  15         e.Cached = time.Now()
  16 }
  17
  18 func (e *Episode) IsStale() bool {
  19         if e == nil {
  20                 return true
  21         }
  22         return time.Now().Sub(e.Cached) > EpisodeCacheDuration
  23 }
  24
  25 // Unique Episode IDentifier.
  26 type EID int
  27
  28 // Retrieves the Episode corresponding to this EID from the cache.
  29 func (eid EID) Episode() *Episode {
  30         var e Episode
  31         if cache.Get(&e, "eid", eid) == nil {
  32                 return &e
  33         }
  34         return nil
  35 }
  36
  37 func cacheEpisode(ep *Episode) {
  38         cache.Set(ep.AID, "aid", "by-eid", ep.EID)
  39         cache.Set(ep, "eid", ep.EID)
  40 }
  41
  42 // Retrieves an Episode by its EID.
  43 //
  44 // If we know which AID owns this EID, then it's equivalent
  45 // to an Anime query. Otherwise, uses both the HTTP and UDP
  46 // APIs to retrieve it.
  47 func (adb *AniDB) EpisodeByID(eid EID) <-chan *Episode {
  48         keys := []cacheKey{"eid", eid}
  49         ch := make(chan *Episode, 1)
  50
  51         ic := make(chan Cacheable, 1)
  52         go func() { ch <- (<-ic).(*Episode); close(ch) }()
  53         if intentMap.Intent(ic, keys...) {
  54                 return ch
  55         }
  56
  57         if !cache.CheckValid(keys...) {
  58                 intentMap.Notify((*Episode)(nil), keys...)
  59                 return ch
  60         }
  61
  62         e := eid.Episode()
  63         if !e.IsStale() {
  64                 intentMap.Notify(e, keys...)
  65                 return ch
  66         }
  67
  68         go func() {
  69                 // The UDP API data is worse than the HTTP API anime data,
  70                 // try and get from the corresponding Anime
  71
  72                 aid := AID(0)
  73                 ok := cache.Get(&aid, "aid", "by-eid", eid) == nil
  74
  75                 udpDone := false
  76
  77                 for i := 0; i < 2; i++ {
  78                         if !ok && udpDone {
  79                                 // couldn't get anime and we already ran the EPISODE query
  80                                 break
  81                         }
  82
  83                         if !ok {
  84                                 // We don't know what the AID is yet.
  85                                 reply := <-adb.udp.SendRecv("EPISODE", paramMap{"eid": eid})
  86
  87                                 if reply.Error() == nil {
  88                                         parts := strings.Split(reply.Lines()[1], "|")
  89
  90                                         if id, err := strconv.ParseInt(parts[1], 10, 32); err == nil {
  91                                                 ok = true
  92                                                 aid = AID(id)
  93                                         }
  94                                 } else if reply.Code() == 340 {
  95                                         cache.MarkInvalid(keys...)
  96                                         cache.Delete(keys...) // deleted EID?
  97                                 } else {
  98                                         break
  99                                 }
 100                                 udpDone = true
 101                         }
 102                         a := <-adb.AnimeByID(AID(aid)) // this caches episodes...
 103                         ep := eid.Episode()            // ...so this is now a cache hit
 104
 105                         if !ep.IsStale() {
 106                                 e = ep
 107                                 break
 108                         } else {
 109                                 ok = false
 110
 111                                 // check to see if we looked in the right AID
 112                                 found := false
 113                                 if a != nil {
 114                                         for _, ep := range a.Episodes {
 115                                                 if eid == ep.EID {
 116                                                         found = true
 117                                                         break
 118                                                 }
 119                                         }
 120                                 }
 121
 122                                 // if found, then it's just that the anime is also stale (offline?)
 123                                 if found {
 124                                         break
 125                                 } else {
 126                                         // otherwise, the EID<->AID map broke
 127                                         cache.Delete("aid", "by-eid", eid)
 128                                 }
 129                         }
 130                 }
 131                 intentMap.Notify(e, keys...)
 132         }()
 133         return ch
 134 }