Skip to content

Commit bb5a6b7

Browse files
authored
fix xorm NewSession uncorrected usages (#774)
1 parent 25663b5 commit bb5a6b7

File tree

6 files changed

+23
-33
lines changed

6 files changed

+23
-33
lines changed

models/access_test.go

Lines changed: 4 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -98,16 +98,12 @@ func TestRepository_RecalculateAccesses(t *testing.T) {
9898
repo1 := AssertExistsAndLoadBean(t, &Repository{ID: 3}).(*Repository)
9999
assert.NoError(t, repo1.GetOwner())
100100

101-
sess := x.NewSession()
102-
defer sess.Close()
103-
_, err := sess.Delete(&Collaboration{UserID: 2, RepoID: 3})
101+
_, err := x.Delete(&Collaboration{UserID: 2, RepoID: 3})
104102
assert.NoError(t, err)
105-
106103
assert.NoError(t, repo1.RecalculateAccesses())
107104

108-
sess = x.NewSession()
109105
access := &Access{UserID: 2, RepoID: 3}
110-
has, err := sess.Get(access)
106+
has, err := x.Get(access)
111107
assert.NoError(t, err)
112108
assert.True(t, has)
113109
assert.Equal(t, AccessModeWrite, access.Mode)
@@ -119,15 +115,11 @@ func TestRepository_RecalculateAccesses2(t *testing.T) {
119115
repo1 := AssertExistsAndLoadBean(t, &Repository{ID: 4}).(*Repository)
120116
assert.NoError(t, repo1.GetOwner())
121117

122-
sess := x.NewSession()
123-
defer sess.Close()
124-
_, err := sess.Delete(&Collaboration{UserID: 4, RepoID: 4})
118+
_, err := x.Delete(&Collaboration{UserID: 4, RepoID: 4})
125119
assert.NoError(t, err)
126-
127120
assert.NoError(t, repo1.RecalculateAccesses())
128121

129-
sess = x.NewSession()
130-
has, err := sess.Get(&Access{UserID: 4, RepoID: 4})
122+
has, err := x.Get(&Access{UserID: 4, RepoID: 4})
131123
assert.NoError(t, err)
132124
assert.False(t, has)
133125
}

models/org.go

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -345,13 +345,15 @@ func getOrgsByUserID(sess *xorm.Session, userID int64, showAll bool) ([]*User, e
345345
// GetOrgsByUserID returns a list of organizations that the given user ID
346346
// has joined.
347347
func GetOrgsByUserID(userID int64, showAll bool) ([]*User, error) {
348-
return getOrgsByUserID(x.NewSession(), userID, showAll)
348+
sess := x.NewSession()
349+
defer sess.Close()
350+
return getOrgsByUserID(sess, userID, showAll)
349351
}
350352

351353
// GetOrgsByUserIDDesc returns a list of organizations that the given user ID
352354
// has joined, ordered descending by the given condition.
353355
func GetOrgsByUserIDDesc(userID int64, desc string, showAll bool) ([]*User, error) {
354-
return getOrgsByUserID(x.NewSession().Desc(desc), userID, showAll)
356+
return getOrgsByUserID(x.Desc(desc), userID, showAll)
355357
}
356358

357359
func getOwnedOrgsByUserID(sess *xorm.Session, userID int64) ([]*User, error) {
@@ -367,14 +369,14 @@ func getOwnedOrgsByUserID(sess *xorm.Session, userID int64) ([]*User, error) {
367369
// GetOwnedOrgsByUserID returns a list of organizations are owned by given user ID.
368370
func GetOwnedOrgsByUserID(userID int64) ([]*User, error) {
369371
sess := x.NewSession()
372+
defer sess.Close()
370373
return getOwnedOrgsByUserID(sess, userID)
371374
}
372375

373376
// GetOwnedOrgsByUserIDDesc returns a list of organizations are owned by
374377
// given user ID, ordered descending by the given condition.
375378
func GetOwnedOrgsByUserIDDesc(userID int64, desc string) ([]*User, error) {
376-
sess := x.NewSession()
377-
return getOwnedOrgsByUserID(sess.Desc(desc), userID)
379+
return getOwnedOrgsByUserID(x.Desc(desc), userID)
378380
}
379381

380382
// GetOrgUsersByUserID returns all organization-user relations by user ID.

models/org_team.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -309,7 +309,7 @@ func UpdateTeam(t *Team, authChanged bool) (err error) {
309309
}
310310

311311
t.LowerName = strings.ToLower(t.Name)
312-
has, err := x.
312+
has, err := sess.
313313
Where("org_id=?", t.OrgID).
314314
And("lower_name=?", t.LowerName).
315315
And("id!=?", t.ID).

models/pull_test.go

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -225,11 +225,9 @@ func TestChangeUsernameInPullRequests(t *testing.T) {
225225
assert.NoError(t, PrepareTestDatabase())
226226
const newUsername = "newusername"
227227
assert.NoError(t, ChangeUsernameInPullRequests("user1", newUsername))
228-
sess := x.NewSession()
229-
defer sess.Close()
230228

231229
prs := make([]*PullRequest, 0, 10)
232-
assert.NoError(t, sess.Where("head_user_name = ?", newUsername).Find(&prs))
230+
assert.NoError(t, x.Where("head_user_name = ?", newUsername).Find(&prs))
233231
assert.Len(t, prs, 2)
234232
for _, pr := range prs {
235233
assert.Equal(t, newUsername, pr.HeadUserName)

models/repo.go

Lines changed: 11 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -1515,16 +1515,16 @@ func DeleteRepository(uid, repoID int64) error {
15151515
return err
15161516
}
15171517

1518-
if err = sess.Commit(); err != nil {
1519-
return fmt.Errorf("Commit: %v", err)
1520-
}
1521-
15221518
if repo.NumForks > 0 {
1523-
if _, err = x.Exec("UPDATE `repository` SET fork_id=0,is_fork=? WHERE fork_id=?", false, repo.ID); err != nil {
1519+
if _, err = sess.Exec("UPDATE `repository` SET fork_id=0,is_fork=? WHERE fork_id=?", false, repo.ID); err != nil {
15241520
log.Error(4, "reset 'fork_id' and 'is_fork': %v", err)
15251521
}
15261522
}
15271523

1524+
if err = sess.Commit(); err != nil {
1525+
return fmt.Errorf("Commit: %v", err)
1526+
}
1527+
15281528
return nil
15291529
}
15301530

@@ -2217,30 +2217,29 @@ func ForkRepository(u *User, oldRepo *Repository, name, desc string) (_ *Reposit
22172217
if err != nil {
22182218
return nil, err
22192219
}
2220-
sessionRelease(sess)
22212220

22222221
// Copy LFS meta objects in new session
2223-
sess = x.NewSession()
2224-
defer sessionRelease(sess)
2225-
if err = sess.Begin(); err != nil {
2222+
sess2 := x.NewSession()
2223+
defer sessionRelease(sess2)
2224+
if err = sess2.Begin(); err != nil {
22262225
return nil, err
22272226
}
22282227

22292228
var lfsObjects []*LFSMetaObject
22302229

2231-
if err = sess.Where("repository_id=?", oldRepo.ID).Find(&lfsObjects); err != nil {
2230+
if err = sess2.Where("repository_id=?", oldRepo.ID).Find(&lfsObjects); err != nil {
22322231
return nil, err
22332232
}
22342233

22352234
for _, v := range lfsObjects {
22362235
v.ID = 0
22372236
v.RepositoryID = repo.ID
2238-
if _, err = sess.Insert(v); err != nil {
2237+
if _, err = sess2.Insert(v); err != nil {
22392238
return nil, err
22402239
}
22412240
}
22422241

2243-
return repo, sess.Commit()
2242+
return repo, sess2.Commit()
22442243
}
22452244

22462245
// GetForks returns all the forks of the repository

models/star.go

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,6 @@ type Star struct {
1414
// StarRepo or unstar repository.
1515
func StarRepo(userID, repoID int64, star bool) error {
1616
sess := x.NewSession()
17-
1817
defer sess.Close()
1918

2019
if err := sess.Begin(); err != nil {

0 commit comments

Comments
 (0)