Adds composite log

Author: ackleymi

log/composite/composite_log.go

@@ -0,0 +1,81 @@
+// Copyright (c) quickfixengine.org  All rights reserved.
+//
+// This file may be distributed under the terms of the quickfixengine.org
+// license as defined by quickfixengine.org and appearing in the file
+// LICENSE included in the packaging of this file.
+//
+// This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING
+// THE WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A
+// PARTICULAR PURPOSE.
+//
+// See http://www.quickfixengine.org/LICENSE for licensing information.
+//
+// Contact [email protected] if any conditions of this licensing
+// are not clear to you.
+
+package composite
+
+import (
+	"github.com/quickfixgo/quickfix"
+)
+
+type compositeLog struct {
+	logs []quickfix.Log
+}
+
+func (l compositeLog) OnIncoming(s []byte) {
+	for _, log := range l.logs {
+		log.OnIncoming(s)
+	}
+}
+
+func (l compositeLog) OnOutgoing(s []byte) {
+	for _, log := range l.logs {
+		log.OnOutgoing(s)
+	}
+}
+
+func (l compositeLog) OnEvent(s string) {
+	for _, log := range l.logs {
+		log.OnEvent(s)
+	}
+}
+
+func (l compositeLog) OnEventf(format string, a ...interface{}) {
+	for _, log := range l.logs {
+		log.OnEventf(format, a)
+	}
+}
+
+type compositeLogFactory struct {
+	logFactories []quickfix.LogFactory
+}
+
+func (clf compositeLogFactory) Create() (quickfix.Log, error) {
+	logs := []quickfix.Log{}
+	for _, lf := range clf.logFactories {
+		log, err := lf.Create()
+		if err != nil {
+			return nil, err
+		}
+		logs = append(logs, log)
+	}
+	return compositeLog{logs}, nil
+}
+
+func (clf compositeLogFactory) CreateSessionLog(sessionID quickfix.SessionID) (quickfix.Log, error) {
+	logs := []quickfix.Log{}
+	for _, lf := range clf.logFactories {
+		log, err := lf.CreateSessionLog(sessionID)
+		if err != nil {
+			return nil, err
+		}
+		logs = append(logs, log)
+	}
+	return compositeLog{logs}, nil
+}
+
+// NewLogFactory creates an instance of LogFactory that writes messages and events to stdout.
+func NewLogFactory(logfactories []quickfix.LogFactory) quickfix.LogFactory {
+	return compositeLogFactory{logfactories}
+}

log/composite/composite_log_test.go

@@ -0,0 +1,109 @@
+// Copyright (c) quickfixengine.org  All rights reserved.
+//
+// This file may be distributed under the terms of the quickfixengine.org
+// license as defined by quickfixengine.org and appearing in the file
+// LICENSE included in the packaging of this file.
+//
+// This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING
+// THE WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A
+// PARTICULAR PURPOSE.
+//
+// See http://www.quickfixengine.org/LICENSE for licensing information.
+//
+// Contact [email protected] if any conditions of this licensing
+// are not clear to you.
+
+package composite
+
+import (
+	"fmt"
+	"log"
+	"os"
+	"path"
+	"strings"
+	"testing"
+	"time"
+
+	_ "github.com/mattn/go-sqlite3"
+	"github.com/quickfixgo/quickfix"
+	"github.com/quickfixgo/quickfix/log/file"
+	"github.com/quickfixgo/quickfix/log/mongo"
+	"github.com/quickfixgo/quickfix/log/screen"
+	"github.com/quickfixgo/quickfix/log/sql"
+	"github.com/stretchr/testify/require"
+	"github.com/stretchr/testify/suite"
+)
+
+// CompositeLogTestSuite runs tests for the MongoLog impl of Log.
+type CompositeLogTestSuite struct {
+	suite.Suite
+	sqlLogRootPath string
+	settings       *quickfix.Settings
+	sessionID      quickfix.SessionID
+}
+
+func (suite *CompositeLogTestSuite) SetupTest() {
+	mongoDbCxn := os.Getenv("MONGODB_TEST_CXN")
+	if len(mongoDbCxn) <= 0 {
+		log.Println("MONGODB_TEST_CXN environment arg is not provided, skipping...")
+		suite.T().SkipNow()
+	}
+	mongoDatabase := "automated_testing_database"
+	mongoReplicaSet := "replicaset"
+
+	// create settings
+	sessionID := quickfix.SessionID{BeginString: "FIX.4.4", SenderCompID: "SENDER", TargetCompID: "TARGET"}
+	logPath := path.Join(os.TempDir(), fmt.Sprintf("TestLogStore-%d", os.Getpid()))
+	suite.sqlLogRootPath = path.Join(os.TempDir(), fmt.Sprintf("SQLLogTestSuite-%d", os.Getpid()))
+	err := os.MkdirAll(suite.sqlLogRootPath, os.ModePerm)
+	require.Nil(suite.T(), err)
+	sqlDriver := "sqlite3"
+	sqlDsn := path.Join(suite.sqlLogRootPath, fmt.Sprintf("%d.db", time.Now().UnixNano()))
+
+	settings, err := quickfix.ParseSettings(strings.NewReader(fmt.Sprintf(`
+[DEFAULT]
+MongoLogConnection=%s
+MongoLogDatabase=%s
+MongoLogReplicaSet=%s
+FileLogPath=%s
+SQLLogDriver=%s
+SQLLogDataSourceName=%s
+SQLLogConnMaxLifetime=14400s
+
+[SESSION]
+BeginString=%s
+SenderCompID=%s
+TargetCompID=%s`, mongoDbCxn, mongoDatabase, mongoReplicaSet, logPath, sqlDriver, sqlDsn, sessionID.BeginString, sessionID.SenderCompID, sessionID.TargetCompID)))
+	require.Nil(suite.T(), err)
+
+	suite.sessionID = sessionID
+	suite.settings = settings
+}
+
+func (suite *CompositeLogTestSuite) TestCreateLogNoSession() {
+
+	mngoLogFactory := mongo.NewLogFactory(suite.settings)
+	sqlLogFactory := sql.NewLogFactory(suite.settings)
+	// create log
+	_, err := NewLogFactory([]quickfix.LogFactory{mngoLogFactory, sqlLogFactory}).Create()
+	require.Nil(suite.T(), err)
+}
+
+func (suite *CompositeLogTestSuite) TestCreateLogSession() {
+
+	screenLogFactory := screen.NewLogFactory()
+	fileLogFactory, err := file.NewLogFactory(suite.settings)
+	require.Nil(suite.T(), err)
+
+	// create log
+	_, err = NewLogFactory([]quickfix.LogFactory{screenLogFactory, fileLogFactory}).CreateSessionLog(suite.sessionID)
+	require.Nil(suite.T(), err)
+}
+
+func (suite *CompositeLogTestSuite) TearDownTest() {
+	os.RemoveAll(suite.sqlLogRootPath)
+}
+
+func TestCompositeLogTestSuite(t *testing.T) {
+	suite.Run(t, new(CompositeLogTestSuite))
+}