urtrator/cache/cache_profiles.go

// URTator - Urban Terror server browser and game launcher, written in
// Go.
//
// Copyright (c) 2016, Stanslav N. a.k.a pztrn (or p0z1tr0n)
// All rights reserved.
//
// Licensed under Terms and Conditions of GNU General Public License
// version 3 or any higher.
// ToDo: put full text of license here.
package cache

import (
    // stdlib
    "fmt"
    "strconv"

    // local
    "github.com/pztrn/urtrator/cachemodels"
    "github.com/pztrn/urtrator/datamodels"
)

func (c *Cache) CreateProfile(name string) {
    fmt.Println("Creating profile " + name)
    _, ok := c.Profiles[name]

    if !ok {
        c.ProfilesMutex.Lock()
        c.Profiles[name] = &cachemodels.Profile{}
        c.Profiles[name].Profile = &datamodels.Profile{}
        c.ProfilesMutex.Unlock()
    }
}

func (c *Cache) deleteProfile(data map[string]string) {
    fmt.Println("Deleting profile " + data["profile_name"])

    _, ok := c.Profiles[data["profile_name"]]
    if ok {
        c.ProfilesMutex.Lock()
        delete(c.Profiles, data["profile_name"])
        c.ProfilesMutex.Unlock()
    }

    _, ok1 := c.Profiles[data["profile_name"]]
    if !ok1 {
        fmt.Println("Profile deleted")
        Database.Unlock()
        Database.Db.MustExec(Database.Db.Rebind("DELETE FROM urt_profiles WHERE name=?"), data["profile_name"])
        Database.Lock()
    } else {
        fmt.Println("Something goes wrong! Profile is still here!")
    }
}

func (c *Cache) FlushProfiles(data map[string]string) {
    fmt.Println("Flushing profiles to database...")

    raw_profiles := []datamodels.Profile{}
    err := Database.Db.Select(&raw_profiles, "SELECT * FROM urt_profiles")
    if err != nil {
        fmt.Println(err.Error())
    }

    cached_profiles := make(map[string]*datamodels.Profile)
    for i := range raw_profiles {
        cached_profiles[raw_profiles[i].Name] = &raw_profiles[i]
    }

    new_profiles := make(map[string]*datamodels.Profile)

    for _, profile := range c.Profiles {
        _, ok := cached_profiles[profile.Profile.Name]
        if !ok {
            fmt.Println("Flushing new profile " + profile.Profile.Name)
            new_profiles[profile.Profile.Name] = &datamodels.Profile{}
            new_profiles[profile.Profile.Name].Name = profile.Profile.Name
            new_profiles[profile.Profile.Name].Version = profile.Profile.Version
            new_profiles[profile.Profile.Name].Binary = profile.Profile.Binary
            new_profiles[profile.Profile.Name].Second_x_session = profile.Profile.Second_x_session
            new_profiles[profile.Profile.Name].Additional_params = profile.Profile.Additional_params
        }
    }

    Database.Unlock()
    tx := Database.Db.MustBegin()
    fmt.Println("Adding new profiles...")
    for _, profile := range new_profiles {
        tx.NamedExec("INSERT INTO urt_profiles (name, version, binary, second_x_session, additional_parameters) VALUES (:name, :version, :binary, :second_x_session, :additional_parameters)", &profile)
    }
    fmt.Println("Updating existing profiles...")
    for _, profile := range cached_profiles {
        tx.NamedExec("UPDATE urt_profiles SET name=:name, version=:version, binary=:binary, second_x_session=:second_x_session, additional_parameters=:additional_parameters WHERE name=:name", &profile)
    }
    tx.Commit()
    Database.Lock()
    fmt.Println("Done")
}

func (c *Cache) LoadProfiles(data map[string]string) {
    fmt.Println("Loading profiles to cache...")

    raw_profiles := []datamodels.Profile{}
    err := Database.Db.Select(&raw_profiles, "SELECT * FROM urt_profiles")
    if err != nil {
        fmt.Println(err.Error())
    }

    for _, profile := range raw_profiles {
        c.Profiles[profile.Name] = &cachemodels.Profile{}
        c.Profiles[profile.Name].Profile = &datamodels.Profile{}
        c.Profiles[profile.Name].Profile.Name = profile.Name
        c.Profiles[profile.Name].Profile.Version = profile.Version
        c.Profiles[profile.Name].Profile.Binary = profile.Binary
        c.Profiles[profile.Name].Profile.Second_x_session = profile.Second_x_session
        c.Profiles[profile.Name].Profile.Additional_params = profile.Additional_params
    }

    fmt.Println("Load completed. Loaded " + strconv.Itoa(len(c.Profiles)) + " profiles.")
}