前言:在此前的一篇博客 通讯录动态版本 中我完成了一个能够动态增加容量的通讯录,当通讯录运行起来的时候,可以给通讯录中增加、删除数据,此时数据是存放在内存中,但是当程序退出的时候,我们会发现通讯录中的数据就不存在了,等下次运行通讯录程序的时候,数据又得重新录入,这样的通讯录使用起来不免会有很多麻烦,所以为了解决这个问题,本次将会带来通讯录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");
}
}