如何在flink中更新KeyedBroadcastProcessFunction中的广播状态？

public class MyBroadcastProcessFunction extends KeyedBroadcastProcessFunction<Key, InputEvent, BroadcastEvent, OutputEvent> {
    
    private MapState<String, BroadcastEvent> broadcastState;
    
    @Override
    public void open(Configuration parameters) throws Exception {
        broadcastState = getRuntimeContext().getMapState(new MapStateDescriptor<>("broadcastState", String.class, BroadcastEvent.class));
    }
    
    @Override
    public void processBroadcastElement(BroadcastEvent value, Context ctx, Collector<OutputEvent> out) throws Exception {
        // 存储广播流中的数据到广播状态
        broadcastState.put(value.getKey(), value);
    }
    
    @Override
    public void processElement(InputEvent value, ReadOnlyContext ctx, Collector<OutputEvent> out) throws Exception {
        // 使用广播状态中的数据
        BroadcastEvent broadcastEvent = broadcastState.get(value.getKey());
        // ...
        
        // 更新广播状态中的数据
        broadcastState.put(value.getKey(), newBroadcastEvent);
        // ...
    }
}