Generate voter data for known elections and simplify MaxVoters (0 = no maximum)
This commit is contained in:
36
internal/models/voters.go
Normal file
36
internal/models/voters.go
Normal file
@ -0,0 +1,36 @@
|
||||
package models
|
||||
|
||||
import (
|
||||
"database/sql"
|
||||
)
|
||||
|
||||
type VoterModelInterface interface {
|
||||
Insert(identity string, electionID int) (int, error)
|
||||
}
|
||||
|
||||
type VoterModel struct {
|
||||
DB *sql.DB
|
||||
}
|
||||
|
||||
func (v *VoterModel) Insert(identity string, electionID int) (int, error) {
|
||||
tx, err := v.DB.Begin()
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
defer tx.Rollback()
|
||||
|
||||
result, err := tx.Exec(`
|
||||
INSERT INTO voters (identity, election_id)
|
||||
VALUES (?, ?)`,
|
||||
identity, electionID)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
|
||||
if err = tx.Commit(); err != nil {
|
||||
return 0, err
|
||||
}
|
||||
|
||||
voterId, err := result.LastInsertId()
|
||||
return int(voterId), nil
|
||||
}
|
Reference in New Issue
Block a user