Removed the IsDebug* crap.
This commit is contained in:
parent
014b42cbb3
commit
e526cff3e7
@ -57,7 +57,7 @@ func poll_players(host string, timeout int) (int, error) {
|
||||
return int(players), nil
|
||||
}
|
||||
|
||||
func PollServers(servers []ServerConfig, timeout int) ([]*RawServerData, error) {
|
||||
func (i *Instance) PollServers(servers []ServerConfig, timeout int) ([]*RawServerData, error) {
|
||||
var wg sync.WaitGroup
|
||||
var tmp []*RawServerData
|
||||
for _, s := range servers {
|
||||
|
||||
@ -19,8 +19,8 @@ var (
|
||||
RE_PLAYERS = regexp.MustCompile(`Logged in: (\d+) player`)
|
||||
)
|
||||
|
||||
func ScrapePage() ([]*RawServerData, error) {
|
||||
data, err := download_data()
|
||||
func (i *Instance) ScrapePage() ([]*RawServerData, error) {
|
||||
data, err := download_data(i.Debug)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
@ -32,10 +32,9 @@ func ScrapePage() ([]*RawServerData, error) {
|
||||
return tmp, nil
|
||||
}
|
||||
|
||||
func download_data() (*goquery.Document, error) {
|
||||
func download_data(debug bool) (*goquery.Document, error) {
|
||||
var r io.Reader
|
||||
if IsDebugging() {
|
||||
fmt.Println("Scraper data source: ./dump.html")
|
||||
if debug {
|
||||
f, err := os.Open("./tmp/dump.html")
|
||||
if err != nil {
|
||||
return nil, err
|
||||
|
||||
@ -27,7 +27,7 @@ func (i *Instance) UpdateServers() {
|
||||
if i.Debug {
|
||||
fmt.Println("\nPolling servers...")
|
||||
}
|
||||
polled, err := PollServers(config.PollServers, config.Timeout)
|
||||
polled, err := i.PollServers(config.PollServers, config.Timeout)
|
||||
if !log_error(err) {
|
||||
for _, s := range polled {
|
||||
i.update_server(tx, s)
|
||||
@ -38,7 +38,7 @@ func (i *Instance) UpdateServers() {
|
||||
if i.Debug {
|
||||
fmt.Println("\nScraping servers...")
|
||||
}
|
||||
scraped, err := ScrapePage()
|
||||
scraped, err := i.ScrapePage()
|
||||
if !log_error(err) {
|
||||
for _, s := range scraped {
|
||||
i.update_server(tx, s)
|
||||
@ -99,7 +99,7 @@ func (i *Instance) update_server(tx *DB, s *RawServerData) {
|
||||
return
|
||||
}
|
||||
|
||||
if IsDebugging() {
|
||||
if i.Debug {
|
||||
fmt.Println(s.Title)
|
||||
}
|
||||
|
||||
|
||||
@ -7,7 +7,6 @@ import (
|
||||
|
||||
var (
|
||||
now = time.Now()
|
||||
debugging bool = false
|
||||
)
|
||||
|
||||
func log_error(err error) bool {
|
||||
@ -24,14 +23,6 @@ func check_error(err error) {
|
||||
}
|
||||
}
|
||||
|
||||
func SetDebug(val bool) {
|
||||
debugging = val
|
||||
}
|
||||
|
||||
func IsDebugging() bool {
|
||||
return debugging
|
||||
}
|
||||
|
||||
func ResetNow() {
|
||||
now = time.Now()
|
||||
}
|
||||
|
||||
@ -11,7 +11,6 @@ import (
|
||||
|
||||
func (i *Instance) Init() {
|
||||
i.DB.InitSchema()
|
||||
SetDebug(i.Debug) // TODO: get rid of this stupid debug thing
|
||||
}
|
||||
|
||||
func (i *Instance) Serve(addr string) error {
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user