package models

import (
	"pro2d/components/db"
	"pro2d/conf"
	"pro2d/protos/pb"
)

type HeroModel struct {
	*db.MgoColl
	Hero *pb.Hero
}
type HeroMap map[string]*HeroModel

func GetHeros(hm HeroMap) map[string]*pb.Hero  {
	h := make(map[string]*pb.Hero)
	for k, v := range hm {
		h[k] = v.Hero
	}
	return h
}

func NewHero(h *pb.Hero) *HeroModel {
	m := &HeroModel{
		MgoColl: db.NewMongoColl(conf.GlobalConf.GameConf.DBName, "hero", db.GetBsonM("id", h.Id), h),
		Hero: h,
	}
	m.Load()
	return m
}