package models
  
  import (

  	"pro2d/protos/pb"

  	"pro2d/src/components/db"
  	"pro2d/src/components/logger"
  	"pro2d/src/utils"

  )
  
  func InitDoc(schema ...interface{})  {
  	for _, s := range schema {

  		coll, keys := utils.FindIndex(s)
  		for  _, index := range keys {
  			db.CreateCollection(coll)

  			logger.Debug("InitDoc collect: %v, createIndex: %s", coll, index)

  			res, err := db.SetUnique(coll, index)

  			if err != nil {

  				logger.Error("InitDoc unique: %s, err: %v", res, err)

  				continue
  			}
  		}
  	}
  }
  
  func InitAccountServerModels()  {
  	var schema []interface{} = []interface{}{

  		pb.Account{},

  	}
  	InitDoc(schema...)
  }
  
  func InitGameServerModels()  {
  	var schema []interface{} = []interface{}{
  		pb.Hero{},
  		pb.Role{},
  		pb.Team{},
  	}
  	InitDoc(schema...)
  }