aiChat/utility/tidb.go

package utility

import (
	"strings"

	"app.yhyue.com/moapp/jybase/mapping"
	. "app.yhyue.com/moapp/jybase/mysql"
	"github.com/gogf/gf/v2/frame/g"
	"github.com/gogf/gf/v2/os/gctx"
)

var tidb *Mysql
var AllDistrict = map[string]bool{}

func init() {
	ctx := gctx.New()
	tidb = NewInit("mysql", strings.TrimPrefix(g.Cfg().MustGet(ctx, "database.default.link").String(), "mysql:"), g.Cfg().MustGet(ctx, "database.default.maxOpen").Int(), g.Cfg().MustGet(ctx, "database.default.maxIdle").Int())
	mapping.BidCodeMapping.Init(tidb)
	for _, v := range mapping.BidCodeMapping.District {
		for kk, _ := range v {
			AllDistrict[kk] = true
		}
	}
}