Register types DMaaP mediator producer

[nonrtric.git] / dmaap-mediator-producer / internal / config / registrator.go

diff --git a/dmaap-mediator-producer/internal/config/registrator.go b/dmaap-mediator-producer/internal/config/registrator.go
new file mode 100644 (file)
index 0000000..f846f9f
--- /dev/null
+++ b/dmaap-mediator-producer/internal/config/registrator.go
@@ -0,0 +1,58 @@
+// -
+//   ========================LICENSE_START=================================
+//   O-RAN-SC
+//   %%
+//   Copyright (C) 2021: Nordix 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.
+//   ========================LICENSE_END===================================
+//
+
+package config
+
+import (
+       "fmt"
+       "net/url"
+
+       log "github.com/sirupsen/logrus"
+
+       "oransc.org/nonrtric/dmaapmediatorproducer/internal/jobtypes"
+       "oransc.org/nonrtric/dmaapmediatorproducer/internal/restclient"
+)
+
+const registerTypePath = "/data-producer/v1/info-types/"
+
+type Registrator interface {
+       RegisterTypes(types []*jobtypes.Type) error
+}
+
+type RegistratorImpl struct {
+       infoCoordinatorAddress string
+}
+
+func NewRegistratorImpl(infoCoordAddr string) *RegistratorImpl {
+       return &RegistratorImpl{
+               infoCoordinatorAddress: infoCoordAddr,
+       }
+}
+
+func (r RegistratorImpl) RegisterTypes(jobTypes []*jobtypes.Type) error {
+       for _, jobType := range jobTypes {
+               body := fmt.Sprintf(`{"info_job_data_schema": %v}`, jobType.Schema)
+               if error := restclient.Put(r.infoCoordinatorAddress+registerTypePath+url.PathEscape(jobType.Name), []byte(body)); error != nil {
+                       return error
+               }
+               log.Debugf("Registered type: %v", jobType)
+       }
+       return nil
+}