db.go

// Code generated by entc, DO NOT EDIT.

package ent

import (
	"context"
	"sync"

	"github.com/BeanWei/li/li-engine/contrib/lient/driver"
	"github.com/gogf/gf/v2/frame/g"
	"github.com/gogf/gf/v2/os/gctx"
	"github.com/gogf/gf/v2/util/gmode"
)

var (
	dbclient     *Client
	dbclientOnce sync.Once
)

func DB() *Client {
	dbclientOnce.Do(func() {
		drv, err := driver.NewDriver()
		if err != nil {
			g.Log().Fatalf(gctx.New(), "[LiEnt Client] - failed init: %+v", err)
		}

		if gmode.IsDevelop() {
			dbclient = NewClient(Driver(drv)).Debug()
		} else {
			dbclient = NewClient(Driver(drv))
		}

		dbclient.Use(func(next Mutator) Mutator {
			return MutateFunc(func(ctx context.Context, m Mutation) (Value, error) {
				v, err := next.Mutate(ctx, m)
				if err != nil {
					g.DB().GetLogger().Error(ctx, err.Error())
					return nil, err
				}
				return v, nil
			})
		})
	})
	return dbclient
}