LeonMMcoset/leonpan
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

feat: upgrade from community to pro / remove unused edges in storage policy

Browse Source
This commit is contained in:
Aaron Liu
2025-04-24 15:25:40 +08:00
parent 9226d0c8ec
commit cc69178310
18 changed files with 24 additions and 499 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
ent/storagepolicy.go
View File

@@ -58,8 +58,6 @@ type StoragePolicy struct {
// StoragePolicyEdges holds the relations/edges for other nodes in the graph.
type StoragePolicyEdges struct {
// Users holds the value of the users edge.
Users []*User `json:"users,omitempty"`
// Groups holds the value of the groups edge.
Groups []*Group `json:"groups,omitempty"`
// Files holds the value of the files edge.
@@ -70,22 +68,13 @@ type StoragePolicyEdges struct {
Node *Node `json:"node,omitempty"`
// loadedTypes holds the information for reporting if a
// type was loaded (or requested) in eager-loading or not.
loadedTypes [5]bool
}
// UsersOrErr returns the Users value or an error if the edge
// was not loaded in eager-loading.
func (e StoragePolicyEdges) UsersOrErr() ([]*User, error) {
if e.loadedTypes[0] {
return e.Users, nil
}
return nil, &NotLoadedError{edge: "users"}
loadedTypes [4]bool
}
// GroupsOrErr returns the Groups value or an error if the edge
// was not loaded in eager-loading.
func (e StoragePolicyEdges) GroupsOrErr() ([]*Group, error) {
if e.loadedTypes[1] {
if e.loadedTypes[0] {
return e.Groups, nil
}
return nil, &NotLoadedError{edge: "groups"}
@@ -94,7 +83,7 @@ func (e StoragePolicyEdges) GroupsOrErr() ([]*Group, error) {
// FilesOrErr returns the Files value or an error if the edge
// was not loaded in eager-loading.
func (e StoragePolicyEdges) FilesOrErr() ([]*File, error) {
if e.loadedTypes[2] {
if e.loadedTypes[1] {
return e.Files, nil
}
return nil, &NotLoadedError{edge: "files"}
@@ -103,7 +92,7 @@ func (e StoragePolicyEdges) FilesOrErr() ([]*File, error) {
// EntitiesOrErr returns the Entities value or an error if the edge
// was not loaded in eager-loading.
func (e StoragePolicyEdges) EntitiesOrErr() ([]*Entity, error) {
if e.loadedTypes[3] {
if e.loadedTypes[2] {
return e.Entities, nil
}
return nil, &NotLoadedError{edge: "entities"}
@@ -112,7 +101,7 @@ func (e StoragePolicyEdges) EntitiesOrErr() ([]*Entity, error) {
// NodeOrErr returns the Node value or an error if the edge
// was not loaded in eager-loading, or loaded but was not found.
func (e StoragePolicyEdges) NodeOrErr() (*Node, error) {
if e.loadedTypes[4] {
if e.loadedTypes[3] {
if e.Node == nil {
// Edge was loaded but was not found.
return nil, &NotFoundError{label: node.Label}
@@ -264,11 +253,6 @@ func (sp *StoragePolicy) Value(name string) (ent.Value, error) {
return sp.selectValues.Get(name)
}
// QueryUsers queries the "users" edge of the StoragePolicy entity.
func (sp *StoragePolicy) QueryUsers() *UserQuery {
return NewStoragePolicyClient(sp.config).QueryUsers(sp)
}
// QueryGroups queries the "groups" edge of the StoragePolicy entity.
func (sp *StoragePolicy) QueryGroups() *GroupQuery {
return NewStoragePolicyClient(sp.config).QueryGroups(sp)
@@ -362,34 +346,28 @@ func (sp *StoragePolicy) String() string {
return builder.String()
}
// SetUsers manually set the edge as loaded state.
func (e *StoragePolicy) SetUsers(v []*User) {
e.Edges.Users = v
e.Edges.loadedTypes[0] = true
}
// SetGroups manually set the edge as loaded state.
func (e *StoragePolicy) SetGroups(v []*Group) {
e.Edges.Groups = v
e.Edges.loadedTypes[1] = true
e.Edges.loadedTypes[0] = true
}
// SetFiles manually set the edge as loaded state.
func (e *StoragePolicy) SetFiles(v []*File) {
e.Edges.Files = v
e.Edges.loadedTypes[2] = true
e.Edges.loadedTypes[1] = true
}
// SetEntities manually set the edge as loaded state.
func (e *StoragePolicy) SetEntities(v []*Entity) {
e.Edges.Entities = v
e.Edges.loadedTypes[3] = true
e.Edges.loadedTypes[2] = true
}
// SetNode manually set the edge as loaded state.
func (e *StoragePolicy) SetNode(v *Node) {
e.Edges.Node = v
e.Edges.loadedTypes[4] = true
e.Edges.loadedTypes[3] = true
}
// StoragePolicies is a parsable slice of StoragePolicy.