gistfile1.erl

with_worker(Fun) when is_function(Fun, 1) ->                                                                                                                                                                  
    {IsFromDict, Worker} =                                                                                                                                                                                    
        case get($riak_pool_worker) of                                                                                                                                                                        
            undefined ->                                                                                                                                                                                      
                Worker = riak_pool:get_worker(),                                                                                                                                                              
                put($riak_pool_worker, Worker),                                                                                                                                                               
                {false, Worker};                                                                                                                                                                              
            Worker ->                                                                                                                                                                                         
                {true, Worker}                                                                                                                                                                                
        end,                                                                                                                                                                                                  
    try                                                                                                                                                                                                       
        Fun(Worker)                                                                                                                                                                                           
    after                                                                                                                                                                                                     
        case IsFromDict of                                                                                                                                                                                    
            true  -> pass;                                                                                                                                                                                    
            false ->                                                                                                                                                                                          
                erase($riak_pool_worker),                                                                                                                                                                     
                catch(riak_pool:free_worker(Worker))                                                                                                                                                          
        end                                                                                                                                                                                                   
    end.