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         if eid < 1 {
  52                 ch <- nil
  53                 close(ch)
  54                 return ch
  55         }
  56
  57         ic := make(chan Cacheable, 1)
  58         go func() { ch <- (<-ic).(*Episode); close(ch) }()
  59         if intentMap.Intent(ic, keys...) {
  60                 return ch
  61         }
  62
  63         if !cache.CheckValid(keys...) {
  64                 intentMap.NotifyClose((*Episode)(nil), keys...)
  65                 return ch
  66         }
  67
  68         e := eid.Episode()
  69         if !e.IsStale() {
  70                 intentMap.NotifyClose(e, keys...)
  71                 return ch
  72         }
  73
  74         go func() {
  75                 // The UDP API data is worse than the HTTP API anime data and
  76                 // might even get truncated on some pathological cases;
  77                 // try and get from the corresponding Anime, which uses the HTTP
  78                 // API episode list.
  79
  80                 aid := AID(0)
  81                 ok := cache.Get(&aid, "aid", "by-eid", eid) == nil
  82
  83                 udpDone := false
  84
  85                 for i := 0; i < 2; i++ {
  86                         if !ok && udpDone {
  87                                 // couldn't get anime and we already ran the EPISODE query
  88                                 break
  89                         }
  90
  91                         if !ok {
  92                                 // We don't know what the AID is yet.
  93                                 reply := <-adb.udp.SendRecv("EPISODE", paramMap{"eid": eid})
  94
  95                                 if reply.Error() == nil {
  96                                         parts := strings.Split(reply.Lines()[1], "|")
  97
  98                                         if id, err := strconv.ParseInt(parts[1], 10, 32); err == nil {
  99                                                 ok = true
 100                                                 aid = AID(id)
 101                                         } else {
 102                                                 break
 103                                         }
 104                                 } else if reply.Code() == 340 {
 105                                         cache.MarkInvalid(keys...)
 106                                         cache.Delete(keys...) // deleted EID?
 107                                         break
 108                                 } else {
 109                                         break
 110                                 }
 111                                 udpDone = true
 112                         }
 113                         a := <-adb.AnimeByID(AID(aid)) // updates the episode cache as well
 114                         ep := a.EpisodeByEID(eid)
 115
 116                         if ep != nil {
 117                                 e = ep
 118                                 break
 119                         } else {
 120                                 // the EID<->AID map broke
 121                                 ok = false
 122                                 cache.Delete("aid", "by-eid", eid)
 123                         }
 124                 }
 125                 intentMap.NotifyClose(e, keys...)
 126         }()
 127         return ch
 128 }