dmaap-mediator-producer/internal/server/server.go

   1 // -
   2 //   ========================LICENSE_START=================================
   3 //   O-RAN-SC
   4 //   %%
   5 //   Copyright (C) 2021: Nordix Foundation
   6 //   %%
   7 //   Licensed under the Apache License, Version 2.0 (the "License");
   8 //   you may not use this file except in compliance with the License.
   9 //   You may obtain a copy of the License at
  10 //
  11 //        http://www.apache.org/licenses/LICENSE-2.0
  12 //
  13 //   Unless required by applicable law or agreed to in writing, software
  14 //   distributed under the License is distributed on an "AS IS" BASIS,
  15 //   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16 //   See the License for the specific language governing permissions and
  17 //   limitations under the License.
  18 //   ========================LICENSE_END===================================
  19 //
  20
  21 package server
  22
  23 import (
  24         "encoding/json"
  25         "fmt"
  26         "io/ioutil"
  27         "net/http"
  28
  29         "oransc.org/nonrtric/dmaapmediatorproducer/internal/jobs"
  30 )
  31
  32 func StatusHandler(w http.ResponseWriter, r *http.Request) {
  33         if r.URL.Path != "/" {
  34                 http.Error(w, "404 not found.", http.StatusNotFound)
  35                 return
  36         }
  37
  38         if r.Method != "GET" {
  39                 http.Error(w, "Method is not supported.", http.StatusMethodNotAllowed)
  40                 return
  41         }
  42
  43         fmt.Fprintf(w, "All is well!")
  44 }
  45
  46 func CreateInfoJobHandler(w http.ResponseWriter, r *http.Request) {
  47         if r.URL.Path != "/producer_simulator/info_job" {
  48                 http.Error(w, "404 not found.", http.StatusNotFound)
  49                 return
  50         }
  51
  52         if r.Method != "POST" {
  53                 http.Error(w, "Method is not supported.", http.StatusMethodNotAllowed)
  54                 return
  55         }
  56
  57         b, readErr := ioutil.ReadAll(r.Body)
  58         if readErr != nil {
  59                 http.Error(w, fmt.Sprintf("Unable to read body due to: %v", readErr), http.StatusBadRequest)
  60                 return
  61         }
  62         jobInfo := jobs.JobInfo{}
  63         if unmarshalErr := json.Unmarshal(b, &jobInfo); unmarshalErr != nil {
  64                 http.Error(w, fmt.Sprintf("Invalid json body. Cause: %v", unmarshalErr), http.StatusBadRequest)
  65                 return
  66         }
  67         if err := jobs.AddJob(jobInfo); err != nil {
  68                 http.Error(w, fmt.Sprintf("Invalid job info. Cause: %v", err), http.StatusBadRequest)
  69         }
  70 }
  71
  72 func CreateServer(port int, handlerFunc func(http.ResponseWriter, *http.Request)) *http.Server {
  73
  74         mux := http.NewServeMux()
  75         mux.HandleFunc("/", handlerFunc)
  76         server := http.Server{
  77                 Addr:    fmt.Sprintf(":%v", port), // :{port}
  78                 Handler: mux,
  79         }
  80         return &server
  81 }