Spark Thrift server memory leak

We are running Thrift Server in standalone mode and we have observed that the heap of the driver is constantly increasing. After analysing the heap dump the issue seems to be that the ElementTrackingStore is constantly increasing due to the addition of RDDOperationGraphWrapper objects that are not cleaned up.

The ElementTrackingStore defines the addTrigger method were you are able to set thresholds in order to perform cleanup but in practice it is used for  ExecutorSummaryWrapper, JobDataWrapper and StageDataWrapper classes by using the following spark properties

• spark.ui.retainedDeadExecutors
• spark.ui.retainedJobs
• spark.ui.retainedStages

So the  RDDOperationGraphWrapper which is been added using the onJobStart method of  AppStatusListener class kvstore.write(uigraph) #line 291

in not cleaned up and it constantly increases causing a memory leak