Skip to main content

BlockingQueue to prevent OOM

Saw an interesting hack when a colleague sent me his code for review. We are a cloud storage company and people upload files and if they upload same file again and again it creates a new version. Some examples are quickbooks or outlook files that will generate multiple versions in a day if you have enabled real time sync on that folder where these files are stored. To optimize bandwidth we use rsync and do a  patch on server to reconstruct these large files but as we save the original file the customer gets charged the full size of the file. This is why customers configure version policy that they would allow 5 versions of the file and if a new one is uploaded we move the oldest to trash.  Now if the customer reduced the versions to keep from 5 to 2 then suddenly we have to delete all these versions.

So earlier to offload processing we had written a rest api that in streaming fashion would return list of deletable versions metadata.

/rest/public/getDeletableVersions   GET.

then a python script would call deleteVersions api in batch.

/rest/public/deleteVersions POST

now this was all complex to test so after 2 years we moved it back to tomcat and rewrote this as a quartz job.

so the programmer reused the code and wrote it as

List<DeletableVersionResponse> deletableVersions = storageService.getDeletableVersions(customerId);



 for(List<DeletableVersionResponse> batch: split(deletableVersions) ) {

   storageService.deleteVersions(batch) ;

}

Problem is that for bigger customers that had 10M+ versions this was causing OOM when we were trying to load all deletable versions.

So I asked the engineer to convert it in such a manner that the api would be
int numDeleted;
while((numDeleted=storageService.deleteNextBatchOfDeletableVersions(customerId))>0) {

}



but this required change in all the layers.



Instead the engineer came up with solution to use BlockingQueue.



So  what he did was

        BlockingQueue<DeletableVersionResponse> deletableVersionResponse = new ArrayBlockingQueue<DeletableVersionResponse>(
                eventBatchSize);
        Future future = executorService.submit(new DeletableVersionRequest(deletableVersionResponse, customerId));
        while (!(deletableVersionResponse.isEmpty() && future.isDone())) {
            processDeletableVersionResponse(user, deletableVersionResponse.poll());
        }

This was a creative way to solve OOM without changing a lot of layers of code.

Comments

Popular posts from this blog

RabbitMQ java clients for beginners

Here is a sample of a consumer and producer example for RabbitMQ. The steps are
Download ErlangDownload Rabbit MQ ServerDownload Rabbit MQ Java client jarsCompile and run the below two class and you are done.
This sample create a Durable Exchange, Queue and a Message. You will have to start the consumer first before you start the for the first time.

For more information on AMQP, Exchanges, Queues, read this excellent tutorial
http://blogs.digitar.com/jjww/2009/01/rabbits-and-warrens/

+++++++++++++++++RabbitMQProducer.java+++++++++++++++++++++++++++
import com.rabbitmq.client.Connection; import com.rabbitmq.client.Channel; import com.rabbitmq.client.*; public class RabbitMQProducer { public static void main(String []args) throws Exception { ConnectionFactory factory = new ConnectionFactory(); factory.setUsername("guest"); factory.setPassword("guest"); factory.setVirtualHost("/"); factory.setHost("127.0.0.1"); factory.setPort(5672); Conne…

Spring query timeout or transaction timeout

If you are using spring to manage transactions then you can specify default transaction timeout using

    <bean id="transactionManager"
        class="org.springframework.jdbc.datasource.DataSourceTransactionManager">
        <property name="dataSource" ref="dataSource" />
        <property name="defaultTimeout" value="30" /> <!--30 sec--->             
    </bean>

or you can override the timeout in the annotation

    @Transactional(readOnly = false, timeout=30)


or if you are doing it programatic transactions then you can do


DataSourceTransactionManager transactionManager = new DataSourceTransactionManager(dataSource);
transactionManager.setDefaultTimeout(30);

 or you can override the timeout for one particular transaction

TransactionTemplate transactionTemplate = new TransactionTemplate();
transactionTemplate.setTimeout(30);

Python adding pid file

I have a thumbnail generator that launches multiple processes and the correct way to shut it down is to send kill -HUP to the parent process. To automate I had to write a pid file from python, it was a piece of cake
def writePidFile(): pid = str(os.getpid()) f = open('thumbnail_rabbit_consumer.pid', 'w') f.write(pid) f.close()