parent
730cf1e15a
commit
be065c5dcb
|
|
@ -0,0 +1,22 @@
|
||||||
|
package config
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"encoding/gob"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Encodable struct{}
|
||||||
|
|
||||||
|
func (e *Encodable) Encode() ([]byte, error) {
|
||||||
|
buff := bytes.NewBuffer(nil)
|
||||||
|
enc := gob.NewEncoder(buff)
|
||||||
|
err := enc.Encode(e)
|
||||||
|
return buff.Bytes(), err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (e *Encodable) Decode(b []byte) error {
|
||||||
|
buff := bytes.NewBuffer(b)
|
||||||
|
dec := gob.NewDecoder(buff)
|
||||||
|
err := dec.Decode(e)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
@ -34,6 +34,7 @@ func TestNew(t *testing.T) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
|
os.Args = []string{"a", "-db", "map"}
|
||||||
if err := New(); err != nil {
|
if err := New(); err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -0,0 +1,7 @@
|
||||||
|
package config
|
||||||
|
|
||||||
|
type Stoppable struct{}
|
||||||
|
|
||||||
|
func (s Stoppable) Stopped() <-chan struct{} {
|
||||||
|
return Values().Ctx.Done()
|
||||||
|
}
|
||||||
9
main.go
9
main.go
|
|
@ -1,8 +1,8 @@
|
||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"local/rssmon2/monitor"
|
|
||||||
"local/rssmon3/config"
|
"local/rssmon3/config"
|
||||||
|
"local/rssmon3/monitor"
|
||||||
"local/rssmon3/server"
|
"local/rssmon3/server"
|
||||||
"log"
|
"log"
|
||||||
"os"
|
"os"
|
||||||
|
|
@ -27,10 +27,7 @@ func main() {
|
||||||
}
|
}
|
||||||
go InterruptAfter(s.Run, sigc)
|
go InterruptAfter(s.Run, sigc)
|
||||||
|
|
||||||
m, err := monitor.New()
|
m := monitor.New()
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
go InterruptAfter(m.Run, sigc)
|
go InterruptAfter(m.Run, sigc)
|
||||||
|
|
||||||
signal.Notify(sigc,
|
signal.Notify(sigc,
|
||||||
|
|
@ -40,7 +37,7 @@ func main() {
|
||||||
syscall.SIGQUIT,
|
syscall.SIGQUIT,
|
||||||
)
|
)
|
||||||
<-sigc
|
<-sigc
|
||||||
can()
|
config.Values().Can()
|
||||||
config.Values().DB.Close()
|
config.Values().DB.Close()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -12,6 +12,7 @@ import (
|
||||||
|
|
||||||
type Item struct {
|
type Item struct {
|
||||||
Key string
|
Key string
|
||||||
|
*config.Encodable
|
||||||
}
|
}
|
||||||
|
|
||||||
const nsLast = "nsLast"
|
const nsLast = "nsLast"
|
||||||
|
|
@ -22,7 +23,8 @@ var forever = time.Duration(time.Hour * 99999)
|
||||||
|
|
||||||
func NewItem(key string, interval time.Duration) (*Item, error) {
|
func NewItem(key string, interval time.Duration) (*Item, error) {
|
||||||
i := &Item{
|
i := &Item{
|
||||||
Key: key,
|
Key: key,
|
||||||
|
Encodable: &config.Encodable{},
|
||||||
}
|
}
|
||||||
|
|
||||||
if err := i.setInterval(interval); err != nil {
|
if err := i.setInterval(interval); err != nil {
|
||||||
|
|
|
||||||
|
|
@ -9,6 +9,7 @@ type Monitor struct {
|
||||||
queue *Queue
|
queue *Queue
|
||||||
Incoming chan *Item
|
Incoming chan *Item
|
||||||
Outgoing chan *Item
|
Outgoing chan *Item
|
||||||
|
config.Stoppable
|
||||||
}
|
}
|
||||||
|
|
||||||
func New() *Monitor {
|
func New() *Monitor {
|
||||||
|
|
@ -22,7 +23,7 @@ func New() *Monitor {
|
||||||
func (m *Monitor) Run() error {
|
func (m *Monitor) Run() error {
|
||||||
for {
|
for {
|
||||||
select {
|
select {
|
||||||
case <-m.stopped():
|
case <-m.Stopped():
|
||||||
return nil
|
return nil
|
||||||
case i := <-m.enqueued():
|
case i := <-m.enqueued():
|
||||||
m.enqueue(i)
|
m.enqueue(i)
|
||||||
|
|
@ -33,10 +34,6 @@ func (m *Monitor) Run() error {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m *Monitor) stopped() <-chan struct{} {
|
|
||||||
return config.Values().Ctx.Done()
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *Monitor) enqueued() chan *Item {
|
func (m *Monitor) enqueued() chan *Item {
|
||||||
return m.Incoming
|
return m.Incoming
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -0,0 +1,110 @@
|
||||||
|
package rss
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"local/rssmon3/config"
|
||||||
|
"log"
|
||||||
|
"net/http"
|
||||||
|
"regexp"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/mmcdole/gofeed"
|
||||||
|
)
|
||||||
|
|
||||||
|
const NSFeeds = "NSFeeds"
|
||||||
|
|
||||||
|
type feed struct {
|
||||||
|
Key string
|
||||||
|
URL string
|
||||||
|
Updated time.Time
|
||||||
|
TitleFilter string
|
||||||
|
ContentFilter string
|
||||||
|
Tags []string
|
||||||
|
*config.Encodable
|
||||||
|
}
|
||||||
|
|
||||||
|
func newFeed(key string) *feed {
|
||||||
|
return &feed{
|
||||||
|
Key: key,
|
||||||
|
Encodable: &config.Encodable{},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *feed) load() error {
|
||||||
|
if f.Key == "" {
|
||||||
|
return errors.New("cannot load nil feed")
|
||||||
|
}
|
||||||
|
db := config.Values().DB
|
||||||
|
b, err := db.Get(f.Key, NSFeeds)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return f.Decode(b)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *feed) pull() error {
|
||||||
|
if f.URL == "" {
|
||||||
|
if err := f.load(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
resp, err := http.Get(f.URL)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
|
|
||||||
|
gofeed, err := gofeed.NewParser().Parse(resp.Body)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
itemTSs := []*time.Time{}
|
||||||
|
for _, i := range gofeed.Items {
|
||||||
|
ts := latestTSPtr(i.UpdatedParsed, i.PublishedParsed)
|
||||||
|
itemTSs = append(itemTSs, &ts)
|
||||||
|
if ts.Before(f.Updated) {
|
||||||
|
log.Println("Skipping old item")
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if ok := regexp.MustCompile(f.TitleFilter).MatchString(i.Title); !ok {
|
||||||
|
log.Println("Skipping bad titled item")
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
item, err := newItem(i, f.ContentFilter)
|
||||||
|
if err != nil {
|
||||||
|
log.Println(err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if err := item.save(); err != nil {
|
||||||
|
log.Println(err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
f.Updated = latestTSPtr(gofeed.PublishedParsed, gofeed.UpdatedParsed)
|
||||||
|
f.Updated = latestTSPtr(append(itemTSs, &f.Updated)...)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *feed) save() error {
|
||||||
|
b, err := f.Encode()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
db := config.Values().DB
|
||||||
|
return db.Set(f.Key, b, NSFeeds)
|
||||||
|
}
|
||||||
|
|
||||||
|
func latestTSPtr(times ...*time.Time) time.Time {
|
||||||
|
var t time.Time
|
||||||
|
for i := range times {
|
||||||
|
if times[i] == nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if times[i].After(t) {
|
||||||
|
t = *times[i]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return t
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,44 @@
|
||||||
|
package rss
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"local/rssmon3/config"
|
||||||
|
"os"
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
func initRSSFeed() {
|
||||||
|
os.Args = []string{"a", "-db", "map"}
|
||||||
|
if err := config.New(); err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestRSSFeedEncodeDecode(t *testing.T) {
|
||||||
|
initRSSFeed()
|
||||||
|
|
||||||
|
f := newFeed("key")
|
||||||
|
f.Updated = time.Now()
|
||||||
|
f.TitleFilter = "a"
|
||||||
|
f.ContentFilter = "b"
|
||||||
|
f.Tags = []string{"c"}
|
||||||
|
|
||||||
|
b, err := f.Encode()
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
g := newFeed("key")
|
||||||
|
if err := g.Decode(b); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
fb, _ := json.Marshal(f)
|
||||||
|
gb, _ := json.Marshal(f)
|
||||||
|
if string(fb) != string(gb) {
|
||||||
|
t.Errorf("%v => %v", *f, *g)
|
||||||
|
}
|
||||||
|
|
||||||
|
t.Logf("%s vs. %s", fb, gb)
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,22 @@
|
||||||
|
package rss
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"local/rssmon3/config"
|
||||||
|
|
||||||
|
"github.com/mmcdole/gofeed"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Item struct {
|
||||||
|
*config.Encodable
|
||||||
|
}
|
||||||
|
|
||||||
|
func newItem(i *gofeed.Item, contentFilter string) (*Item, error) {
|
||||||
|
return &Item{
|
||||||
|
Encodable: &config.Encodable{},
|
||||||
|
}, errors.New("not impl")
|
||||||
|
}
|
||||||
|
|
||||||
|
func (i *Item) save() error {
|
||||||
|
return errors.New("not impl")
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,44 @@
|
||||||
|
package rss
|
||||||
|
|
||||||
|
import (
|
||||||
|
"local/rssmon3/config"
|
||||||
|
"local/rssmon3/monitor"
|
||||||
|
"log"
|
||||||
|
)
|
||||||
|
|
||||||
|
const nsFeeds = "nsFeeds"
|
||||||
|
|
||||||
|
type RSS struct {
|
||||||
|
items chan *monitor.Item
|
||||||
|
config.Stoppable
|
||||||
|
}
|
||||||
|
|
||||||
|
func New(items chan *monitor.Item) *RSS {
|
||||||
|
return &RSS{
|
||||||
|
items: items,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (rss *RSS) Run() error {
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case <-rss.Stopped():
|
||||||
|
return nil
|
||||||
|
case i := <-rss.items:
|
||||||
|
if err := rss.update(i); err != nil {
|
||||||
|
log.Println(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (rss *RSS) update(item *monitor.Item) error {
|
||||||
|
f := newFeed(item.Key)
|
||||||
|
if err := f.pull(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if err := f.save(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1 @@
|
||||||
|
bf5140d93694bf40f561841cdee34065d4cc2455
|
||||||
|
|
@ -2,13 +2,11 @@ package server
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
"encoding/json"
|
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
"local/router"
|
"local/router"
|
||||||
"local/rssmon3/config"
|
"local/rssmon3/config"
|
||||||
"local/sandbox/selenium/copart/copart/auction"
|
|
||||||
"local/storage"
|
"local/storage"
|
||||||
"log"
|
"log"
|
||||||
"net/http"
|
"net/http"
|
||||||
|
|
@ -68,11 +66,7 @@ func (s *Server) auctions(w http.ResponseWriter, r *http.Request) {
|
||||||
r.ParseForm()
|
r.ParseForm()
|
||||||
switch strings.ToLower(r.Method) {
|
switch strings.ToLower(r.Method) {
|
||||||
case "get":
|
case "get":
|
||||||
if len(r.Form) > 0 {
|
foo = s.get
|
||||||
foo = s.query
|
|
||||||
} else {
|
|
||||||
foo = s.get
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
foo(w, r)
|
foo(w, r)
|
||||||
}
|
}
|
||||||
|
|
@ -96,70 +90,3 @@ func (s *Server) get(w http.ResponseWriter, r *http.Request) {
|
||||||
}
|
}
|
||||||
io.Copy(w, bytes.NewBuffer(value))
|
io.Copy(w, bytes.NewBuffer(value))
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *Server) query(w http.ResponseWriter, r *http.Request) {
|
|
||||||
log.Println("QUERY", r.URL.Path)
|
|
||||||
list, err := s.list()
|
|
||||||
if err != nil {
|
|
||||||
s.error(w, r, err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
form := r.Form
|
|
||||||
for _, value := range form {
|
|
||||||
for _, chunk := range value {
|
|
||||||
iteration := []string{}
|
|
||||||
for _, name := range list {
|
|
||||||
if strings.Contains(strings.ToLower(name), strings.ToLower(chunk)) {
|
|
||||||
iteration = append(iteration, name)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
list = iteration
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if len(list) < 1 {
|
|
||||||
s.notFound(w, r)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
carsJSON, err := s.gatherCarsTruncatedJSON(list)
|
|
||||||
if err != nil {
|
|
||||||
s.error(w, r, fmt.Errorf("gather cars(%v): %v", list, err))
|
|
||||||
return
|
|
||||||
}
|
|
||||||
w.Write(carsJSON)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *Server) list() ([]string, error) {
|
|
||||||
db := config.Values().DB
|
|
||||||
list := []string{}
|
|
||||||
if b, err := db.Get("LIST"); err == storage.ErrNotFound {
|
|
||||||
list = []string{}
|
|
||||||
db.Set("LIST", []byte("[]"))
|
|
||||||
} else if err != nil {
|
|
||||||
return nil, fmt.Errorf("bad list in storage: %v", err)
|
|
||||||
} else if err := json.Unmarshal(b, &list); err != nil {
|
|
||||||
return nil, fmt.Errorf("bad list in storage: %v", err)
|
|
||||||
}
|
|
||||||
return list, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *Server) gatherCarsTruncatedJSON(keys []string) ([]byte, error) {
|
|
||||||
cars := []*auction.Car{}
|
|
||||||
k := 0
|
|
||||||
for _, key := range keys {
|
|
||||||
k += 1
|
|
||||||
if k > 50 {
|
|
||||||
cars = append(cars, &auction.Car{Title: "Max results reached, please search more specifically"})
|
|
||||||
break
|
|
||||||
}
|
|
||||||
b, err := config.Values().DB.Get(key)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
c := auction.NewCar()
|
|
||||||
if err := c.Decode(b); err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
cars = append(cars, c)
|
|
||||||
}
|
|
||||||
return json.Marshal(cars)
|
|
||||||
}
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue