X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;ds=sidebyside;f=pkg%2Frestful%2Frestful.go;fp=pkg%2Frestful%2Frestful.go;h=9b784a13e2586d3fddb7e5171c762ed744a1cf0a;hb=41f03638ee74f924d160328b3291aae61740cbdd;hp=9c9e861e47dd0980fd80f548f62a7f029bf7ceda;hpb=9960403a919d3aeb771d7504376a6cb5f7a6e3d1;p=ric-plt%2Fricdms.git

diff --git a/pkg/restful/restful.go b/pkg/restful/restful.go
index 9c9e861..9b784a1 100644
--- a/pkg/restful/restful.go
+++ b/pkg/restful/restful.go
@@ -20,6 +20,9 @@
 package restful
 
 import (
+	"fmt"
+	"io"
+	"io/ioutil"
 	"log"
 	"os"
 
@@ -34,6 +37,7 @@ import (
 	"gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/resthooks"
 	"gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/ricdms"
 	"github.com/go-openapi/loads"
+	"github.com/go-openapi/runtime"
 	"github.com/go-openapi/runtime/middleware"
 )
 
@@ -93,6 +97,25 @@ func (r *Restful) setupHandler() {
 		return resp
 	})
 
+	api.ApplicationZipProducer = runtime.ProducerFunc(func(w io.Writer, data interface{}) error {
+		if zp, ok := data.(io.ReadCloser); ok {
+			defer zp.Close()
+			b, err := ioutil.ReadAll(zp)
+
+			if err != nil {
+				ricdms.Logger.Error("error: %v", err)
+				return err
+			}
+			_, err = w.Write(b)
+
+			if err != nil {
+				ricdms.Logger.Error("error: %v", err)
+				return err
+			}
+			return nil
+		}
+		return fmt.Errorf("not support")
+	})
 	r.api = api
 }