9
9
"xorm.io/xorm"
10
10
)
11
11
12
- func AddPrimaryEmail2EmailAddress (x * xorm.Engine ) ( err error ) {
12
+ func AddPrimaryEmail2EmailAddress (x * xorm.Engine ) error {
13
13
type User struct {
14
14
ID int64 `xorm:"pk autoincr"`
15
15
Email string `xorm:"NOT NULL"`
@@ -26,12 +26,12 @@ func AddPrimaryEmail2EmailAddress(x *xorm.Engine) (err error) {
26
26
}
27
27
28
28
// Add lower_email and is_primary columns
29
- if err = x .Table ("email_address" ).Sync2 (new (EmailAddress1 )); err != nil {
30
- return
29
+ if err : = x .Table ("email_address" ).Sync2 (new (EmailAddress1 )); err != nil {
30
+ return err
31
31
}
32
32
33
- if _ , err = x .Exec ("UPDATE email_address SET lower_email=LOWER(email), is_primary=?" , false ); err != nil {
34
- return
33
+ if _ , err : = x .Exec ("UPDATE email_address SET lower_email=LOWER(email), is_primary=?" , false ); err != nil {
34
+ return err
35
35
}
36
36
37
37
type EmailAddress struct {
@@ -44,8 +44,8 @@ func AddPrimaryEmail2EmailAddress(x *xorm.Engine) (err error) {
44
44
}
45
45
46
46
// change lower_email as unique
47
- if err = x .Sync2 (new (EmailAddress )); err != nil {
48
- return
47
+ if err : = x .Sync2 (new (EmailAddress )); err != nil {
48
+ return err
49
49
}
50
50
51
51
sess := x .NewSession ()
@@ -55,34 +55,33 @@ func AddPrimaryEmail2EmailAddress(x *xorm.Engine) (err error) {
55
55
56
56
for start := 0 ; ; start += batchSize {
57
57
users := make ([]* User , 0 , batchSize )
58
- if err = sess .Limit (batchSize , start ).Find (& users ); err != nil {
59
- return
58
+ if err : = sess .Limit (batchSize , start ).Find (& users ); err != nil {
59
+ return err
60
60
}
61
61
if len (users ) == 0 {
62
62
break
63
63
}
64
64
65
65
for _ , user := range users {
66
- var exist bool
67
- exist , err = sess .Where ("email=?" , user .Email ).Table ("email_address" ).Exist ()
66
+ exist , err := sess .Where ("email=?" , user .Email ).Table ("email_address" ).Exist ()
68
67
if err != nil {
69
- return
68
+ return err
70
69
}
71
70
if ! exist {
72
- if _ , err = sess .Insert (& EmailAddress {
71
+ if _ , err : = sess .Insert (& EmailAddress {
73
72
UID : user .ID ,
74
73
Email : user .Email ,
75
74
LowerEmail : strings .ToLower (user .Email ),
76
75
IsActivated : user .IsActive ,
77
76
IsPrimary : true ,
78
77
}); err != nil {
79
- return
78
+ return err
80
79
}
81
80
} else {
82
- if _ , err = sess .Where ("email=?" , user .Email ).Cols ("is_primary" ).Update (& EmailAddress {
81
+ if _ , err : = sess .Where ("email=?" , user .Email ).Cols ("is_primary" ).Update (& EmailAddress {
83
82
IsPrimary : true ,
84
83
}); err != nil {
85
- return
84
+ return err
86
85
}
87
86
}
88
87
}
0 commit comments