init.go 753 Bytes
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 {
				continue
			}

			utils.Sugar.Debugf("InitDoc collect: %v, createIndex: %s", coll, key)
			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...)
}