retrive pods and logs by gateway

Signed-off-by: Roland.Ma <rolandma@kubesphere.io>

pkg/kapis/gateway/v1alpha1/handler.go

@@ -17,30 +17,44 @@ limitations under the License.
 package v1alpha1
 
 import (
+	"context"
+	"fmt"
+
 	"github.com/emicklei/go-restful"
-	"kubesphere.io/api/gateway/v1alpha1"
+	corev1 "k8s.io/api/core/v1"
+	"k8s.io/apiserver/pkg/util/flushwriter"
+	"k8s.io/client-go/kubernetes"
+	"k8s.io/client-go/kubernetes/scheme"
 	"sigs.k8s.io/controller-runtime/pkg/cache"
 	"sigs.k8s.io/controller-runtime/pkg/client"
 
+	"kubesphere.io/api/gateway/v1alpha1"
+
 	"kubesphere.io/kubesphere/pkg/api"
 	"kubesphere.io/kubesphere/pkg/apiserver/query"
+	"kubesphere.io/kubesphere/pkg/informers"
 	operator "kubesphere.io/kubesphere/pkg/models/gateway"
+
 	servererr "kubesphere.io/kubesphere/pkg/server/errors"
 	"kubesphere.io/kubesphere/pkg/simple/client/gateway"
+	conversionsv1 "kubesphere.io/kubesphere/pkg/utils/conversions/core/v1"
 )
 
 type handler struct {
 	options *gateway.Options
 	gw      operator.GatewayOperator
+	factory informers.InformerFactory
 }
 
 //newHandler create an instance of the handler
-func newHandler(options *gateway.Options, cache cache.Cache, client client.Client) *handler {
+func newHandler(options *gateway.Options, cache cache.Cache, client client.Client, factory informers.InformerFactory, k8sClient kubernetes.Interface) *handler {
+	conversionsv1.RegisterConversions(scheme.Scheme)
 	// Do not register Gateway scheme globally. Which will cause conflict in ks-controller-manager.
 	v1alpha1.AddToScheme(client.Scheme())
 	return &handler{
 		options: options,
-		gw:      operator.NewGatewayOperator(client, cache, options),
+		factory: factory,
+		gw:      operator.NewGatewayOperator(client, cache, options, factory, k8sClient),
 	}
 }
 
@@ -126,3 +140,36 @@ func (h *handler) List(request *restful.Request, response *restful.Response) {
 
 	response.WriteEntity(result)
 }
+
+func (h *handler) ListPods(request *restful.Request, response *restful.Response) {
+	queryParam := query.ParseQueryParameter(request)
+	ns := request.PathParameter("namespace")
+
+	result, err := h.gw.GetPods(ns, queryParam)
+	if err != nil {
+		api.HandleError(response, request, err)
+		return
+	}
+
+	response.WriteEntity(result)
+}
+
+func (h *handler) PodLog(request *restful.Request, response *restful.Response) {
+
+	podNamespace := request.PathParameter("namespace")
+	podID := request.PathParameter("pod")
+
+	query := request.Request.URL.Query()
+	logOptions := &corev1.PodLogOptions{}
+	if err := scheme.ParameterCodec.DecodeParameters(query, corev1.SchemeGroupVersion, logOptions); err != nil {
+		api.HandleError(response, request, fmt.Errorf("unable to decode query"))
+		return
+	}
+
+	fw := flushwriter.Wrap(response.ResponseWriter)
+	err := h.gw.GetPodLogs(context.TODO(), podNamespace, podID, logOptions, fw)
+	if err != nil {
+		api.HandleError(response, request, err)
+		return
+	}
+}

pkg/kapis/gateway/v1alpha1/register.go

@@ -22,6 +22,7 @@ import (
 	"github.com/emicklei/go-restful"
 	restfulspec "github.com/emicklei/go-restful-openapi"
 	"k8s.io/apimachinery/pkg/runtime/schema"
+	"k8s.io/client-go/kubernetes"
 	"kubesphere.io/api/gateway/v1alpha1"
 	"sigs.k8s.io/controller-runtime/pkg/cache"
 	"sigs.k8s.io/controller-runtime/pkg/client"
@@ -29,16 +30,17 @@ import (
 	"kubesphere.io/kubesphere/pkg/api"
 	"kubesphere.io/kubesphere/pkg/apiserver/runtime"
 	"kubesphere.io/kubesphere/pkg/constants"
+	"kubesphere.io/kubesphere/pkg/informers"
 	"kubesphere.io/kubesphere/pkg/server/errors"
 	"kubesphere.io/kubesphere/pkg/simple/client/gateway"
 )
 
 var GroupVersion = schema.GroupVersion{Group: "gateway.kubesphere.io", Version: "v1alpha1"}
 
-func AddToContainer(container *restful.Container, options *gateway.Options, cache cache.Cache, client client.Client) error {
+func AddToContainer(container *restful.Container, options *gateway.Options, cache cache.Cache, client client.Client, factory informers.InformerFactory, k8sClient kubernetes.Interface) error {
 	ws := runtime.NewWebService(GroupVersion)
 
-	handler := newHandler(options, cache, client)
+	handler := newHandler(options, cache, client, factory, k8sClient)
 
 	// register gateway apis
 	ws.Route(ws.POST("/namespaces/{namespace}/gateways").
@@ -85,6 +87,21 @@ func AddToContainer(container *restful.Container, options *gateway.Options, cach
 		Returns(http.StatusOK, api.StatusOK, v1alpha1.Gateway{}).
 		Metadata(restfulspec.KeyOpenAPITags, []string{constants.GatewayTag}))
 
+	ws.Route(ws.GET("/namespaces/{namespace}/gateways/{gateway}/pods").
+		To(handler.ListPods).
+		Doc("Retrieve gateways workload pods.").
+		Param(ws.PathParameter("namespace", "the watching namespace of the gateway")).
+		Returns(http.StatusOK, api.StatusOK, v1alpha1.Gateway{}).
+		Metadata(restfulspec.KeyOpenAPITags, []string{constants.GatewayTag}))
+
+	ws.Route(ws.GET("/namespaces/{namespace}/gateways/{gateway}/pods/{pod}/log").
+		To(handler.PodLog).
+		Doc("Retrieve log of the gateway's pod").
+		Param(ws.PathParameter("namespace", "the watching namespace of the gateway")).
+		Param(ws.PathParameter("pod", "the pod name of the gateway")).
+		Returns(http.StatusOK, api.StatusOK, v1alpha1.Gateway{}).
+		Metadata(restfulspec.KeyOpenAPITags, []string{constants.GatewayTag}))
+
 	container.Add(ws)
 	return nil
 }