Vladimir Hodakov
b8226d8aa8
Recent game update changed pokememes view in pokedeks, so we need to reflect it by updating parser. Introducing DataCache - a silver bullet for eliminating lags linked to database queries. Less queries, more in RAM, faster work. Needs testing in production environment.
46 lines
1.1 KiB
Go
46 lines
1.1 KiB
Go
// i2_bot – Instinct PokememBro Bot
|
||
// Copyright (c) 2018 Vladimir "fat0troll" Hodakov
|
||
|
||
package datacache
|
||
|
||
import (
|
||
"errors"
|
||
"git.wtfteam.pro/fat0troll/i2_bot/lib/dbmapping"
|
||
"strconv"
|
||
)
|
||
|
||
func (dc *DataCache) initElements() {
|
||
c.Log.Info("Initializing Elements storage...")
|
||
dc.elements = make(map[int]*dbmapping.Element)
|
||
}
|
||
|
||
func (dc *DataCache) loadElements() {
|
||
c.Log.Info("Load current Elements data from database to DataCache...")
|
||
elements := []dbmapping.Element{}
|
||
err := c.Db.Select(&elements, "SELECT * FROM elements")
|
||
if err != nil {
|
||
// This is critical error and we need to stop immediately!
|
||
c.Log.Fatal(err.Error())
|
||
}
|
||
|
||
dc.elementsMutex.Lock()
|
||
for i := range elements {
|
||
dc.elements[elements[i].ID] = &elements[i]
|
||
}
|
||
c.Log.Info("Loaded elements in DataCache: " + strconv.Itoa(len(dc.elements)))
|
||
dc.elementsMutex.Unlock()
|
||
}
|
||
|
||
func (dc *DataCache) findElementIDBySymbol(symbol string) (int, error) {
|
||
dc.elementsMutex.Lock()
|
||
for i := range dc.elements {
|
||
if dc.elements[i].Symbol == symbol {
|
||
dc.elementsMutex.Unlock()
|
||
return i, nil
|
||
}
|
||
}
|
||
|
||
dc.elementsMutex.Unlock()
|
||
return 0, errors.New("There is no element with symbol = " + symbol)
|
||
}
|