package models
  
  import (

  	"pro2d/protos/pb"

  	"pro2d/src/components/db"

  	"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
  }