ACE/tests/Map_Test.h

   1 /* -*- C++ -*- */
   2
   3 //=============================================================================
   4 /**
   5  *  @file    Map_Test.h
   6  *
   7  * This file has the class definitions needed for template generation in
   8  * Map_Test.cpp.
   9  *
  10  *  @author Irfan Pyarali <irfan@cs.wustl.edu>
  11  */
  12 //=============================================================================
  13
  14
  15 #ifndef ACE_TESTS_MAP_TEST_H
  16 #define ACE_TESTS_MAP_TEST_H
  17
  18 #include "ace/OS_NS_string.h"
  19
  20 #if !defined (ACE_LACKS_PRAGMA_ONCE)
  21 # pragma once
  22 #endif /* ACE_LACKS_PRAGMA_ONCE */
  23
  24 #include "ace/Active_Map_Manager.h"
  25 #include "ace/Containers.h"
  26
  27 // Key data type.
  28 typedef ACE_Array<char> KEY;
  29
  30 /**
  31  * @class Key_Generator
  32  *
  33  * @brief Defines a key generator.
  34  *
  35  * This class is used in adapters of maps that do not produce keys.
  36  */
  37 class Key_Generator
  38 {
  39 public:
  40   Key_Generator ()
  41     : counter_ (0)
  42     {
  43     }
  44
  45   int operator() (KEY &key)
  46     {
  47       // Keep original information in the key intact.
  48       size_t original_size = key.size ();
  49
  50       // Size of this counter key.
  51       const size_t counter_key_size = sizeof this->counter_;
  52
  53       // Resize to accommodate both the original data and the new key.
  54       key.size (counter_key_size + original_size);
  55
  56       // Add new key data.
  57       ACE_OS::memcpy (&key[original_size],
  58                       &++this->counter_,
  59                       counter_key_size);
  60
  61       // Success.
  62       return 0;
  63     }
  64
  65 private:
  66   u_long counter_;
  67 };
  68
  69 class Hash_Key
  70 {
  71 public:
  72   u_long operator () (const KEY &key) const
  73     {
  74       // Recover system generated part of key.
  75       u_long value;
  76       size_t counter_key_size = sizeof (u_long);
  77
  78       // Copy system key from user key.
  79       ACE_OS::memcpy (&value,
  80                       &key[key.size () - counter_key_size],
  81                       sizeof value);
  82
  83       // Return the system key value as the hash value.
  84       return value;
  85     }
  86 };
  87
  88 class Key_Adapter
  89 {
  90 public:
  91   int encode (const KEY &original_key,
  92               const ACE_Active_Map_Manager_Key &active_key,
  93               KEY &modified_key)
  94     {
  95       // Keep original information in the key intact.
  96       modified_key = original_key;
  97       size_t original_size = modified_key.size ();
  98
  99       // Size of active key.
 100       size_t active_key_size = active_key.size ();
 101
 102       // Resize to accommodate both the original data and the new active key.
 103       modified_key.size (active_key_size + original_size);
 104
 105       // Copy active key data into user key.
 106       active_key.encode (&modified_key[original_size]);
 107
 108       // Success.
 109       return 0;
 110     }
 111
 112   int decode (const KEY &modified_key,
 113               ACE_Active_Map_Manager_Key &active_key)
 114     {
 115       // Read the active key data from the back of the key.
 116       size_t active_key_size = active_key.size ();
 117       size_t original_size = modified_key.size () - active_key_size;
 118
 119       // Read off value of index and generation.
 120       active_key.decode (&modified_key[original_size]);
 121
 122       // Success.
 123       return 0;
 124     }
 125
 126   int decode (const KEY &modified_key,
 127               KEY &original_key)
 128     {
 129       // Read the original user key data from the front of the
 130       // modified key.
 131       size_t active_key_size = ACE_Active_Map_Manager_Key::size ();
 132
 133       // Copy all the data.
 134       original_key = modified_key;
 135
 136       // Resize to ignore active key data.
 137       original_key.size (original_key.size () - active_key_size);
 138
 139       // Success.
 140       return 0;
 141     }
 142 };
 143
 144 #endif /* ACE_TESTS_MAP_TEST_H */