Adds config option to crawl only specific shops. (crawler)

1 files changed, 11 insertions, 1 deletions

diff --git a/crawler/shops.go b/crawler/shops.go
index 1babc9d..be9bd53 100644
--- a/crawler/shops.go
+++ b/crawler/shops.go
@@ -1,6 +1,8 @@
 package main
 
 import (
+	"strings"
+
 	log "github.com/Sirupsen/logrus"
 )
 
@@ -96,7 +98,15 @@ func (app *App) getShops() ([]Shop, error) {
 
 	Shops := []Shop{}
 
-	query := `SELECT id,name,short_url,url,logo_url,shipping_costs,free_shipping FROM shop`
+	var shop_query string
+	if len(app.Config.ShopIDs) > 0 {
+		shopIDs := strings.Join(app.Config.ShopIDs, `", "`)
+		if shopIDs != "" {
+			shop_query = " WHERE id IN (" + shopIDs + ")"
+		}
+
+	}
+	query := `SELECT id,name,short_url,url,logo_url,shipping_costs,free_shipping FROM shop ` + shop_query
 
 	rows, err := app.DB.Queryx(query)
 	if err != nil {