Actually use pg
This commit is contained in:
parent
0f1bd4cac3
commit
c3664ef3e0
14
postgres/db.go
Normal file
14
postgres/db.go
Normal file
@ -0,0 +1,14 @@
|
|||||||
|
package postgres
|
||||||
|
|
||||||
|
import "gopkg.in/pg.v5"
|
||||||
|
|
||||||
|
// Connect to a database
|
||||||
|
func Connect(server string, user string, password string, database string) *pg.DB {
|
||||||
|
db := pg.Connect(&pg.Options{
|
||||||
|
User: user,
|
||||||
|
Password: password,
|
||||||
|
Addr: server,
|
||||||
|
Database: database,
|
||||||
|
})
|
||||||
|
return db
|
||||||
|
}
|
@ -1,22 +1,20 @@
|
|||||||
package postgres
|
package postgres
|
||||||
|
|
||||||
import (
|
import "git.javil.eu/jacob1123/budgeteer"
|
||||||
"database/sql"
|
|
||||||
|
|
||||||
"git.javil.eu/jacob1123/budgeteer"
|
import "gopkg.in/pg.v5"
|
||||||
)
|
|
||||||
|
|
||||||
// UserService represents a PostgreSQL implementation of myapp.UserService.
|
// UserService represents a PostgreSQL implementation of myapp.UserService.
|
||||||
type UserService struct {
|
type UserService struct {
|
||||||
DB *sql.DB
|
DB *pg.DB
|
||||||
}
|
}
|
||||||
|
|
||||||
// User returns a user for a given id.
|
// User returns a user for a given id.
|
||||||
func (s *UserService) User(id int) (*budgeteer.User, error) {
|
func (s *UserService) User(id int) (*budgeteer.User, error) {
|
||||||
var u budgeteer.User
|
u := &budgeteer.User{ID: id}
|
||||||
row := s.DB.QueryRow(`SELECT id, email, password, name FROM users WHERE id = $1`, id)
|
err := s.DB.Select(&u)
|
||||||
if err := row.Scan(&u.ID, &u.Email, &u.Password, &u.Name); err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
return &u, nil
|
return u, nil
|
||||||
}
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user