// List returns a list of factors (weights) per endpoint group (i.e. site) func List(r *http.Request, cfg config.Config) (int, http.Header, []byte, error) { //STANDARD DECLARATIONS START code := http.StatusOK h := http.Header{} output := []byte("") err := error(nil) contentType := "text/xml" charset := "utf-8" //STANDARD DECLARATIONS END contentType, err = respond.ParseAcceptHeader(r) h.Set("Content-Type", fmt.Sprintf("%s; charset=%s", contentType, charset)) if err != nil { code = http.StatusNotAcceptable output, _ = respond.MarshalContent(respond.NotAcceptableContentType, contentType, "", " ") return code, h, output, err } tenantDbConfig, err := authentication.AuthenticateTenant(r.Header, cfg) if err != nil { output = []byte(http.StatusText(http.StatusUnauthorized)) code = http.StatusUnauthorized //If wrong api key is passed we return UNAUTHORIZED http status h.Set("Content-Type", fmt.Sprintf("%s; charset=%s", contentType, charset)) return code, h, output, err } session, err := mongo.OpenSession(tenantDbConfig) if err != nil { code = http.StatusInternalServerError return code, h, output, err } defer mongo.CloseSession(session) results := []FactorsOutput{} err = mongo.Find(session, tenantDbConfig.Db, "weights", nil, "name", &results) if err != nil { code = http.StatusInternalServerError return code, h, output, err } output, err = createView(results, contentType) //Render the results into XML format if err != nil { code = http.StatusInternalServerError return code, h, output, err } mongo.CloseSession(session) return code, h, output, err }
// List existing recomputations func List(r *http.Request, cfg config.Config) (int, http.Header, []byte, error) { //STANDARD DECLARATIONS START code := http.StatusOK h := http.Header{} output := []byte("") err := error(nil) // contentType := "application/json" charset := "utf-8" //STANDARD DECLARATIONS END urlValues := r.URL.Query() contentType := r.Header.Get("Accept") contentType, err = respond.ParseAcceptHeader(r) h.Set("Content-Type", fmt.Sprintf("%s; charset=%s", contentType, charset)) if err != nil { code = http.StatusNotAcceptable output, _ = respond.MarshalContent(respond.NotAcceptableContentType, contentType, "", " ") return code, h, output, err } tenantDbConfig, err := authentication.AuthenticateTenant(r.Header, cfg) if err != nil { output, _ = respond.MarshalContent(respond.UnauthorizedMessage, contentType, "", " ") code = http.StatusUnauthorized //If wrong api key is passed we return UNAUTHORIZED http status return code, h, output, err } filter := IncomingRecomputation{ StartTime: urlValues.Get("start_time"), EndTime: urlValues.Get("end_time"), Reason: urlValues.Get("reason"), Report: urlValues.Get("report"), } session, err := mongo.OpenSession(tenantDbConfig) if err != nil { code = http.StatusInternalServerError return code, h, output, err } results := []MongoInterface{} err = mongo.Find(session, tenantDbConfig.Db, recomputationsColl, filter, "timestamp", &results) if err != nil { code = http.StatusInternalServerError return code, h, output, err } output, err = createListView(results, contentType) return code, h, output, err }
// List returns a list of factors (weights) per endpoint group (i.e. site) func List(r *http.Request, cfg config.Config) (int, http.Header, []byte, error) { //STANDARD DECLARATIONS START code := http.StatusOK h := http.Header{} output := []byte("") err := error(nil) contentType := "text/xml" charset := "utf-8" //STANDARD DECLARATIONS END contentType, err = respond.ParseAcceptHeader(r) h.Set("Content-Type", fmt.Sprintf("%s; charset=%s", contentType, charset)) if err != nil { code = http.StatusNotAcceptable output, _ = respond.MarshalContent(respond.NotAcceptableContentType, contentType, "", " ") return code, h, output, err } // Grab Tenant DB configuration from context tenantDbConfig := context.Get(r, "tenant_conf").(config.MongoConfig) session, err := mongo.OpenSession(tenantDbConfig) if err != nil { code = http.StatusInternalServerError return code, h, output, err } defer mongo.CloseSession(session) results := []FactorsOutput{} err = mongo.Find(session, tenantDbConfig.Db, "weights", nil, "name", &results) if err != nil { code = http.StatusInternalServerError return code, h, output, err } output, err = createView(results, contentType) //Render the results into XML format if err != nil { code = http.StatusInternalServerError return code, h, output, err } mongo.CloseSession(session) return code, h, output, err }
// SubmitRecomputation insert a new pending recomputation in the tenants database func SubmitRecomputation(r *http.Request, cfg config.Config) (int, http.Header, []byte, error) { //STANDARD DECLARATIONS START code := http.StatusAccepted h := http.Header{} output := []byte("") err := error(nil) charset := "utf-8" //STANDARD DECLARATIONS END contentType, err := respond.ParseAcceptHeader(r) h.Set("Content-Type", fmt.Sprintf("%s; charset=%s", contentType, charset)) if err != nil { code = http.StatusNotAcceptable output, _ = respond.MarshalContent(respond.NotAcceptableContentType, contentType, "", " ") return code, h, output, err } tenantDbConfig, err := authentication.AuthenticateTenant(r.Header, cfg) if err != nil { output, _ = respond.MarshalContent(respond.UnauthorizedMessage, contentType, "", " ") code = http.StatusUnauthorized //If wrong api key is passed we return UNAUTHORIZED http status return code, h, output, err } session, err := mongo.OpenSession(tenantDbConfig) if err != nil { code = http.StatusInternalServerError return code, h, output, err } var recompSubmission IncomingRecomputation // urlValues := r.URL.Query() body, err := ioutil.ReadAll(io.LimitReader(r.Body, cfg.Server.ReqSizeLimit)) if err != nil { panic(err) } if err := r.Body.Close(); err != nil { panic(err) } if err := json.Unmarshal(body, &recompSubmission); err != nil { code = 422 // unprocessable entity output = []byte("Unprocessable JSON") return code, h, output, err } now := time.Now() recomputation := MongoInterface{ ID: mongo.NewUUID(), RequesterName: tenantDbConfig.User, RequesterEmail: tenantDbConfig.Email, StartTime: recompSubmission.StartTime, EndTime: recompSubmission.EndTime, Reason: recompSubmission.Reason, Report: recompSubmission.Report, Exclude: recompSubmission.Exclude, Timestamp: now.Format("2006-01-02 15:04:05"), Status: "pending", } err = mongo.Insert(session, tenantDbConfig.Db, recomputationsColl, recomputation) if err != nil { panic(err) } output, err = createSubmitView(recomputation, contentType, r) return code, h, output, err }
// Create function is used to implement the create tenant request. // The request is an http POST request with the tenant description // provided as json structure in the request body func Create(r *http.Request, cfg config.Config) (int, http.Header, []byte, error) { //STANDARD DECLARATIONS START code := http.StatusOK h := http.Header{} output := []byte("") err := error(nil) contentType := "text/xml" charset := "utf-8" //STANDARD DECLARATIONS END // Content Negotiation contentType, err = respond.ParseAcceptHeader(r) h.Set("Content-Type", fmt.Sprintf("%s; charset=%s", contentType, charset)) // if authentication procedure fails then // return unauthorized http status if authentication.AuthenticateAdmin(r.Header, cfg) == false { output, _ = respond.MarshalContent(respond.UnauthorizedMessage, contentType, "", " ") code = http.StatusUnauthorized //If wrong api key is passed we return UNAUTHORIZED http status return code, h, output, err } // Try ingest request body body, err := ioutil.ReadAll(io.LimitReader(r.Body, cfg.Server.ReqSizeLimit)) if err != nil { panic(err) } if err := r.Body.Close(); err != nil { panic(err) } incoming := Tenant{} // Parse body json if err := json.Unmarshal(body, &incoming); err != nil { output, _ = respond.MarshalContent(respond.BadRequestBadJSON, contentType, "", " ") code = http.StatusBadRequest return code, h, output, err } // Try to open the mongo session session, err := mongo.OpenSession(cfg.MongoDB) defer session.Close() if err != nil { code = http.StatusInternalServerError return code, h, output, err } // Check if name exists sameName := []Tenant{} filter := bson.M{"info.name": incoming.Info.Name} err = mongo.Find(session, cfg.MongoDB.Db, "tenants", filter, "name", &sameName) if len(sameName) > 0 { code = http.StatusConflict output, err = createMsgView("Tenant with same name already exists", code) return code, h, output, err } // Generate new id incoming.ID = mongo.NewUUID() incoming.Info.Created = time.Now().Format("2006-01-02 15:04:05") incoming.Info.Updated = incoming.Info.Created err = mongo.Insert(session, cfg.MongoDB.Db, "tenants", incoming) if err != nil { panic(err) } // Create view of the results output, err = createRefView(incoming, "Tenant was succesfully created", 201, r) //Render the results into JSON code = http.StatusCreated return code, h, output, err }
// Delete function used to implement remove tenant request func Delete(r *http.Request, cfg config.Config) (int, http.Header, []byte, error) { //STANDARD DECLARATIONS START code := http.StatusOK h := http.Header{} output := []byte("") err := error(nil) contentType := "text/xml" charset := "utf-8" //STANDARD DECLARATIONS END // Content Negotiation contentType, err = respond.ParseAcceptHeader(r) h.Set("Content-Type", fmt.Sprintf("%s; charset=%s", contentType, charset)) vars := mux.Vars(r) // if authentication procedure fails then // return unauthorized if authentication.AuthenticateAdmin(r.Header, cfg) == false { output, _ = respond.MarshalContent(respond.UnauthorizedMessage, contentType, "", " ") code = http.StatusUnauthorized //If wrong api key is passed we return UNAUTHORIZED http status return code, h, output, err } // Try to open the mongo session session, err := mongo.OpenSession(cfg.MongoDB) defer session.Close() if err != nil { code = http.StatusInternalServerError return code, h, output, err } filter := bson.M{"id": vars["ID"]} // Retrieve Results from database results := []Tenant{} err = mongo.Find(session, cfg.MongoDB.Db, "tenants", filter, "name", &results) if err != nil { code = http.StatusInternalServerError return code, h, output, err } // Check if nothing found if len(results) < 1 { output, _ = respond.MarshalContent(respond.NotFound, contentType, "", " ") code = http.StatusNotFound return code, h, output, err } mongo.Remove(session, cfg.MongoDB.Db, "tenants", filter) if err != nil { code = http.StatusInternalServerError return code, h, output, err } // Create view of the results output, err = createMsgView("Tenant Successfully Deleted", 200) //Render the results into JSON if err != nil { code = http.StatusInternalServerError return code, h, output, err } h.Set("Content-Type", fmt.Sprintf("%s; charset=%s", contentType, charset)) return code, h, output, err }
// Update function used to implement update tenant request. // This is an http PUT request that gets a specific tenant's name // as a urlvar parameter input and a json structure in the request // body in order to update the datastore document for the specific // tenant func Update(r *http.Request, cfg config.Config) (int, http.Header, []byte, error) { //STANDARD DECLARATIONS START code := http.StatusOK h := http.Header{} output := []byte("") err := error(nil) contentType := "text/xml" charset := "utf-8" //STANDARD DECLARATIONS END // Content Negotiation contentType, err = respond.ParseAcceptHeader(r) h.Set("Content-Type", fmt.Sprintf("%s; charset=%s", contentType, charset)) vars := mux.Vars(r) // if authentication procedure fails then // return unauthorized if authentication.AuthenticateAdmin(r.Header, cfg) == false { output, _ = respond.MarshalContent(respond.UnauthorizedMessage, contentType, "", " ") code = http.StatusUnauthorized //If wrong api key is passed we return UNAUTHORIZED http status return code, h, output, err } incoming := Tenant{} // ingest body data body, err := ioutil.ReadAll(io.LimitReader(r.Body, cfg.Server.ReqSizeLimit)) if err != nil { panic(err) } if err := r.Body.Close(); err != nil { panic(err) } // parse body json if err := json.Unmarshal(body, &incoming); err != nil { output, _ = respond.MarshalContent(respond.BadRequestBadJSON, contentType, "", " ") code = http.StatusBadRequest return code, h, output, err } // Try to open the mongo session session, err := mongo.OpenSession(cfg.MongoDB) defer session.Close() if err != nil { code = http.StatusInternalServerError return code, h, output, err } // create filter to retrieve specific profile with id filter := bson.M{"id": vars["ID"]} incoming.ID = vars["ID"] // Retrieve Results from database results := []Tenant{} err = mongo.Find(session, cfg.MongoDB.Db, "tenants", filter, "name", &results) if err != nil { code = http.StatusInternalServerError return code, h, output, err } // Check if nothing found if len(results) < 1 { output, _ = respond.MarshalContent(respond.NotFound, contentType, "", " ") code = http.StatusNotFound return code, h, output, err } // If user chose to change name - check if name already exists if results[0].Info.Name != incoming.Info.Name { sameName := []Tenant{} filter = bson.M{"info.name": incoming.Info.Name} err = mongo.Find(session, cfg.MongoDB.Db, "tenants", filter, "name", &sameName) if len(sameName) > 1 { code = http.StatusConflict output, err = createMsgView("Tenant with same name already exists", code) return code, h, output, err } } // run the update query incoming.Info.Created = results[0].Info.Created incoming.Info.Updated = time.Now().Format("2006-01-02 15:04:05") filter = bson.M{"id": vars["ID"]} err = mongo.Update(session, cfg.MongoDB.Db, "tenants", filter, incoming) if err != nil { code = http.StatusInternalServerError return code, h, output, err } // Create view for response message output, err = createMsgView("Tenant successfully updated", 200) //Render the results into JSON code = http.StatusOK return code, h, output, err }
// ListOne function implement an http GET request that accepts // a name parameter urlvar and retrieves information only for the // specific tenant func ListOne(r *http.Request, cfg config.Config) (int, http.Header, []byte, error) { //STANDARD DECLARATIONS START code := http.StatusOK h := http.Header{} output := []byte("") err := error(nil) contentType := "text/xml" charset := "utf-8" //STANDARD DECLARATIONS END vars := mux.Vars(r) // Content Negotiation contentType, err = respond.ParseAcceptHeader(r) h.Set("Content-Type", fmt.Sprintf("%s; charset=%s", contentType, charset)) if err != nil { code = http.StatusNotAcceptable output, _ = respond.MarshalContent(respond.NotAcceptableContentType, contentType, "", " ") return code, h, output, err } // if authentication procedure fails then // return unauthorized http status if authentication.AuthenticateAdmin(r.Header, cfg) == false { output, _ = respond.MarshalContent(respond.UnauthorizedMessage, contentType, "", " ") code = http.StatusUnauthorized //If wrong api key is passed we return UNAUTHORIZED http status return code, h, output, err } // Try to open the mongo session session, err := mongo.OpenSession(cfg.MongoDB) defer session.Close() if err != nil { code = http.StatusInternalServerError return code, h, output, err } // Create structure to hold query results results := []Tenant{} // Create a simple query object to query by id query := bson.M{"id": vars["ID"]} // Query collection tenants for the specific tenant id err = mongo.Find(session, cfg.MongoDB.Db, "tenants", query, "name", &results) if err != nil { code = http.StatusInternalServerError return code, h, output, err } // Check if nothing found if len(results) < 1 { output, _ = respond.MarshalContent(respond.NotFound, contentType, "", " ") code = http.StatusNotFound return code, h, output, err } // After successfully retrieving the db results // call the createView function to render them into idented xml output, err = createListView(results, "Success", code) if err != nil { code = http.StatusInternalServerError return code, h, output, err } h.Set("Content-Type", fmt.Sprintf("%s; charset=%s", contentType, charset)) return code, h, output, err }
// ListOne function that implements the http GET request that retrieves // all avaiable report information func ListOne(r *http.Request, cfg config.Config) (int, http.Header, []byte, error) { //STANDARD DECLARATIONS START code := http.StatusOK h := http.Header{} output := []byte("") err := error(nil) contentType := "text/xml" charset := "utf-8" //STANDARD DECLARATIONS END contentType, err = respond.ParseAcceptHeader(r) h.Set("Content-Type", fmt.Sprintf("%s; charset=%s", contentType, charset)) if err != nil { code = http.StatusNotAcceptable output, _ = respond.MarshalContent(respond.NotAcceptableContentType, contentType, "", " ") return code, h, output, err } tenantDbConfig, err := authentication.AuthenticateTenant(r.Header, cfg) if err != nil { output, _ = respond.MarshalContent(respond.UnauthorizedMessage, contentType, "", " ") code = http.StatusUnauthorized //If wrong api key is passed we return UNAUTHORIZED http status return code, h, output, err } //Extracting urlvar "name" from url path id := mux.Vars(r)["id"] // Try to open the mongo session session, err := mongo.OpenSession(tenantDbConfig) defer session.Close() if err != nil { code = http.StatusInternalServerError return code, h, output, err } // Create structure for storing query results result := MongoInterface{} // Create a simple query object to query by name query := bson.M{"id": id} // Query collection tenants for the specific tenant name err = mongo.FindOne(session, tenantDbConfig.Db, reportsColl, query, &result) // If query returned zero result then no tenant matched this name, // abort and notify user accordingly if err != nil { code = http.StatusNotFound output, _ := ReportNotFound(contentType) h.Set("Content-Type", fmt.Sprintf("%s; charset=%s", contentType, charset)) return code, h, output, err } // After successfully retrieving the db results // call the createView function to render them into idented xml output, err = createView([]MongoInterface{result}, contentType) if err != nil { code = http.StatusInternalServerError return code, h, output, err } code = http.StatusOK h.Set("Content-Type", fmt.Sprintf("%s; charset=%s", contentType, charset)) return code, h, output, err }
//Update function to update contents of an existing metric profile func Update(r *http.Request, cfg config.Config) (int, http.Header, []byte, error) { //STANDARD DECLARATIONS START code := http.StatusOK h := http.Header{} output := []byte("") err := error(nil) charset := "utf-8" //STANDARD DECLARATIONS END vars := mux.Vars(r) contentType, err := respond.ParseAcceptHeader(r) h.Set("Content-Type", fmt.Sprintf("%s; charset=%s", contentType, charset)) if err != nil { code = http.StatusNotAcceptable output, _ = respond.MarshalContent(respond.NotAcceptableContentType, contentType, "", " ") return code, h, output, err } tenantDbConfig, err := authentication.AuthenticateTenant(r.Header, cfg) if err != nil { output, _ = respond.MarshalContent(respond.UnauthorizedMessage, contentType, "", " ") code = http.StatusUnauthorized //If wrong api key is passed we return UNAUTHORIZED http status return code, h, output, err } incoming := OpsProfile{} // ingest body data body, err := ioutil.ReadAll(io.LimitReader(r.Body, cfg.Server.ReqSizeLimit)) if err != nil { panic(err) } if err := r.Body.Close(); err != nil { panic(err) } // parse body json if err := json.Unmarshal(body, &incoming); err != nil { output, _ = respond.MarshalContent(respond.BadRequestBadJSON, contentType, "", " ") code = 400 return code, h, output, err } session, err := mongo.OpenSession(tenantDbConfig) defer mongo.CloseSession(session) if err != nil { code = http.StatusInternalServerError return code, h, output, err } // create filter to retrieve specific profile with id filter := bson.M{"id": vars["ID"]} incoming.ID = vars["ID"] // Retrieve Results from database results := []OpsProfile{} err = mongo.Find(session, tenantDbConfig.Db, "operations_profiles", filter, "name", &results) if err != nil { panic(err) } // Check if nothing found if len(results) < 1 { output, _ = respond.MarshalContent(respond.NotFound, contentType, "", " ") code = 404 return code, h, output, err } // Validate States var errList []string errList = append(errList, incoming.validateDuplicates()...) errList = append(errList, incoming.validateStates()...) errList = append(errList, incoming.validateMentions()...) if len(errList) > 0 { output, err = createErrView("Validation Error", 422, errList) code = 422 return code, h, output, err } // run the update query err = mongo.Update(session, tenantDbConfig.Db, "operations_profiles", filter, incoming) if err != nil { code = http.StatusInternalServerError return code, h, output, err } // Create view for response message output, err = createMsgView("Operations Profile successfully updated", 200) //Render the results into JSON code = 200 return code, h, output, err }
//Create a new metric profile func Create(r *http.Request, cfg config.Config) (int, http.Header, []byte, error) { //STANDARD DECLARATIONS START code := http.StatusOK h := http.Header{} output := []byte("") err := error(nil) charset := "utf-8" //STANDARD DECLARATIONS END contentType, err := respond.ParseAcceptHeader(r) h.Set("Content-Type", fmt.Sprintf("%s; charset=%s", contentType, charset)) if err != nil { code = http.StatusNotAcceptable output, _ = respond.MarshalContent(respond.NotAcceptableContentType, contentType, "", " ") return code, h, output, err } tenantDbConfig, err := authentication.AuthenticateTenant(r.Header, cfg) if err != nil { output, _ = respond.MarshalContent(respond.UnauthorizedMessage, contentType, "", " ") code = http.StatusUnauthorized //If wrong api key is passed we return UNAUTHORIZED http status return code, h, output, err } session, err := mongo.OpenSession(tenantDbConfig) defer mongo.CloseSession(session) if err != nil { code = http.StatusInternalServerError return code, h, output, err } incoming := OpsProfile{} // Try ingest request body body, err := ioutil.ReadAll(io.LimitReader(r.Body, cfg.Server.ReqSizeLimit)) if err != nil { panic(err) } if err := r.Body.Close(); err != nil { panic(err) } // Parse body json if err := json.Unmarshal(body, &incoming); err != nil { output, _ = respond.MarshalContent(respond.BadRequestBadJSON, contentType, "", " ") code = 400 return code, h, output, err } // Validate States var errList []string errList = append(errList, incoming.validateDuplicates()...) errList = append(errList, incoming.validateStates()...) errList = append(errList, incoming.validateMentions()...) if len(errList) > 0 { output, err = createErrView("Validation Error", 422, errList) code = 422 return code, h, output, err } // Generate new id incoming.ID = mongo.NewUUID() err = mongo.Insert(session, tenantDbConfig.Db, "operations_profiles", incoming) if err != nil { panic(err) } // Create view of the results output, err = createRefView(incoming, "Operations Profile successfully created", 201, r) //Render the results into JSON code = 201 return code, h, output, err }
// List the existing metric profiles for the tenant making the request // Also there is an optional url param "name" to filter results by func List(r *http.Request, cfg config.Config) (int, http.Header, []byte, error) { //STANDARD DECLARATIONS START code := http.StatusOK h := http.Header{} output := []byte("") err := error(nil) contentType := "text/xml" charset := "utf-8" //STANDARD DECLARATIONS END // Content Negotiation contentType, err = respond.ParseAcceptHeader(r) h.Set("Content-Type", fmt.Sprintf("%s; charset=%s", contentType, charset)) if err != nil { code = http.StatusNotAcceptable output, _ = respond.MarshalContent(respond.NotAcceptableContentType, contentType, "", " ") return code, h, output, err } urlValues := r.URL.Query() // Tenant Authentication tenantDbConfig, err := authentication.AuthenticateTenant(r.Header, cfg) if err != nil { output, _ = respond.MarshalContent(respond.UnauthorizedMessage, contentType, "", " ") code = http.StatusUnauthorized //If wrong api key is passed we return UNAUTHORIZED http status return code, h, output, err } // Open session to tenant database session, err := mongo.OpenSession(tenantDbConfig) defer mongo.CloseSession(session) if err != nil { code = http.StatusInternalServerError return code, h, output, err } // Retrieve Results from database var filter interface{} if len(urlValues["name"]) > 0 { filter = bson.M{"name": urlValues["name"][0]} } else { filter = nil } results := []MongoInterface{} err = mongo.Find(session, tenantDbConfig.Db, "metric_profiles", filter, "name", &results) if err != nil { code = http.StatusInternalServerError return code, h, output, err } // Create view of the results output, err = createListView(results, "Success", code) //Render the results into JSON if err != nil { code = http.StatusInternalServerError return code, h, output, err } h.Set("Content-Type", fmt.Sprintf("%s; charset=%s", contentType, charset)) return code, h, output, err }
// Create function is used to implement the create report request. // The request is an http POST request with the report description // provided as json structure in the request body func Create(r *http.Request, cfg config.Config) (int, http.Header, []byte, error) { //STANDARD DECLARATIONS START code := http.StatusCreated h := http.Header{} output := []byte("") err := error(nil) contentType := "text/xml" charset := "utf-8" //STANDARD DECLARATIONS END contentType, err = respond.ParseAcceptHeader(r) h.Set("Content-Type", fmt.Sprintf("%s; charset=%s", contentType, charset)) if err != nil { code = http.StatusNotAcceptable output, _ = respond.MarshalContent(respond.NotAcceptableContentType, contentType, "", " ") return code, h, output, err } tenantDbConfig, err := authentication.AuthenticateTenant(r.Header, cfg) if err != nil { output, _ = respond.MarshalContent(respond.UnauthorizedMessage, contentType, "", " ") code = http.StatusUnauthorized //If wrong api key is passed we return UNAUTHORIZED http status return code, h, output, err } //Reading the json input from the request body reqBody, err := ioutil.ReadAll(io.LimitReader(r.Body, cfg.Server.ReqSizeLimit)) if err != nil { return code, h, output, err } input := MongoInterface{} //Unmarshalling the json input into byte form err = json.Unmarshal(reqBody, &input) // Check if json body is malformed if err != nil { output, _ := respond.MarshalContent(respond.MalformedJSONInput, contentType, "", " ") code = http.StatusBadRequest h.Set("Content-Type", fmt.Sprintf("%s; charset=%s", contentType, charset)) return code, h, output, err } // Try to open the mongo session session, err := mongo.OpenSession(tenantDbConfig) defer session.Close() if err != nil { code = http.StatusInternalServerError return code, h, output, err } // Validate profiles given in report validationErrors := input.ValidateProfiles(session.DB(tenantDbConfig.Db)) if len(validationErrors) > 0 { code = 422 out := respond.UnprocessableEntity out.Errors = validationErrors output = out.MarshalTo(contentType) return code, h, output, err } // Prepare structure for storing query results results := []MongoInterface{} // Check if report with the same name exists in datastore query := searchName(input.Info.Name) err = mongo.Find(session, tenantDbConfig.Db, reportsColl, query, "name", &results) if err != nil { code = http.StatusInternalServerError return code, h, output, err } // If results are returned for the specific name // then we already have an existing report and we must // abort creation notifing the user if len(results) > 0 { // Name was found so print the error message in xml out := respond.ResponseMessage{ Status: respond.StatusResponse{ Message: "Report with the same name already exists", Code: strconv.Itoa(http.StatusConflict), }} output, _ = respond.MarshalContent(out, contentType, "", " ") code = http.StatusConflict h.Set("Content-Type", fmt.Sprintf("%s; charset=%s", contentType, charset)) return code, h, output, err } input.Info.Created = time.Now().Format("2006-01-02 15:04:05") input.Info.Updated = input.Info.Created input.ID = mongo.NewUUID() // If no report exists with this name create a new one err = mongo.Insert(session, tenantDbConfig.Db, reportsColl, input) if err != nil { code = http.StatusInternalServerError return code, h, output, err } // Notify user that the report has been created. In xml style selfLink := "https://" + r.Host + r.URL.Path + "/" + input.ID output, err = SubmitSuccesful(input, contentType, selfLink) if err != nil { code = http.StatusInternalServerError return code, h, output, err } h.Set("Content-Type", fmt.Sprintf("%s; charset=%s", contentType, charset)) return code, h, output, err }
// Delete function used to implement remove report request func Delete(r *http.Request, cfg config.Config) (int, http.Header, []byte, error) { //STANDARD DECLARATIONS START code := http.StatusOK h := http.Header{} output := []byte("") err := error(nil) contentType := "text/xml" charset := "utf-8" //STANDARD DECLARATIONS END contentType, err = respond.ParseAcceptHeader(r) h.Set("Content-Type", fmt.Sprintf("%s; charset=%s", contentType, charset)) if err != nil { code = http.StatusNotAcceptable output, _ = respond.MarshalContent(respond.NotAcceptableContentType, contentType, "", " ") return code, h, output, err } tenantDbConfig, err := authentication.AuthenticateTenant(r.Header, cfg) if err != nil { output, _ = respond.MarshalContent(respond.UnauthorizedMessage, contentType, "", " ") code = http.StatusUnauthorized //If wrong api key is passed we return UNAUTHORIZED http status return code, h, output, err } //Extracting record id from url id := mux.Vars(r)["id"] // Try to open the mongo session session, err := mongo.OpenSession(tenantDbConfig) defer session.Close() if err != nil { code = http.StatusInternalServerError return code, h, output, err } // We search by name and delete the document in db query := bson.M{"id": id} info, err := mongo.Remove(session, tenantDbConfig.Db, reportsColl, query) if err != nil { if err.Error() != "not found" { code = http.StatusInternalServerError return code, h, output, err } //Render the response into XML code = http.StatusNotFound output, err = ReportNotFound(contentType) return code, h, output, err } // info.Removed > 0 means that many documents have been removed // If deletion took place we notify user accordingly. // Else we notify that no tenant matched the specific name if info.Removed > 0 { code = http.StatusOK output, err = respond.CreateResponseMessage("Report was successfully deleted", "200", contentType) } else { code = http.StatusNotFound output, err = ReportNotFound(contentType) } //Render the response into XML if err != nil { code = http.StatusInternalServerError return code, h, output, err } return code, h, output, err }
// Update function used to implement update report request. // This is an http PUT request that gets a specific report's name // as a urlvar parameter input and a json structure in the request // body in order to update the datastore document for the specific // report func Update(r *http.Request, cfg config.Config) (int, http.Header, []byte, error) { //STANDARD DECLARATIONS START code := http.StatusOK h := http.Header{} output := []byte("") err := error(nil) contentType := "text/xml" charset := "utf-8" //STANDARD DECLARATIONS END contentType, err = respond.ParseAcceptHeader(r) h.Set("Content-Type", fmt.Sprintf("%s; charset=%s", contentType, charset)) if err != nil { code = http.StatusNotAcceptable output, _ = respond.MarshalContent(respond.NotAcceptableContentType, contentType, "", " ") return code, h, output, err } tenantDbConfig, err := authentication.AuthenticateTenant(r.Header, cfg) if err != nil { output, _ = respond.MarshalContent(respond.UnauthorizedMessage, contentType, "", " ") code = http.StatusUnauthorized //If wrong api key is passed we return UNAUTHORIZED http status return code, h, output, err } //Extracting report name from url id := mux.Vars(r)["id"] //Reading the json input reqBody, err := ioutil.ReadAll(r.Body) input := MongoInterface{} //Unmarshalling the json input into byte form err = json.Unmarshal(reqBody, &input) if err != nil { // User provided malformed json input data output, _ := respond.MarshalContent(respond.MalformedJSONInput, contentType, "", " ") code = http.StatusBadRequest h.Set("Content-Type", fmt.Sprintf("%s; charset=%s", contentType, charset)) return code, h, output, err } sanitizedInput := bson.M{ "$set": bson.M{ // "info": bson.M{ "info.name": input.Info.Name, "info.description": input.Info.Description, "info.updated": time.Now().Format("2006-01-02 15:04:05"), // }, "profiles": input.Profiles, "filter_tags": input.Tags, "topology_schema": input.Topology, }} // Try to open the mongo session session, err := mongo.OpenSession(tenantDbConfig) defer session.Close() if err != nil { code = http.StatusInternalServerError return code, h, output, err } // Validate profiles given in report validationErrors := input.ValidateProfiles(session.DB(tenantDbConfig.Db)) if len(validationErrors) > 0 { code = 422 out := respond.UnprocessableEntity out.Errors = validationErrors output = out.MarshalTo(contentType) return code, h, output, err } // We search by name and update query := bson.M{"id": id} err = mongo.Update(session, tenantDbConfig.Db, reportsColl, query, sanitizedInput) if err != nil { if err.Error() != "not found" { code = http.StatusInternalServerError return code, h, output, err } //Render the response into XML code = http.StatusNotFound output, err = ReportNotFound(contentType) return code, h, output, err } //Render the response into XML output, err = respond.CreateResponseMessage("Report was successfully updated", "200", contentType) if err != nil { code = http.StatusInternalServerError return code, h, output, err } code = http.StatusOK return code, h, output, err }
// ListOne handles the listing of one specific profile based on its given id func ListOne(r *http.Request, cfg config.Config) (int, http.Header, []byte, error) { //STANDARD DECLARATIONS START code := http.StatusOK h := http.Header{} output := []byte("") err := error(nil) contentType := "text/xml" charset := "utf-8" //STANDARD DECLARATIONS END // Content Negotiation contentType, err = respond.ParseAcceptHeader(r) h.Set("Content-Type", fmt.Sprintf("%s; charset=%s", contentType, charset)) vars := mux.Vars(r) if err != nil { code = http.StatusNotAcceptable output, _ = respond.MarshalContent(respond.NotAcceptableContentType, contentType, "", " ") return code, h, output, err } // Tenant Authentication tenantDbConfig, err := authentication.AuthenticateTenant(r.Header, cfg) if err != nil { output, _ = respond.MarshalContent(respond.UnauthorizedMessage, contentType, "", " ") code = http.StatusUnauthorized //If wrong api key is passed we return UNAUTHORIZED http status return code, h, output, err } // Open session to tenant database session, err := mongo.OpenSession(tenantDbConfig) defer mongo.CloseSession(session) if err != nil { code = http.StatusInternalServerError return code, h, output, err } filter := bson.M{"id": vars["ID"]} // Retrieve Results from database results := []OpsProfile{} err = mongo.Find(session, tenantDbConfig.Db, "operations_profiles", filter, "name", &results) if err != nil { code = http.StatusInternalServerError return code, h, output, err } // Check if nothing found if len(results) < 1 { output, _ = respond.MarshalContent(respond.NotFound, contentType, "", " ") code = 404 return code, h, output, err } // Create view of the results output, err = createListView(results, "Success", code) //Render the results into JSON if err != nil { code = http.StatusInternalServerError return code, h, output, err } h.Set("Content-Type", fmt.Sprintf("%s; charset=%s", contentType, charset)) return code, h, output, err }
// ListMetricTimelines returns a list of metric timelines func ListMetricTimelines(r *http.Request, cfg config.Config) (int, http.Header, []byte, error) { //STANDARD DECLARATIONS START code := http.StatusOK h := http.Header{} output := []byte("List Metric Timelines") err := error(nil) contentType := "application/xml" charset := "utf-8" //STANDARD DECLARATIONS END contentType, err = respond.ParseAcceptHeader(r) h.Set("Content-Type", fmt.Sprintf("%s; charset=%s", contentType, charset)) if err != nil { code = http.StatusNotAcceptable output, _ = respond.MarshalContent(respond.NotAcceptableContentType, contentType, "", " ") return code, h, output, err } // Parse the request into the input urlValues := r.URL.Query() vars := mux.Vars(r) parsedStart, parsedEnd, errs := respond.ValidateDateRange(urlValues.Get("start_time"), urlValues.Get("end_time")) if len(errs) > 0 { code = http.StatusBadRequest output = respond.CreateFailureResponseMessage("Bad Request", "400", errs).MarshalTo(contentType) } input := InputParams{ parsedStart, parsedEnd, vars["report_name"], vars["group_type"], vars["group_name"], vars["service_name"], vars["endpoint_name"], vars["metric_name"], contentType, } // Call authenticateTenant to check the api key and retrieve // the correct tenant db conf tenantDbConfig, err := authentication.AuthenticateTenant(r.Header, cfg) if err != nil { output = []byte(http.StatusText(http.StatusUnauthorized)) code = http.StatusUnauthorized //If wrong api key is passed we return UNAUTHORIZED http status h.Set("Content-Type", fmt.Sprintf("%s; charset=%s", contentType, charset)) return code, h, output, err } // Mongo Session results := []DataOutput{} session, err := mongo.OpenSession(tenantDbConfig) defer mongo.CloseSession(session) metricCollection := session.DB(tenantDbConfig.Db).C("status_metrics") // Query the detailed metric results reportID, err := mongo.GetReportID(session, tenantDbConfig.Db, input.report) if err != nil { code = http.StatusInternalServerError return code, h, output, err } err = metricCollection.Find(prepareQuery(input, reportID)).All(&results) if err != nil { code = http.StatusInternalServerError return code, h, output, err } output, err = createView(results, input) //Render the results into XML format h.Set("Content-Type", fmt.Sprintf("%s; charset=%s", contentType, charset)) return code, h, output, err }
// List function that implements the http GET request that retrieves // all avaiable tenant information func List(r *http.Request, cfg config.Config) (int, http.Header, []byte, error) { //STANDARD DECLARATIONS START code := http.StatusOK h := http.Header{} output := []byte("") err := error(nil) contentType := "text/xml" charset := "utf-8" //STANDARD DECLARATIONS END // Content Negotiation contentType, err = respond.ParseAcceptHeader(r) h.Set("Content-Type", fmt.Sprintf("%s; charset=%s", contentType, charset)) if err != nil { code = http.StatusNotAcceptable output, _ = respond.MarshalContent(respond.NotAcceptableContentType, contentType, "", " ") return code, h, output, err } // if authentication procedure fails then // return unauthorized http status if authentication.AuthenticateAdmin(r.Header, cfg) == false { output, _ = respond.MarshalContent(respond.UnauthorizedMessage, contentType, "", " ") code = http.StatusUnauthorized //If wrong api key is passed we return UNAUTHORIZED http status return code, h, output, err } // Try to open the mongo session session, err := mongo.OpenSession(cfg.MongoDB) defer session.Close() if err != nil { code = http.StatusInternalServerError return code, h, output, err } // Create structure for storing query results results := []Tenant{} // Query tenant collection for all available documents. // nil query param == match everything err = mongo.Find(session, cfg.MongoDB.Db, "tenants", nil, "name", &results) if err != nil { code = http.StatusInternalServerError return code, h, output, err } // After successfully retrieving the db results // call the createView function to render them into idented xml output, err = createListView(results, "Success", code) if err != nil { code = http.StatusInternalServerError return code, h, output, err } h.Set("Content-Type", fmt.Sprintf("%s; charset=%s", contentType, charset)) return code, h, output, err }
// ListServiceFlavorResults is responsible for handling request to list service flavor results func ListServiceFlavorResults(r *http.Request, cfg config.Config) (int, http.Header, []byte, error) { //STANDARD DECLARATIONS START code := http.StatusOK h := http.Header{} output := []byte("") err := error(nil) contentType := "application/xml" charset := "utf-8" //STANDARD DECLARATIONS END contentType, err = respond.ParseAcceptHeader(r) h.Set("Content-Type", fmt.Sprintf("%s; charset=%s", contentType, charset)) if err != nil { code = http.StatusNotAcceptable output, _ = respond.MarshalContent(respond.NotAcceptableContentType, contentType, "", " ") return code, h, output, err } // Parse the request into the input urlValues := r.URL.Query() vars := mux.Vars(r) tenantDbConfig, err := authentication.AuthenticateTenant(r.Header, cfg) if err != nil { if err.Error() == "Unauthorized" { code = http.StatusUnauthorized out := respond.UnauthorizedMessage output = out.MarshalTo(contentType) return code, h, output, err } code = http.StatusInternalServerError return code, h, output, err } session, err := mongo.OpenSession(tenantDbConfig) defer mongo.CloseSession(session) if err != nil { code = http.StatusInternalServerError return code, h, output, err } report := reports.MongoInterface{} err = mongo.FindOne(session, tenantDbConfig.Db, "reports", bson.M{"info.name": vars["report_name"]}, &report) if err != nil { code = http.StatusBadRequest message := "The report with the name " + vars["report_name"] + " does not exist" output, err := createErrorMessage(message, contentType) //Render the response into XML or JSON h.Set("Content-Type", fmt.Sprintf("%s; charset=%s", contentType, charset)) return code, h, output, err } input := serviceFlavorResultQuery{ basicQuery: basicQuery{ Name: vars["service_type"], Granularity: urlValues.Get("granularity"), Format: contentType, StartTime: urlValues.Get("start_time"), EndTime: urlValues.Get("end_time"), Report: report, Vars: vars, }, EndpointGroup: vars["lgroup_name"], } tenantDB := session.DB(tenantDbConfig.Db) errs := input.Validate(tenantDB) if len(errs) > 0 { out := respond.BadRequestSimple out.Errors = errs output = out.MarshalTo(contentType) code = 400 return code, h, output, err } if vars["lgroup_type"] != report.GetEndpointGroupType() { code = http.StatusBadRequest message := "The report " + vars["report_name"] + " does not define endpoint group type: " + vars["lgroup_type"] + ". Try using " + report.GetEndpointGroupType() + " instead." output, err := createErrorMessage(message, contentType) //Render the response into XML or JSON h.Set("Content-Type", fmt.Sprintf("%s; charset=%s", contentType, charset)) return code, h, output, err } results := []ServiceFlavorInterface{} if err != nil { code = http.StatusInternalServerError return code, h, output, err } // Construct the query to mongodb based on the input filter := bson.M{ "date": bson.M{"$gte": input.StartTimeInt, "$lte": input.EndTimeInt}, "report": report.ID, } if input.Name != "" { filter["name"] = input.Name } if input.EndpointGroup != "" { filter["supergroup"] = input.EndpointGroup } // Select the granularity of the search daily/monthly if input.Granularity == "daily" { customForm[0] = "20060102" customForm[1] = "2006-01-02" query := DailyServiceFlavor(filter) err = mongo.Pipe(session, tenantDbConfig.Db, "service_ar", query, &results) } else if input.Granularity == "monthly" { customForm[0] = "200601" customForm[1] = "2006-01" query := MonthlyServiceFlavor(filter) err = mongo.Pipe(session, tenantDbConfig.Db, "service_ar", query, &results) } // mongo.Find(session, tenantDbConfig.Db, "endpoint_group_ar", bson.M{}, "_id", &results) if err != nil { code = http.StatusInternalServerError return code, h, output, err } output, err = createServiceFlavorResultView(results, report, input.Format) if err != nil { code = http.StatusInternalServerError return code, h, output, err } return code, h, output, err }
// List function that implements the http GET request that retrieves // all avaiable report information func List(r *http.Request, cfg config.Config) (int, http.Header, []byte, error) { //STANDARD DECLARATIONS START code := http.StatusOK h := http.Header{} output := []byte("") err := error(nil) contentType := "text/xml" charset := "utf-8" //STANDARD DECLARATIONS END urlValues := r.URL.Query() contentType, err = respond.ParseAcceptHeader(r) h.Set("Content-Type", fmt.Sprintf("%s; charset=%s", contentType, charset)) if err != nil { code = http.StatusNotAcceptable output, _ = respond.MarshalContent(respond.NotAcceptableContentType, contentType, "", " ") return code, h, output, err } tenantDbConfig, err := authentication.AuthenticateTenant(r.Header, cfg) if err != nil { output, _ = respond.MarshalContent(respond.UnauthorizedMessage, contentType, "", " ") code = http.StatusUnauthorized //If wrong api key is passed we return UNAUTHORIZED http status return code, h, output, err } // Try to open the mongo session session, err := mongo.OpenSession(cfg.MongoDB) defer session.Close() if err != nil { code = http.StatusInternalServerError return code, h, output, err } query := bson.M{} if urlValues.Get("name") != "" { query["info.name"] = urlValues["name"] } // Create structure for storing query results results := []MongoInterface{} // Query tenant collection for all available documents. // nil query param == match everything err = mongo.Find(session, tenantDbConfig.Db, reportsColl, nil, "id", &results) if err != nil { code = http.StatusInternalServerError return code, h, output, err } // After successfully retrieving the db results // call the createView function to render them into idented xml output, err = createView(results, contentType) if err != nil { code = http.StatusInternalServerError return code, h, output, err } h.Set("Content-Type", fmt.Sprintf("%s; charset=%s", contentType, charset)) return code, h, output, err }