hdkv/i2_bot
Archived
1
Fork 0
Code Releases Activity

Fix for non-finding weapons

Browse Source
This commit is contained in:
Vladimir Hodakov 2018-01-30 18:41:23 +04:00
parent 78c882347d
commit 7bcaca66aa
2 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
lib/datacache/weapons.go
View File

@ -7,6 +7,7 @@ import (
"errors" "errors"
"git.wtfteam.pro/fat0troll/i2_bot/lib/dbmapping" "git.wtfteam.pro/fat0troll/i2_bot/lib/dbmapping"
"strconv" "strconv"
"strings"
) )
func (dc *DataCache) initWeapons() { func (dc *DataCache) initWeapons() {
@ -37,7 +38,7 @@ func (dc *DataCache) loadWeapons() {
func (dc *DataCache) GetWeaponTypeByName(name string) (*dbmapping.Weapon, error) { func (dc *DataCache) GetWeaponTypeByName(name string) (*dbmapping.Weapon, error) {
dc.weaponsMutex.Lock() dc.weaponsMutex.Lock()
for i := range dc.weapons { for i := range dc.weapons {
if dc.weapons[i].Name == name { if strings.HasPrefix(dc.weapons[i].Name, name) {
dc.weaponsMutex.Unlock() dc.weaponsMutex.Unlock()
return dc.weapons[i], nil return dc.weapons[i], nil
} }

6
lib/users/parsers.go
View File

@ -247,7 +247,11 @@ func (u *Users) ParseProfile(update *tgbotapi.Update, playerRaw *dbmapping.Playe
profileRaw.Exp = expInt profileRaw.Exp = expInt
profileRaw.EggExp = eggexpInt profileRaw.EggExp = eggexpInt
profileRaw.Power = powerInt profileRaw.Power = powerInt
profileRaw.WeaponID = weaponRaw.ID if weaponRaw != nil {
profileRaw.WeaponID = weaponRaw.ID
} else {
profileRaw.WeaponID = 0
}
profileRaw.Crystalls = crystallsInt profileRaw.Crystalls = crystallsInt
profileRaw.CreatedAt = time.Now().UTC() profileRaw.CreatedAt = time.Now().UTC()