feat recorder: add rotate options

recorder/file.go

@@ -2,7 +2,7 @@ package recorder
 
 import (
 	"context"
-	"os"
+	"io"
 
 	"github.com/go-gost/core/recorder"
 )
@@ -20,40 +20,36 @@ func SepRecorderOption(sep string) FileRecorderOption {
 }
 
 type fileRecorder struct {
-	filename string
-	sep      string
+	out io.WriteCloser
+	sep string
 }
 
 // FileRecorder records data to file.
-func FileRecorder(filename string, opts ...FileRecorderOption) recorder.Recorder {
+func FileRecorder(out io.WriteCloser, opts ...FileRecorderOption) recorder.Recorder {
 	var options fileRecorderOptions
 	for _, opt := range opts {
 		opt(&options)
 	}
 
 	return &fileRecorder{
-		filename: filename,
-		sep:      options.sep,
+		out: out,
+		sep: options.sep,
 	}
 }
 
 func (r *fileRecorder) Record(ctx context.Context, b []byte, opts ...recorder.RecordOption) error {
-	f, err := os.OpenFile(r.filename, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
-	if err != nil {
+	if _, err := r.out.Write(b); err != nil {
 		return err
 	}
-	defer f.Close()
 
-	if _, err = f.Write(b); err != nil {
-		return err
-	}
 	if r.sep != "" {
-		_, err := f.WriteString(r.sep)
+		_, err := io.WriteString(r.out, r.sep)
 		return err
 	}
+
 	return nil
 }
 
 func (r *fileRecorder) Close() error {
-	return nil
+	return r.out.Close()
 }