#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, "d-d-d d:d:d", (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; }
