#RI-3612-add inifinity score in zset

Author: AmirAllayarovSofteq

redisinsight/api/src/modules/browser/__mocks__/z-set.ts

@@ -8,12 +8,21 @@ import { SortOrder } from 'src/constants';
 
 export const mockZSetMemberDto: ZSetMemberDto = {
   name: Buffer.from('member1'),
-  score: 0,
+  score: '-inf',
 };
 export const mockZSetMemberDto2: ZSetMemberDto = {
   name: Buffer.from('member2'),
+  score: 0,
+};
+export const mockZSetMemberDto3: ZSetMemberDto = {
+  name: Buffer.from('member3'),
   score: 2,
 };
+
+export const mockZSetMemberDto4: ZSetMemberDto = {
+  name: Buffer.from('member4'),
+  score: 'inf',
+};
 export const mockGetMembersDto: GetZSetMembersDto = {
   keyName: Buffer.from('zSet'),
   offset: 0,
@@ -28,7 +37,7 @@ export const mockSearchMembersDto: SearchZSetMembersDto = {
 };
 export const mockAddMembersDto: AddMembersToZSetDto = {
   keyName: mockGetMembersDto.keyName,
-  members: [mockZSetMemberDto, mockZSetMemberDto2],
+  members: [mockZSetMemberDto, mockZSetMemberDto2, mockZSetMemberDto3, mockZSetMemberDto4],
 };
 export const mockUpdateMemberDto: UpdateMemberInZSetDto = {
   keyName: mockGetMembersDto.keyName,
@@ -39,10 +48,14 @@ export const mockMembersForZAddCommand = [
   mockZSetMemberDto.name,
   mockZSetMemberDto2.score,
   mockZSetMemberDto2.name,
+  mockZSetMemberDto3.score,
+  mockZSetMemberDto3.name,
+  mockZSetMemberDto4.score,
+  mockZSetMemberDto4.name,
 ];
 export const mockDeleteMembersDto: DeleteMembersFromZSetDto = {
   keyName: mockAddMembersDto.keyName,
-  members: [mockZSetMemberDto.name, mockZSetMemberDto2.name],
+  members: [mockZSetMemberDto.name, mockZSetMemberDto2.name, mockZSetMemberDto3.name, mockZSetMemberDto4.name],
 };
 export const getZSetMembersInAscResponse = {
   keyName: mockGetMembersDto.keyName,

redisinsight/api/src/modules/browser/dto/z-set.dto.ts

@@ -12,6 +12,7 @@ import {
   ValidateNested
 } from 'class-validator';
 import { Type } from 'class-transformer';
+import { isNumber } from 'lodash';
 import { SortOrder } from 'src/constants';
 import {
   DeleteMembersFromSetDto,
@@ -65,6 +66,24 @@ export class GetZSetMembersDto extends KeyDto {
   sortOrder: SortOrder;
 }
 
+export class ScoreNumberValidation {
+  @ApiProperty({
+    description: 'Infinity score values',
+    type: String,
+  })
+  @IsString()
+  score: string;
+}
+
+export class ScoreStringValidation {
+  @ApiProperty({
+    description: 'Common score value',
+    type: Number,
+  })
+  @IsNumber({ maxDecimalPlaces: 15 })
+  score: number;
+}
+
 export class ZSetMemberDto {
   @ApiProperty({
     type: String,
@@ -77,12 +96,11 @@ export class ZSetMemberDto {
 
   @ApiProperty({
     description: 'Member score value.',
-    type: Number,
+    type: Number || String,
     default: 1,
   })
   @IsDefined()
-  @IsNumber({ maxDecimalPlaces: 15 })
-  // @Type(() => Number)
+  @Type((val) => isNumber(val) ? ScoreNumberValidation : ScoreStringValidation)
   score: number | string;
 }
 

redisinsight/api/src/modules/browser/services/z-set-business/z-set-business.service.spec.ts

@@ -184,7 +184,7 @@ describe('ZSetBusinessService', () => {
           BrowserToolZSetCommands.ZRange,
           expect.anything(),
         )
-        .mockResolvedValue(['member1', '0', 'member2', '2']);
+        .mockResolvedValue(['member1', '-inf', 'member2', '0', 'member3', '2', 'member4', 'inf']);
 
       const result = await service.getMembers(
         mockClientOptions,
@@ -199,7 +199,7 @@ describe('ZSetBusinessService', () => {
           BrowserToolZSetCommands.ZRevRange,
           expect.anything(),
         )
-        .mockResolvedValue(['member2', '2', 'member1', '0']);
+        .mockResolvedValue(['member4', 'inf', 'member3', '2', 'member2', '0', 'member1', '-inf']);
 
       const result = await service.getMembers(mockClientOptions, {
         ...mockGetMembersDto,
@@ -485,7 +485,7 @@ describe('ZSetBusinessService', () => {
           BrowserToolZSetCommands.ZScan,
           expect.anything(),
         )
-        .mockResolvedValue([0, ['member1', '0', 'member2', '2']]);
+        .mockResolvedValue([0, ['member1', '-inf', 'member2', '0', 'member3', '2', 'member4', 'inf']]);
 
       const result = await service.searchMembers(
         mockClientOptions,

redisinsight/api/src/modules/browser/services/z-set-business/z-set-business.service.ts

@@ -277,8 +277,10 @@ export class ZSetBusinessService {
           BrowserToolZSetCommands.ZScore,
           [keyName, member],
         );
+        const formattedScore = isNaN(parseFloat(score)) ? String(score) : parseFloat(score);
+
         if (!isNull(score)) {
-          result.members.push(plainToClass(ZSetMemberDto, { name: member, score }));
+          result.members.push(plainToClass(ZSetMemberDto, { name: member, score: formattedScore }));
         }
       } else {
         const scanResult = await this.scanZSet(clientOptions, dto);
@@ -405,7 +407,7 @@ export class ZSetBusinessService {
     const result: ZSetMemberDto[] = [];
     while (reply.length) {
       const member = reply.splice(0, 2);
-      const score = isNaN(parseFloat(member[1])) ? member[1] : parseFloat(member[1]);
+      const score = isNaN(parseFloat(member[1])) ? String(member[1]) : parseFloat(member[1]);
       result.push(plainToClass(ZSetMemberDto, {
         name: member[0],
         score,

redisinsight/ui/src/pages/browser/components/zset-details/ZSetDetails.spec.tsx

@@ -12,13 +12,14 @@ jest.mock('uiSrc/slices/browser/zset', () => {
     setZsetInitialState: jest.fn,
     zsetDataSelector: jest.fn().mockReturnValue({
       ...defaultState,
-      total: 3,
+      total: 4,
       key: 'z',
       keyName: 'z',
       members: [
         { name: { type: 'Buffer', data: [49] }, score: 1 },
         { name: { type: 'Buffer', data: [50] }, score: 2 },
         { name: { type: 'Buffer', data: [51] }, score: 3 },
+        { name: { type: 'Buffer', data: [52] }, score: 'inf' },
       ],
     }),
     updateZsetScoreStateSelector: jest.fn().mockReturnValue(defaultState.updateScore),
@@ -52,6 +53,16 @@ describe('ZSetDetails', () => {
     expect(screen.getByTestId(/zset-edit-button-1/)).toBeInTheDocument()
   })
 
+  it('should render disabled edit button', () => {
+    render(<ZSetDetails {...instance(mockedProps)} />)
+    expect(screen.getByTestId(/zset-edit-button-4/)).toBeDisabled()
+  })
+
+  it('should render enabled edit button', () => {
+    render(<ZSetDetails {...instance(mockedProps)} />)
+    expect(screen.getByTestId(/zset-edit-button-3/)).not.toBeDisabled()
+  })
+
   it('should render editor after click edit button and able to change value', () => {
     render(<ZSetDetails {...instance(mockedProps)} />)
     fireEvent.click(screen.getAllByTestId(/zset-edit-button/)[0])

redisinsight/ui/src/pages/browser/components/zset-details/ZSetDetails.tsx

@@ -1,6 +1,6 @@
 import React, { useCallback, useEffect, useRef, useState } from 'react'
 import { useDispatch, useSelector } from 'react-redux'
-import { toNumber } from 'lodash'
+import { toNumber, isNumber } from 'lodash'
 import cx from 'classnames'
 import { EuiButtonIcon, EuiProgress, EuiText, EuiToolTip } from '@elastic/eui'
 import { CellMeasurerCache } from 'react-virtualized'
@@ -39,8 +39,8 @@ import InlineItemEditor from 'uiSrc/components/inline-item-editor/InlineItemEdit
 import { IColumnSearchState, ITableColumn } from 'uiSrc/components/virtual-table/interfaces'
 import { StopPropagation } from 'uiSrc/components/virtual-table'
 import { getColumnWidth } from 'uiSrc/components/virtual-grid'
-import { AddMembersToZSetDto, SearchZSetMembersResponse } from 'apiSrc/modules/browser/dto'
 import { stringToBuffer } from 'uiSrc/utils/formatters/bufferFormatters'
+import { AddMembersToZSetDto, SearchZSetMembersResponse } from 'apiSrc/modules/browser/dto'
 import PopoverDelete from '../popover-delete/PopoverDelete'
 
 import styles from './styles.module.scss'
@@ -326,20 +326,22 @@ const ZSetDetails = (props: Props) => {
       minWidth: 100,
       maxWidth: 100,
       absoluteWidth: 100,
-      render: function Actions(_act: any, { name: nameItem }: IZsetMember) {
+      render: function Actions(_act: any, { name: nameItem, score }: IZsetMember) {
         const name = bufferToString(nameItem, viewFormat)
         return (
           <StopPropagation>
             <div className="value-table-actions">
-              <EuiButtonIcon
-                iconType="pencil"
-                aria-label="Edit field"
-                className="editFieldBtn"
-                color="primary"
-                disabled={updateLoading}
-                onClick={() => handleEditMember(nameItem, true)}
-                data-testid={`zset-edit-button-${name}`}
-              />
+              <EuiToolTip content={!isNumber(score) ? 'Use CLI or Workbench to edit the score' : null}>
+                <EuiButtonIcon
+                  iconType="pencil"
+                  aria-label="Edit field"
+                  className="editFieldBtn"
+                  color="primary"
+                  disabled={updateLoading || !isNumber(score)}
+                  onClick={() => handleEditMember(nameItem, true)}
+                  data-testid={`zset-edit-button-${name}`}
+                />
+              </EuiToolTip>
               <PopoverDelete
                 header={createDeleteFieldHeader(nameItem)}
                 text={createDeleteFieldMessage(key ?? '')}