kopia lustrzana https://github.com/cblgh/lieu
improve results from /random
rodzic
4ba6437f0a
commit
6df1a379e4
7
cli.go
7
cli.go
|
@ -90,6 +90,13 @@ func main() {
|
||||||
util.DatabaseDoesNotExist(config.Data.Database)
|
util.DatabaseDoesNotExist(config.Data.Database)
|
||||||
}
|
}
|
||||||
interactiveMode(config.Data.Database)
|
interactiveMode(config.Data.Database)
|
||||||
|
case "random":
|
||||||
|
exists := util.CheckFileExists(config.Data.Database)
|
||||||
|
if !exists {
|
||||||
|
util.DatabaseDoesNotExist(config.Data.Database)
|
||||||
|
}
|
||||||
|
db := database.InitDB(config.Data.Database)
|
||||||
|
fmt.Println(database.GetRandomPage(db))
|
||||||
case "host":
|
case "host":
|
||||||
exists := util.CheckFileExists(config.Data.Database)
|
exists := util.CheckFileExists(config.Data.Database)
|
||||||
if !exists {
|
if !exists {
|
||||||
|
|
|
@ -110,11 +110,28 @@ func GetWordCount(db *sql.DB) int {
|
||||||
return countQuery(db, "inv_index")
|
return countQuery(db, "inv_index")
|
||||||
}
|
}
|
||||||
|
|
||||||
func GetRandomPage(db *sql.DB) string {
|
func GetRandomDomain(db *sql.DB) string {
|
||||||
rows, err := db.Query("SELECT url FROM pages ORDER BY RANDOM() LIMIT 1;")
|
rows, err := db.Query("SELECT domain FROM domains ORDER BY RANDOM() LIMIT 1;")
|
||||||
util.Check(err)
|
util.Check(err)
|
||||||
defer rows.Close()
|
defer rows.Close()
|
||||||
|
|
||||||
|
var domain string
|
||||||
|
for rows.Next() {
|
||||||
|
err = rows.Scan(&domain)
|
||||||
|
util.Check(err)
|
||||||
|
}
|
||||||
|
return domain
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetRandomPage(db *sql.DB) string {
|
||||||
|
domain := GetRandomDomain(db)
|
||||||
|
stmt, err := db.Prepare("SELECT url FROM pages WHERE domain = ? ORDER BY RANDOM() LIMIT 1;")
|
||||||
|
defer stmt.Close()
|
||||||
|
util.Check(err)
|
||||||
|
|
||||||
|
rows, err := stmt.Query(domain)
|
||||||
|
defer rows.Close()
|
||||||
|
|
||||||
var link string
|
var link string
|
||||||
for rows.Next() {
|
for rows.Next() {
|
||||||
err = rows.Scan(&link)
|
err = rows.Scan(&link)
|
||||||
|
|
Ładowanie…
Reference in New Issue