TnbDBaseFile.h

[詳解]

#pragma once
#include "TnbFile.h"
#include "TnbGrid.h"
 
 
 
//TNB Library
namespace TNB
{
 
 
 
#pragma pack(push)
#pragma pack(1)
 
 
 
class CDBaseFile : public CGrid
{
    DEFSUPER(CGrid);
public:
 
    bool LoadFile(LPCTSTR lpszFileName)
    {
        _super::AllDeleteRecord();
        CFileReader fr;
        if ( fr.Open(lpszFileName) )
        {
            try
            {
                m_Load(fr);
            }
            catch ( CTnbException& e )
            {
                e.OnCatch();
                _super::AllDeleteRecord();
                return false;
            }
            catch ( ... )
            {
                throw;
            }
            return true;
        }
        return false;
    }
 
    bool SaveFile(LPCTSTR lpszFileName) const
    {
        CFileWriter fw;
        if ( fw.New(lpszFileName) )
        {
            try
            {
                m_Save(fw);
            }
            catch ( CTnbException& e )
            {
                e.OnCatch();
                return false;
            }
            catch ( ... )
            {
                throw;
            }
            return true;
        }
        return false;
    }
 
private:
    struct THead
    {
        BYTE    version         :3;     
        BYTE    withMemoFile4   :1;     
        BYTE    withSqlTable    :3;     
        BYTE    withMemoFile    :1;     
        BYTE    year;                   
        BYTE    month;                  
        BYTE    day;                    
        DWORD   recordCount;            
        WORD    headBytes;              
        WORD    recordBytes;            
        BYTE    padding1[2];            
        BYTE    transaction;            
        BYTE    encryptionFlag;         
        BYTE    padding2[12];           
        BYTE    withMdxFile;            
        BYTE    language;               
        BYTE    padding3[2];            
        THead(void)
        {
            Zero(*this);
            version = 3;
        }
        void SetTime(const SYSTEMTIME& st)
        {
            year = static_cast<BYTE>(st.wYear - 1900);
            month = static_cast<BYTE>(st.wMonth);
            day = static_cast<BYTE>(st.wDay);
        }
    };
    C_ASSERT ( sizeof(THead) == 32 ); 
 
    struct TField
    {
        char    name[11];               
        char    type;                   
        BYTE    padding1[4];            
        BYTE    fieldLength;            
        BYTE    decimalLength;          
        BYTE    padding2[2];            
        BYTE    wordId;                 
        BYTE    padding3[10];           
        BYTE    withMdxField;           
        TField(void)
        {
            Zero(*this);
        }
        void SetName(LPCTSTR lpsz)
        {
            Zero(name);
            CAscii s = lpsz;
            if ( s.GetLength() > 10 )
            {
                s = s.Left(10);
            }
            STRLIB::Copy(name, s);
        }
    };
    C_ASSERT ( sizeof(TField) == 32 ); 
 
    void m_Load(const IReader& r)
    {
        //=== ヘッダ読込み
        THead hd;
        r.ReadExactly(sizeof(THead), &hd);
        if ( hd.version != 3 )
        {
            throw CNotSupportException();
        }
        //=== フィールド読込み
        size_t fieldCount = (hd.headBytes - sizeof(THead)) / sizeof(TField); //フィールド数
        size_t recordBytesTemp = 0;
        loop ( i, fieldCount )
        {
            TField fd;
            r.ReadExactly(sizeof(TField), &fd);
            TFieldInfo fi(CStr(fd.name), fd.fieldLength, fd.decimalLength, static_cast<EFieldType>(fd.type));
            if ( _super::AddField(fi) == INVALID_INDEX )
            {
                throw CAppendFailureException();
            }
            recordBytesTemp += fd.fieldLength;
        }
        BYTE b;
        r.ReadExactly(sizeof(BYTE), &b);
        if ( b != 0x0D )
        {
            throw CInvalidParamException();
        }
        recordBytesTemp++;
        if ( recordBytesTemp != hd.recordBytes )
        {
            TRACE2("ヘッダのレコードサイズ(%d)と各フィールドに書かれているサイズの合計(%d)が異なります。\n", hd.recordBytes, recordBytesTemp);
            throw CInvalidParamException();
        }
        //=== レコード読込み
        loop ( i, hd.recordCount )
        {
            CWorkMemT<char> buf(hd.recordBytes);    //レコードのバイト数
            char* B = buf.Ref();
            r.ReadExactly(hd.recordBytes, B);
            if ( B[0] != 0x20 )
            {
                continue; //sp以外はデリートレコード
            }
            B++;
            INDEX rec = _super::AddRecord();
            ASSERT( rec != INVALID_INDEX );
            loop ( i, fieldCount )
            {
                TFieldInfo fi;
                EFieldKind k = _super::GetFieldInfo(fi, i);
                ASSERT( k != EK_Invalid );
                CAscii a;
                a.SetFromLeft(B, fi.width);
                if ( ! _super::SetStringRecord(rec, i, CStr(a)) )
                {
                    throw CInvalidParamException();
                }
                B += fi.width;
            }
        }
        //=== 終端
        r.ReadExactly(sizeof(BYTE), &b);
        if ( b != 0x1A )
        {
            throw CInvalidParamException();
        }
        _super::Dump();
    }
 
    void m_Save(IWriter& _w) const
    {
        //=== ヘッダ書き出し
        THead hd;
        {
            SYSTEMTIME st;
            ::GetLocalTime(&st);
            hd.SetTime(st);
            size_t rb = 0;
            loop ( i, _super::GetFieldCount() )
            {
                TFieldInfo fi;
                _super::GetFieldInfo(fi, i);
                rb += fi.width;
            }
            hd.recordCount = ToDword(_super::GetRecordCount());
            hd.recordBytes = ToWord(rb + 1);
            hd.headBytes = ToWord(_super::GetFieldCount() * sizeof(TField) + sizeof(THead) + 1);
        }
        _w.Write(sizeof(THead), &hd);
        //=== フィールド書き出し
        CWorkMemT<size_t> widths(_super::GetFieldCount());
        loop ( i, _super::GetFieldCount() )
        {
            TFieldInfo fi;
            if ( _super::GetFieldInfo(fi, i) == EK_Invalid )
            {
                throw CInvalidParamException();
            }
            widths[i] = fi.width;
            TField fd;
            fd.SetName(fi.name);
            fd.fieldLength = static_cast<BYTE>(fi.width);
            fd.decimalLength = static_cast<BYTE>(fi.decimals);
            fd.type = static_cast<char>(fi.type);
            _w.Write(sizeof(TField), &fd);
        }
        BYTE b = 0x0D;
        _w.Write(sizeof(BYTE), &b);
        //=== レコード書き出し
        loop ( i, GetRecordCount() )
        {
            CWorkMemT<char> buf(hd.recordBytes);    //レコードのバイト数
            char* B = buf.Ref();
            *B++ = 0x20;
            INDEX rec = i;
            size_t len = 1;
            loop ( j, _super::GetFieldCount() )
            {
                size_t l = widths[j];
                CAscii a = _super::GetStringRecord(rec, j, true) + CStr::Lineup(' ', l);
                a = a.Left(l);
                TNB::MemCopy(B, a, l);
                B += l;
                len += l;
            }
            ASSERT ( len == hd.recordBytes );
            _w.Write(hd.recordBytes, buf.Ref());
        }
        //=== 終端
        b = 0x1A;
        _w.Write(sizeof(BYTE), &b);
    }
 
    friend class CDBaseFileTest;
};
 
 
 
#pragma pack(pop)
 
 
 
}; // TNB