Skip to content

Commit d1dc9cb

Browse files
authored
Refactor the function RemoveOrgUser (#27582)
This PR is a small refactor to merge removeOrgUser into RemoveOrgUser.
1 parent ae396ac commit d1dc9cb

File tree

2 files changed

+12
-20
lines changed

2 files changed

+12
-20
lines changed

models/org.go

Lines changed: 11 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -14,12 +14,11 @@ import (
1414
repo_model "code.gitea.io/gitea/models/repo"
1515
)
1616

17-
func removeOrgUser(ctx context.Context, orgID, userID int64) error {
17+
// RemoveOrgUser removes user from given organization.
18+
func RemoveOrgUser(ctx context.Context, orgID, userID int64) error {
1819
ou := new(organization.OrgUser)
1920

20-
sess := db.GetEngine(ctx)
21-
22-
has, err := sess.
21+
has, err := db.GetEngine(ctx).
2322
Where("uid=?", userID).
2423
And("org_id=?", orgID).
2524
Get(ou)
@@ -52,7 +51,13 @@ func removeOrgUser(ctx context.Context, orgID, userID int64) error {
5251
}
5352
}
5453

55-
if _, err := sess.ID(ou.ID).Delete(ou); err != nil {
54+
ctx, committer, err := db.TxContext(ctx)
55+
if err != nil {
56+
return err
57+
}
58+
defer committer.Close()
59+
60+
if _, err := db.GetEngine(ctx).ID(ou.ID).Delete(ou); err != nil {
5661
return err
5762
} else if _, err = db.Exec(ctx, "UPDATE `user` SET num_members=num_members-1 WHERE id=?", orgID); err != nil {
5863
return err
@@ -74,7 +79,7 @@ func removeOrgUser(ctx context.Context, orgID, userID int64) error {
7479
}
7580

7681
if len(repoIDs) > 0 {
77-
if _, err = sess.
82+
if _, err = db.GetEngine(ctx).
7883
Where("user_id = ?", userID).
7984
In("repo_id", repoIDs).
8085
Delete(new(access_model.Access)); err != nil {
@@ -93,18 +98,5 @@ func removeOrgUser(ctx context.Context, orgID, userID int64) error {
9398
}
9499
}
95100

96-
return nil
97-
}
98-
99-
// RemoveOrgUser removes user from given organization.
100-
func RemoveOrgUser(ctx context.Context, orgID, userID int64) error {
101-
ctx, committer, err := db.TxContext(ctx)
102-
if err != nil {
103-
return err
104-
}
105-
defer committer.Close()
106-
if err := removeOrgUser(ctx, orgID, userID); err != nil {
107-
return err
108-
}
109101
return committer.Commit()
110102
}

models/org_team.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -502,7 +502,7 @@ func removeInvalidOrgUser(ctx context.Context, userID, orgID int64) error {
502502
}); err != nil {
503503
return err
504504
} else if count == 0 {
505-
return removeOrgUser(ctx, orgID, userID)
505+
return RemoveOrgUser(ctx, orgID, userID)
506506
}
507507
return nil
508508
}

0 commit comments

Comments
 (0)