Skip to content

Commit 59ae164

Browse files
authored
fix(storage): use s3 provider config flag (#85)
1 parent b8f97cc commit 59ae164

2 files changed

Lines changed: 6 additions & 1 deletion

File tree

cmd/lite/lite.go

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -202,6 +202,9 @@ func init() {
202202
if err := viper.BindEnv("clickhouse-database", "CLICKHOUSE_DATABASE"); err != nil {
203203
bindError(err)
204204
}
205+
if err := viper.BindEnv("s3-provider", "S3_PROVIDER"); err != nil {
206+
bindError(err)
207+
}
205208

206209
rootCmd.AddCommand(migrate.RootCmd)
207210
migrate.RootCmd.AddCommand(

pkg/storage/storage_strategy.go

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@ import (
55
"io"
66

77
"github.com/fivemanage/lite/pkg/storage/s3"
8+
"github.com/spf13/viper"
89
)
910

1011
type StorageLayer interface {
@@ -16,7 +17,8 @@ type StorageLayer interface {
1617
func New(provider string) StorageLayer {
1718
switch provider {
1819
case "s3":
19-
return s3.New()
20+
s3Provider := viper.GetString("s3-provider")
21+
return s3.New(s3Provider)
2022
}
2123

2224
return nil

0 commit comments

Comments
 (0)