コード例 #1
0
// Handlers start here
func getAttributes(params martini.Params, writer http.ResponseWriter, db *mgo.Database) (int, string) {
	resource := strings.ToLower(params["resource"])
	setJsonResponseHeader(writer)

	attrs := resourceAttributes{}
	err := db.C(dbInfo.Collection).Find(bson.M{"resource": resource}).One(&attrs)

	if err == nil {
		return http.StatusOK, utils.JsonString(attrs)
	} else {
		return http.StatusNotFound, utils.JsonString(errorMsg{"No attributes found for the resource: " + resource})
	}
}
コード例 #2
0
func addAttribute(attr attribute, err binding.Errors, params martini.Params, writer http.ResponseWriter, db *mgo.Database) (int, string) {
	setJsonResponseHeader(writer)

	if err.Count() > 0 {
		return http.StatusConflict, utils.JsonString(errorMsg{err.Overall["missing-requirement"]})
	}

	resource := strings.ToLower(params["resource"])
	query := bson.M{"resource": resource}
	update := mgo.Change{Upsert: true, Update: bson.M{"$addToSet": bson.M{"attributes": attr}}}

	if _, dbErr := db.C(dbInfo.Collection).Find(query).Apply(update, &attr); dbErr != nil {
		return http.StatusConflict, utils.JsonString(errorMsg{dbErr.Error()})
	}

	return http.StatusOK, "{}"
}