Refactor code

x/registry/keeper/genesis.go

@@ -62,7 +62,7 @@ func (k *Keeper) ExportGenesis(ctx sdk.Context) (*registry.GenesisState, error)
 		return nil, err
 	}
 
-	records, err := k.ListRecords(ctx)
+	records, _, err := k.PaginatedListRecords(ctx, nil)
 	if err != nil {
 		return nil, err
 	}

x/registry/keeper/keeper.go

@@ -186,10 +186,12 @@ func (k Keeper) HasRecord(ctx sdk.Context, id string) (bool, error) {
 	return has, nil
 }
 
-// ListRecords - get all records.
+// PaginatedListRecords - get all records with optional pagination.
-func (k Keeper) ListRecords(ctx sdk.Context) ([]registrytypes.Record, error) {
+func (k Keeper) PaginatedListRecords(ctx sdk.Context, pagination *query.PageRequest) ([]registrytypes.Record, *query.PageResponse, error) {
 	var records []registrytypes.Record
+	var pageResp *query.PageResponse
 
+	if pagination == nil {
 		err := k.Records.Walk(ctx, nil, func(key string, value registrytypes.Record) (bool, error) {
 			if err := k.populateRecordNames(ctx, &value); err != nil {
 				return true, err
@@ -199,15 +201,11 @@ func (k Keeper) ListRecords(ctx sdk.Context) ([]registrytypes.Record, error) {
 			return false, nil
 		})
 		if err != nil {
-		return nil, err
+			return nil, nil, err
 		}
-
+	} else {
-	return records, nil
+		var err error
-}
+		records, pageResp, err = query.CollectionPaginate(ctx, k.Records, pagination, func(key string, value registrytypes.Record) (registrytypes.Record, error) {
-
-// PaginatedListRecords - get all records with pagination.
-func (k Keeper) PaginatedListRecords(ctx sdk.Context, pagination *query.PageRequest) ([]registrytypes.Record, *query.PageResponse, error) {
-	records, pageResp, err := query.CollectionPaginate(ctx, k.Records, pagination, func(key string, value registrytypes.Record) (registrytypes.Record, error) {
 			if err := k.populateRecordNames(ctx, &value); err != nil {
 				return registrytypes.Record{}, err
 			}
@@ -217,6 +215,7 @@ func (k Keeper) PaginatedListRecords(ctx sdk.Context, pagination *query.PageRequ
 		if err != nil {
 			return nil, nil, err
 		}
+	}
 
 	return records, pageResp, nil
 }
@@ -259,58 +258,18 @@ func (k Keeper) GetRecordsByBondId(ctx sdk.Context, bondId string) ([]registryty
 	return records, nil
 }
 
-// RecordsFromAttributes gets a list of records whose attributes match all provided values
-func (k Keeper) RecordsFromAttributes(
-	ctx sdk.Context,
-	attributes []*registrytypes.QueryRecordsRequest_KeyValueInput,
-	all bool,
-) ([]registrytypes.Record, error) {
-	resultRecordIds := []string{}
-	for i, attr := range attributes {
-		suffix, err := QueryValueToJSON(attr.Value)
-		if err != nil {
-			return nil, err
-		}
-		mapKey := collections.Join(attr.Key, string(suffix))
-		recordIds, err := k.getAttributeMapping(ctx, mapKey)
-		if err != nil {
-			return nil, err
-		}
-
-		if i == 0 {
-			resultRecordIds = recordIds
-		} else {
-			resultRecordIds = getIntersection(recordIds, resultRecordIds)
-		}
-	}
-
-	records := []registrytypes.Record{}
-	for _, id := range resultRecordIds {
-		record, err := k.GetRecordById(ctx, id)
-		if err != nil {
-			return nil, err
-		}
-		if record.Deleted {
-			continue
-		}
-		if !all && len(record.Names) == 0 {
-			continue
-		}
-		records = append(records, record)
-	}
-
-	return records, nil
-}
-
 // PaginatedRecordsFromAttributes gets a list of records whose attributes match all provided values
-// with pagination.
+// with optional pagination.
 func (k Keeper) PaginatedRecordsFromAttributes(
 	ctx sdk.Context,
 	attributes []*registrytypes.QueryRecordsRequest_KeyValueInput,
 	all bool,
 	pagination *query.PageRequest,
 ) ([]registrytypes.Record, *query.PageResponse, error) {
-	resultRecordIds := []string{}
+	var resultRecordIds []string
+	var pageResp *query.PageResponse
+
+	filteredRecordIds := []string{}
 	for i, attr := range attributes {
 		suffix, err := QueryValueToJSON(attr.Value)
 		if err != nil {
@@ -323,16 +282,20 @@ func (k Keeper) PaginatedRecordsFromAttributes(
 		}
 
 		if i == 0 {
-			resultRecordIds = recordIds
+			filteredRecordIds = recordIds
 		} else {
-			resultRecordIds = getIntersection(recordIds, resultRecordIds)
+			filteredRecordIds = getIntersection(recordIds, filteredRecordIds)
 		}
 	}
 
-	paginatedResultRecordIds, pageRes := paginate(resultRecordIds, pagination)
+	if pagination != nil {
+		resultRecordIds, pageResp = paginate(filteredRecordIds, pagination)
+	} else {
+		resultRecordIds = filteredRecordIds
+	}
 
 	records := []registrytypes.Record{}
-	for _, id := range paginatedResultRecordIds {
+	for _, id := range resultRecordIds {
 		record, err := k.GetRecordById(ctx, id)
 		if err != nil {
 			return nil, nil, err
@@ -346,7 +309,7 @@ func (k Keeper) PaginatedRecordsFromAttributes(
 		records = append(records, record)
 	}
 
-	return records, pageRes, nil
+	return records, pageResp, nil
 }
 
 // TODO not recursive, and only should be if we want to support querying with whole sub-objects,

x/registry/keeper/query_server.go

@@ -43,22 +43,12 @@ func (qs queryServer) Records(c context.Context, req *registrytypes.QueryRecords
 	var pageResp *query.PageResponse
 	var err error
 	if len(attributes) > 0 {
-		if req.Pagination != nil {
 		records, pageResp, err = qs.k.PaginatedRecordsFromAttributes(ctx, attributes, all, req.Pagination)
-		} else {
-			records, err = qs.k.RecordsFromAttributes(ctx, attributes, all)
-		}
-
 		if err != nil {
 			return nil, err
 		}
 	} else {
-		if req.Pagination != nil {
 		records, pageResp, err = qs.k.PaginatedListRecords(ctx, req.Pagination)
-		} else {
-			records, err = qs.k.ListRecords(ctx)
-		}
-
 		if err != nil {
 			return nil, err
 		}