rewrite load

This commit is contained in:
w1ndyb0y
2020-12-23 14:23:55 +08:00
parent 33eab0a6c6
commit 7a21e09048
+17 -5
View File
@@ -94,10 +94,24 @@ func LoadUnprocessed() {
db := getDb()
defer db.Close()
for {
telegrams := getTelegram(db)
if len(telegrams) < 1 {
break
}
for _, t := range telegrams {
if WriteToClient(t.text) == nil {
processed(db, t)
}
}
}
}
func getTelegram(db *sql.DB) []telegram {
var telegrams []telegram
num := countTelegram()
log.Println("unprocessed : ", num)
if num < 1 {
break
return telegrams
}
log.Println("try to load unprocessed telegram ", num)
rows, err := db.Query(load)
@@ -107,11 +121,9 @@ func LoadUnprocessed() {
var t telegram
err := rows.Scan(&t.id, &t.text)
checkDbErr(err, "error in scan telegram")
if WriteToClient(t.text) == nil {
processed(db, t)
}
}
telegrams = append(telegrams, t)
}
return telegrams
}
func processed(db *sql.DB, t telegram) {