63 lines
979 B
Go
63 lines
979 B
Go
package services
|
|
|
|
import (
|
|
"database/sql"
|
|
"errors"
|
|
|
|
"go-sjles-pta-vote/server/db"
|
|
"go-sjles-pta-vote/server/models"
|
|
)
|
|
|
|
func CreatePoll(poll *models.Poll) (*models.Poll, error) {
|
|
new_poll := models.Poll{}
|
|
|
|
db_conn, err := db.Connect()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer db.Close()
|
|
|
|
get_stmt, err := db_conn.Prepare(`
|
|
SELECT id
|
|
FROM polls
|
|
WHERE question == $1
|
|
`)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer get_stmt.Close()
|
|
|
|
var id int
|
|
err = get_stmt.QueryRow(poll.Question).Scan(&id)
|
|
if err != sql.ErrNoRows {
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return nil, errors.New("Already exists")
|
|
}
|
|
|
|
stmt, err := db_conn.Prepare(`
|
|
INSERT INTO polls (
|
|
question,
|
|
expires_at
|
|
) VALUES (
|
|
$1,
|
|
$2
|
|
) RETURNING ID;
|
|
`)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
defer stmt.Close()
|
|
|
|
res, err := stmt.Exec(poll.Question, poll.ExpiresAt)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
new_poll.ID, err = res.LastInsertId()
|
|
|
|
return &new_poll, err
|
|
} |