mcs/class/corlib/System.Collections.Concurrent/ObjectPool.cs

   1 // ObjectPool.cs
   2 //
   3 // Copyright (c) 2011 Novell
   4 //
   5 // Authors:
   6 //      Jérémie "garuma" Laval
   7 //
   8 // Permission is hereby granted, free of charge, to any person obtaining a copy
   9 // of this software and associated documentation files (the "Software"), to deal
  10 // in the Software without restriction, including without limitation the rights
  11 // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  12 // copies of the Software, and to permit persons to whom the Software is
  13 // furnished to do so, subject to the following conditions:
  14 //
  15 // The above copyright notice and this permission notice shall be included in
  16 // all copies or substantial portions of the Software.
  17 //
  18 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  19 // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  20 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  21 // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  22 // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  23 // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  24 // THE SOFTWARE.
  25 //
  26 //
  27
  28 #if NET_4_0 || MOBILE || BOOTSTRAP_NET_4_0 || INSIDE_SYSTEM_WEB
  29
  30 using System;
  31 using System.Threading;
  32 using System.Collections;
  33 using System.Collections.Generic;
  34 using System.Runtime.Serialization;
  35
  36 namespace System.Collections.Concurrent
  37 {
  38         internal abstract class ObjectPool<T> where T : class
  39         {
  40                 const int capacity = 20;
  41                 const int bit = 0x8000000;
  42
  43                 readonly T[] buffer;
  44                 int addIndex;
  45                 int removeIndex;
  46
  47                 public ObjectPool ()
  48                 {
  49                         buffer = new T[capacity];
  50                         for (int i = 0; i < capacity; i++)
  51                                 buffer[i] = Creator ();
  52                         addIndex = capacity - 1;
  53                 }
  54
  55                 protected abstract T Creator ();
  56
  57                 public T Take ()
  58                 {
  59                         if ((addIndex & ~bit) - 1 == removeIndex)
  60                                 return Creator ();
  61
  62                         int i;
  63                         T result;
  64                         int tries = 3;
  65
  66                         do {
  67                                 i = removeIndex;
  68                                 if ((addIndex & ~bit) - 1 == i || tries == 0)
  69                                         return Creator ();
  70                                 result = buffer[i % capacity];
  71                         } while (Interlocked.CompareExchange (ref removeIndex, i + 1, i) != i && --tries > -1);
  72
  73                         return result;
  74                 }
  75
  76                 public void Release (T obj)
  77                 {
  78                         if (obj == null || addIndex - removeIndex >= capacity - 1)
  79                                 return;
  80
  81                         int i;
  82                         int tries = 3;
  83                         do {
  84                                 do {
  85                                         i = addIndex;
  86                                 } while ((i & bit) > 0);
  87                                 if (i - removeIndex >= capacity - 1)
  88                                         return;
  89                         } while (Interlocked.CompareExchange (ref addIndex, i + 1 + bit, i) != i && --tries > 0);
  90
  91                         buffer[i % capacity] = obj;
  92                         addIndex = addIndex - bit;
  93                 }
  94         }
  95 }
  96
  97 #endif