oceanbase

1
// Code generated by entc, DO NOT EDIT.
2

3
package ent
4

5
import (
6
	"context"
7
)
8

9
type clientCtxKey struct{}
10

11
// FromContext returns a Client stored inside a context, or nil if there isn't one.
12
func FromContext(ctx context.Context) *Client {
13
	c, _ := ctx.Value(clientCtxKey{}).(*Client)
14
	return c
15
}
16

17
// NewContext returns a new context with the given Client attached.
18
func NewContext(parent context.Context, c *Client) context.Context {
19
	return context.WithValue(parent, clientCtxKey{}, c)
20
}
21

22
type txCtxKey struct{}
23

24
// TxFromContext returns a Tx stored inside a context, or nil if there isn't one.
25
func TxFromContext(ctx context.Context) *Tx {
26
	tx, _ := ctx.Value(txCtxKey{}).(*Tx)
27
	return tx
28
}
29

30
// NewTxContext returns a new context with the given Tx attached.
31
func NewTxContext(parent context.Context, tx *Tx) context.Context {
32
	return context.WithValue(parent, txCtxKey{}, tx)
33
}
34