117 lines
2.6 KiB
Go
117 lines
2.6 KiB
Go
package models
|
|
|
|
import "database/sql"
|
|
|
|
// Beatmap represents a single beatmap (difficulty) on osu!.
|
|
type Beatmap struct {
|
|
ID int `json:"BeatmapID"`
|
|
ParentSetID int
|
|
DiffName string
|
|
FileMD5 string
|
|
Mode int
|
|
BPM float64
|
|
AR float32
|
|
OD float32
|
|
CS float32
|
|
HP float32
|
|
TotalLength int
|
|
HitLength int
|
|
Playcount int
|
|
Passcount int
|
|
MaxCombo int
|
|
DifficultyRating float64
|
|
}
|
|
|
|
const beatmapFields = `
|
|
id, parent_set_id, diff_name, file_md5, mode, bpm,
|
|
ar, od, cs, hp, total_length, hit_length,
|
|
playcount, passcount, max_combo, difficulty_rating`
|
|
|
|
func readBeatmapsFromRows(rows *sql.Rows, capacity int) ([]Beatmap, error) {
|
|
var err error
|
|
bms := make([]Beatmap, 0, capacity)
|
|
for rows.Next() {
|
|
var b Beatmap
|
|
err = rows.Scan(
|
|
&b.ID, &b.ParentSetID, &b.DiffName, &b.FileMD5, &b.Mode, &b.BPM,
|
|
&b.AR, &b.OD, &b.CS, &b.HP, &b.TotalLength, &b.HitLength,
|
|
&b.Playcount, &b.Passcount, &b.MaxCombo, &b.DifficultyRating,
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
bms = append(bms, b)
|
|
}
|
|
|
|
return bms, rows.Err()
|
|
}
|
|
|
|
func inClause(length int) string {
|
|
if length <= 0 {
|
|
return ""
|
|
}
|
|
b := make([]byte, length*3-2)
|
|
for i := 0; i < length; i++ {
|
|
b[i*3] = '?'
|
|
if i != length-1 {
|
|
b[i*3+1] = ','
|
|
b[i*3+2] = ' '
|
|
}
|
|
}
|
|
return string(b)
|
|
}
|
|
|
|
func sIntToSInterface(i []int) []interface{} {
|
|
args := make([]interface{}, len(i))
|
|
for idx, id := range i {
|
|
args[idx] = id
|
|
}
|
|
return args
|
|
}
|
|
|
|
// FetchBeatmaps retrieves a list of beatmap knowing their IDs.
|
|
func FetchBeatmaps(db *sql.DB, ids ...int) ([]Beatmap, error) {
|
|
if len(ids) == 0 {
|
|
return nil, nil
|
|
}
|
|
|
|
q := `SELECT ` + beatmapFields + ` FROM beatmaps WHERE id IN (` + inClause(len(ids)) + `)`
|
|
|
|
rows, err := db.Query(q, sIntToSInterface(ids)...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return readBeatmapsFromRows(rows, len(ids))
|
|
}
|
|
|
|
// CreateBeatmaps adds beatmaps in the database.
|
|
func CreateBeatmaps(db *sql.DB, bms ...Beatmap) error {
|
|
if len(bms) == 0 {
|
|
return nil
|
|
}
|
|
|
|
q := `INSERT INTO beatmaps(` + beatmapFields + `) VALUES `
|
|
const valuePlaceholder = `(
|
|
?, ?, ?, ?, ?, ?,
|
|
?, ?, ?, ?, ?, ?,
|
|
?, ?, ?, ?
|
|
)`
|
|
|
|
args := make([]interface{}, 0, 15*4)
|
|
for idx, bm := range bms {
|
|
if idx != 0 {
|
|
q += ", "
|
|
}
|
|
q += valuePlaceholder
|
|
args = append(args,
|
|
bm.ID, bm.ParentSetID, bm.DiffName, bm.FileMD5, bm.Mode, bm.BPM,
|
|
bm.AR, bm.OD, bm.CS, bm.HP, bm.TotalLength, bm.HitLength,
|
|
bm.Playcount, bm.Passcount, bm.MaxCombo, bm.DifficultyRating,
|
|
)
|
|
}
|
|
|
|
_, err := db.Exec(q, args...)
|
|
return err
|
|
}
|