React-data-grid :根据行的值更改行颜色

import ReactDataGrid from 'react-data-grid';
import 'react-data-grid/dist/react-data-grid.css';

const columns = [
  { key: 'id', name: 'ID' },
  { key: 'name', name: 'Name' },
  { key: 'age', name: 'Age' }
];

const rows = [
  { id: 1, name: 'John', age: 25 },
  { id: 2, name: 'Alice', age: 30 },
  { id: 3, name: 'Bob', age: 35 }
];

const getRowStyle = (row) => {
  if (row.age < 30) {
    return { backgroundColor: 'green' };
  } else if (row.age > 30) {
    return { backgroundColor: 'red' };
  }
  return null;
};

const rowGetter = (i) => rows[i];

const CustomDataGrid = () => (
  <ReactDataGrid
    columns={columns}
    rowGetter={rowGetter}
    rowsCount={rows.length}
    getRowStyle={getRowStyle}
  />
);