|
| 1 | +import { CommandHandler, EventBus, ICommandHandler } from '@nestjs/cqrs'; |
| 2 | +import { Logger } from '@nestjs/common'; |
| 3 | +import { SaveGameResultsCommand } from 'gameserver/command/SaveGameResults/save-game-results.command'; |
| 4 | +import { MatchmakingMode } from 'gateway/shared-types/matchmaking-mode'; |
| 5 | +import FinishedMatchEntity from 'gameserver/model/finished-match.entity'; |
| 6 | +import PlayerInMatchEntity from 'gameserver/model/player-in-match.entity'; |
| 7 | +import { MatchRecordedEvent } from 'gameserver/event/match-recorded.event'; |
| 8 | +import { GameSessionFinishedEvent } from 'gateway/events/game-session-finished.event'; |
| 9 | +import { InjectRepository } from '@nestjs/typeorm'; |
| 10 | +import { DataSource, Repository } from 'typeorm'; |
| 11 | +import { GameServerSessionEntity } from 'gameserver/model/game-server-session.entity'; |
| 12 | + |
| 13 | +@CommandHandler(SaveGameResultsCommand) |
| 14 | +export class SaveGameResultsHandler |
| 15 | + implements ICommandHandler<SaveGameResultsCommand> |
| 16 | +{ |
| 17 | + private readonly logger = new Logger(SaveGameResultsHandler.name); |
| 18 | + |
| 19 | + constructor( |
| 20 | + @InjectRepository(FinishedMatchEntity) |
| 21 | + private readonly finishedMatchRepository: Repository<FinishedMatchEntity>, |
| 22 | + @InjectRepository(PlayerInMatchEntity) |
| 23 | + private readonly playerInMatchRepository: Repository<PlayerInMatchEntity>, |
| 24 | + @InjectRepository(GameServerSessionEntity) |
| 25 | + private readonly gameServerSessionModelRepository: Repository<GameServerSessionEntity>, |
| 26 | + private readonly ebus: EventBus, |
| 27 | + private readonly datasource: DataSource, |
| 28 | + ) {} |
| 29 | + |
| 30 | + async execute({ event }: SaveGameResultsCommand) { |
| 31 | + const existingRecordedMatch = await this.finishedMatchRepository.exists({ |
| 32 | + where: { |
| 33 | + id: event.matchId, |
| 34 | + }, |
| 35 | + }); |
| 36 | + |
| 37 | + if (existingRecordedMatch) { |
| 38 | + this.logger.warn("Tried to save already existing match", { |
| 39 | + match_id: event.matchId |
| 40 | + }); |
| 41 | + return; |
| 42 | + } |
| 43 | + |
| 44 | + let modeOverride = |
| 45 | + (event.type === MatchmakingMode.UNRANKED || |
| 46 | + event.type === MatchmakingMode.RANKED) && |
| 47 | + event.players.length !== 10 |
| 48 | + ? MatchmakingMode.BOTS |
| 49 | + : event.type; |
| 50 | + |
| 51 | + const m = await this.datasource.transaction<FinishedMatchEntity>( |
| 52 | + async (entityManager) => { |
| 53 | + const m = new FinishedMatchEntity( |
| 54 | + event.matchId, |
| 55 | + event.winner, |
| 56 | + new Date(event.timestamp * 1000).toISOString(), |
| 57 | + event.gameMode, |
| 58 | + modeOverride, |
| 59 | + event.duration, |
| 60 | + event.server, |
| 61 | + ); |
| 62 | + |
| 63 | + m.externalMatchId = event.externalMatchId; |
| 64 | + |
| 65 | + await entityManager.save(m); |
| 66 | + |
| 67 | + const pims = event.players.map((t) => { |
| 68 | + const pim = new PlayerInMatchEntity(); |
| 69 | + |
| 70 | + pim.match = m; |
| 71 | + // kda |
| 72 | + pim.kills = t.kills; |
| 73 | + pim.deaths = t.deaths; |
| 74 | + pim.assists = t.assists; |
| 75 | + |
| 76 | + // creeps |
| 77 | + pim.denies = t.denies; |
| 78 | + pim.last_hits = t.last_hits; |
| 79 | + |
| 80 | + // pm |
| 81 | + pim.gpm = t.gpm; |
| 82 | + pim.xpm = t.xpm; |
| 83 | + pim.hero_damage = t.heroDamage; |
| 84 | + pim.hero_healing = t.heroHealing; |
| 85 | + pim.tower_damage = t.towerDamage; |
| 86 | + |
| 87 | + pim.abandoned = t.abandoned; |
| 88 | + pim.gold = t.networth; |
| 89 | + |
| 90 | + pim.item0 = t.item0; |
| 91 | + pim.item1 = t.item1; |
| 92 | + pim.item2 = t.item2; |
| 93 | + pim.item3 = t.item3; |
| 94 | + pim.item4 = t.item4; |
| 95 | + pim.item5 = t.item5; |
| 96 | + |
| 97 | + pim.level = t.level; |
| 98 | + pim.playerId = t.steam_id; |
| 99 | + pim.team = t.team; |
| 100 | + pim.hero = t.hero; |
| 101 | + return pim; |
| 102 | + }); |
| 103 | + |
| 104 | + await entityManager.save(pims); |
| 105 | + |
| 106 | + return m; |
| 107 | + }, |
| 108 | + ); |
| 109 | + |
| 110 | + await this.ebus.publish( |
| 111 | + new MatchRecordedEvent( |
| 112 | + event.matchId, |
| 113 | + event.winner, |
| 114 | + event.duration, |
| 115 | + m.matchmaking_mode, |
| 116 | + m.game_mode, |
| 117 | + event.timestamp, |
| 118 | + event.server, |
| 119 | + event.players, |
| 120 | + ), |
| 121 | + ); |
| 122 | + |
| 123 | + const runningSession = await this.gameServerSessionModelRepository.findOne({ |
| 124 | + where: { |
| 125 | + url: event.server, |
| 126 | + }, |
| 127 | + }); |
| 128 | + if (runningSession) { |
| 129 | + await this.gameServerSessionModelRepository.delete(runningSession.url); |
| 130 | + this.ebus.publish( |
| 131 | + new GameSessionFinishedEvent( |
| 132 | + runningSession.url, |
| 133 | + runningSession.matchId, |
| 134 | + runningSession.matchInfoJson, |
| 135 | + ), |
| 136 | + ); |
| 137 | + } |
| 138 | + } |
| 139 | +} |
0 commit comments