前言：在此前的一篇博客通訊錄動態版本中我完成了一個能夠動態增加容量的通訊錄，當通訊錄運行起來的時候，可以給通訊錄中增加、洗掉資料，此時資料是存放在記憶體中，但是當程式退出的時候，我們會發現通訊錄中的資料就不存在了，等下次運行通訊錄程式的時候，資料又得重新錄入，這樣的通訊錄使用起來不免會有很多麻煩，所以為了解決這個問題，本次將會帶來通訊錄2.0版，即可以用檔案保存的版本，接下來我們一起來看看吧！

在一次性就能學會的檔案操作！介紹了許多對檔案的操作，本次就會用到這些操作來優化我們的通訊錄，

文章目錄

保存通訊錄資料到檔案
加載資料到通訊錄
代碼總覽

保存通訊錄資料到檔案

為了保證通訊錄在第二次打開的時候仍有第一次輸入的資料，我們需要在第一次關閉通訊錄前就將其中的資料保存到檔案中，所以我們定義了一個函式 SaveContact 來保存資料到檔案，定義為：

void SaveContact(Contact* pc)
{
	FILE* pf = fopen("Contact.dat", "w");//創建并以寫入的方式打開檔案
	if (pf == NULL)
	{
		perror("Contact.dat");
		return;
	}
	int i = 0;
	for (i = 0; i < pc->sz; i++)
	{
		fwrite(pc->data + i, sizeof(PeoInfo), 1, pf);//利用回圈將資料保存到檔案中
	}
	fclose(pf);
	pf = NULL;
}

加載資料到通訊錄

當我們保存了上一次通訊錄的資料，那么在我們下一次使用通訊錄時，我們應該先將上一次通訊錄保存的資料加載到通訊錄才行，所以我們需要對初始化通訊錄的函式進行相應的調整，即在初始化時首加載資料，定義的函式為 LoadContact ，我們還需要注意一點的是在加載資料的程序中如果加載的資料容量大于原本的容量，就會加載失敗，為了避免這種情況，在加載前需先判斷是否需要擴容，所以定義函式 CheckCapacity 用來判斷，
兩函式分別如下：

void CheckCapacity(Contact* pc)
{
	if (pc->sz == pc->capacity)
	{
		PeoInfo* ptr = (PeoInfo*)realloc(pc->data, (ADD_CAP + pc->capacity) * sizeof(PeoInfo));
		if (ptr != NULL)
		{
			pc->data = ptr;
			pc->capacity += ADD_CAP;
			printf("增容成功!\n");
		}
		else
		{
			printf("增容失敗\n");
			return;
		}
	}
}

void LoadContact(Contact * pc)
{
	FILE* pf = fopen("Contact.dat", "r");//以讀的方式打開檔案
	if (pf == NULL)
	{
		perror("Contact.dat");
		return;
	}
	PeoInfo tmp = { 0 };
	while (fread(&tmp, sizeof(PeoInfo), 1, pf))//每次加載一個資料到通訊錄，當通訊錄中的資料被加
	                                            //載玩后，fread函式就會回傳0.此時回圈就自然結束
	{
		CheckCapacity(pc);//加載前首先判斷是否需要擴容
		pc->data[pc->sz] = tmp;
		pc->sz++;
	}
	fclose(pf);
	pf = NULL;
}

代碼總覽

同樣的分為三個檔案，分別如下：
contact.h

#include<stdio.h>
#include<string.h>
#include<stdlib.h>

#define MAX_NAME 20
#define SEX 10
#define MAX_ADDR 100
#define MAX_TELE 12
#define MAX_SZ 3
#define ADD_CAP 2
#define MAX 1000

typedef struct PeoInfo
{
	char name[MAX_NAME];
	char sex[SEX];
	int  age;
	char tele[MAX_TELE];
	char addr[MAX_ADDR];	
} PeoInfo;  //重定義為了使用方便，下同

typedef struct Contact
{
	PeoInfo* data;
	int sz;
	int capacity;
}Contact;  //重定義

void InitContact(Contact* pc);  //初始化通訊錄

void AddContact(Contact* pc);  //增加聯系人

void PrintContact(const Contact* pc);  //列印通訊錄

void DelContact(Contact* pc);  //洗掉聯系人

void SearchContact(Contact* pc);  //查找聯系人

void ModifyContact(Contact* pc);  //修改聯系人

void DestoryContact(Contact* pc);  //銷毀通訊錄

void SaveContact(Contact* pc);  //保存通訊錄

void LoadContact(Contact* pc);  //加載通訊錄

void CheckCapacity(Contact* pc);   //判斷通訊錄

test.c

#include"contact.h"  //參考頭檔案
void menu()
{
	printf("***************************\n");
	printf("***   1.Add     2.Del   ***\n");
	printf("***   3.Search  4.Modify***\n");
	printf("***        5.Print      ***\n");
	printf("***        0.Exit       ***\n");
	printf("***************************\n");
}

enum Option //將每個功能的名稱定義為一個列舉型別的成員，使辨識度更高
{
	EXIT,
	ADD,
	DEL,
	SEARCH,
	MODIFY,
	PRINT,
};
 
int main()
{
	Contact con;
	InitContact(&con);
	int input = 0;
	do
	{
		menu();
		scanf("%d", &input);
		switch (input)
		{
		    case ADD:
				AddContact(&con);//添加聯系人
				break;
			case DEL:
				DelContact(&con);//洗掉聯系人
				break;
			case SEARCH:
				SearchContact(&con);//查找聯系人
				break;
			case MODIFY:
				ModifyContact(&con);//修改聯系人
				break;
			case PRINT:
				PrintContact(&con);//顯示通訊錄
				break;
			case EXIT:
			    SaveContact(&con);//銷毀之前首先保存通訊錄里的資料
			    
				DestoryContact(&con);//銷毀通訊錄
				printf("退出通訊錄\n");
				break;
			default:
				printf("選擇錯誤，請重新選擇:");
				break;
		}
	} while (input);
	return 0;
}

contact.c

#include"contact.h"
void CheckCapacity(Contact* pc)
{
	if (pc->sz == pc->capacity)
	{
		PeoInfo* ptr = (PeoInfo*)realloc(pc->data, (ADD_CAP + pc->capacity) * sizeof(PeoInfo));
		if (ptr != NULL)
		{
			pc->data = ptr;
			pc->capacity += ADD_CAP;
			printf("增容成功!\n");
		}
		else
		{
			printf("增容失敗\n");
			return;
		}
	}
}

void LoadContact(Contact * pc)
{
	FILE* pf = fopen("Contact.dat", "r");
	if (pf == NULL)
	{
		perror("Contact.dat");
		return;
	}
	PeoInfo tmp = { 0 };
	while (fread(&tmp, sizeof(PeoInfo), 1, pf))
	{
		CheckCapacity(pc);
		pc->data[pc->sz] = tmp;
		pc->sz++;
	}
	fclose(pf);
	pf = NULL;
}

void InitContact(Contact* pc)
{
	pc->data = (PeoInfo*)malloc(MAX_SZ * sizeof(PeoInfo));
	if (pc->data == NULL)
	{
		perror("InitContact");
		return;
	}
	pc->sz = 0;
	pc->capacity = MAX_SZ;
	LoadContact(pc);
}

void DestoryContact(Contact* pc)
{
	free(pc->data);
	pc->data = NULL;
	pc->sz = 0;
	pc->capacity = 0;
}


//保存通訊錄
void SaveContact(Contact* pc)
{
	FILE* pf = fopen("Contact.dat", "w");
	if (pf == NULL)
	{
		perror("Contact.dat");
		return;
	}
	int i = 0;
	for (i = 0; i < pc->sz; i++)
	{
		fwrite(pc->data + i, sizeof(PeoInfo), 1, pf);
	}
	fclose(pf);
	pf = NULL;
}

void AddContact(Contact* pc)
{
	CheckCapacity(pc);
	
	printf("請輸入姓名:>");
	scanf("%s", pc->data[pc->sz].name);
	printf("請輸入性別:>");
	scanf("%s", pc->data[pc->sz].sex);
	printf("請輸入年齡:>");
	scanf("%d", &(pc->data[pc->sz].age));
	printf("請輸入電話:>");
	scanf("%s", pc->data[pc->sz].tele);
	printf("請輸入地址:>");
	scanf("%s", pc->data[pc->sz].addr);
	pc->sz++;
	printf("增加成功!\n");
}

void PrintContact(const Contact* pc)
{
	printf("%-20s\t%-5s\t%-5s\t%-12s\t%-20s\t\n", "姓名", "性別", "年齡", "電話", "地址");
	int i = 0;
	for (i = 0; i < pc->sz; i++)
	{
		printf("%-20s\t%-5s\t%-5d\t%-12s\t%-20s\t\n",
			pc->data[i].name,
			pc->data[i].sex,
			pc->data[i].age,
			pc->data[i].tele,
			pc->data[i].addr);
	}
}

static int FindContact(const Contact* pc, char name[])
{
	int i = 0;
	for (i = 0; i < pc->sz; i++)
	{
		if (strcmp(name, (pc->data[i]).name) == 0)
			return i;
	}
	return -1;
}

void DelContact(Contact* pc)
{
	char name[MAX_NAME] = { 0 };
	printf("請輸入要洗掉的聯系人:>");
	scanf("%s", name);
	int x = FindContact(pc, name);
	if (x < 0)
	{
		printf("沒找到聯系人\n");
	}
	else
	{
		int i = 0;
		for (i = x; x < pc->sz - 1; x++)
		{
			pc->data[x] = pc->data[x + 1];
		}
		pc->sz--;
		printf("洗掉成功！\n");
	}
}

void SearchContact(Contact* pc)
{
	char name[MAX_NAME] = { 0 };
	printf("請輸入要查找的聯系人:>");
	scanf("%s", name);
	int x = FindContact(pc, name);
	if (x < 0)
	{
		printf("沒找到聯系人\n");
	}
	else
	{
		printf("%-20s\t%-5s\t%-5s\t%-12s\t%-20s\t\n", "姓名", "性別", "年齡", "電話", "地址");
		printf("%-20s\t%-5s\t%-5d\t%-12s\t%-20s\t\n",
			pc->data[x].name,
			pc->data[x].sex,
			pc->data[x].age,
			pc->data[x].tele,
			pc->data[x].addr);
	}
}

void ModifyContact(Contact* pc)
{
	char name[MAX_NAME] = { 0 };
	printf("請輸入要修改的聯系人:>");
	scanf("%s", name);
	int x = FindContact(pc, name);
	if (x < 0)
	{
		printf("沒找到聯系人\n");
	}
	else
	{
		printf("請輸入姓名:>");
		scanf("%s", pc->data[x].name);
		printf("請輸入性別:>");
		scanf("%s", pc->data[x].sex);
		printf("請輸入年齡:>");
		scanf("%d", &(pc->data[x].age));
		printf("請輸入電話:>");
		scanf("%s", pc->data[x].tele);
		printf("請輸入地址:>");
		scanf("%s", pc->data[x].addr);
		printf("修改成功!\n");
	}
}

轉載請註明出處，本文鏈接：https://www.uj5u.com/ruanti/301565.html

標籤：其他

上一篇：字串庫函式及其模擬實作【中篇】--- strncpy +strncat + strncmp

下一篇：模型部署的開源機器學習工具 MLflow | 每日工具推薦

通訊錄2.0版本（檔案保存版）

文章目錄

保存通訊錄資料到檔案

加載資料到通訊錄

代碼總覽