72 lines
1.8 KiB
Go
72 lines
1.8 KiB
Go
package db
|
|
|
|
import (
|
|
"database/sql"
|
|
|
|
"github.com/CPunch/gopenfusion/protocol"
|
|
)
|
|
|
|
type Inventory struct {
|
|
PlayerID int
|
|
Slot int
|
|
ID int
|
|
Type int
|
|
Opt int
|
|
TimeLimit int
|
|
}
|
|
|
|
// start && end are both inclusive
|
|
func (db *DBHandler) GetPlayerInventorySlots(PlayerID int, start int, end int) ([]protocol.SItemBase, error) {
|
|
rows, err := db.Query("SELECT PlayerID, Slot, ID, Type, Opt, TimeLimit FROM Inventory WHERE Slot BETWEEN ? AND ? AND PlayerID = ?", start, end, PlayerID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var inven []Inventory
|
|
for rows.Next() {
|
|
item := Inventory{}
|
|
|
|
if err := rows.Scan(
|
|
&item.PlayerID, &item.Slot, &item.ID, &item.Type, &item.Opt, &item.TimeLimit); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
inven = append(inven, item)
|
|
}
|
|
|
|
// populate an SItemBase array
|
|
items := make([]protocol.SItemBase, end-start)
|
|
for _, item := range inven {
|
|
items[item.Slot-start] = protocol.SItemBase{
|
|
IID: int16(item.ID),
|
|
IType: int16(item.Type),
|
|
IOpt: int32(item.Opt),
|
|
ITimeLimit: int32(item.TimeLimit),
|
|
}
|
|
}
|
|
return items, nil
|
|
}
|
|
|
|
// start is inclusive
|
|
func (db *DBHandler) SetPlayerInventorySlots(PlayerID int, start int, items []protocol.SItemBase) error {
|
|
return db.Transaction(func(tx *sql.Tx) error {
|
|
// delete inventory slots
|
|
_, err := db.Query("DELETE FROM Inventory WHERE Slot BETWEEN ? AND ? AND PlayerID = ?", start, start+len(items)-1, PlayerID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// insert inventory
|
|
for i, item := range items {
|
|
if item.IID != 0 {
|
|
_, err := db.Query("INSERT INTO Inventory (PlayerID, Slot, ID, Type, Opt, TimeLimit) VALUES (?, ?, ?, ?, ?, ?)", PlayerID, start+i, item.IID, item.IType, item.IOpt, item.ITimeLimit)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
})
|
|
}
|