Always store primary email address into email_address table and also the state (#15956)
* Always store primary email address into email_address table and also the state * Add lower_email to not convert email to lower as what's added * Fix fixture * Fix tests * Use BeforeInsert to save lower email * Fix v180 migration * fix tests * Fix test * Remove wrong submited codes * Fix test * Fix test * Fix test * Add test for v181 migration * remove change user's email to lower * Revert change on user's email column * Fix lower email * Fix test * Fix testtokarchuk/v1.17
Lunny Xiao
3 years ago
committed by
GitHub
parent
21cde5c439
commit
b9d611e917
@ -1,35 +1,279 @@ |
|||||||
- |
- |
||||||
id: 1 |
id: 1 |
||||||
uid: 1 |
uid: 11 |
||||||
email: user11@example.com |
email: user11@example.com |
||||||
|
lower_email: user11@example.com |
||||||
is_activated: false |
is_activated: false |
||||||
|
is_primary: true |
||||||
|
|
||||||
- |
- |
||||||
id: 2 |
id: 2 |
||||||
uid: 1 |
uid: 12 |
||||||
email: user12@example.com |
email: user12@example.com |
||||||
is_activated: false |
lower_email: user12@example.com |
||||||
|
is_activated: true |
||||||
|
is_primary: true |
||||||
|
|
||||||
- |
- |
||||||
id: 3 |
id: 3 |
||||||
uid: 2 |
uid: 2 |
||||||
email: user2@example.com |
email: user2@example.com |
||||||
|
lower_email: user2@example.com |
||||||
is_activated: true |
is_activated: true |
||||||
|
is_primary: true |
||||||
|
|
||||||
- |
- |
||||||
id: 4 |
id: 4 |
||||||
uid: 2 |
uid: 21 |
||||||
email: user21@example.com |
email: user21@example.com |
||||||
is_activated: false |
lower_email: user21@example.com |
||||||
|
is_activated: true |
||||||
|
is_primary: true |
||||||
|
|
||||||
- |
- |
||||||
id: 5 |
id: 5 |
||||||
uid: 9999999 |
uid: 9999999 |
||||||
email: user9999999@example.com |
email: user9999999@example.com |
||||||
|
lower_email: user9999999@example.com |
||||||
is_activated: true |
is_activated: true |
||||||
|
is_primary: false |
||||||
|
|
||||||
- |
- |
||||||
id: 6 |
id: 6 |
||||||
uid: 10 |
uid: 10 |
||||||
|
email: user10@example.com |
||||||
|
lower_email: user10@example.com |
||||||
|
is_activated: true |
||||||
|
is_primary: true |
||||||
|
|
||||||
|
- |
||||||
|
id: 7 |
||||||
|
uid: 10 |
||||||
email: user101@example.com |
email: user101@example.com |
||||||
|
lower_email: user101@example.com |
||||||
|
is_activated: true |
||||||
|
is_primary: false |
||||||
|
|
||||||
|
- |
||||||
|
id: 8 |
||||||
|
uid: 9 |
||||||
|
email: user9@example.com |
||||||
|
lower_email: user9@example.com |
||||||
|
is_activated: false |
||||||
|
is_primary: true |
||||||
|
|
||||||
|
- |
||||||
|
id: 9 |
||||||
|
uid: 1 |
||||||
|
email: user1@example.com |
||||||
|
lower_email: user1@example.com |
||||||
|
is_activated: true |
||||||
|
is_primary: true |
||||||
|
|
||||||
|
- |
||||||
|
id: 10 |
||||||
|
uid: 3 |
||||||
|
email: user3@example.com |
||||||
|
lower_email: user3@example.com |
||||||
|
is_activated: true |
||||||
|
is_primary: true |
||||||
|
|
||||||
|
- |
||||||
|
id: 11 |
||||||
|
uid: 4 |
||||||
|
email: user4@example.com |
||||||
|
lower_email: user4@example.com |
||||||
|
is_activated: true |
||||||
|
is_primary: true |
||||||
|
|
||||||
|
- |
||||||
|
id: 12 |
||||||
|
uid: 5 |
||||||
|
email: user5@example.com |
||||||
|
lower_email: user5@example.com |
||||||
is_activated: true |
is_activated: true |
||||||
|
is_primary: true |
||||||
|
|
||||||
|
- |
||||||
|
id: 13 |
||||||
|
uid: 6 |
||||||
|
email: user6@example.com |
||||||
|
lower_email: user6@example.com |
||||||
|
is_activated: true |
||||||
|
is_primary: true |
||||||
|
|
||||||
|
- |
||||||
|
id: 14 |
||||||
|
uid: 7 |
||||||
|
email: user7@example.com |
||||||
|
lower_email: user7@example.com |
||||||
|
is_activated: true |
||||||
|
is_primary: true |
||||||
|
|
||||||
|
- |
||||||
|
id: 15 |
||||||
|
uid: 8 |
||||||
|
email: user8@example.com |
||||||
|
lower_email: user8@example.com |
||||||
|
is_activated: true |
||||||
|
is_primary: true |
||||||
|
|
||||||
|
- |
||||||
|
id: 16 |
||||||
|
uid: 13 |
||||||
|
email: user13@example.com |
||||||
|
lower_email: user13@example.com |
||||||
|
is_activated: true |
||||||
|
is_primary: true |
||||||
|
|
||||||
|
- |
||||||
|
id: 17 |
||||||
|
uid: 14 |
||||||
|
email: user14@example.com |
||||||
|
lower_email: user14@example.com |
||||||
|
is_activated: true |
||||||
|
is_primary: true |
||||||
|
|
||||||
|
- |
||||||
|
id: 18 |
||||||
|
uid: 15 |
||||||
|
email: user15@example.com |
||||||
|
lower_email: user15@example.com |
||||||
|
is_activated: true |
||||||
|
is_primary: true |
||||||
|
|
||||||
|
- |
||||||
|
id: 19 |
||||||
|
uid: 16 |
||||||
|
email: user16@example.com |
||||||
|
lower_email: user16@example.com |
||||||
|
is_activated: true |
||||||
|
is_primary: true |
||||||
|
|
||||||
|
- |
||||||
|
id: 20 |
||||||
|
uid: 17 |
||||||
|
email: user17@example.com |
||||||
|
lower_email: user17@example.com |
||||||
|
is_activated: true |
||||||
|
is_primary: true |
||||||
|
|
||||||
|
- |
||||||
|
id: 21 |
||||||
|
uid: 18 |
||||||
|
email: user18@example.com |
||||||
|
lower_email: user18@example.com |
||||||
|
is_activated: true |
||||||
|
is_primary: true |
||||||
|
|
||||||
|
- |
||||||
|
id: 22 |
||||||
|
uid: 19 |
||||||
|
email: user19@example.com |
||||||
|
lower_email: user19@example.com |
||||||
|
is_activated: true |
||||||
|
is_primary: true |
||||||
|
|
||||||
|
- |
||||||
|
id: 23 |
||||||
|
uid: 20 |
||||||
|
email: user20@example.com |
||||||
|
lower_email: user20@example.com |
||||||
|
is_activated: true |
||||||
|
is_primary: true |
||||||
|
|
||||||
|
- |
||||||
|
id: 24 |
||||||
|
uid: 22 |
||||||
|
email: limited_org@example.com |
||||||
|
lower_email: limited_org@example.com |
||||||
|
is_activated: true |
||||||
|
is_primary: true |
||||||
|
|
||||||
|
- |
||||||
|
id: 25 |
||||||
|
uid: 23 |
||||||
|
email: privated_org@example.com |
||||||
|
lower_email: privated_org@example.com |
||||||
|
is_activated: true |
||||||
|
is_primary: true |
||||||
|
|
||||||
|
- |
||||||
|
id: 26 |
||||||
|
uid: 24 |
||||||
|
email: user24@example.com |
||||||
|
lower_email: user24@example.com |
||||||
|
is_activated: true |
||||||
|
is_primary: true |
||||||
|
|
||||||
|
- |
||||||
|
id: 27 |
||||||
|
uid: 25 |
||||||
|
email: org25@example.com |
||||||
|
lower_email: org25@example.com |
||||||
|
is_activated: true |
||||||
|
is_primary: true |
||||||
|
|
||||||
|
- |
||||||
|
id: 28 |
||||||
|
uid: 26 |
||||||
|
email: org26@example.com |
||||||
|
lower_email: org26@example.com |
||||||
|
is_activated: true |
||||||
|
is_primary: true |
||||||
|
|
||||||
|
- |
||||||
|
id: 29 |
||||||
|
uid: 27 |
||||||
|
email: user27@example.com |
||||||
|
lower_email: user27@example.com |
||||||
|
is_activated: true |
||||||
|
is_primary: true |
||||||
|
|
||||||
|
- |
||||||
|
id: 30 |
||||||
|
uid: 28 |
||||||
|
email: user28@example.com |
||||||
|
lower_email: user28@example.com |
||||||
|
is_activated: true |
||||||
|
is_primary: true |
||||||
|
|
||||||
|
- |
||||||
|
id: 31 |
||||||
|
uid: 29 |
||||||
|
email: user29@example.com |
||||||
|
lower_email: user29@example.com |
||||||
|
is_activated: true |
||||||
|
is_primary: true |
||||||
|
|
||||||
|
- |
||||||
|
id: 32 |
||||||
|
uid: 30 |
||||||
|
email: user30@example.com |
||||||
|
lower_email: user30@example.com |
||||||
|
is_activated: true |
||||||
|
is_primary: true |
||||||
|
|
||||||
|
- |
||||||
|
id: 33 |
||||||
|
uid: 1 |
||||||
|
email: user1-2@example.com |
||||||
|
lower_email: user1-2@example.com |
||||||
|
is_activated: true |
||||||
|
is_primary: false |
||||||
|
|
||||||
|
- |
||||||
|
id: 34 |
||||||
|
uid: 1 |
||||||
|
email: user1-3@example.com |
||||||
|
lower_email: user1-3@example.com |
||||||
|
is_activated: true |
||||||
|
is_primary: false |
||||||
|
|
||||||
|
- |
||||||
|
id: 35 |
||||||
|
uid: 2 |
||||||
|
email: user2-2@example.com |
||||||
|
lower_email: user2-2@example.com |
||||||
|
is_activated: false |
||||||
|
is_primary: false |
||||||
@ -0,0 +1,92 @@ |
|||||||
|
// Use of this source code is governed by a MIT-style
|
||||||
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
|
package migrations |
||||||
|
|
||||||
|
import ( |
||||||
|
"strings" |
||||||
|
|
||||||
|
"xorm.io/xorm" |
||||||
|
) |
||||||
|
|
||||||
|
func addPrimaryEmail2EmailAddress(x *xorm.Engine) (err error) { |
||||||
|
type User struct { |
||||||
|
ID int64 `xorm:"pk autoincr"` |
||||||
|
Email string `xorm:"NOT NULL"` |
||||||
|
IsActive bool `xorm:"INDEX"` // Activate primary email
|
||||||
|
} |
||||||
|
|
||||||
|
type EmailAddress1 struct { |
||||||
|
ID int64 `xorm:"pk autoincr"` |
||||||
|
UID int64 `xorm:"INDEX NOT NULL"` |
||||||
|
Email string `xorm:"UNIQUE NOT NULL"` |
||||||
|
LowerEmail string |
||||||
|
IsActivated bool |
||||||
|
IsPrimary bool `xorm:"DEFAULT(false) NOT NULL"` |
||||||
|
} |
||||||
|
|
||||||
|
// Add lower_email and is_primary columns
|
||||||
|
if err = x.Table("email_address").Sync2(new(EmailAddress1)); err != nil { |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
if _, err = x.Exec("UPDATE email_address SET lower_email=LOWER(email), is_primary=?", false); err != nil { |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
type EmailAddress struct { |
||||||
|
ID int64 `xorm:"pk autoincr"` |
||||||
|
UID int64 `xorm:"INDEX NOT NULL"` |
||||||
|
Email string `xorm:"UNIQUE NOT NULL"` |
||||||
|
LowerEmail string `xorm:"UNIQUE NOT NULL"` |
||||||
|
IsActivated bool |
||||||
|
IsPrimary bool `xorm:"DEFAULT(false) NOT NULL"` |
||||||
|
} |
||||||
|
|
||||||
|
// change lower_email as unique
|
||||||
|
if err = x.Sync2(new(EmailAddress)); err != nil { |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
sess := x.NewSession() |
||||||
|
defer sess.Close() |
||||||
|
|
||||||
|
const batchSize = 100 |
||||||
|
|
||||||
|
for start := 0; ; start += batchSize { |
||||||
|
users := make([]*User, 0, batchSize) |
||||||
|
if err = sess.Limit(batchSize, start).Find(&users); err != nil { |
||||||
|
return |
||||||
|
} |
||||||
|
if len(users) == 0 { |
||||||
|
break |
||||||
|
} |
||||||
|
|
||||||
|
for _, user := range users { |
||||||
|
var exist bool |
||||||
|
exist, err = sess.Where("email=?", user.Email).Table("email_address").Exist() |
||||||
|
if err != nil { |
||||||
|
return |
||||||
|
} |
||||||
|
if !exist { |
||||||
|
if _, err = sess.Insert(&EmailAddress{ |
||||||
|
UID: user.ID, |
||||||
|
Email: user.Email, |
||||||
|
LowerEmail: strings.ToLower(user.Email), |
||||||
|
IsActivated: user.IsActive, |
||||||
|
IsPrimary: true, |
||||||
|
}); err != nil { |
||||||
|
return |
||||||
|
} |
||||||
|
} else { |
||||||
|
if _, err = sess.Where("email=?", user.Email).Cols("is_primary").Update(&EmailAddress{ |
||||||
|
IsPrimary: true, |
||||||
|
}); err != nil { |
||||||
|
return |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
return nil |
||||||
|
} |
||||||
@ -0,0 +1,54 @@ |
|||||||
|
// Copyright 2021 The Gitea Authors. All rights reserved.
|
||||||
|
// Use of this source code is governed by a MIT-style
|
||||||
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
|
package migrations |
||||||
|
|
||||||
|
import ( |
||||||
|
"strings" |
||||||
|
"testing" |
||||||
|
|
||||||
|
"github.com/stretchr/testify/assert" |
||||||
|
) |
||||||
|
|
||||||
|
func Test_addPrimaryEmail2EmailAddress(t *testing.T) { |
||||||
|
type User struct { |
||||||
|
ID int64 |
||||||
|
Email string |
||||||
|
IsActive bool |
||||||
|
} |
||||||
|
|
||||||
|
// Prepare and load the testing database
|
||||||
|
x, deferable := prepareTestEnv(t, 0, new(User)) |
||||||
|
if x == nil || t.Failed() { |
||||||
|
defer deferable() |
||||||
|
return |
||||||
|
} |
||||||
|
defer deferable() |
||||||
|
|
||||||
|
err := addPrimaryEmail2EmailAddress(x) |
||||||
|
assert.NoError(t, err) |
||||||
|
|
||||||
|
type EmailAddress struct { |
||||||
|
ID int64 `xorm:"pk autoincr"` |
||||||
|
UID int64 `xorm:"INDEX NOT NULL"` |
||||||
|
Email string `xorm:"UNIQUE NOT NULL"` |
||||||
|
LowerEmail string `xorm:"UNIQUE NOT NULL"` |
||||||
|
IsActivated bool |
||||||
|
IsPrimary bool `xorm:"DEFAULT(false) NOT NULL"` |
||||||
|
} |
||||||
|
|
||||||
|
var users = make([]User, 0, 20) |
||||||
|
err = x.Find(&users) |
||||||
|
assert.NoError(t, err) |
||||||
|
|
||||||
|
for _, user := range users { |
||||||
|
var emailAddress EmailAddress |
||||||
|
has, err := x.Where("lower_email=?", strings.ToLower(user.Email)).Get(&emailAddress) |
||||||
|
assert.NoError(t, err) |
||||||
|
assert.True(t, has) |
||||||
|
assert.True(t, emailAddress.IsPrimary) |
||||||
|
assert.EqualValues(t, user.IsActive, emailAddress.IsActivated) |
||||||
|
assert.EqualValues(t, user.ID, emailAddress.UID) |
||||||
|
} |
||||||
|
} |
||||||
Loading…
Reference in new issue