devstack/authorizer
4
0
Fork 0
Code Pull Requests Actions Activity

Add query for verification requests

Browse Source 
Resolves #28
This commit is contained in:
Lakhan Samani
2021-07-18 09:52:54 +05:30
parent c6cbcd2e66
commit e2fc610762
14 changed files with 187 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
server/db/db.go
View File

@@ -18,9 +18,10 @@ type Manager interface {
GetUsers() ([]User, error)
GetUserByEmail(email string) (User, error)
UpdateVerificationTime(verifiedAt int64, id uint) error
AddVerification(verification Verification) (Verification, error)
GetVerificationByToken(token string) (Verification, error)
AddVerification(verification VerificationRequest) (VerificationRequest, error)
GetVerificationByToken(token string) (VerificationRequest, error)
DeleteToken(email string) error
GetVerificationRequests() ([]VerificationRequest, error)
}
type manager struct {
@@ -50,7 +51,7 @@ func init() {
if err != nil {
log.Fatal("Failed to init db:", err)
} else {
db.AutoMigrate(&User{}, &Verification{})
db.AutoMigrate(&User{}, &VerificationRequest{})
}
Mgr = &manager{db: db}

21
server/db/verification.go → server/db/verificationRequests.go
View File

@@ -6,7 +6,7 @@ import (
"gorm.io/gorm/clause"
)
type Verification struct {
type VerificationRequest struct {
ID uint `gorm:"primaryKey"`
Token string `gorm:"index"`
Identifier string
@@ -17,7 +17,7 @@ type Verification struct {
}
// AddVerification function to add verification record
func (mgr *manager) AddVerification(verification Verification) (Verification, error) {
func (mgr *manager) AddVerification(verification VerificationRequest) (VerificationRequest, error) {
result := mgr.db.Clauses(clause.OnConflict{
Columns: []clause.Column{{Name: "email"}},
DoUpdates: clause.AssignmentColumns([]string{"token", "identifier", "expires_at"}),
@@ -29,8 +29,8 @@ func (mgr *manager) AddVerification(verification Verification) (Verification, er
return verification, nil
}
func (mgr *manager) GetVerificationByToken(token string) (Verification, error) {
var verification Verification
func (mgr *manager) GetVerificationByToken(token string) (VerificationRequest, error) {
var verification VerificationRequest
result := mgr.db.Where("token = ?", token).First(&verification)
if result.Error != nil {
@@ -42,7 +42,7 @@ func (mgr *manager) GetVerificationByToken(token string) (Verification, error) {
}
func (mgr *manager) DeleteToken(email string) error {
var verification Verification
var verification VerificationRequest
result := mgr.db.Where("email = ?", email).Delete(&verification)
if result.Error != nil {
@@ -52,3 +52,14 @@ func (mgr *manager) DeleteToken(email string) error {
return nil
}
// GetUsers function to get all users
func (mgr *manager) GetVerificationRequests() ([]VerificationRequest, error) {
var verificationRequests []VerificationRequest
result := mgr.db.Find(&verificationRequests)
if result.Error != nil {
log.Println(result.Error)
return verificationRequests, result.Error
}
return verificationRequests, nil
}