server/resource/v1/history_resource.go

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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.
 */

package v1

import (
	"net/http"

	"github.com/apache/servicecomb-kie/pkg/common"
	"github.com/apache/servicecomb-kie/pkg/model"
	"github.com/apache/servicecomb-kie/server/datasource"
	"github.com/go-chassis/cari/config"

	goRestful "github.com/emicklei/go-restful"
	"github.com/go-chassis/go-chassis/v2/server/restful"
	"github.com/go-chassis/openlog"
)

// HistoryResource TODO
type HistoryResource struct {
}

// GetRevisions search key only by label
func (r *HistoryResource) GetRevisions(context *restful.Context) {
	var err error
	kvID := context.ReadPathParameter(common.PathParamKVID)
	offsetStr := context.ReadQueryParameter(common.QueryParamOffset)
	limitStr := context.ReadQueryParameter(common.QueryParamLimit)
	offset, limit, err := checkPagination(offsetStr, limitStr)
	if err != nil {
		WriteErrResponse(context, config.ErrInvalidParams, err.Error())
		return
	}
	if kvID == "" {
		openlog.Error("kv id is nil")
		WriteErrResponse(context, config.ErrRequiredRecordId, "kv_id must not be empty")
		return
	}
	revisions, err := datasource.GetBroker().GetHistoryDao().GetHistory(context.Ctx, kvID,
		context.ReadPathParameter(common.PathParameterProject),
		ReadDomain(context.Ctx),
		datasource.WithOffset(offset),
		datasource.WithLimit(limit))
	if err != nil {
		if err == datasource.ErrRevisionNotExist {
			WriteErrResponse(context, config.ErrRecordNotExists, err.Error())
			return
		}
		WriteError(context, err)
		return
	}
	err = writeResponse(context, revisions)
	if err != nil {
		openlog.Error(err.Error())
	}
}

// GetPollingData get the record of the get or list history
func (r *HistoryResource) GetPollingData(context *restful.Context) {
	query := &model.PollingDetail{}
	sessionID := context.ReadQueryParameter(common.QueryParamSessionID)
	if sessionID != "" {
		query.SessionID = sessionID
	}
	sessionCroup := context.ReadQueryParameter(common.QueryParamSessionGroup)
	if sessionCroup != "" {
		query.SessionGroup = sessionCroup
	}
	revision := context.ReadQueryParameter(common.QueryParamRev)
	if revision != "" {
		query.Revision = revision
	}
	ip := context.ReadQueryParameter(common.QueryParamIP)
	if ip != "" {
		query.IP = ip
	}
	urlPath := context.ReadQueryParameter(common.QueryParamURLPath)
	if urlPath != "" {
		query.URLPath = urlPath
	}
	userAgent := context.ReadQueryParameter(common.QueryParamUserAgent)
	if userAgent != "" {
		query.UserAgent = userAgent
	}
	domain := ReadDomain(context.Ctx)
	if domain == "" {
		WriteErrResponse(context, config.ErrInternal, common.MsgDomainMustNotBeEmpty)
		return
	}
	query.Domain = domain
	query.Project = context.ReadPathParameter(common.PathParameterProject)
	records, err := datasource.GetBroker().GetTrackDao().GetPollingDetail(context.Ctx, query)
	if err != nil {
		if err == datasource.ErrRecordNotExists {
			WriteErrResponse(context, config.ErrRecordNotExists, err.Error())
			return
		}
		WriteErrResponse(context, config.ErrInternal, err.Error())
		return
	}
	resp := &model.PollingDataResponse{}
	resp.Data = records
	resp.Total = len(records)
	err = writeResponse(context, resp)
	if err != nil {
		openlog.Error(err.Error())
	}
}

// URLPatterns defined config operations
func (r *HistoryResource) URLPatterns() []restful.Route {
	return []restful.Route{
		{
			Method:       http.MethodGet,
			Path:         "/v1/{project}/kie/revision/{kv_id}",
			ResourceFunc: r.GetRevisions,
			FuncDesc:     "get all revisions by key id",
			Parameters: []*restful.Parameters{
				DocPathProject, DocPathKeyID,
			},
			Returns: []*restful.Returns{
				{
					Code:  http.StatusOK,
					Model: model.DocResponseGetKey{},
				},
			},
			Consumes: []string{goRestful.MIME_JSON},
			Produces: []string{goRestful.MIME_JSON},
		},

		{
			Method:       http.MethodGet,
			Path:         "/v1/{project}/kie/track",
			ResourceFunc: r.GetPollingData,
			FuncDesc:     "get polling tracks of clients of kie server",
			Parameters: []*restful.Parameters{
				DocPathProject, DocQuerySessionIDParameters, DocQueryIPParameters, DocQueryURLPathParameters, DocQueryUserAgentParameters,
			},
			Returns: []*restful.Returns{
				{
					Code:    http.StatusOK,
					Message: "true",
					Model:   []model.PollingDataResponse{},
				},
			},
			Consumes: []string{goRestful.MIME_JSON},
			Produces: []string{goRestful.MIME_JSON},
		},
	}
}