mirror of https://github.com/chubin/wttr.in
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
89 lines
1.6 KiB
89 lines
1.6 KiB
package ip
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"path/filepath"
|
|
|
|
"github.com/samonzeweb/godb"
|
|
"github.com/samonzeweb/godb/adapters/sqlite"
|
|
|
|
"github.com/chubin/wttr.in/internal/util"
|
|
)
|
|
|
|
//nolint:cyclop
|
|
func (c *Cache) ConvertCache() error {
|
|
dbfile := c.config.Geo.IPCacheDB
|
|
|
|
err := util.RemoveFileIfExists(dbfile)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
db, err := godb.Open(sqlite.Adapter, dbfile)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = createTable(db, "Address")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
log.Println("listing cache entries...")
|
|
files, err := filepath.Glob(filepath.Join(c.config.Geo.IPCache, "*"))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
log.Printf("going to convert %d entries\n", len(files))
|
|
|
|
block := []Address{}
|
|
for i, file := range files {
|
|
ip := filepath.Base(file)
|
|
loc, err := c.Read(ip)
|
|
if err != nil {
|
|
log.Println("invalid entry for", ip)
|
|
|
|
continue
|
|
}
|
|
|
|
block = append(block, *loc)
|
|
|
|
if i%1000 != 0 || i == 0 {
|
|
continue
|
|
}
|
|
|
|
err = db.BulkInsert(&block).Do()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
block = []Address{}
|
|
log.Println("converted", i+1, "entries")
|
|
}
|
|
|
|
// inserting the rest.
|
|
err = db.BulkInsert(&block).Do()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
log.Println("converted", len(files), "entries")
|
|
|
|
return nil
|
|
}
|
|
|
|
func createTable(db *godb.DB, tableName string) error {
|
|
createTable := fmt.Sprintf(
|
|
`create table %s (
|
|
name text not null primary key,
|
|
fullName text not null,
|
|
lat text not null,
|
|
long text not null);
|
|
`, tableName)
|
|
|
|
_, err := db.CurrentDB().Exec(createTable)
|
|
|
|
return err
|
|
}
|