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...)
}