Adds: Custome end points structure

cmd/root.go

@@ -23,8 +23,8 @@ import (
 	"github.com/spf13/pflag"
 
 	"github.com/elastic/apm-server/beater"
+	"github.com/elastic/apm-server/custom_server"
 	"github.com/elastic/apm-server/idxmgmt"
-	_ "github.com/elastic/apm-server/include"
 	"github.com/elastic/beats/libbeat/cmd"
 	"github.com/elastic/beats/libbeat/cmd/instance"
 	"github.com/elastic/beats/libbeat/common"
@@ -44,6 +44,21 @@ const IdxPattern = "apm"
 var RootCmd *cmd.BeatsRootCmd
 
 func init() {
+
+	// r := mux.NewRouter()
+
+	// r.HandleFunc("/test", func(w http.ResponseWriter, r *http.Request) {
+	// 	// vars := mux.Vars(r)
+	// 	// title := vars["title"]
+	// 	// page := vars["page"]
+
+	// 	fmt.Fprintf(w, "You've requested the book: on page \n")
+	// }).Methods("GET")
+
+	// go http.ListenAndServe(":9000", r)
+
+	s := custom_server.NewServer()
+	s.Start()
 	overrides := common.MustNewConfigFrom(map[string]interface{}{
 		"logging": map[string]interface{}{
 			"metrics": map[string]interface{}{

custom_server/custom_router.go

@@ -0,0 +1,57 @@
+package custom_server
+
+import (
+	"net/http"
+
+	"github.com/gorilla/mux"
+)
+
+type CustomRouter struct {
+}
+
+func NewUserRouter(router *mux.Router) *mux.Router {
+	customRouter := CustomRouter{}
+
+	router.HandleFunc("/", customRouter.createUserHandler).Methods("GET")
+	// router.HandleFunc("/{username}", userRouter.getUserHandler).Methods("GET")
+	return router
+}
+
+func (ur *CustomRouter) createUserHandler(w http.ResponseWriter, r *http.Request) {
+	// user, err := decodeUser(r)
+	// if err != nil {
+	// 	Error(w, http.StatusBadRequest, "Invalid request payload")
+	// 	return
+	// }
+
+	// err = ur.userService.Create(&user)
+	// if err != nil {
+	// 	Error(w, http.StatusInternalServerError, err.Error())
+	// 	return
+	// }
+	Json(w, http.StatusOK, "")
+}
+
+// func (ur *userRouter) getUserHandler(w http.ResponseWriter, r *http.Request) {
+// 	vars := mux.Vars(r)
+// 	log.Println(vars)
+// 	username := vars["username"]
+
+// 	user, err := ur.userService.GetByUsername(username)
+// 	if err != nil {
+// 		Error(w, http.StatusNotFound, err.Error())
+// 		return
+// 	}
+
+// 	Json(w, http.StatusOK, user)
+// }
+
+// func decodeUser(ur *http.Request) (root.User, error) {
+// 	var u root.User
+// 	if r.Body == nil {
+// 		return u, errors.New("no request body")
+// 	}
+// 	decoder := json.NewDecoder(r.Body)
+// 	err := decoder.Decode(&u)
+// 	return u, err
+// }

custom_server/custom_server.go

@@ -0,0 +1,31 @@
+package custom_server
+
+import (
+	"log"
+	"net/http"
+	"os"
+
+	"github.com/gorilla/handlers"
+	"github.com/gorilla/mux"
+)
+
+type CustomServer struct {
+	router *mux.Router
+}
+
+func NewServer() *CustomServer {
+	s := CustomServer{router: mux.NewRouter()}
+	NewUserRouter(s.newSubrouter("/custom_end_point"))
+	return &s
+}
+
+func (s *CustomServer) Start() {
+	log.Println("Listening on port 8080")
+	if err := http.ListenAndServe(":8080", handlers.LoggingHandler(os.Stdout, s.router)); err != nil {
+		log.Fatal("http.ListenAndServe: ", err)
+	}
+}
+
+func (s *CustomServer) newSubrouter(path string) *mux.Router {
+	return s.router.PathPrefix(path).Subrouter()
+}

custom_server/response.go

@@ -0,0 +1,18 @@
+package custom_server
+
+import (
+	"encoding/json"
+	"net/http"
+)
+
+func Error(w http.ResponseWriter, code int, message string) {
+	Json(w, code, map[string]string{"error": message})
+}
+
+func Json(w http.ResponseWriter, code int, payload interface{}) {
+	response, _ := json.Marshal(payload)
+
+	w.Header().Set("Content-Type", "application/json")
+	w.WriteHeader(code)
+	w.Write(response)
+}

main.go

@@ -23,6 +23,7 @@ import (
 	"os"
 
 	"github.com/elastic/apm-server/cmd"
+
 )
 
 func main() {