DataTable: keep focus on filter input

[majagrubic]

I am using Prime React Datatable. I would love to keep built-in controls for filtering and pagination, but have filtering and pagination server side.
I am running into a small problem: when a user inputs something in the filtering input, a new request is made to the server, which reloads the data in the table (expected). However, when that happens, the focus from the input field is lost; in order to continue typingm the user needs to click back to the field they were previously typing into; this leads to a very poor user experience.

Is there a simple way to keep focus on the filter input fields when data reloads, without writing my own controls?

Simplified code:

const [filters, setFilters] = useState(initFilterValues);
const [values, setValues] = useState(initialData);

const onFilter = (e: DataTableStateEvent) => {
    const newFilters = { ...filters, ...e.filters };
    setFilters(newFilters);
};

useEffect(() => {
   // fetch data from server
   // const data = await response.json();
   setValues(data);
}, [filters]);

return(
  <DataTable
            values={values}
            filters={filters}
            onFilter={onFilter}
            filterDisplay="row"
            lazy
  >
    <Column field="col1" filter={true} />
    <Column field="col2" filter={true} />
    <Column field="col3" filter={true} />
  </DataTable>
);