From e061a3dfadf56277402364075ceb92b0597ded6c Mon Sep 17 00:00:00 2001
From: YuQing <384681@qq.com>
Date: Wed, 16 Mar 2022 11:48:00 +0800
Subject: [PATCH] add file src/sf_buffered_writer.h

---
 .gitignore               |  1 +
 src/Makefile.in          |  2 +-
 src/sf_buffered_writer.h | 92 ++++++++++++++++++++++++++++++++++++++++
 3 files changed, 94 insertions(+), 1 deletion(-)
 create mode 100644 src/sf_buffered_writer.h

diff --git a/.gitignore b/.gitignore
index d7c1fbe..ea67af2 100644
--- a/.gitignore
+++ b/.gitignore
@@ -31,3 +31,4 @@ src/Makefile
 
 # other
 *.swp
+*.swo
diff --git a/src/Makefile.in b/src/Makefile.in
index 15ab52a..accf6c0 100644
--- a/src/Makefile.in
+++ b/src/Makefile.in
@@ -9,7 +9,7 @@ TOP_HEADERS = sf_types.h sf_global.h sf_define.h sf_nio.h sf_service.h \
               sf_func.h sf_util.h sf_configs.h sf_proto.h sf_binlog_writer.h \
               sf_cluster_cfg.h sf_sharding_htable.h sf_connection_manager.h  \
               sf_serializer.h sf_binlog_index.h sf_file_writer.h \
-              sf_ordered_writer.h sf_iov.h
+              sf_ordered_writer.h sf_buffered_writer.h sf_iov.h
 
 IDEMP_SERVER_HEADER = idempotency/server/server_types.h \
                       idempotency/server/server_channel.h  \
diff --git a/src/sf_buffered_writer.h b/src/sf_buffered_writer.h
new file mode 100644
index 0000000..37d70a2
--- /dev/null
+++ b/src/sf_buffered_writer.h
@@ -0,0 +1,92 @@
+/*
+ * Copyright (c) 2020 YuQing <384681@qq.com>
+ *
+ * This program is free software: you can use, redistribute, and/or modify
+ * it under the terms of the GNU Affero General Public License, version 3
+ * or later ("AGPL"), as published by the Free Software Foundation.
+ *
+ * This program is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program. If not, see <https://www.gnu.org/licenses/>.
+ */
+
+//sf_buffered_writer.h
+
+#ifndef _SF_BUFFERED_WRITER_H_
+#define _SF_BUFFERED_WRITER_H_
+
+#include "sf_types.h"
+
+typedef struct {
+    int fd;
+    const char *filename;
+    SFBinlogBuffer buffer;
+} SFBufferedWriter;
+
+#define sf_buffered_writer_init(writer, filename) \
+    sf_buffered_writer_init_ex(writer, filename, 1024 * 1024)
+
+#define SF_BUFFERED_WRITER_LENGTH(bw) ((bw).buffer.current - (bw).buffer.buff)
+#define SF_BUFFERED_WRITER_REMAIN(bw) ((bw).buffer.end - (bw).buffer.current)
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+    static inline int sf_buffered_writer_init_ex(SFBufferedWriter *writer,
+            const char *filename, const int buffer_size)
+    {
+        int result;
+
+        writer->filename = filename;
+        writer->fd = open(filename, O_WRONLY | O_CREAT | O_TRUNC, 0644);
+        if (writer->fd < 0) {
+            result = errno != 0 ? errno : EIO;
+            logError("file: "__FILE__", line: %d, "
+                    "open file %s fail, errno: %d, error info: %s",
+                    __LINE__, filename, result, STRERROR(result));
+            return result;
+        }
+
+        if ((result=sf_binlog_buffer_init(&writer->buffer, buffer_size)) != 0) {
+            return result;
+        }
+        writer->buffer.end = writer->buffer.buff + writer->buffer.size;
+        return 0;
+    }
+
+    static inline int sf_buffered_writer_save(SFBufferedWriter *writer)
+    {
+        int result;
+        int length;
+
+        length = writer->buffer.current - writer->buffer.buff;
+        if (fc_safe_write(writer->fd, writer->buffer.buff, length) != length) {
+            result = errno != 0 ? errno : EIO;
+            logError("file: "__FILE__", line: %d, "
+                    "write to file %s fail, errno: %d, error info: %s",
+                    __LINE__, writer->filename, result, STRERROR(result));
+            return result;
+        }
+
+        writer->buffer.current = writer->buffer.buff;
+        return 0;
+    }
+
+    static inline void sf_buffered_writer_destroy(SFBufferedWriter *writer)
+    {
+        if (writer->fd >= 0) {
+            close(writer->fd);
+            writer->fd = -1;
+        }
+        sf_binlog_buffer_destroy(&writer->buffer);
+    }
+
+#ifdef __cplusplus
+}
+#endif
+
+#endif