gameServer/OfficialServer/updateRating.hs
author unc0rr
Sat, 14 Nov 2015 22:19:05 +0300
changeset 11381 437a60995fe1
parent 11380 ff0fa38bdb18
child 11390 36e1bbb6ecea
permissions -rw-r--r--
Rating updater, alpha version
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
11380
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
     1
{-# LANGUAGE ScopedTypeVariables, OverloadedStrings #-}
11358
55360683db75 Start work on rating calculator
unc0rr
parents:
diff changeset
     2
module Main where
55360683db75 Start work on rating calculator
unc0rr
parents:
diff changeset
     3
11380
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
     4
import Data.Maybe
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
     5
import Data.TConfig
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
     6
import qualified Data.ByteString.Char8 as B
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
     7
import Database.MySQL.Simple
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
     8
import Database.MySQL.Simple.QueryResults
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
     9
import Database.MySQL.Simple.Result
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    10
import Control.Monad
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    11
import Control.Exception
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    12
import System.IO
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    13
import qualified  Data.Map as Map
11381
437a60995fe1 Rating updater, alpha version
unc0rr
parents: 11380
diff changeset
    14
import Data.Time.Clock
11380
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    15
------
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    16
import OfficialServer.Glicko2
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    17
11358
55360683db75 Start work on rating calculator
unc0rr
parents:
diff changeset
    18
11380
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    19
queryEpochDates = "SELECT epoch, todatetime, todatetime + INTERVAL 1 week FROM rating_epochs WHERE epoch = (SELECT MAX(epoch) FROM rating_epochs)"
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    20
queryPreviousRatings = "SELECT v.userid, v.rating, v.rd, v.volatility FROM rating_values as v WHERE (v.epoch = (SELECT MAX(epoch) FROM rating_epochs))"
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    21
queryGameResults =
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    22
        "SELECT \
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    23
        \     p.userid \
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    24
        \     , p.place \
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    25
        \     , COALESCE(vp.rating, 1500) \
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    26
        \     , COALESCE(vp.rd, 350) \
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    27
        \     , COALESCE(vp.volatility, 0.06) \
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    28
        \     , COALESCE(vo.rating, 1500) \
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    29
        \     , COALESCE(vo.rd, 350) \
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    30
        \     , COALESCE(vo.volatility, 0.06) \
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    31
        \ FROM \
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    32
        \     (SELECT epoch, todatetime FROM rating_epochs WHERE epoch = (SELECT MAX(epoch) FROM rating_epochs)) as e \
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    33
        \     JOIN rating_games as g ON (g.time BETWEEN e.todatetime AND e.todatetime + INTERVAL 1 WEEK - INTERVAL 1 SECOND) \
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    34
        \     JOIN rating_players as p ON (p.gameid = g.id) \
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    35
        \     JOIN rating_players as o ON (p.gameid = o.gameid AND p.userid <> o.userid AND (p.place = 0 OR (p.place <> o.place))) \
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    36
        \     LEFT OUTER JOIN rating_values as vp ON (vp.epoch = e.epoch AND vp.userid = p.userid) \
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    37
        \     LEFT OUTER JOIN rating_values as vo ON (vo.epoch = e.epoch AND vo.userid = o.userid) \
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    38
        \ GROUP BY p.userid, p.gameid, p.place \
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    39
        \ ORDER BY p.userid"
11381
437a60995fe1 Rating updater, alpha version
unc0rr
parents: 11380
diff changeset
    40
insertNewRatings = "INSERT INTO rating_values (userid, epoch, rating, rd, volatility) VALUES (?, ?, ?, ?, ?)"
437a60995fe1 Rating updater, alpha version
unc0rr
parents: 11380
diff changeset
    41
insertNewEpoch = "INSERT INTO rating_epochs (epoch, todatetime) VALUES (?, ?)"
11380
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    42
11381
437a60995fe1 Rating updater, alpha version
unc0rr
parents: 11380
diff changeset
    43
mergeRatingData :: Map.Map Int (RatingData, [GameData]) -> [(Int, (RatingData, [GameData]))] -> Map.Map Int (RatingData, [GameData])
437a60995fe1 Rating updater, alpha version
unc0rr
parents: 11380
diff changeset
    44
mergeRatingData m s = foldr (unc0rry (Map.insertWith mf)) m s
437a60995fe1 Rating updater, alpha version
unc0rr
parents: 11380
diff changeset
    45
    where
437a60995fe1 Rating updater, alpha version
unc0rr
parents: 11380
diff changeset
    46
        mf (rd, gds) (_, gds2) = (rd, gds ++ gds2)
437a60995fe1 Rating updater, alpha version
unc0rr
parents: 11380
diff changeset
    47
        unc0rry f (a, b) c = f a b c
437a60995fe1 Rating updater, alpha version
unc0rr
parents: 11380
diff changeset
    48
11380
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    49
calculateRatings dbConn = do
11381
437a60995fe1 Rating updater, alpha version
unc0rr
parents: 11380
diff changeset
    50
    [(epochNum :: Int, fromDate :: UTCTime, toDate :: UTCTime)] <- query_ dbConn queryEpochDates
11380
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    51
    initRatingData <- (Map.fromList . map fromDBrating) `fmap` query_ dbConn queryPreviousRatings
11381
437a60995fe1 Rating updater, alpha version
unc0rr
parents: 11380
diff changeset
    52
    gameData <- map fromGameResult `fmap` query_ dbConn queryGameResults
437a60995fe1 Rating updater, alpha version
unc0rr
parents: 11380
diff changeset
    53
    let mData = map getNewRating . Map.toList $ mergeRatingData initRatingData gameData
437a60995fe1 Rating updater, alpha version
unc0rr
parents: 11380
diff changeset
    54
    executeMany dbConn insertNewRatings $ map (toInsert epochNum) mData
437a60995fe1 Rating updater, alpha version
unc0rr
parents: 11380
diff changeset
    55
    execute dbConn insertNewEpoch (epochNum + 1, toDate)
11380
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    56
    return ()
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    57
    where
11381
437a60995fe1 Rating updater, alpha version
unc0rr
parents: 11380
diff changeset
    58
        toInsert e (i, RatingData r rd v) = (i, e + 1, r, rd, v)
437a60995fe1 Rating updater, alpha version
unc0rr
parents: 11380
diff changeset
    59
        getNewRating (a, d) = (a, uncurry calcNewRating d)
437a60995fe1 Rating updater, alpha version
unc0rr
parents: 11380
diff changeset
    60
        convPlace :: Int -> Double
437a60995fe1 Rating updater, alpha version
unc0rr
parents: 11380
diff changeset
    61
        convPlace 0 = 0.5
437a60995fe1 Rating updater, alpha version
unc0rr
parents: 11380
diff changeset
    62
        convPlace 1 = 1.0
437a60995fe1 Rating updater, alpha version
unc0rr
parents: 11380
diff changeset
    63
        convPlace 2 = 0.0
437a60995fe1 Rating updater, alpha version
unc0rr
parents: 11380
diff changeset
    64
        convPlace _ = error "Incorrect place value"
11380
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    65
        fromDBrating (a, b, c, d) = (a, (RatingData b c d, []))
11381
437a60995fe1 Rating updater, alpha version
unc0rr
parents: 11380
diff changeset
    66
        fromGameResult (pid, place, prating, pRD, pvol, orating, oRD, ovol) =
437a60995fe1 Rating updater, alpha version
unc0rr
parents: 11380
diff changeset
    67
            (pid,
437a60995fe1 Rating updater, alpha version
unc0rr
parents: 11380
diff changeset
    68
                (RatingData prating pRD pvol
437a60995fe1 Rating updater, alpha version
unc0rr
parents: 11380
diff changeset
    69
                , [GameData (RatingData orating oRD ovol) $ convPlace place]))
11380
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    70
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    71
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    72
data DBConnectInfo = DBConnectInfo {
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    73
    dbHost
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    74
    , dbName
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    75
    , dbLogin
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    76
    , dbPassword :: B.ByteString
11358
55360683db75 Start work on rating calculator
unc0rr
parents:
diff changeset
    77
    }
55360683db75 Start work on rating calculator
unc0rr
parents:
diff changeset
    78
11380
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    79
cfgFileName :: String
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    80
cfgFileName = "hedgewars-server.ini"
11358
55360683db75 Start work on rating calculator
unc0rr
parents:
diff changeset
    81
55360683db75 Start work on rating calculator
unc0rr
parents:
diff changeset
    82
11380
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    83
readServerConfig :: ConnectInfo -> IO ConnectInfo
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    84
readServerConfig ci = do
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    85
    cfg <- readConfig cfgFileName
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    86
    return $ ci{
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    87
        connectHost = value "dbHost" cfg
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    88
        , connectDatabase = value "dbName" cfg
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    89
        , connectUser = value "dbLogin" cfg
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    90
        , connectPassword = value "dbPassword" cfg
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    91
    }
11358
55360683db75 Start work on rating calculator
unc0rr
parents:
diff changeset
    92
    where
11380
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    93
        value n c = fromJust2 n $ getValue n c
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    94
        fromJust2 n Nothing = error $ "Missing config entry " ++ n
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    95
        fromJust2 _ (Just a) = a
11358
55360683db75 Start work on rating calculator
unc0rr
parents:
diff changeset
    96
11380
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    97
dbConnectionLoop mySQLConnectionInfo =
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    98
    Control.Exception.handle (\(e :: SomeException) -> hPutStrLn stderr $ show e) $
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
    99
        bracket
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
   100
            (connect mySQLConnectionInfo)
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
   101
            close
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
   102
            calculateRatings
11358
55360683db75 Start work on rating calculator
unc0rr
parents:
diff changeset
   103
11380
ff0fa38bdb18 Some WIP
unc0rr
parents: 11359
diff changeset
   104
main = readServerConfig defaultConnectInfo >>= dbConnectionLoop