🚧 根据参数按需初始化 S3 Store

Author: uglyer

cmd/ha_sqlited/ha_sqlited.go

@@ -5,6 +5,7 @@ import (
 	hadb "github.com/uglyer/ha-sqlite/db"
 	"github.com/uglyer/ha-sqlite/log"
 	"github.com/uglyer/ha-sqlite/proto"
+	"github.com/uglyer/ha-sqlite/s3"
 	"google.golang.org/grpc"
 	"net"
 )
@@ -25,10 +26,20 @@ func main() {
 	if err != nil {
 		log.Fatal(fmt.Sprintf("failed to listen (%s): %v", port, err))
 	}
-	store, err := hadb.NewHaSqliteDBManagerWithConfig(config.HaSqlite)
+	var store *hadb.HaSqliteDBManager
+	if config.S3.Enabled {
+		s3Store, err := s3.NewS3Client(config.S3)
+		if err != nil {
+			log.Fatal(fmt.Sprintf("failed to create s3 client: %v", err))
+		}
+		store, err = hadb.NewHaSqliteDBManagerWithConfigAndS3(config.HaSqlite, s3Store)
+	} else {
+		store, err = hadb.NewHaSqliteDBManagerWithConfig(config.HaSqlite)
+	}
 	if err != nil {
 		log.Fatal(fmt.Sprintf("failed to create db manager: %v", err))
 	}
+
 	s := grpc.NewServer()
 	proto.RegisterDBServer(s, store)
 	go func() {

db/ha_sqlite_db_manager.go

@@ -7,6 +7,7 @@ import (
 	"github.com/uglyer/ha-sqlite/db/store"
 	"github.com/uglyer/ha-sqlite/log"
 	"github.com/uglyer/ha-sqlite/proto"
+	"github.com/uglyer/ha-sqlite/s3"
 	"path"
 	"sync"
 	"time"
@@ -21,6 +22,7 @@ type HaSqliteDBManager struct {
 	dbLockedMap       map[int64]int
 	defaultOnApplyWal func(b []byte) error
 	config            *HaSqliteConfig
+	s3Store           s3.S3Store
 }
 
 var defaultHaSqliteConfig = &HaSqliteConfig{
@@ -36,6 +38,10 @@ func NewHaSqliteDBManager() (*HaSqliteDBManager, error) {
 }
 
 func NewHaSqliteDBManagerWithConfig(config *HaSqliteConfig) (*HaSqliteDBManager, error) {
+	return NewHaSqliteDBManagerWithConfigAndS3(config, nil)
+}
+
+func NewHaSqliteDBManagerWithConfigAndS3(config *HaSqliteConfig, s3Store s3.S3Store) (*HaSqliteDBManager, error) {
 	s, err := store.NewHaSqliteDBStoreWithDataSourceName(config.ManagerDBPath)
 	if err != nil {
 		return nil, err
@@ -47,7 +53,8 @@ func NewHaSqliteDBManagerWithConfig(config *HaSqliteConfig) (*HaSqliteDBManager,
 		defaultOnApplyWal: func(b []byte) error {
 			return nil
 		},
-		config: config,
+		config:  config,
+		s3Store: s3Store,
 	}, nil
 }