func (uc SubscriptionController) GetSubscription(w http.ResponseWriter, r *http.Request) {
vars := mux.Vars(r)
id := vars["id"]
// Verify id is ObjectId, otherwise bail
if !bson.IsObjectIdHex(id) {
w.WriteHeader(404)
return
}
// Grab id
oid := bson.ObjectIdHex(id)
// Stub subscription
u := models.Subscription{}
// Fetch subscription
if err := uc.session.DB("prizma").C("subscriptions").FindId(oid).One(&u); err != nil {
w.WriteHeader(404)
return
}
// Marshal provided interface into JSON structure
uj, _ := json.Marshal(u)
// Write content-type, statuscode, payload
w.Header().Set("Content-Type", "application/json")
w.WriteHeader(200)
fmt.Fprintf(w, "%s", uj)
}
func (uc PublicationController) ListUserPublications(w http.ResponseWriter, r *http.Request) {
vars := mux.Vars(r)
id := vars["id"]
// Verify id is ObjectId, otherwise bail
if !bson.IsObjectIdHex(id) {
w.WriteHeader(404)
return
}
// Grab id
oid := bson.ObjectIdHex(id)
// Stub publication
results := make([]models.Publication, 0)
// Fetch publications
if err := uc.session.DB("prizma").C("publications").Find(bson.M{"publisherId": oid}).All(&results); err != nil {
w.WriteHeader(404)
return
}
// Marshal provided interface into JSON structure
uj, _ := json.Marshal(results)
// Write content-type, statuscode, payload
w.Header().Set("Content-Type", "application/json")
w.WriteHeader(200)
fmt.Fprintf(w, "%s", uj)
}
// RemovePublication removes an existing publication resource
func (uc PublicationController) UpdatePublication(w http.ResponseWriter, r *http.Request) {
// Grab id
vars := mux.Vars(r)
id := vars["pid"]
// Stub an publication to be populated from the body
u := models.Publication{}
// Populate the publication data
json.NewDecoder(r.Body).Decode(&u)
// Verify id is ObjectId, otherwise bail
if !bson.IsObjectIdHex(id) {
w.WriteHeader(404)
return
}
// Grab id
oid := bson.ObjectIdHex(id)
// update publication
if err := uc.session.DB("prizma").C("publications").UpdateId(oid, u); err != nil {
w.WriteHeader(404)
return
}
// Write status
w.WriteHeader(200)
}
// RemoveSubscription removes an existing subscription resource
func (uc SubscriptionController) RemoveSubscription(w http.ResponseWriter, r *http.Request) {
// Grab id
vars := mux.Vars(r)
id := vars["id"]
// Verify id is ObjectId, otherwise bail
if !bson.IsObjectIdHex(id) {
w.WriteHeader(404)
return
}
// Grab id
oid := bson.ObjectIdHex(id)
// Remove subscription
if err := uc.session.DB("prizma").C("subscriptions").RemoveId(oid); err != nil {
w.WriteHeader(404)
return
}
// Write status
w.WriteHeader(200)
}
func GetById(s *mgo.Session, e models.Entity, collection string, w http.ResponseWriter, r *http.Request) {
vars := mux.Vars(r)
id := vars["id"]
// Verify id is ObjectId, otherwise bail
if !bson.IsObjectIdHex(id) {
w.WriteHeader(404)
return
}
// Grab id
oid := bson.ObjectIdHex(id)
// Fetch user
if err := s.DB("prizma").C(collection).FindId(oid).One(&e); err != nil {
w.WriteHeader(404)
return
}
// Marshal provided interface into JSON structure
uj, _ := json.Marshal(e)
// Write content-type, statuscode, payload
w.Header().Set("Content-Type", "application/json")
w.WriteHeader(200)
fmt.Fprintf(w, "%s", uj)
}
// PurgeMissing removes from collections any state that refers to transaction
// documents that for whatever reason have been lost from the system (removed
// by accident or lost in a hard crash, for example).
//
// This method should very rarely be needed, if at all, and should never be
// used during the normal operation of an application. Its purpose is to put
// a system that has seen unavoidable corruption back in a working state.
func (r *Runner) PurgeMissing(collections ...string) error {
type M map[string]interface{}
type S []interface{}
type TDoc struct {
Id interface{} "_id"
TxnQueue []string "txn-queue"
}
found := make(map[bson.ObjectId]bool)
sort.Strings(collections)
for _, collection := range collections {
c := r.tc.Database.C(collection)
iter := c.Find(nil).Select(bson.M{"_id": 1, "txn-queue": 1}).Iter()
var tdoc TDoc
for iter.Next(&tdoc) {
for _, txnToken := range tdoc.TxnQueue {
txnId := bson.ObjectIdHex(txnToken[:24])
if found[txnId] {
continue
}
if r.tc.FindId(txnId).One(nil) == nil {
found[txnId] = true
continue
}
logf("WARNING: purging from document %s/%v the missing transaction id %s", collection, tdoc.Id, txnId)
err := c.UpdateId(tdoc.Id, M{"$pull": M{"txn-queue": M{"$regex": "^" + txnId.Hex() + "_*"}}})
if err != nil {
return fmt.Errorf("error purging missing transaction %s: %v", txnId.Hex(), err)
}
}
}
if err := iter.Close(); err != nil {
return fmt.Errorf("transaction queue iteration error for %s: %v", collection, err)
}
}
type StashTDoc struct {
Id docKey "_id"
TxnQueue []string "txn-queue"
}
iter := r.sc.Find(nil).Select(bson.M{"_id": 1, "txn-queue": 1}).Iter()
var stdoc StashTDoc
for iter.Next(&stdoc) {
for _, txnToken := range stdoc.TxnQueue {
txnId := bson.ObjectIdHex(txnToken[:24])
if found[txnId] {
continue
}
if r.tc.FindId(txnId).One(nil) == nil {
found[txnId] = true
continue
}
logf("WARNING: purging from stash document %s/%v the missing transaction id %s", stdoc.Id.C, stdoc.Id.Id, txnId)
err := r.sc.UpdateId(stdoc.Id, M{"$pull": M{"txn-queue": M{"$regex": "^" + txnId.Hex() + "_*"}}})
if err != nil {
return fmt.Errorf("error purging missing transaction %s: %v", txnId.Hex(), err)
}
}
}
if err := iter.Close(); err != nil {
return fmt.Errorf("transaction stash iteration error: %v", err)
}
return nil
}
func (tt token) id() bson.ObjectId { return bson.ObjectIdHex(string(tt[:24])) }
