storage/manager.go at main · goforj/storage · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
package storage

import (
	"context"
	"fmt"
)

// Manager holds named storage disks.
// @group Manager
//
// Example: keep a manager for later disk lookups
//
//	mgr, _ := storage.New(storage.Config{
//		Default: "local",
//		Disks: map[storage.DiskName]storage.DriverConfig{
//			"local": localstorage.Config{Root: "/tmp/storage-manager"},
//		},
//	})
//	_ = mgr
type Manager struct {
	defaultDisk DiskName
	disks       map[DiskName]Storage
}

// New constructs a Manager and eagerly initializes all disks.
// @group Manager
//
// Example: build a manager with named disks
//
//	mgr, _ := storage.New(storage.Config{
//		Default: "local",
//		Disks: map[storage.DiskName]storage.DriverConfig{
//			"local":  localstorage.Config{Root: "/tmp/storage-local"},
//			"assets": localstorage.Config{Root: "/tmp/storage-assets", Prefix: "public"},
//		},
//	})
//	_ = mgr
func New(cfg Config) (*Manager, error) {
	if cfg.Default == "" {
		return nil, fmt.Errorf("storage: default disk is required")
	}
	if len(cfg.Disks) == 0 {
		return nil, fmt.Errorf("storage: at least one disk is required")
	}

	disks := make(map[DiskName]Storage, len(cfg.Disks))
	for name, driverCfg := range cfg.Disks {
		driverName, diskCfg, err := resolveDriverConfig(driverCfg)
		if err != nil {
			return nil, fmt.Errorf("storage: initialize disk %q: %w", name, err)
		}
		factory, ok := lookupDriver(driverName)
		if !ok {
			return nil, fmt.Errorf("storage: unknown driver %q for disk %q", driverName, name)
		}
		d, err := factory(context.Background(), diskCfg)
		if err != nil {
			return nil, fmt.Errorf("storage: initialize disk %q: %w", name, err)
		}
		disks[name] = d
	}

	return &Manager{
		defaultDisk: cfg.Default,
		disks:       disks,
	}, nil
}

// Default returns the default disk or panics if misconfigured.
// @group Manager
//
// Example: get the default disk
//
//	mgr, _ := storage.New(storage.Config{
//		Default: "local",
//		Disks: map[storage.DiskName]storage.DriverConfig{
//			"local": localstorage.Config{Root: "/tmp/storage-default"},
//		},
//	})
//
//	fs := mgr.Default()
//	fmt.Println(fs != nil)
//	// Output: true
func (m *Manager) Default() Storage {
	d, ok := m.disks[m.defaultDisk]
	if !ok {
		panic("storage: default disk misconfigured")
	}
	return d
}

// Disk returns a named disk or an error if it does not exist.
// @group Manager
//
// Example: get a named disk
//
//	mgr, _ := storage.New(storage.Config{
//		Default: "local",
//		Disks: map[storage.DiskName]storage.DriverConfig{
//			"local":   localstorage.Config{Root: "/tmp/storage-default"},
//			"uploads": localstorage.Config{Root: "/tmp/storage-uploads"},
//		},
//	})
//
//	fs, _ := mgr.Disk("uploads")
//	fmt.Println(fs != nil)
//	// Output: true
func (m *Manager) Disk(name DiskName) (Storage, error) {
	d, ok := m.disks[name]
	if !ok {
		return nil, fmt.Errorf("storage: disk %q not found", name)
	}
	return d, nil
}