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)
|
||||
}
|
||||
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":
|
||||
exists := util.CheckFileExists(config.Data.Database)
|
||||
if !exists {
|
||||
|
|
|
@ -110,11 +110,28 @@ func GetWordCount(db *sql.DB) int {
|
|||
return countQuery(db, "inv_index")
|
||||
}
|
||||
|
||||
func GetRandomPage(db *sql.DB) string {
|
||||
rows, err := db.Query("SELECT url FROM pages ORDER BY RANDOM() LIMIT 1;")
|
||||
func GetRandomDomain(db *sql.DB) string {
|
||||
rows, err := db.Query("SELECT domain FROM domains ORDER BY RANDOM() LIMIT 1;")
|
||||
util.Check(err)
|
||||
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
|
||||
for rows.Next() {
|
||||
err = rows.Scan(&link)
|
||||
|
|
Ładowanie…
Reference in New Issue