Add tournament reports storage and karma changing function

lib/datacache/datacacheinterface/datacacheinterface.go

@@ -34,6 +34,7 @@ type DataCacheInterface interface {
 	GetUserRolesInSquads(userID int) []dbmapping.SquadPlayerFull
 
 	AddPlayer(player *dbmapping.Player) (int, error)
+	ChangePlayerKarma(addition int, playerID int) (*dbmapping.Player, error)
 	GetOrCreatePlayerByTelegramID(telegramID int) (*dbmapping.Player, error)
 	GetPlayerByID(playerID int) (*dbmapping.Player, error)
 	GetPlayerByTelegramID(telegramID int) (*dbmapping.Player, error)
@@ -45,6 +46,8 @@ type DataCacheInterface interface {
 	GetProfileByID(profileID int) (*dbmapping.Profile, error)
 	GetProfileByPlayerID(playerID int) (*dbmapping.Profile, error)
 
+	AddTournamentReport(tournamentNumber int, playerID int, target string) (int, error)
+
 	GetElementByID(elementID int) (*datamapping.Element, error)
 	FindElementIDBySymbol(symbol string) (int, error)
 

lib/datacache/exported.go

@@ -29,6 +29,10 @@ type DataCache struct {
 	// Note: int in this array for player ID, not for profile ID
 	currentProfiles      map[int]*dbmapping.Profile
 	currentProfilesMutex sync.Mutex
+	// Tournament reports
+	tournamentReports                      map[int]*dbmapping.TournamentReport
+	tournamentReportsByTournamentAndPlayer map[int]map[int]*dbmapping.TournamentReport
+	tournamentReportsMutex                 sync.Mutex
 
 	// Chats
 	chats      map[int]*dbmapping.Chat
@@ -84,6 +88,8 @@ func (dc *DataCache) Init() {
 	dc.loadPlayers()
 	dc.initProfiles()
 	dc.loadProfiles()
+	dc.initTournamentReports()
+	dc.loadTournamentReports()
 	dc.initChats()
 	dc.loadChats()
 	dc.initSquads()

lib/datacache/players.go

@@ -58,6 +58,25 @@ func (dc *DataCache) AddPlayer(player *dbmapping.Player) (int, error) {
 	return insertedPlayer.ID, nil
 }
 
+// ChangePlayerKarma changes karma value for user
+func (dc *DataCache) ChangePlayerKarma(addition int, playerID int) (*dbmapping.Player, error) {
+	if dc.players[playerID] == nil {
+		return nil, errors.New("There is no player with ID = " + strconv.Itoa(playerID))
+	}
+
+	dc.playersMutex.Lock()
+	dc.players[playerID].Karma += addition
+	dc.players[playerID].UpdatedAt = time.Now().UTC()
+	dc.playersMutex.Unlock()
+
+	_, err := c.Db.NamedExec("UPDATE `players` SET karma=:karma, updated_at=:updated_at WHERE id=:id", dc.players[playerID])
+	if err != nil {
+		c.Log.Error(err.Error())
+	}
+
+	return dc.players[playerID], nil
+}
+
 // GetOrCreatePlayerByTelegramID finds user by Telegram ID and creates one if not exist
 func (dc *DataCache) GetOrCreatePlayerByTelegramID(telegramID int) (*dbmapping.Player, error) {
 	c.Log.Info("DataCache: Getting player with Telegram ID=", telegramID)

lib/datacache/tournament_reports.go

@@ -0,0 +1,95 @@
+// i2_bot – Instinct PokememBro Bot
+// Copyright (c) 2018 Vladimir "fat0troll" Hodakov
+
+package datacache
+
+import (
+	"errors"
+	"strconv"
+	"time"
+
+	"source.wtfteam.pro/i2_bot/i2_bot/lib/dbmapping"
+)
+
+func (dc *DataCache) initTournamentReports() {
+	c.Log.Info("Initializing Tournament Reports storage...")
+	dc.tournamentReports = make(map[int]*dbmapping.TournamentReport)
+	dc.tournamentReportsByTournamentAndPlayer = make(map[int]map[int]*dbmapping.TournamentReport)
+}
+
+func (dc *DataCache) loadTournamentReports() {
+	c.Log.Info("Load current TournamentReports data from database to DataCache...")
+	tournamentReports := []dbmapping.TournamentReport{}
+	err := c.Db.Select(&tournamentReports, "SELECT * FROM tournament_reports")
+	if err != nil {
+		// This is critical error and we need to stop immediately!
+		c.Log.Fatal(err.Error())
+	}
+
+	dc.tournamentReportsMutex.Lock()
+	for i := range tournamentReports {
+		dc.tournamentReports[tournamentReports[i].ID] = &tournamentReports[i]
+		if dc.tournamentReportsByTournamentAndPlayer[tournamentReports[i].TournamentNumber] == nil {
+			dc.tournamentReportsByTournamentAndPlayer[tournamentReports[i].TournamentNumber] = make(map[int]*dbmapping.TournamentReport)
+		}
+
+		dc.tournamentReportsByTournamentAndPlayer[tournamentReports[i].TournamentNumber][tournamentReports[i].PlayerID] = &tournamentReports[i]
+	}
+	c.Log.Info("Loaded tournament reports in DataCache: " + strconv.Itoa(len(dc.tournamentReports)))
+	dc.tournamentReportsMutex.Unlock()
+}
+
+// External functions
+
+// AddTournamentReport creates new tournament report in database
+func (dc *DataCache) AddTournamentReport(tournamentNumber int, playerID int, target string) (int, error) {
+	if tournamentNumber < 3551 {
+		return 0, errors.New("Can't save tournament from the very past")
+	}
+	if dc.players[playerID] == nil {
+		return 0, errors.New("There is no player with ID = " + strconv.Itoa(playerID))
+	}
+	if target != "O" && target != "M" && target != "A" {
+		return 0, errors.New("Malformed report")
+	}
+
+	dc.tournamentReportsMutex.Lock()
+	if dc.tournamentReportsByTournamentAndPlayer[tournamentNumber] != nil {
+		if dc.tournamentReportsByTournamentAndPlayer[tournamentNumber][playerID] != nil {
+			dc.tournamentReportsMutex.Unlock()
+			return dc.tournamentReportsByTournamentAndPlayer[tournamentNumber][playerID].ID, errors.New("There is already this report in database")
+		}
+	} else {
+		dc.tournamentReportsByTournamentAndPlayer[tournamentNumber] = make(map[int]*dbmapping.TournamentReport)
+	}
+	dc.tournamentReportsMutex.Unlock()
+
+	report := dbmapping.TournamentReport{}
+	report.PlayerID = playerID
+	report.TournamentNumber = tournamentNumber
+	report.Target = target
+	report.CreatedAt = time.Now().UTC()
+
+	c.Log.Debug("Saving report...")
+	_, err := c.Db.NamedExec("INSERT INTO tournament_reports VALUES(NULL, :player_id, :tournament_number, :target, :created_at)", &tournamentNumber)
+	if err != nil {
+		c.Log.Error(err.Error())
+		return 0, err
+	}
+
+	newReport := dbmapping.TournamentReport{}
+	err = c.Db.Get(&newReport, "SELECT * FROM tournament_reports WHERE player_id=? AND tournament_number=?", report.PlayerID, report.TournamentNumber)
+	if err != nil {
+		c.Log.Error(err.Error())
+		return 0, err
+	}
+
+	dc.tournamentReportsMutex.Lock()
+	dc.tournamentReports[newReport.ID] = &newReport
+	dc.tournamentReportsByTournamentAndPlayer[tournamentNumber][playerID] = &newReport
+	dc.tournamentReportsMutex.Unlock()
+
+	dc.ChangePlayerKarma(3, newReport.PlayerID)
+
+	return newReport.ID, nil
+}