Skip to content

Commit

Permalink
[FEATURE] Charts | Change charts time-unit to bigger timespan opensea…
Browse files Browse the repository at this point in the history
…rch-project#164 (opensearch-project#204) (opensearch-project#205)

* [FEATURE] Charts | Change charts time-unit to bigger timespan opensearch-project#164

Signed-off-by: Jovan Cvetkovic <[email protected]>

* [FEATURE] Charts | Change charts time-unit to bigger timespan opensearch-project#164

Signed-off-by: Jovan Cvetkovic <[email protected]>

Signed-off-by: Jovan Cvetkovic <[email protected]>
(cherry picked from commit a915f85)

Co-authored-by: Jovan Cvetkovic <[email protected]>
Signed-off-by: AWSHurneyt <[email protected]>
  • Loading branch information
2 people authored and AWSHurneyt committed Feb 22, 2023
1 parent 3c36271 commit 7e87fa4
Show file tree
Hide file tree
Showing 5 changed files with 231 additions and 32 deletions.
16 changes: 14 additions & 2 deletions public/pages/Alerts/containers/Alerts/Alerts.tsx
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,11 @@ import { FieldValueSelectionFilterConfigType } from '@elastic/eui/src/components
import dateMath from '@elastic/datemath';
import React, { Component } from 'react';
import { ContentPanel } from '../../../../components/ContentPanel';
import { getAlertsVisualizationSpec } from '../../../Overview/utils/helpers';
import {
getAlertsVisualizationSpec,
getChartTimeUnit,
getDateFormatByTimeUnit,
} from '../../../Overview/utils/helpers';
import moment from 'moment';
import {
ALERT_STATE,
Expand Down Expand Up @@ -70,6 +74,7 @@ export interface AlertsState {
filteredAlerts: AlertItem[];
detectors: { [key: string]: Detector };
loading: boolean;
timeUnit: string;
}

const groupByOptions = [
Expand All @@ -93,6 +98,7 @@ export default class Alerts extends Component<AlertsProps, AlertsState> {
filteredAlerts: [],
detectors: {},
loading: false,
timeUnit: 'yearmonthdatehoursminutes',
};
}

Expand Down Expand Up @@ -218,7 +224,10 @@ export default class Alerts extends Component<AlertsProps, AlertsState> {
};
});

return getAlertsVisualizationSpec(visData, this.state.groupBy);
return getAlertsVisualizationSpec(visData, this.state.groupBy, {
timeUnit: this.state.timeUnit,
dateFormat: getDateFormatByTimeUnit(this.state.startTime, this.state.endTime),
});
}

createGroupByControl(): React.ReactNode {
Expand Down Expand Up @@ -297,10 +306,13 @@ export default class Alerts extends Component<AlertsProps, AlertsState> {
if (recentlyUsedRanges.length > MAX_RECENTLY_USED_TIME_RANGES)
recentlyUsedRanges = recentlyUsedRanges.slice(0, MAX_RECENTLY_USED_TIME_RANGES);
const endTime = start === end ? DEFAULT_DATE_RANGE.end : end;

const timeUnit = getChartTimeUnit(start, endTime);
this.setState({
startTime: start,
endTime: endTime,
recentlyUsedRanges: recentlyUsedRanges,
timeUnit,
});
};

Expand Down
18 changes: 15 additions & 3 deletions public/pages/Findings/containers/Findings/Findings.tsx
Original file line number Diff line number Diff line change
Expand Up @@ -29,11 +29,15 @@ import {
MAX_RECENTLY_USED_TIME_RANGES,
OS_NOTIFICATION_PLUGIN,
} from '../../../../utils/constants';
import { getFindingsVisualizationSpec } from '../../../Overview/utils/helpers';
import {
getChartTimeUnit,
getDateFormatByTimeUnit,
getFindingsVisualizationSpec,
} from '../../../Overview/utils/helpers';
import { CoreServicesContext } from '../../../../components/core_services';
import { Finding } from '../../models/interfaces';
import { Detector } from '../../../../../models/interfaces';
import { FeatureChannelList } from '../../../../../server/models/interfaces/Notifications';
import { FeatureChannelList } from '../../../../../server/models/interfaces';
import {
getNotificationChannels,
parseNotificationChannelsToOptions,
Expand All @@ -46,6 +50,7 @@ import {
} from '../../../../utils/helpers';
import { DetectorHit, RuleSource } from '../../../../../server/models/interfaces';
import { NotificationsStart } from 'opensearch-dashboards/public';
import { ruleSeverity } from '../../../Rules/utils/constants';

interface FindingsProps extends RouteComponentProps {
detectorService: DetectorsService;
Expand All @@ -68,6 +73,7 @@ interface FindingsState {
groupBy: FindingsGroupByType;
filteredFindings: FindingItemType[];
plugins: string[];
timeUnit: string;
}

interface FindingVisualizationData {
Expand Down Expand Up @@ -103,6 +109,7 @@ export default class Findings extends Component<FindingsProps, FindingsState> {
groupBy: 'logType',
filteredFindings: [],
plugins: [],
timeUnit: 'yearmonthdatehoursminutes',
};
}

Expand Down Expand Up @@ -234,10 +241,12 @@ export default class Findings extends Component<FindingsProps, FindingsState> {
if (recentlyUsedRanges.length > MAX_RECENTLY_USED_TIME_RANGES)
recentlyUsedRanges = recentlyUsedRanges.slice(0, MAX_RECENTLY_USED_TIME_RANGES);
const endTime = start === end ? DEFAULT_DATE_RANGE.end : end;
const timeUnit = getChartTimeUnit(start, endTime);
this.setState({
startTime: start,
endTime: endTime,
recentlyUsedRanges: recentlyUsedRanges,
timeUnit,
});
};

Expand All @@ -256,7 +265,10 @@ export default class Findings extends Component<FindingsProps, FindingsState> {
});
});

return getFindingsVisualizationSpec(visData, this.state.groupBy);
return getFindingsVisualizationSpec(visData, this.state.groupBy, {
timeUnit: this.state.timeUnit,
dateFormat: getDateFormatByTimeUnit(this.state.startTime, this.state.endTime),
});
}

createGroupByControl(): React.ReactNode {
Expand Down
15 changes: 15 additions & 0 deletions public/pages/Overview/utils/constants.ts
Original file line number Diff line number Diff line change
Expand Up @@ -9,3 +9,18 @@ export const summaryGroupByOptions = [
];

export const moreLink = 'https://opensearch.org/docs/latest/security-analytics/';

/**
* Time axis' timeUnit map
* for each time search unit there is a mapped chart timeUnit
*/
export const TimeUnitsMap: {
[key: string]: string;
} = {
minutes: 'yearmonthdatehoursminutes',
hours: 'yearmonthdatehoursminutes',
days: 'yearmonthdatehours',
weeks: 'yearmonthdatehours',
months: 'yearmonthdatehours',
years: 'yearmonthdate',
};
59 changes: 59 additions & 0 deletions public/pages/Overview/utils/helper.test.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,59 @@
import { getChartTimeUnit, getDateFormatByTimeUnit } from './helpers';
import { TimeUnitsMap } from './constants';

describe('helper utilities spec', () => {
describe('tests getDateFormatByTimeUnit function', () => {
const yearFormat = '%Y-%m-%d';
const dayFormat = '%H:%M:%S';
const fullFormat = '%Y-%m-%d %H:%M';

const timeFormats: {
[key: string]: string;
} = {
'now-15m': dayFormat,
'now-15h': fullFormat,
'now-15d': fullFormat,
'now-2M': yearFormat,
'now-2y': fullFormat,
};

it(` - function should return default format ${fullFormat} if dates are not valid`, () => {
expect(getDateFormatByTimeUnit('', '')).toBe(fullFormat);
});

for (const [start, format] of Object.entries(timeFormats)) {
it(` - function should return ${format} if start date is ${start}`, () => {
expect(getDateFormatByTimeUnit(start, 'now')).toBe(format);
});
}
});

describe('tests getChartTimeUnit function', () => {
const defaultTimeUnit = 'yearmonthdatehoursminutes';
it(' - function should return default timeUnit if fn params are invalid', () => {
expect(getChartTimeUnit('', '')).toBe(defaultTimeUnit);
});

it(' - function should return default timeUnit if one is passed as param', () => {
const defaultFormat = 'yearmonthdate';
expect(getChartTimeUnit('', '', defaultFormat)).toBe(defaultFormat);
});

const timeUnits: {
[key: string]: string;
} = {
minutes: 'now-15m',
hours: 'now-15h',
days: 'now-15d',
weeks: 'now-15w',
months: 'now-5M',
years: 'now-15y',
};

for (const [unit, start] of Object.entries(timeUnits)) {
it(` - function should return ${TimeUnitsMap[unit]} if unit is ${unit}`, () => {
expect(getChartTimeUnit(start, 'now')).toBe(TimeUnitsMap[unit]);
});
}
});
});
Loading

0 comments on commit 7e87fa4

Please sign in to comment.