上一篇中，我們講了如何自己定義一個負責文件操作的功能集合，今天我們就利用這一組功能完成保存記錄的動作。

文件讀寫操作

在上一篇的基礎上，我們稍作修改加入到我們的通訊錄工程中來。代碼如下：

• MyFile.h

    #ifndef __MY_FILE_H__
    #define __MY_FILE_H__
    
    #include <stdio.h>
    
    #define FOR_READ        0
    #define FOR_WRITE       1
    
    FILE* fp;
    
    extern int OpenFile(char* pFileName, int nOperate);
    extern void CloseFile();
    
    extern char* ReadLine();
    extern int WriteLine(char* pLine);
    
    #endif
  

• MyFile.c

    #include "MyFile.h"
    
    #include "string.h"
    
    #define BUF_SIZE 100
    char pBuf[BUF_SIZE];
    
    int OpenFile(char* pFileName, int nOperate)
    {
        switch (nOperate)
        {
        case FOR_READ:
            fp = fopen(pFileName, "r");
            break;
        case FOR_WRITE:
            fp = fopen(pFileName, "w+");
            break;
        default:
            break;
        }
    
        if (fp == NULL)
        {
            return 0;
        }
        else
        {
            return 1;
        }
    }
    
    void CloseFile()
    {
        fflush(fp);
        fclose(fp);
    }
    
    char* ReadLine()
    {
        pBuf[0] = 0;
    
        if(feof(fp))
        {
            return NULL;
        }
        else
        {
            if (NULL == fgets(pBuf, BUF_SIZE, fp))
            {
                return NULL;
            }
            else
            {
                pBuf[strlen(pBuf) - 1] = 0;
            }
            
        }
    
        return pBuf;
    }
    
    int WriteLine(char* pLine)
    {
        fwrite(pLine, sizeof(char), strlen(pLine), fp);
        fwrite("\n", sizeof(char), 1, fp);
    
        return 1;
    }
  

注意，這段代碼和上一篇中的有所不同，主要是我們在ReadLine()函數的實現中采用了fgets()這個函數。這個函數的特點是讀取字符串時遇到換行符會自動停止，與我們的需求比較類似。

數據保存功能

當我們在程序中錄入了一些通訊錄條目之后，我們需要在退出前將這些數據保存在文件中。在main.c文件中，我們添加一個函數調用，函數名是SavedRecords()。

這個函數實現如下：

void SavedRecords()
{
    if (OpenFile("book.txt", FOR_WRITE) != 1)
    {
        return;
    }

    ListNode* pNode;
    for (pNode = g_pL; pNode->_pR != NULL; pNode = pNode->_pNext)
    {
        pNode->_pR;
        WriteLine(pNode->_pR->_pStrName->pBuf);
        WriteLine(pNode->_pR->_pStrTel->pBuf);
        WriteLine(pNode->_pR->_pStrPS->pBuf);
        WriteLine("----------");
    }

    CloseFile();
}

這個函數主要執行三個動作：

• 打開一個名為book.txt的文件，注意使用寫模式打開
• 遍歷鏈表，之后調用WriteLine()函數把每條記錄的每個條目寫入book.txt文件
• 關閉文件

在文件中保存數據時，我們用“-----”來分隔不同的記錄。就這樣，數據保存功能實現了。

數據恢復功能

保存了數據之后，在每次啟動程序時，需要重新把這些數據讀取出來保存進數據鏈表中。我們用一個函數LoadRecords()來實現。

在圖中程序啟動的位置，加入這個函數的調用。具體代碼如下：

void LoadRecords()
{
    Record* pR = NULL;

    if (OpenFile("book.txt", FOR_READ) != 1)
    {
        return;
    }

    char* pLine = NULL;
    while ((pLine = ReadLine()) != NULL)
    {
        if (pLine[0] == 0)
        {
            break;
        }
        
        RecordInit(&pR);

        RecordSetName(pR, pLine);

        pLine = ReadLine();
        RecordSetTel(pR, pLine);

        pLine = ReadLine();
        RecordSetPS(pR, pLine);
        
        pLine = ReadLine();

        ListAdd(pR);
    }

    CloseFile();
}

這段代碼也很簡單，通過文件操作接口把book.txt文件中的數據讀出，填進Record數據結構，之后添加到數據鏈表中。此時，執行查找方法打印出全部的數據，看看是不是之前的數據都被恢復了呢？

這樣，我們一個完整的項目就完成了。

完整的代碼請在下面這個路徑中獲取:https://github.com/breakerthb/TelBook.git

我是天花板，讓我們一起在軟件開發中自我迭代。
如有任何問題，歡迎與我聯系。

上一篇：21天C語言代碼訓練營（第十五天）
下一篇：21天C語言代碼訓練營（第十七天）