66 lines
1.4 KiB
Go
66 lines
1.4 KiB
Go
package data
|
|
|
|
import (
|
|
json "github.com/json-iterator/go"
|
|
"github.com/oylshe1314/framework/util"
|
|
)
|
|
|
|
type LotteryConfig struct {
|
|
Id int `json:"id"`
|
|
PoolId int `json:"pool_id"`
|
|
BoxId int `json:"box_id"`
|
|
ItemId int `json:"item_id"`
|
|
ItemNum int `json:"item_num"`
|
|
ItemWeight int `json:"item_weight"`
|
|
}
|
|
|
|
type LotteryBox struct {
|
|
PoolId int
|
|
BoxId int
|
|
Weights int
|
|
Tables []*LotteryConfig
|
|
}
|
|
|
|
type LotteryConfigTable struct {
|
|
l []*LotteryConfig
|
|
m map[uint64]*LotteryBox
|
|
}
|
|
|
|
func (this *LotteryConfigTable) load(buf []byte) error {
|
|
var err = json.Unmarshal(buf, &this.l)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
this.m = make(map[uint64]*LotteryBox)
|
|
for i := range this.l {
|
|
var key = util.Compose2uint32(uint32(this.l[i].PoolId), uint32(this.l[i].BoxId))
|
|
var box = this.m[key]
|
|
if box == nil {
|
|
box = &LotteryBox{
|
|
PoolId: this.l[i].PoolId,
|
|
BoxId: this.l[i].BoxId,
|
|
Weights: this.l[i].ItemWeight,
|
|
Tables: []*LotteryConfig{this.l[i]},
|
|
}
|
|
this.m[key] = box
|
|
} else {
|
|
box.Weights += this.l[i].ItemWeight
|
|
box.Tables = append(box.Tables, this.l[i])
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (this *LotteryConfigTable) List() []*LotteryConfig {
|
|
return this.l
|
|
}
|
|
|
|
//func (this *LotteryConfigTable) Get(i int) *LotteryConfig {
|
|
// return this.l[i]
|
|
//}
|
|
|
|
func (this *LotteryConfigTable) Find(poolId, boxId int) *LotteryBox {
|
|
return this.m[util.Compose2uint32(uint32(poolId), uint32(boxId))]
|
|
}
|