added authentication among other things. considered working at this point.
This commit is contained in:
37
models/user.go
Normal file
37
models/user.go
Normal file
@@ -0,0 +1,37 @@
|
||||
package models
|
||||
|
||||
import (
|
||||
"database/sql"
|
||||
"log"
|
||||
)
|
||||
|
||||
type User struct {
|
||||
Id int
|
||||
Username string
|
||||
PasswordHash string
|
||||
}
|
||||
|
||||
var db *sql.DB
|
||||
|
||||
func SetDB(database *sql.DB) {
|
||||
db = database
|
||||
}
|
||||
|
||||
func CreateUser(username, passwordHash string) error {
|
||||
_, err := db.Exec("INSERT INTO users (username, password_hash) VALUES (?, ?)", username, passwordHash)
|
||||
return err
|
||||
}
|
||||
|
||||
func GetUserByUsername(username string) *User {
|
||||
row := db.QueryRow("SELECT id, username, password_hash FROM users WHERE username = ?", username)
|
||||
|
||||
var user User
|
||||
err := row.Scan(&user.Id, &user.Username, &user.PasswordHash)
|
||||
if err != nil {
|
||||
if err != sql.ErrNoRows {
|
||||
log.Println("DB error:", err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
return &user
|
||||
}
|
||||
Reference in New Issue
Block a user