budgl/models/people.go

39 lines
782 B
Go

package models
type People struct {
PeopID int
PeopName string
}
func GetPeople() ([]*People, error) {
rows, err := db.Query("SELECT * FROM people")
if err != nil {
return nil, err
}
defer rows.Close()
ppls := make([]*People, 0)
for rows.Next() {
ppl := new(People)
err := rows.Scan(&ppl.PeopID, &ppl.PeopName)
if err != nil {
return nil, err
}
ppls = append(ppls, ppl)
}
if err = rows.Err(); err != nil {
return nil, err
}
return ppls, nil
}
func AddPeople() error {
sqlStat := "INSERT INTO people (peopname) VALUES ($1) RETURNING uid"
var uid int
err := db.QueryRow(sqlStat).Scan(&uid)
if err != nil {
return err
}
return nil
}