/
booking.go
93 lines (76 loc) · 1.99 KB
/
booking.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
package qrtickets
import (
"google.golang.org/appengine"
"google.golang.org/appengine/datastore"
"time"
"net/http"
"golang.org/x/net/context"
)
// Booking - A performer scheduled to perform at an event
type Booking struct {
EventKey datastore.Key `json:"event"`
PerformerKey datastore.Key `json:"performer"`
SetTime time.Time `json:"setTime"`
DatastoreKey datastore.Key `json:"booking_id" datastore:"-"`
}
// AddBooking - Associate a performer to an event
func AddBooking(w http.ResponseWriter, r *http.Request) {
ctx := appengine.NewContext(r)
const timeformat = "2006-01-02 15:04:05 -0700"
st, _ := time.Parse(timeformat, r.FormValue("set_time"))
// Create the Venue object
p1 := Booking{
SetTime: st,
}
// Load the Performer Key
if len(r.FormValue("performer")) > 0 {
performer, err := datastore.DecodeKey(r.FormValue("performer"))
if err != nil {
JSONError(&w, err.Error())
return
}
p1.PerformerKey = *performer
}
// Load the Event Key
if len(r.FormValue("event")) > 0 {
event, err := datastore.DecodeKey(r.FormValue("event"))
if err != nil {
JSONError(&w, err.Error())
return
}
p1.EventKey = *event
}
// Add the booking to the Datastore
k, err := p1.Store(ctx)
if err != nil {
JSONError(&w, err.Error())
return
}
p1.DatastoreKey = k
return
}
// Store - Stores the current booking into Google datastore
func (e *Booking) Store(ctx context.Context) (datastore.Key, error) {
var k *datastore.Key
// See if a key exists, or if a new one is required
if e.DatastoreKey.Incomplete() {
k = datastore.NewIncompleteKey(ctx, "Booking", nil)
} else {
k = &e.DatastoreKey
}
// Stash the entry in the datastore
key, err := datastore.Put(ctx, k, e)
if err != nil {
panic(err)
}
return *key, nil
}
// Load - Loads the booking from Google datastore into the booking object
func (e *Booking) Load(ctx context.Context, k datastore.Key) error {
err := datastore.Get(ctx, &k, e)
e.DatastoreKey = k
if err != nil {
return err
}
return nil
}