使用用户名或电子邮件地址登录
Logging in with username or email address
我正在尝试改进我的 Go 应用程序中处理登录的代码块。它接收一个 login
值和一个 password
值,然后检查数据库中的 username
字段或 email
字段以查看是否可以找到用户。如果 login
包含 @
符号,则将使用电子邮件。
所以我目前有:
if strings.Contains(login, "@") {
err := collection("users").Find(bson.M{"email" : login}).One(&user)
if err != nil { //no user found
api.WriteError(w, 400, "USER_NOT_FOUND", "No user found with email address: " + login)
return
}
} else {
err := collection("users").Find(bson.M{"username" : login}).One(&user)
if err != nil { //no user found
api.WriteError(w, 400, "USER_NOT_FOUND", "No user found with username: " + login)
return
}
}
//go on and check password
但是你可以看到这不是很有效,因为我不得不处理同一个错误两次。我也试过:
if strings.Contains(login.Identifier, "@") {
err := collection("users").Find(bson.M{}).One(&user)
} else {
err := collection("users").Find(bson.M{"username" : login}).One(&user)
}
if err != nil { //no user found
api.WriteError(w, 400, "USER_NOT_FOUND", "No user found with email address: " + login)
return
}
//go on and check password
但 err
并不总是存在。
我考虑过以某种方式使 {"username" : login}
部分成为一个字符串变量,但我似乎无法让它工作。
是这样的吗?
field := "username"
if strings.Contains(login.Identifier, "@") {
field = "email"
}
err := collection("users").Find(bson.M{field: login}).One(&user)
if err != nil {
api.WriteError(w, 400, "USER_NOT_FOUND", "No user found with " + field + ": " + login)
}
我正在尝试改进我的 Go 应用程序中处理登录的代码块。它接收一个 login
值和一个 password
值,然后检查数据库中的 username
字段或 email
字段以查看是否可以找到用户。如果 login
包含 @
符号,则将使用电子邮件。
所以我目前有:
if strings.Contains(login, "@") {
err := collection("users").Find(bson.M{"email" : login}).One(&user)
if err != nil { //no user found
api.WriteError(w, 400, "USER_NOT_FOUND", "No user found with email address: " + login)
return
}
} else {
err := collection("users").Find(bson.M{"username" : login}).One(&user)
if err != nil { //no user found
api.WriteError(w, 400, "USER_NOT_FOUND", "No user found with username: " + login)
return
}
}
//go on and check password
但是你可以看到这不是很有效,因为我不得不处理同一个错误两次。我也试过:
if strings.Contains(login.Identifier, "@") {
err := collection("users").Find(bson.M{}).One(&user)
} else {
err := collection("users").Find(bson.M{"username" : login}).One(&user)
}
if err != nil { //no user found
api.WriteError(w, 400, "USER_NOT_FOUND", "No user found with email address: " + login)
return
}
//go on and check password
但 err
并不总是存在。
我考虑过以某种方式使 {"username" : login}
部分成为一个字符串变量,但我似乎无法让它工作。
是这样的吗?
field := "username"
if strings.Contains(login.Identifier, "@") {
field = "email"
}
err := collection("users").Find(bson.M{field: login}).One(&user)
if err != nil {
api.WriteError(w, 400, "USER_NOT_FOUND", "No user found with " + field + ": " + login)
}