package models

import (
	"pro2d/components/db"
	"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(id int64) *HeroModel {
	h := &pb.Hero{
		Id:         id,
	}
	m := &HeroModel{
		MgoColl: db.NewMongoColl("hero", db.GetBsonM("id", h.Id), h),
		Hero: h,
	}
	return m
}