package models

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

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(strconv.Itoa(int(id)), h),
		Hero: h,
	}
	return m
}