dmaap-mediator-producer/stub/ics/ics.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 main
  22
  23 import (
  24         "flag"
  25         "fmt"
  26         "io/ioutil"
  27         "net/http"
  28
  29         "github.com/gorilla/mux"
  30 )
  31
  32 func main() {
  33         port := flag.Int("port", 8434, "The port this stub will listen on")
  34         flag.Parse()
  35         fmt.Println("Starting ICS stub on port ", *port)
  36
  37         r := mux.NewRouter()
  38         r.HandleFunc("/data-producer/v1/info-types/{typeId}", handleTypeRegistration).Methods(http.MethodPut, http.MethodPut)
  39         r.HandleFunc("/data-producer/v1/info-producers/{producerId}", handleProducerRegistration).Methods(http.MethodPut, http.MethodPut)
  40         fmt.Println(http.ListenAndServe(fmt.Sprintf(":%v", *port), r))
  41 }
  42
  43 func handleTypeRegistration(w http.ResponseWriter, r *http.Request) {
  44         vars := mux.Vars(r)
  45         id, ok := vars["typeId"]
  46         if ok {
  47                 fmt.Printf("Registered type %v with schema: %v\n", id, readBody(r))
  48         }
  49 }
  50
  51 func handleProducerRegistration(w http.ResponseWriter, r *http.Request) {
  52         vars := mux.Vars(r)
  53         id, ok := vars["producerId"]
  54         if ok {
  55                 fmt.Printf("Registered producer %v with data: %v\n", id, readBody(r))
  56         }
  57 }
  58
  59 func readBody(r *http.Request) string {
  60         b, readErr := ioutil.ReadAll(r.Body)
  61         if readErr != nil {
  62                 return fmt.Sprintf("Unable to read body due to: %v", readErr)
  63         }
  64         return string(b)
  65 }