mirror of
https://github.com/ent/ent.git
synced 2026-05-22 09:31:45 +03:00
entc/gen: check runtime import before executing privacy rules (#1054)
This commit is contained in:
@@ -295,6 +295,9 @@ func (uq *UserQuery) prepareQuery(ctx context.Context) error {
|
||||
}
|
||||
uq.sql = prev
|
||||
}
|
||||
if user.Policy == nil {
|
||||
return errors.New("ent: uninitialized user.Policy (forgotten import ent/runtime?)")
|
||||
}
|
||||
if err := user.Policy.EvalQuery(ctx, uq); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
@@ -370,6 +370,9 @@ func (gq *GroupQuery) prepareQuery(ctx context.Context) error {
|
||||
}
|
||||
gq.sql = prev
|
||||
}
|
||||
if group.Policy == nil {
|
||||
return errors.New("ent: uninitialized group.Policy (forgotten import ent/runtime?)")
|
||||
}
|
||||
if err := group.Policy.EvalQuery(ctx, gq); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
@@ -295,6 +295,9 @@ func (tq *TenantQuery) prepareQuery(ctx context.Context) error {
|
||||
}
|
||||
tq.sql = prev
|
||||
}
|
||||
if tenant.Policy == nil {
|
||||
return errors.New("ent: uninitialized tenant.Policy (forgotten import ent/runtime?)")
|
||||
}
|
||||
if err := tenant.Policy.EvalQuery(ctx, tq); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
@@ -370,6 +370,9 @@ func (uq *UserQuery) prepareQuery(ctx context.Context) error {
|
||||
}
|
||||
uq.sql = prev
|
||||
}
|
||||
if user.Policy == nil {
|
||||
return errors.New("ent: uninitialized user.Policy (forgotten import ent/runtime?)")
|
||||
}
|
||||
if err := user.Policy.EvalQuery(ctx, uq); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user