Save if view was made by bot
This commit is contained in:
parent
023aa22096
commit
f59b999899
11
database.go
11
database.go
|
@ -32,7 +32,7 @@ func initDatabase() (e error) {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
e = migrateDatabase(db.sqlDB)
|
e = migrateDatabase(db.sqlDB)
|
||||||
db.trackingStmt, e = db.sqlDB.Prepare("insert into views(url, ref, useragent) values(:url, :ref, :ua)")
|
db.trackingStmt, e = db.sqlDB.Prepare("insert into views(url, ref, useragent, bot) values(:url, :ref, :ua, :bot)")
|
||||||
if e != nil {
|
if e != nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
@ -59,12 +59,17 @@ func trackView(urlString string, ref string, ua string) {
|
||||||
parsedRef, _ := url.Parse(ref)
|
parsedRef, _ := url.Parse(ref)
|
||||||
ref = parsedRef.Hostname()
|
ref = parsedRef.Hostname()
|
||||||
}
|
}
|
||||||
|
bot := 0
|
||||||
if ua != "" {
|
if ua != "" {
|
||||||
// Parse Useragent
|
// Parse Useragent
|
||||||
uaName, uaVersion := user_agent.New(ua).Browser()
|
userAgent := user_agent.New(ua)
|
||||||
|
if userAgent.Bot() {
|
||||||
|
bot = 1
|
||||||
|
}
|
||||||
|
uaName, uaVersion := userAgent.Browser()
|
||||||
ua = uaName + " " + uaVersion
|
ua = uaName + " " + uaVersion
|
||||||
}
|
}
|
||||||
_, e := db.trackingStmt.Exec(sql.Named("url", urlString), sql.Named("ref", ref), sql.Named("ua", ua))
|
_, e := db.trackingStmt.Exec(sql.Named("url", urlString), sql.Named("ref", ref), sql.Named("ua", ua), sql.Named("bot", bot))
|
||||||
if e != nil {
|
if e != nil {
|
||||||
fmt.Println("Inserting into DB failed:", e)
|
fmt.Println("Inserting into DB failed:", e)
|
||||||
}
|
}
|
||||||
|
|
|
@ -0,0 +1,3 @@
|
||||||
|
-- +migrate Up
|
||||||
|
ALTER TABLE views
|
||||||
|
ADD COLUMN bot INTEGER DEFAULT 0 NOT NULL;
|
Reference in New Issue