← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to mysys/cached_directory.cc

  • Committer: Brian Aker
  • Date: 2009-07-16 22:37:01 UTC
  • mto: This revision was merged to the branch mainline in revision 1100.
  • Revision ID: brian@gaz-20090716223701-vbbbo8dmgd2ljqqo
Refactor TableShare has to be behind class.

Show diffs side-by-side

added added

removed removed

Lines of Context:
mysys/cached_directory.cc
25
25
 *   Implementation of CachedDirectory class.
26
26
 */
27
27
 
28
 
#include "config.h"
29
 
 
30
 
#include <sys/types.h>
31
 
#include <sys/stat.h>
32
 
#include <unistd.h>
33
 
 
34
 
#include <strings.h>
35
 
#include <limits.h>
36
 
 
37
 
#include "drizzled/cached_directory.h"
 
28
#include "cached_directory.h"
38
29
 
39
30
using namespace std;
40
31
 
41
 
namespace drizzled
42
 
{
43
 
 
44
 
CachedDirectory::CachedDirectory() : 
45
 
  error(0)
46
 
{
47
 
}
48
 
 
49
 
 
50
 
CachedDirectory::CachedDirectory(const string &in_path) :
51
 
  error(0)
52
 
{
53
 
  // TODO: Toss future exception
 
32
 
 
33
CachedDirectory::CachedDirectory(const string &in_path)
 
34
  : error(0)
 
35
{
54
36
  (void) open(in_path);
55
37
}
56
38
 
57
 
 
58
 
CachedDirectory::CachedDirectory(const string& in_path, set<string>& allowed_exts) :
59
 
  error(0)
60
 
{
61
 
  // TODO: Toss future exception
62
 
  (void) open(in_path, allowed_exts);
63
 
}
64
 
 
65
 
CachedDirectory::CachedDirectory(const string& in_path, enum CachedDirectory::FILTER filter) :
66
 
  error(0)
67
 
{
68
 
  set<string> empty;
69
 
  // TODO: Toss future exception
70
 
  (void) open(in_path, empty, filter);
71
 
}
72
 
 
73
 
 
74
39
CachedDirectory::~CachedDirectory()
75
40
{
76
 
  for (Entries::iterator p= entries.begin(); p != entries.end(); ++p)
 
41
  Entries::iterator p= entries.begin();
 
42
  while (p != entries.end())
77
43
  {
78
44
    if (*p)
79
45
      delete *p;
 
46
    ++p;
80
47
  }
81
 
  entries.clear();
82
48
}
83
49
 
84
50
bool CachedDirectory::open(const string &in_path)
85
51
{
86
 
  set<string> empty;
87
 
 
88
 
  return open(in_path, empty);
89
 
}
90
 
 
91
 
bool CachedDirectory::open(const string &in_path, set<string> &allowed_exts)
92
 
{
93
 
  return open(in_path, allowed_exts, CachedDirectory::NONE);
94
 
}
95
 
 
96
 
bool CachedDirectory::open(const string &in_path, set<string> &allowed_exts, enum CachedDirectory::FILTER filter)
97
 
{
98
 
  DIR *dirp= opendir(in_path.c_str());
99
 
 
100
 
  if (dirp == NULL)
 
52
  size_t size;
 
53
  DIR *dirp;
 
54
 
 
55
  if ((dirp= opendir(in_path.c_str())) == NULL)
101
56
  {
102
57
    error= errno;
103
58
    return false;
104
59
  }
105
60
 
106
 
  path= in_path;
 
61
  /*
 
62
   * The readdir_r() call on Solaris operates a bit differently from other
 
63
   * systems in that the dirent structure must be allocated along with enough
 
64
   * space to contain the filename (see man page for readdir_r on Solaris).
 
65
   */
107
66
 
108
 
  union {
109
 
    dirent entry;
110
 
#ifdef __sun
111
 
    /*
112
 
     * The readdir_r() call on Solaris operates a bit differently from other
113
 
     * systems in that the dirent structure must be allocated along with enough
114
 
     * space to contain the filename (see man page for readdir_r on Solaris).
115
 
     * Instead of dynamically try to allocate this buffer, just set the max
116
 
     * name for a path instead.
117
 
     */
118
 
    char space[sizeof(dirent) + PATH_MAX + 1];
 
67
#ifdef SOLARIS
 
68
  size= sizeof(dirent) + pathconf(in_path.c_str(), _PC_NAME_MAX);
 
69
#else
 
70
  size= sizeof(dirent);
119
71
#endif
120
 
  } buffer;
 
72
 
 
73
  dirent *entry= (dirent *) malloc(size);
 
74
 
 
75
  if (entry == NULL)
 
76
  {
 
77
    error= errno;
 
78
    closedir(dirp);
 
79
    return false;
 
80
  }
121
81
 
122
82
  int retcode;
123
83
  dirent *result;
124
84
 
125
 
  while ((retcode= readdir_r(dirp, &buffer.entry, &result)) == 0 &&
126
 
         result != NULL)
 
85
  retcode= readdir_r(dirp, entry, &result);
 
86
  while ((retcode == 0) && (result != NULL))
127
87
  {
128
 
    if (! allowed_exts.empty())
129
 
    {
130
 
      char *ptr= rindex(result->d_name, '.');
131
 
 
132
 
      if (ptr)
133
 
      {
134
 
        set<string>::iterator it;
135
 
        it= allowed_exts.find(ptr);
136
 
 
137
 
        if (it != allowed_exts.end())
138
 
        {
139
 
          entries.push_back(new Entry(result->d_name));
140
 
        }
141
 
      }
142
 
    }
143
 
    else
144
 
    {
145
 
      switch (filter)
146
 
      {
147
 
      case DIRECTORY:
148
 
        {
149
 
          struct stat entrystat;
150
 
 
151
 
          if (result->d_name[0] == '.')
152
 
            continue;
153
 
 
154
 
          stat(result->d_name, &entrystat);
155
 
 
156
 
          if (S_ISDIR(entrystat.st_mode))
157
 
          {
158
 
            entries.push_back(new Entry(result->d_name));
159
 
          }
160
 
        }
161
 
        break;
162
 
      case FILE:
163
 
        {
164
 
          struct stat entrystat;
165
 
 
166
 
          stat(result->d_name, &entrystat);
167
 
 
168
 
          if (S_ISREG(entrystat.st_mode))
169
 
          {
170
 
            entries.push_back(new Entry(result->d_name));
171
 
          }
172
 
        }
173
 
        break;
174
 
      case NONE:
175
 
      case MAX:
176
 
        entries.push_back(new Entry(result->d_name));
177
 
        break;
178
 
      }
179
 
    }
 
88
    entries.push_back(new Entry(entry->d_name));
 
89
    retcode= readdir_r(dirp, entry, &result);
180
90
  }
181
91
    
182
92
  closedir(dirp);
 
93
  free(entry);
 
94
 
183
95
  error= retcode;
184
96
 
185
97
  return error == 0;
186
98
}
187
 
 
188
 
} /* namespace drizzled */