feat(getallassets) : add support for include_fields in request to lim…

…it the response.

Makefile

@@ -26,7 +26,7 @@ help:           ##@help show this help
 NAME="github.com/goto/compass"
 VERSION=$(shell git describe --always --tags 2>/dev/null)
 COVERFILE="/tmp/compass.coverprofile"
-PROTON_COMMIT := "a6b2821e8ddd1127a63d3b376f860990d58931da"
+PROTON_COMMIT := "0db56803e2f75591fe9d1cd461122ebcbaeda8c6"
 
 TOOLS_MOD_DIR = ./tools
 TOOLS_DIR = $(abspath ./.tools)

core/asset/filter.go

@@ -7,31 +7,35 @@ import (
 )
 
 type Filter struct {
-	Types         []Type
-	Services      []string
-	Size          int
-	Offset        int
-	SortBy        string `validate:"omitempty,oneof=name type service created_at updated_at"`
-	SortDirection string `validate:"omitempty,oneof=asc desc"`
-	QueryFields   []string
-	Query         string
-	Data          map[string][]string
+	Types           []Type
+	Services        []string
+	Size            int
+	Offset          int
+	SortBy          string `validate:"omitempty,oneof=name type service created_at updated_at"`
+	SortDirection   string `validate:"omitempty,oneof=asc desc"`
+	QueryFields     []string
+	Query           string
+	Data            map[string][]string
+	IncludeFields   []string
+	IncludeUserInfo bool
 }
 
 func (f *Filter) Validate() error {
 	return validator.ValidateStruct(f)
 }
 
 type filterBuilder struct {
-	types         string
-	services      string
-	q             string
-	qFields       string
-	data          map[string]string
-	size          int
-	offset        int
-	sortBy        string
-	sortDirection string
+	types           string
+	services        string
+	q               string
+	qFields         string
+	data            map[string]string
+	size            int
+	offset          int
+	sortBy          string
+	sortDirection   string
+	includeFields   []string
+	includeUserInfo bool
 }
 
 func NewFilterBuilder() *filterBuilder {
@@ -83,13 +87,25 @@ func (fb *filterBuilder) SortDirection(sortDirection string) *filterBuilder {
 	return fb
 }
 
+func (fb *filterBuilder) IncludeFields(includeFields []string) *filterBuilder {
+	fb.includeFields = includeFields
+	return fb
+}
+
+func (fb *filterBuilder) IncludeUserInfo(includeUserInfo bool) *filterBuilder {
+	fb.includeUserInfo = includeUserInfo
+	return fb
+}
+
 func (fb *filterBuilder) Build() (Filter, error) {
 	flt := Filter{
-		Size:          fb.size,
-		Offset:        fb.offset,
-		SortBy:        fb.sortBy,
-		SortDirection: fb.sortDirection,
-		Query:         fb.q,
+		Size:            fb.size,
+		Offset:          fb.offset,
+		SortBy:          fb.sortBy,
+		SortDirection:   fb.sortDirection,
+		Query:           fb.q,
+		IncludeFields:   fb.includeFields,
+		IncludeUserInfo: fb.includeUserInfo,
 	}
 
 	if len(fb.data) != 0 {

internal/server/v1beta1/asset.go

@@ -64,6 +64,8 @@ func (server *APIServer) GetAllAssets(ctx context.Context, req *compassv1beta1.G
 		SortBy(req.GetSort()).
 		SortDirection(req.GetDirection()).
 		Data(req.GetData()).
+		IncludeFields(req.IncludeFields).
+		IncludeUserInfo(req.IncludeUserInfo).
 		Build()
 	if err != nil {
 		return nil, status.Error(codes.InvalidArgument, bodyParserErrorMsg(err))

internal/store/postgres/asset_repository.go

@@ -26,20 +26,54 @@ type AssetRepository struct {
 
 // GetAll retrieves list of assets with filters
 func (r *AssetRepository) GetAll(ctx context.Context, flt asset.Filter) ([]asset.Asset, error) {
-	builder := r.getAssetSQL().Offset(uint64(flt.Offset))
-	size := flt.Size
+	fields := flt.IncludeFields
+
+	var builder sq.SelectBuilder
+	if len(fields) == 0 {
+		builder = r.getAssetSQL().Offset(uint64(flt.Offset))
+	} else {
+		ifIdNotIncluded := true
+		for _, field := range fields {
+			if strings.ToLower(field) == "a.id" {
+				ifIdNotIncluded = false
+			}
+		}
+
+		if ifIdNotIncluded {
+			fields = append(fields, "a.id")
+		}
 
-	if size > 0 {
-		builder = r.getAssetSQL().Limit(uint64(size)).Offset(uint64(flt.Offset))
+		if flt.IncludeUserInfo {
+			userFields := []string{
+				"u.id as \"updated_by.id\"",
+				"u.uuid as \"updated_by.uuid\"",
+				"u.email as \"updated_by.email\"",
+				"u.provider as \"updated_by.provider\"",
+				"u.created_at as \"updated_by.created_at\"",
+				"u.updated_at as \"updated_by.updated_at\"",
+			}
+			fields = append(fields, userFields...)
+		}
+
+		builder = sq.Select(fields...).From("assets as a").
+			LeftJoin("users u ON a.updated_by = u.id").
+			Offset(uint64(flt.Offset))
+	}
+
+	if flt.Size > 0 {
+		builder = builder.Limit(uint64(flt.Size))
 	}
+
 	builder = r.BuildFilterQuery(builder, flt)
 	builder = r.buildOrderQuery(builder, flt)
+
 	query, args, err := builder.PlaceholderFormat(sq.Dollar).ToSql()
 	if err != nil {
 		return nil, fmt.Errorf("error building query: %w", err)
 	}
 
 	var ams []*AssetModel
+
 	err = r.client.db.SelectContext(ctx, &ams, query, args...)
 	if err != nil {
 		return nil, fmt.Errorf("error getting asset list: %w", err)

proto/compass.swagger.yaml

@@ -94,6 +94,18 @@ paths:
           in: query
           required: false
           type: boolean
+        - name: include_fields
+          in: query
+          required: false
+          type: array
+          items:
+            type: string
+          collectionFormat: multi
+        - name: include_user_info
+          description: if true user information will be added in the response else not
+          in: query
+          required: false
+          type: boolean
       tags:
         - Asset
     put: