logrotate.cpp
上传用户:yhdzpy8989
上传日期:2007-06-13
资源大小:13604k
文件大小:5k
源码类别:

生物技术

开发平台:

C/C++

  1. /*
  2.  * ===========================================================================
  3.  * PRODUCTION $Log: logrotate.cpp,v $
  4.  * PRODUCTION Revision 1000.2  2004/06/01 19:40:14  gouriano
  5.  * PRODUCTION PRODUCTION: UPGRADED [GCC34_MSVC7] Dev-tree R1.8
  6.  * PRODUCTION
  7.  * ===========================================================================
  8.  */
  9. /*  $Id: logrotate.cpp,v 1000.2 2004/06/01 19:40:14 gouriano Exp $
  10. * ===========================================================================
  11. *
  12. *                            PUBLIC DOMAIN NOTICE
  13. *               National Center for Biotechnology Information
  14. *
  15. *  This software/database is a "United States Government Work" under the
  16. *  terms of the United States Copyright Act.  It was written as part of
  17. *  the author's official duties as a United States Government employee and
  18. *  thus cannot be copyrighted.  This software/database is freely available
  19. *  to the public for use. The National Library of Medicine and the U.S.
  20. *  Government have not placed any restriction on its use or reproduction.
  21. *
  22. *  Although all reasonable efforts have been taken to ensure the accuracy
  23. *  and reliability of the software and data, the NLM and the U.S.
  24. *  Government do not and cannot warrant the performance or results that
  25. *  may be obtained by using this software or data. The NLM and the U.S.
  26. *  Government disclaim all warranties, express or implied, including
  27. *  warranties of performance, merchantability or fitness for any particular
  28. *  purpose.
  29. *
  30. *  Please cite the author in any work or product based on this material.
  31. *
  32. * ===========================================================================
  33. *
  34. * Author:  Aaron Ucko, NCBI
  35. *
  36. * File Description:
  37. *   File streams supporting log rotation
  38. *
  39. * ===========================================================================
  40. */
  41. #include <ncbi_pch.hpp>
  42. #include <util/logrotate.hpp>
  43. #include <corelib/ncbifile.hpp>
  44. BEGIN_NCBI_SCOPE
  45. CRotatingLogStreamBuf::CRotatingLogStreamBuf(CRotatingLogStream* stream,
  46.                                              const string&       filename,
  47.                                              CT_OFF_TYPE         limit,
  48.                                              IOS_BASE::openmode  mode)
  49.     : m_Stream(stream),
  50.       m_FileName(filename),
  51.       m_Size(0),
  52.       m_Limit(limit),
  53.       m_Mode(mode)
  54. {
  55.     open(m_FileName.c_str(), m_Mode);
  56.     m_Size = seekoff(0, IOS_BASE::cur, IOS_BASE::out);
  57. }
  58. void CRotatingLogStreamBuf::Rotate(void)
  59. {
  60.     close();
  61.     string old_name = m_FileName; // Copy in case x_BackupName mutates.
  62.     string new_name = m_Stream->x_BackupName(m_FileName);
  63.     if ( !new_name.empty() ) {
  64.         CFile(new_name).Remove();
  65.         CFile(old_name).Rename(new_name);
  66.     }
  67.     open(m_FileName.c_str(), m_Mode);
  68.     m_Size = seekoff(0, IOS_BASE::cur, IOS_BASE::out);
  69. }
  70. // The use of new_size in overflow and sync is to avoid
  71. // double-counting when one calls the other.  (Which, if either, is
  72. // actually lower-level seems to vary with compiler.)
  73. CT_INT_TYPE CRotatingLogStreamBuf::overflow(CT_INT_TYPE c)
  74. {
  75.     // The only operators CT_POS_TYPE reliably seems to support
  76.     // are += and -=, so stick to those. :-/
  77.     CT_POS_TYPE new_size = m_Size;
  78.     new_size += pptr() - pbase();
  79.     if ( !CT_EQ_INT_TYPE(c, CT_EOF) ) {
  80.         new_size += 1;
  81.     }
  82.     // Perform output first, in case switching files discards data.
  83.     CT_INT_TYPE result = CNcbiFilebuf::overflow(c);
  84.     // Don't assume the buffer's actually empty; some implementations
  85.     // seem to handle the case of pptr() being null by setting the
  86.     // pointers and writing c to the buffer but not actually flushing
  87.     // it to disk. :-/
  88.     new_size -= pptr() - pbase();
  89.     m_Size = new_size;
  90.     if (m_Size - CT_POS_TYPE(0) >= m_Limit) {
  91.         Rotate();
  92.     }
  93.     return result;
  94. }
  95. int CRotatingLogStreamBuf::sync(void)
  96. {
  97.     // Perform output first, in case switching files discards data.
  98.     CT_POS_TYPE new_size = m_Size;
  99.     new_size += pptr() - pbase();
  100.     int result = CNcbiFilebuf::sync();
  101.     // pptr() ought to equal pbase() now, but just in case...
  102.     new_size -= pptr() - pbase();
  103.     m_Size = new_size;
  104.     if (m_Size - CT_POS_TYPE(0) >= m_Limit) {
  105.         Rotate();
  106.     }
  107.     return result;
  108. }
  109. string CRotatingLogStream::x_BackupName(string& name)
  110. {
  111. #if defined(NCBI_OS_UNIX) && !defined(NCBI_OS_CYGWIN)
  112.     return name + CurrentTime().AsString(".Y-M-D-Z-h:m:s");
  113. #else
  114.     // Colons are special; avoid them.
  115.     return name + CurrentTime().AsString(".Y-M-D-Z-h-m-s");
  116. #endif
  117. }
  118. END_NCBI_SCOPE
  119. /*
  120. * ===========================================================================
  121. *
  122. * $Log: logrotate.cpp,v $
  123. * Revision 1000.2  2004/06/01 19:40:14  gouriano
  124. * PRODUCTION: UPGRADED [GCC34_MSVC7] Dev-tree R1.8
  125. *
  126. * Revision 1.8  2004/05/17 21:06:02  gorelenk
  127. * Added include of PCH ncbi_pch.hpp
  128. *
  129. * Revision 1.7  2004/02/19 22:57:56  ucko
  130. * Accommodate stricter implementations of CT_POS_TYPE.
  131. *
  132. * Revision 1.6  2003/03/27 23:17:41  vakatov
  133. * Ident
  134. *
  135. * Revision 1.5  2003/03/26 18:45:16  kans
  136. * add initialize default for m_Size (RS)
  137. *
  138. * Revision 1.4  2003/02/13 01:00:04  ucko
  139. * Limit CT_POS_TYPE operations to += and -=.  (Sigh.)
  140. *
  141. * Revision 1.3  2003/02/12 19:59:11  ucko
  142. * Parenthesize pointer subtractions.
  143. *
  144. * Revision 1.2  2003/02/12 16:41:08  ucko
  145. * Always supply CRotatingLogStreamBuf::sync, and avoid double-counting
  146. * via run-time rather than compile-time logic.
  147. *
  148. * Revision 1.1  2002/11/25 17:21:00  ucko
  149. * Add support for automatic log rotation (CRotatingLogStream)
  150. *
  151. * ===========================================================================
  152. */