C语言实现写入日志文件

C语言实现写入日志文件

[cpp] view
plaincopy

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

#include <time.h>

#ifdef WIN32

#include <windows.h>

#else

#include <unistd.h> // linux下头文件

#endif

#define FILE_MAX_SIZE (1024*1024)

/*

获得当前时间字符串

@param buffer [out]: 时间字符串

@return 空

*/

void get_local_time(char*
buffer)

{

time_t rawtime;

struct tm*
timeinfo;

time(&rawtime);

timeinfo = localtime(&rawtime);

sprintf(buffer, "%04d-%02d-%02d %02d:%02d:%02d",

(timeinfo->tm_year+1900), timeinfo->tm_mon, timeinfo->tm_mday,

timeinfo->tm_hour, timeinfo->tm_min, timeinfo->tm_sec);

}

/*

获得文件大小

@param filename [in]: 文件名

@return 文件大小

*/

long get_file_size(char*
filename)

{

long length = 0;

FILE *fp = NULL;

fp = fopen(filename, "rb");

if (fp != NULL)

{

fseek(fp, 0, SEEK_END);

length = ftell(fp);

}

if (fp != NULL)

{

fclose(fp);

fp = NULL;

}

return length;

}

/*

写入日志文件

@param filename [in]: 日志文件名

@param max_size [in]: 日志文件大小限制

@param buffer [in]: 日志内容

@param buf_size [in]: 日志内容大小

@return 空

*/

void write_log_file(char*
filename,long max_size, char*
buffer, unsigned buf_size)

{

if (filename != NULL && buffer != NULL)

{

// 文件超过最大限制, 删除

long length = get_file_size(filename);

if (length > max_size)

{

unlink(filename); // 删除文件

}

// 写日志

{

FILE *fp;

fp = fopen(filename, "at+");

if (fp != NULL)

{

char now[32];

memset(now, 0, sizeof(now));

get_local_time(now);

fwrite(now, strlen(now)+1, 1, fp);

fwrite(buffer, buf_size, 1, fp);

fclose(fp);

fp = NULL;

}

}

}

}

int main(int argc,char**
argv)

{

int i;

for (i=0; i<10; ++i)

{

char buffer[32];

memset(buffer, 0, sizeof(buffer));

sprintf(buffer, "====> %d\n", i);

write_log_file("log.txt", FILE_MAX_SIZE, buffer, strlen(buffer));

#ifdef WIN32

Sleep(100); // 毫秒

#else

sleep(1); // 秒

#endif

}

// system("pause");

return 0;

}  

转自：点击打开链接