go-pkg-rss/databaseHandler.go

64 lines
1.5 KiB
Go
Raw Normal View History

package feeder
import (
"fmt"
)
type databaseHandler struct {
db *database
itemhandler ItemHandler
chanhandler ChannelHandler
}
func (d *databaseHandler) ProcessItems(f *Feed, ch *Channel, items []*Item) {
fmt.Println("Processing items")
var newitems []*Item
for _, item := range items {
if d.db.request <- item.Key(); !<-d.db.response {
newitems = append(newitems, item)
}
}
if len(newitems) > 0 && d.itemhandler != nil {
d.itemhandler.ProcessItems(f, ch, newitems)
}
// No items to process, may as well end here
}
func (d *databaseHandler) ProcessChannels(f *Feed, ch []*Channel) {
fmt.Println("Processing channels")
var newchannels []*Channel
for _, channel := range ch {
if d.db.request <- channel.Key(); !<-d.db.response {
newchannels = append(newchannels, channel)
}
}
if len(newchannels) > 0 && d.chanhandler != nil {
d.chanhandler.ProcessChannels(f, newchannels)
}
// No channels to process, may as well end here
}
func NewDatabaseHandler(handler Handler) Handler {
database := new(databaseHandler)
database.db = NewDatabase()
database.itemhandler = handler
database.chanhandler = handler
return database
}
func NewDatabaseItemHandler(itemhandler ItemHandler) ItemHandler {
database := new(databaseHandler)
database.db = NewDatabase()
database.itemhandler = itemhandler
return database
}
func NewDatabaseChannelHandler(chanhandler ChannelHandler) ChannelHandler {
database := new(databaseHandler)
database.db = NewDatabase()
database.chanhandler = chanhandler
return database
}