[VOL-3678] First implementation of the BBSim-sadis-server
Change-Id: I5077a8f861f4cc6af9759f31a4a415042c05eba3
diff --git a/internal/core/server.go b/internal/core/server.go
new file mode 100644
index 0000000..8a5dfd6
--- /dev/null
+++ b/internal/core/server.go
@@ -0,0 +1,107 @@
+/*
+ * Copyright 2018-present Open Networking Foundation
+
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+
+ * http://www.apache.org/licenses/LICENSE-2.0
+
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package core
+
+import (
+ "context"
+ "encoding/json"
+ "fmt"
+ "github.com/gorilla/mux"
+ "github.com/opencord/voltha-lib-go/v4/pkg/log"
+ "net/http"
+ "sync"
+)
+
+type Server struct {
+ store *Store
+}
+
+func NewServer(store *Store) *Server {
+ return &Server{
+ store: store,
+ }
+}
+
+func (s *Server) StartSadisServer(wg *sync.WaitGroup) {
+ defer wg.Done()
+ ctx := context.Background()
+
+ addr := "0.0.0.0:8080"
+
+ router := mux.NewRouter().StrictSlash(true)
+ router.HandleFunc("/subscribers/{ID}", s.serveEntry)
+ router.HandleFunc("/profiles/{ID}", s.serveBWPEntry)
+
+ logger.Fatal(ctx, http.ListenAndServe(addr, router))
+}
+
+func (s Server) serveEntry(w http.ResponseWriter, r *http.Request) {
+ vars := mux.Vars(r)
+ id := vars["ID"]
+
+ ctx := context.TODO()
+ logger.Debugw(ctx, "received-sadis-entry-request", log.Fields{"id": id})
+
+ w.Header().Set("Content-Type", "application/json")
+
+ if olt, err := s.store.getOlt(r.Context(), id); err == nil {
+ w.WriteHeader(http.StatusOK)
+ _ = json.NewEncoder(w).Encode(olt)
+ logger.Infow(ctx, "responded-to-sadis-olt-entry-request", log.Fields{"id": id})
+ return
+ }
+
+ if onu, err := s.store.getOnu(r.Context(), id); err == nil {
+ w.WriteHeader(http.StatusOK)
+ _ = json.NewEncoder(w).Encode(onu)
+ logger.Infow(ctx, "responded-to-sadis-onu-entry-request", log.Fields{"id": id})
+ return
+ }
+
+ w.WriteHeader(http.StatusNotFound)
+ msg := make(map[string]interface{})
+ msg["statusCode"] = http.StatusNotFound
+ msg["message"] = fmt.Sprintf("Entry with ID %s not found.", id)
+ _ = json.NewEncoder(w).Encode(msg)
+
+ logger.Warnw(ctx, "sadis-entry-not-found", log.Fields{"id": id})
+}
+
+func (s Server) serveBWPEntry(w http.ResponseWriter, r *http.Request) {
+ vars := mux.Vars(r)
+ id := vars["ID"]
+
+ ctx := context.TODO()
+ logger.Debugw(ctx, "received-sadis-bandwidthprofile-request", log.Fields{"id": id})
+
+ w.Header().Set("Content-Type", "application/json")
+
+ if bp, err := s.store.getBp(r.Context(), id); err == nil {
+ w.WriteHeader(http.StatusOK)
+ _ = json.NewEncoder(w).Encode(bp)
+ logger.Infow(ctx, "responded-to-sadis-bandwidthprofile-request", log.Fields{"id": id})
+ return
+ }
+
+ w.WriteHeader(http.StatusNotFound)
+ msg := make(map[string]interface{})
+ msg["statusCode"] = http.StatusNotFound
+ msg["message"] = fmt.Sprintf("BandwidthProfile with ID %s not found.", id)
+ _ = json.NewEncoder(w).Encode(msg)
+
+ logger.Warnw(ctx, "sadis-bandwidthprofile-not-found", log.Fields{"id": id})
+}