added ability to get single tranx with id

This commit is contained in:
Derek McQuay 2017-02-07 21:28:37 -08:00
parent c2f8d2eca0
commit 2bb365206e
No known key found for this signature in database
GPG Key ID: 92A7BC0C86B0B91A
2 changed files with 43 additions and 3 deletions

25
api.go
View File

@ -4,6 +4,7 @@ import (
"encoding/json" "encoding/json"
"log" "log"
"net/http" "net/http"
"strings"
"time" "time"
) )
@ -195,14 +196,32 @@ func (s *Server) tranx(w http.ResponseWriter, req *http.Request) {
http.Error(w, string(b), http.StatusBadRequest) http.Error(w, string(b), http.StatusBadRequest)
return return
case "GET": case "GET":
tranxs, err := s.db.getTranxs() searchreq := req.URL.Path[len(prefix["tranx"]):]
if len(searchreq) == 0 {
tranxs, err := s.db.getTranxs()
if err != nil {
log.Printf("%+v", err)
b, _ := json.Marshal(NewFailure(err.Error()))
http.Error(w, string(b), http.StatusInternalServerError)
return
}
json.NewEncoder(w).Encode(tranxs)
return
}
if searchreq[len(searchreq)-1] != '/' {
http.Redirect(w, req, prefix["tranx"]+searchreq+"/", http.StatusMovedPermanently)
return
}
searchReqParsed := strings.Split(searchreq, "/")
t, err := s.db.getTranx(searchReqParsed[0])
if err != nil { if err != nil {
log.Printf("%+v", err) log.Printf("%+v", err)
b, _ := json.Marshal(NewFailure(err.Error())) b, _ := json.Marshal(NewFailure(err.Error()))
http.Error(w, string(b), http.StatusInternalServerError) http.Error(w, string(b), http.StatusBadRequest)
return return
} }
json.NewEncoder(w).Encode(tranxs) json.NewEncoder(w).Encode(t)
case "POST": case "POST":
t := tranx{} t := tranx{}
err := json.NewDecoder(req.Body).Decode(&t) err := json.NewDecoder(req.Body).Decode(&t)

21
db.go
View File

@ -127,6 +127,27 @@ func (d *DB) getTranxs() ([]tranx, error) {
return results, nil return results, nil
} }
func (d *DB) getTranx(i string) (tranx, error) {
result := tranx{}
row := d.db.QueryRow("SELECT id, cost, store, info, date, user_id, category_id FROM tranx WHERE id = $1",
i,
)
err := row.Scan(
&result.ID,
&result.Cost,
&result.Store,
&result.Info,
&result.Date,
&result.User_ID,
&result.Category_ID,
)
if err != nil {
return tranx{}, err
}
return result, nil
}
//func (d *DB) checkOwner(filename, client string) error { //func (d *DB) checkOwner(filename, client string) error {
// row := d.db.QueryRowx("SELECT client FROM pics WHERE filename = $1", filename) // row := d.db.QueryRowx("SELECT client FROM pics WHERE filename = $1", filename)
// var owner string // var owner string