package models
  
  import (
  	"pro2d/components/db"
  	"pro2d/protos/pb"
  	"pro2d/utils"
  )
  
  func InitDoc(schema ...interface{})  {
  	for _, s := range schema {
  		for coll, key := range utils.FindIndex(s) {
  			if err := db.CreateCollection(coll); err != nil {
  				utils.Sugar.Errorf("InitDoc err: %v", err)
  				continue
  			}
  
  			res, err := db.SetUnique(coll, key)
  			if err != nil {
  				utils.Sugar.Errorf("InitDoc unique: %s, err: %v", res, err)
  				continue
  			}
  		}
  	}
  }
  
  func InitAccountServerModels()  {
  	var schema []interface{} = []interface{}{
  		pb.AccountInfo{},
  	}
  	InitDoc(schema...)
  }
  
  func InitGameServerModels()  {
  	var schema []interface{} = []interface{}{
  		pb.Hero{},
  		pb.Role{},
  		pb.Team{},
  	}
  	InitDoc(schema...)
  }