前言
在之前我们已经写过两个版本的通讯录了:
第一个是静态版本的通讯录,该通讯录一运行起来就开辟了一大块内存空间,用于存储1000个联系人的信息,并且运行结束数据也就销毁了。该通讯录比较死板,存储的联系人信息最多就是1000个,你若是没有那么多联系人信息需要存储,那么空间就白白浪费掉了,你若是还有更多的联系人信息需要存储,该通讯录也不能再增加容量。
第二个是可以根据用户需要存储的联系人个数,按需开辟内存空间以供用户使用的动态版本的通讯录,该通讯录解决了第一个通讯录浪费内存空间的问题,但是仍然没有解决保存数据的问题。
任何一个程序都应该拥有保存数据的功能,不论是办公软件,还是游戏,不然这就是一个“一次性”软件,用了一次之后下一次就要“从头来过”,谁会使用这样的软件呢?
这次,我们就通过文件操作方面的知识,对动态版本的通讯录进行进一步改造,让它变成一个可以保存数据的通讯录。
保存通讯录内数据到文件
要实现这个功能,你首先得知道如何运用文件操作函数。进入函数体,先打开一个文件如果打开成功,就向该文件传输数据,最后关闭文件即可。
//保存通讯录内信息到文件中
void SaveContact(struct Contact* ps)
{
//打开文件
FILE* pf = fopen("contact.dat", "wb");
if (pf == NULL)
{
printf("保存失败\n");
return;
}
//向文件传输数据
int i = 0;
for (i = 0; i < ps->size; i++)
{
fwrite(ps->data + i, sizeof(struct PeoInfo), 1, pf);
}
//关闭文件
fclose(pf);
pf = NULL;
printf("保存成功\n");
}
这个功能的实现中,我们除了用到了打开文件的函数fopen和关闭文件的函数fclose之外,我们用了一个叫做fwrite的函数,该函数可以将数据以二进制的形式输入到文件中。
这样做有一个好处就是当别人打开这个文件时,并不知道里面存放的是什么信息,因为以二进制形式存储的数据是我们无法读懂的:
所以只有在通讯录中才能看到相关联系人的信息内容。
加载数据到通讯录
当我们保存了上一次通讯录的数据,那么在我们下一次使用通讯录时,我们应该先将上一次通讯录保存的数据加载到通讯录才行,所以我们需要对初始化通讯录的函数进行相应的调整。
//初始化通讯录
void InitContact(struct Contact* ps)
{
ps->data = (struct PeoInfo*)malloc(INIT_CAP * sizeof(struct PeoInfo));
if (ps->data == NULL)//开辟失败
return;
ps->size = 0;//一开始通讯录为空
ps->capacity = INIT_CAP;//一开始通讯录的只能存放3个联系人的信息,若不够再扩容
//加载文件
LodeContact(ps);
}
我们可以在之前初始化通讯录的后面加上一个函数,该函数的功能就是将上一次通讯录保存的数据从文件中加载到通讯录中来。
//检测通讯录状态,若已满,则扩容,若增容时开辟空间失败,则返回1
static int CheckCapacity(struct Contact* ps)
{
if (ps->size == ps->capacity)//通讯录人数已达上限
{
//增容
struct PeoInfo* ptr = (struct PeoInfo*)realloc(ps->data, (ps->capacity + 2)*sizeof(struct PeoInfo));
if (ptr != NULL)
{
ps->data = ptr;
ps->capacity += 2;
return 0;//增容成功
}
else
{
return 1;//增容失败
}
}
return 0;
}
//加载文件
void LodeContact(struct Contact* ps)
{
//打开文件
FILE* pf = fopen("contact.dat", "rb");
if (pf == NULL)
{
printf("加载文件失败\n");
return ;
}
//加载文件内容到通讯录
struct PeoInfo tmp = { 0 };
while ( fread(&tmp, sizeof(struct PeoInfo), 1, pf))
{
//检测是否需要增容
int flag = CheckCapacity(ps);
//检测通讯录状态,若已满,则扩容,若增容时开辟空间失败,则返回1
if (flag == 1)
{
printf("加载文件失败\n");
return;
}
//存储元素
ps->data[ps->size] = tmp;
ps->size++;
}
//关闭文件
fclose(pf);
pf = NULL;
}
因为我们保存数据到文件的时候是以二进制的形式保存进去的,所以当我们要将数据拿出来的时候也要以二进制的形式才能将数据正确输出。于是我们就用到了fread函数,该函数与fwrite函数是成对使用的,它的功能就是将数据以二进制的形式从文件中输出。
这样看来,要实现保存数据的功能其实并没有想象中那么困难,我们只需要在原来的基础上增加两段代码即可,保存数据到文件的代码和加载文件数据到程序的代码。
代码总览
contact.h
#define _CRT_SECURE_NO_WARNINGS 1
#define INIT_CAP 3
//通讯录初始容量
#define MAX_NAME 15
#define MAX_SEX 5
#define MAX_TELE 12
#define MAX_ADDRESS 20
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
enum Option
{
EXIT,//0
ADD,//1
DELETE,//2
SREACH,//3
MODIFY,//4
SHOW,//5
SORT,//6
SAVE//7
};//增加代码可读性
struct PeoInfo
{
char name[MAX_NAME];
int age;
char sex[MAX_SEX];
int tele[MAX_TELE];
char address[MAX_ADDRESS];
};//每一个联系人的信息内容
//通讯录类型
struct Contact
{
struct PeoInfo* data;//指向一块动态开辟的空间
int size;//记录当前已有联系人的个数
int capacity;//当前通讯录的最大容量
};
//初始化通讯录
void InitContact(struct Contact* ps);
//添加一个联系人的信息
void AddContact(struct Contact* ps);
//删除一个联系人的信息
void DeleteContact(struct Contact* ps);
//查找指定联系人并打印其信息
void SreachContact(const struct Contact* ps);
//修改指定联系人的信息
void ModifyContact(struct Contact* ps);
//打印通讯录中的全部信息
void ShowContact(const struct Contact* ps);
//通过名字排序通讯录中联系人的先后顺序
void SortContact(struct Contact* ps);
//销毁通讯录数据
void DestroyContact(struct Contact* ps);
//保存通讯录内信息到文件中
void SaveContact(struct Contact* ps);
contact.c
#include "contact.h"
//检测通讯录状态,若已满,则扩容,若增容时开辟空间失败,则返回1
static int CheckCapacity(struct Contact* ps)
{
if (ps->size == ps->capacity)//通讯录人数已达上限
{
//增容
struct PeoInfo* ptr = (struct PeoInfo*)realloc(ps->data, (ps->capacity + 2)*sizeof(struct PeoInfo));
if (ptr != NULL)
{
ps->data = ptr;
ps->capacity += 2;
return 0;//增容成功
}
else
{
return 1;//增容失败
}
}
return 0;
}
//加载文件
void LodeContact(struct Contact* ps)
{
//打开文件
FILE* pf = fopen("contact.dat", "rb");
if (pf == NULL)
{
printf("加载文件失败\n");
return ;
}
//加载文件内容到通讯录
struct PeoInfo tmp = { 0 };
while ( fread(&tmp, sizeof(struct PeoInfo), 1, pf))
{
//检测是否需要增容
int flag = CheckCapacity(ps);
//检测通讯录状态,若已满,则扩容,若增容时开辟空间失败,则返回1
if (flag == 1)
{
printf("加载文件失败\n");
return;
}
//存储元素
ps->data[ps->size] = tmp;
ps->size++;
}
//关闭文件
fclose(pf);
pf = NULL;
}
//初始化通讯录
void InitContact(struct Contact* ps)
{
ps->data = (struct PeoInfo*)malloc(INIT_CAP * sizeof(struct PeoInfo));
if (ps->data == NULL)//开辟失败
return;
ps->size = 0;//一开始通讯录为空
ps->capacity = INIT_CAP;//一开始通讯录的只能存放3个联系人的信息,若不够再扩容
//加载文件
LodeContact(ps);
}
//添加一个联系人的信息
void AddContact(struct Contact* ps)
{
int flag = CheckCapacity(ps);
if (flag == 1)
{
printf("内存空间不足,增容失败\n");
return;
}
//检测通讯录状态,若已满,则扩容,若增容时开辟空间失败,则返回1
//增加数据
printf("请输入姓名:>");
scanf("%s", ps->data[ps->size].name);
printf("请输入年龄:>");
scanf("%d", &(ps->data[ps->size].age));
printf("请输入性别:>");
scanf("%s", ps->data[ps->size].sex);
printf("请输入电话:>");
scanf("%s", ps->data[ps->size].tele);
printf("请输入地址:>");
scanf("%s", ps->data[ps->size].address);
ps->size++;//通讯录中联系人个数+1
printf("添加成功\n");
}
//通过名字查找联系人,找到了返回下标,没找到返回-1
static int FindByName(char name[MAX_NAME],const struct Contact* ps)
{
int i = 0;
for (i = 0; i < ps->size; i++)
{
if (strcmp(ps->data[i].name, name) == 0)
return i;//找到了返回下标
}
return -1;//找不到的情况
}
//删除一个联系人的信息
void DeleteContact(struct Contact* ps)
{
char name[MAX_NAME];
printf("请输入要删除联系人的姓名:>");
scanf("%s", name);
//查找
int pos = FindByName(name, ps);//找到了返回下标,没找到返回-1
//删除
if (pos == -1)
{
printf("查无此人\n");
}
else
{
int j = 0;
for (j = pos; j < ps->size - 1; j++)
{
ps->data[j] = ps->data[j + 1];
}//从要删除的联系人位置开始,后一位联系人信息覆盖前一个联系人信息
printf("删除成功\n");
ps->size--;//通讯录中联系人个数-1
}
}
//查找指定联系人并打印其信息
void SreachContact(const struct Contact* ps)
{
char name[MAX_NAME];
printf("请输入要查找联系人的姓名:>");
scanf("%s", name);
int pos = FindByName(name, ps);//找到了返回下标,没找到返回-1
if (pos == -1)
{
printf("查无此人\n");
}
else
{
printf("%-15s\t%-4s\t%-5s\t%-12s\t%-20s\n", "姓名", "年龄", "性别", "电话", "住址");
printf("%-15s\t%-4d\t%-5s\t%-12s\t%-20s\n",
ps->data[pos].name,
ps->data[pos].age,
ps->data[pos].sex,
ps->data[pos].tele,
ps->data[pos].address);
}//打印该联系人的信息内容
}
//修改指定联系人的信息
void ModifyContact(struct Contact* ps)
{
char name[MAX_NAME];
printf("请输入要修改的联系人的姓名:>");
scanf("%s", name);
int pos = FindByName(name, ps);//找到了返回下标,没找到返回-1
if (pos == -1)
{
printf("查无此人\n");
}
else
{
printf("请输入姓名:>");
scanf("%s", ps->data[pos].name);
printf("请输入年龄:>");
scanf("%d", &(ps->data[pos].age));
printf("请输入性别:>");
scanf("%s", ps->data[pos].sex);
printf("请输入电话:>");
scanf("%s", ps->data[pos].tele);
printf("请输入地址:>");
scanf("%s", ps->data[pos].address);
printf("修改成功\n");
}//修改联系人信息,即将该联系人信息重新录入
}
//打印通讯录中的全部信息
void ShowContact(const struct Contact* ps)
{
if (ps->size == 0)
{
printf("通讯录为空\n");
}//通讯录中联系人个数为0
else
{
printf("%-15s\t%-4s\t%-5s\t%-12s\t%-20s\n", "姓名", "年龄", "性别", "电话", "住址");
//打印信息栏
int i = 0;
for (i = 0; i < ps->size; i++)
{
printf("%-15s\t%-4d\t%-5s\t%-12s\t%-20s\n",
ps->data[i].name,
ps->data[i].age,
ps->data[i].sex,
ps->data[i].tele,
ps->data[i].address);
}//打印通讯录全部信息内容
}
}
//自定义的比较函数
int CmpByName(const void* e1, const void* e2)
{
return strcmp((const char*)e1, (const char*)e2);
}
//通过名字排序通讯录中联系人的先后顺序
void SortContact(struct Contact* ps)
{
qsort(ps->data, ps->size, sizeof(struct PeoInfo), CmpByName);//排序
}
//销毁通讯录数据
void DestroyContact(struct Contact* ps)
{
//释放动态开辟空间
free(ps->data);
ps->data = NULL;
}
//保存通讯录内信息到文件中
void SaveContact(struct Contact* ps)
{
//打开文件
FILE* pf = fopen("contact.dat", "wb");
if (pf == NULL)
{
printf("保存失败\n");
return;
}
//向文件传输数据
int i = 0;
for (i = 0; i < ps->size; i++)
{
fwrite(ps->data + i, sizeof(struct PeoInfo), 1, pf);
}
//关闭文件
fclose(pf);
pf = NULL;
printf("保存成功\n");
}
test.c
#include "contact.h"
void menu()
{
printf("|-----------------------|\n");
printf("| contact |\n");
printf("| 1.Add 2.Delete |\n");
printf("| 3.Search 4.Modify |\n");
printf("| 5.Show 6.Sort |\n");
printf("| 7.Save 0.Exit |\n");
printf("|-----------------------|\n");
}//打印菜单
int main()
{
int input = 0;
//创建通讯录
struct Contact con;//con就是通讯录,里面包含data指针,size和capacity
//初始化通讯录
InitContact(&con);
do
{
menu();
printf("请选择:>");
scanf("%d", &input);
switch (input)
{
case ADD:
AddContact(&con);
break;
case DELETE:
DeleteContact(&con);
break;
case SREACH:
SreachContact(&con);
break;
case MODIFY:
ModifyContact(&con);
break;
case SHOW:
ShowContact(&con);
break;
case SORT:
SortContact(&con);
break;
case SAVE:
SaveContact(&con);
break;
case EXIT:
DestroyContact(&con);
printf("退出通讯录\n");
break;
default:
printf("选择错误,请重新选择\n");
break;
}
} while (input);
return 0;
}