You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
132 lines
3.4 KiB
Go
132 lines
3.4 KiB
Go
/*
|
|
Copyright (C) 2024 Pagefault Games
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU Affero General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU Affero General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Affero General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
package db
|
|
|
|
import (
|
|
"database/sql"
|
|
"encoding/hex"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"time"
|
|
|
|
_ "github.com/go-sql-driver/mysql"
|
|
)
|
|
|
|
var handle *sql.DB
|
|
|
|
func Init(username, password, protocol, address, database string) error {
|
|
var err error
|
|
|
|
handle, err = sql.Open("mysql", username+":"+password+"@"+protocol+"("+address+")/"+database)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to open database connection: %s", err)
|
|
}
|
|
|
|
if protocol == "unix" {
|
|
handle.SetMaxOpenConns(1000)
|
|
} else {
|
|
handle.SetMaxOpenConns(200)
|
|
}
|
|
|
|
handle.SetConnMaxIdleTime(time.Second * 30)
|
|
handle.SetConnMaxLifetime(time.Minute)
|
|
|
|
tx, err := handle.Begin()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
tx.Exec("CREATE TABLE IF NOT EXISTS systemSaveData (uuid BINARY(16) PRIMARY KEY, data LONGBLOB, timestamp TIMESTAMP)")
|
|
tx.Exec("CREATE TABLE IF NOT EXISTS sessionSaveData (uuid BINARY(16), slot TINYINT, data LONGBLOB, timestamp TIMESTAMP, PRIMARY KEY (uuid, slot))")
|
|
err = tx.Commit()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
// TODO temp code
|
|
entries, err := os.ReadDir("userdata")
|
|
if err != nil {
|
|
log.Fatalln(err)
|
|
return nil
|
|
}
|
|
|
|
for _, entry := range entries {
|
|
if !entry.IsDir() {
|
|
continue
|
|
}
|
|
|
|
uuidString := entry.Name()
|
|
uuid, err := hex.DecodeString(uuidString)
|
|
if err != nil {
|
|
log.Printf("failed to decode uuid: %s", err)
|
|
continue
|
|
}
|
|
|
|
var count int
|
|
err = handle.QueryRow("SELECT COUNT(*) FROM systemSaveData WHERE uuid = ?", uuid).Scan(&count)
|
|
if err != nil || count != 0 {
|
|
continue
|
|
}
|
|
|
|
// store new system data
|
|
systemData, err := LegacyReadSystemSaveData(uuid)
|
|
if err != nil {
|
|
log.Printf("failed to read system save data for %v: %s", uuidString, err)
|
|
continue
|
|
}
|
|
|
|
err = StoreSystemSaveData(uuid, systemData)
|
|
if err != nil {
|
|
log.Fatalf("failed to store system save data for %v: %s\n", uuidString, err)
|
|
continue
|
|
}
|
|
|
|
// delete old system data
|
|
err = os.Remove("userdata/" + uuidString + "/system.pzs")
|
|
if err != nil {
|
|
log.Fatalf("failed to remove legacy system save data for %v: %s", uuidString, err)
|
|
}
|
|
|
|
for i := 0; i < 5; i++ {
|
|
sessionData, err := LegacyReadSessionSaveData(uuid, i)
|
|
if err != nil {
|
|
log.Printf("failed to read session save data %v for %v: %s", i, uuidString, err)
|
|
continue
|
|
}
|
|
|
|
// store new session data
|
|
err = StoreSessionSaveData(uuid, sessionData, i)
|
|
if err != nil {
|
|
log.Fatalf("failed to store session save data for %v: %s\n", uuidString, err)
|
|
}
|
|
|
|
// delete old session data
|
|
filename := "session"
|
|
if i != 0 {
|
|
filename += fmt.Sprintf("%d", i)
|
|
}
|
|
err = os.Remove(fmt.Sprintf("userdata/%s/%s.pzs", uuidString, filename))
|
|
if err != nil {
|
|
log.Fatalf("failed to remove legacy session save data %v for %v: %s", i, uuidString, err)
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|