added db.go file
This commit is contained in:
Derek McQuay
parent
9f98331c59
commit
5985527302
351
db.go
Normal file
351
db.go
Normal file
@ -0,0 +1,351 @@
|
||||
package chipmunk
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
|
||||
"github.com/jmoiron/sqlx"
|
||||
_ "github.com/lib/pq"
|
||||
)
|
||||
|
||||
type DB struct {
|
||||
db *sqlx.DB
|
||||
}
|
||||
|
||||
func NewDB(dbhost, dbname string) (*DB, error) {
|
||||
var err error
|
||||
config := fmt.Sprintf(
|
||||
"dbname=%s host=%s sslmode=disable",
|
||||
dbname,
|
||||
dbhost,
|
||||
)
|
||||
db, err := sqlx.Connect(
|
||||
"postgres",
|
||||
config,
|
||||
)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
d := &DB{db}
|
||||
d.initializeDB()
|
||||
return d, nil
|
||||
}
|
||||
|
||||
func (d *DB) initializeDB() {
|
||||
// XXX ignoring errors
|
||||
d.db.Exec(createdb)
|
||||
d.db.Exec(primeCategories)
|
||||
}
|
||||
|
||||
//func (d *DB) checkOwner(filename, client string) error {
|
||||
// row := d.db.QueryRowx("SELECT client FROM pics WHERE filename = $1", filename)
|
||||
// var owner string
|
||||
// err := row.Scan(&owner)
|
||||
// if err == sql.ErrNoRows {
|
||||
// return errors.New("file not in DB")
|
||||
// }
|
||||
// if owner != strings.ToLower(client) {
|
||||
// return errors.New("filename did not match owner")
|
||||
// }
|
||||
// return nil
|
||||
//}
|
||||
//
|
||||
//func (d *DB) getPic(filename string) (Pic, error) {
|
||||
// p := Pic{}
|
||||
// err := d.db.QueryRowx(
|
||||
// `
|
||||
// SELECT
|
||||
// filename,
|
||||
// category_id as category,
|
||||
// client,
|
||||
// categorized,
|
||||
// reported
|
||||
// FROM pics
|
||||
// WHERE
|
||||
// filename = $1
|
||||
// LIMIT 1
|
||||
// `,
|
||||
// filename,
|
||||
// ).StructScan(&p)
|
||||
// logs.Debug("%+v", p)
|
||||
// if err == sql.ErrNoRows {
|
||||
// return Pic{}, errors.New("pic not found")
|
||||
// }
|
||||
// return p, nil
|
||||
//}
|
||||
//
|
||||
//func (d *DB) votesForPic(filename string) ([]int, error) {
|
||||
// results := []int{}
|
||||
// rows, err := d.db.Queryx(`SELECT vote FROM votes WHERE filename = $1`, filename)
|
||||
// if err != nil {
|
||||
// return []int{}, err
|
||||
// }
|
||||
// for rows.Next() {
|
||||
// var result int
|
||||
// err := rows.Scan(&result)
|
||||
// if err != nil {
|
||||
// return []int{}, err
|
||||
// }
|
||||
// results = append(results, result)
|
||||
// }
|
||||
// return results, nil
|
||||
//}
|
||||
//
|
||||
//func (d *DB) closeVoting(filename string) error {
|
||||
// _, err := d.db.Exec(
|
||||
// `UPDATE pics SET categorized = true WHERE filename = $1`,
|
||||
// filename,
|
||||
// )
|
||||
// return err
|
||||
//}
|
||||
//
|
||||
//func (d *DB) votingClosed(filename string) (bool, error) {
|
||||
// row := d.db.QueryRow("SELECT categorized FROM pics WHERE filename = $1", filename)
|
||||
// already := false
|
||||
// err := row.Scan(&already)
|
||||
// return already, err
|
||||
//}
|
||||
//
|
||||
//type Token string
|
||||
//
|
||||
//func (d *DB) tokenForFile(filename string) (Token, error) {
|
||||
// row := d.db.QueryRow(`
|
||||
// SELECT
|
||||
// clients.token
|
||||
// FROM pics, clients
|
||||
// WHERE
|
||||
// pics.client = clients.id
|
||||
// AND pics.filename = $1
|
||||
// `,
|
||||
// filename,
|
||||
// )
|
||||
// token := sql.NullString{}
|
||||
// err := row.Scan(&token)
|
||||
// return Token(token.String), err
|
||||
//}
|
||||
//
|
||||
//type Pic struct {
|
||||
// Filename string `json:"filename"`
|
||||
// Category int `json:"category"`
|
||||
// Client string `json:"client"`
|
||||
// Categorized bool `json:"categorized"db:"categorized"`
|
||||
// Reported bool `json:"reported"db:"reported"`
|
||||
//}
|
||||
//
|
||||
//type PicDetails struct {
|
||||
// // did the request work; this goes along with HTTP status codes
|
||||
// Success bool `json:"success"`
|
||||
//
|
||||
// Filename string `json:"filename"`
|
||||
//
|
||||
// Message string `json:"message"`
|
||||
// Works bool `json:"works"`
|
||||
// Categorized bool `json:"categorized"`
|
||||
//}
|
||||
//
|
||||
//func (d *DB) getPicDetails(filename string) (PicDetails, error) {
|
||||
// pd := PicDetails{
|
||||
// Success: false,
|
||||
// Filename: filename,
|
||||
// Message: "unknown ranking",
|
||||
// }
|
||||
// votes, err := d.votesForPic(filename)
|
||||
// if err != nil {
|
||||
// return pd, err
|
||||
// }
|
||||
// message, works, done, err := categorize(votes)
|
||||
// if err != nil {
|
||||
// return pd, err
|
||||
// }
|
||||
// pd.Success = true
|
||||
// pd.Message = message
|
||||
// pd.Works = works
|
||||
// pd.Categorized = done
|
||||
// return pd, nil
|
||||
//}
|
||||
//
|
||||
//type VoteReq struct {
|
||||
// Filename string `json:"filename"`
|
||||
// Client string `json:"client"`
|
||||
// Value int `json:"value"`
|
||||
//}
|
||||
//
|
||||
//type Category struct {
|
||||
// Id int `json:"id"db:"id"`
|
||||
// Name string `json:"name"db:"name"`
|
||||
//}
|
||||
//
|
||||
//func (d *DB) getCategories() (map[string]int, error) {
|
||||
// results := map[string]int{}
|
||||
// rows, err := d.db.Queryx("SELECT id, name FROM categories")
|
||||
// if err != nil {
|
||||
// return nil, err
|
||||
// }
|
||||
// for rows.Next() {
|
||||
// var result Category
|
||||
// err := rows.StructScan(&result)
|
||||
// if err != nil {
|
||||
// return nil, err
|
||||
// }
|
||||
// results[result.Name] = result.Id
|
||||
// }
|
||||
// return results, nil
|
||||
//}
|
||||
//
|
||||
//func (d *DB) flagPic(filename string, reporter string) (int, error) {
|
||||
// logs.Debug("(%s, %s)", filename, reporter)
|
||||
//
|
||||
// var count int
|
||||
// row := d.db.QueryRow(
|
||||
// "SELECT COUNT(*) FROM reports WHERE filename = $1 AND reporter = $2",
|
||||
// filename,
|
||||
// reporter,
|
||||
// )
|
||||
// err := row.Scan(&count)
|
||||
// if err != nil {
|
||||
// return http.StatusInternalServerError, err
|
||||
// }
|
||||
// if count > 0 {
|
||||
// return http.StatusBadRequest, fmt.Errorf("%s already reported %s", reporter, filename)
|
||||
// }
|
||||
//
|
||||
// _, err = d.db.Exec(
|
||||
// `INSERT INTO reports (filename, reporter) VALUES ($1, $2)`,
|
||||
// filename,
|
||||
// reporter,
|
||||
// )
|
||||
// if err != nil {
|
||||
// return http.StatusBadRequest, err
|
||||
// }
|
||||
//
|
||||
// var culprit string
|
||||
// err = d.db.Get(&culprit, `
|
||||
// SELECT
|
||||
// client
|
||||
// FROM pics, clients
|
||||
// WHERE
|
||||
// pics.client = clients.id AND
|
||||
// pics.filename = $1
|
||||
// `,
|
||||
// filename,
|
||||
// )
|
||||
// if err != nil {
|
||||
// return http.StatusInternalServerError, err
|
||||
// }
|
||||
//
|
||||
// _, err = d.db.Exec(`
|
||||
// UPDATE
|
||||
// clients
|
||||
// SET
|
||||
// infractions = infractions + 1
|
||||
// WHERE
|
||||
// clients.id = $1
|
||||
// `,
|
||||
// culprit,
|
||||
// )
|
||||
// if err != nil {
|
||||
// return http.StatusInternalServerError, err
|
||||
// }
|
||||
//
|
||||
// _, err = d.db.Exec(`
|
||||
// UPDATE
|
||||
// pics
|
||||
// SET
|
||||
// reported = true
|
||||
// WHERE
|
||||
// filename = $1
|
||||
// `,
|
||||
// filename,
|
||||
// )
|
||||
// if err != nil {
|
||||
// return http.StatusInternalServerError, err
|
||||
// }
|
||||
// return http.StatusCreated, nil
|
||||
//}
|
||||
//
|
||||
//type client struct {
|
||||
// Id string
|
||||
// Token sql.NullString
|
||||
// Infractions int
|
||||
// db *sqlx.DB
|
||||
//}
|
||||
//
|
||||
//const reportThreshold = 3
|
||||
//
|
||||
//func (c *client) isBlocked() bool {
|
||||
// if c.Infractions >= reportThreshold {
|
||||
// return true
|
||||
// }
|
||||
// return false
|
||||
//}
|
||||
//
|
||||
//func (c *client) MarshalJSON() ([]byte, error) {
|
||||
// r := struct {
|
||||
// Id string `json:"id"`
|
||||
// Infractions int `json:"infractions"`
|
||||
// }{
|
||||
// Id: c.Id,
|
||||
// Infractions: c.Infractions,
|
||||
// }
|
||||
//
|
||||
// return json.Marshal(r)
|
||||
//}
|
||||
//
|
||||
//func (c *client) SetToken(token string) error {
|
||||
// _, err := c.db.Exec(
|
||||
// `UPDATE clients SET token = $1 WHERE id = $2`,
|
||||
// token,
|
||||
// c.Id,
|
||||
// )
|
||||
// if err != nil {
|
||||
// return err
|
||||
// }
|
||||
// c.Token = sql.NullString{
|
||||
// String: token,
|
||||
// Valid: true,
|
||||
// }
|
||||
// return nil
|
||||
//}
|
||||
//
|
||||
//// getOrCreateClient performs db operations requisite to fetch current info or
|
||||
//// create a new client.
|
||||
////
|
||||
//// returns:
|
||||
//// *client, created or not, error
|
||||
//func (d *DB) getOrCreateClient(clientId string) (*client, bool, error) {
|
||||
// var created bool
|
||||
// c := &client{
|
||||
// db: d.db,
|
||||
// }
|
||||
// err := d.db.QueryRowx(
|
||||
// "SELECT * FROM clients WHERE id = $1",
|
||||
// clientId,
|
||||
// ).StructScan(c)
|
||||
// if err == sql.ErrNoRows {
|
||||
// _, err = d.db.Exec(
|
||||
// `INSERT INTO clients (id) VALUES ($1)`,
|
||||
// clientId,
|
||||
// )
|
||||
// if err != nil {
|
||||
// return nil, false, fmt.Errorf(
|
||||
// "problem inserting client: %s: %+v",
|
||||
// clientId,
|
||||
// err,
|
||||
// )
|
||||
// }
|
||||
// created = true
|
||||
// err := d.db.QueryRowx(
|
||||
// "SELECT * FROM clients WHERE id = $1",
|
||||
// clientId,
|
||||
// ).StructScan(c)
|
||||
// if err != nil {
|
||||
// return nil, true, err
|
||||
// }
|
||||
// }
|
||||
// return c, created, nil
|
||||
//}
|
||||
//
|
||||
//type UploadResp struct {
|
||||
// Success bool `json:"success"`
|
||||
// Sha string `json:"sha1"`
|
||||
//}
|
||||
Loading…
Reference in New Issue
Block a user