This repository has been archived by the owner on Jul 12, 2023. It is now read-only.
/
config.go
84 lines (72 loc) · 3.13 KB
/
config.go
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
// Copyright 2020 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
// Package export defines the handlers for managing exposure key exporting.
package export
import (
"time"
"github.com/google/exposure-notifications-server/internal/setup"
"github.com/google/exposure-notifications-server/internal/storage"
"github.com/google/exposure-notifications-server/pkg/database"
"github.com/google/exposure-notifications-server/pkg/keys"
"github.com/google/exposure-notifications-server/pkg/observability"
"github.com/google/exposure-notifications-server/pkg/secrets"
)
// Compile-time check to assert this config matches requirements.
var (
_ setup.BlobstoreConfigProvider = (*Config)(nil)
_ setup.DatabaseConfigProvider = (*Config)(nil)
_ setup.KeyManagerConfigProvider = (*Config)(nil)
_ setup.SecretManagerConfigProvider = (*Config)(nil)
_ setup.ObservabilityExporterConfigProvider = (*Config)(nil)
)
// Config represents the configuration and associated environment variables for
// the export components.
type Config struct {
Database database.Config
KeyManager keys.Config
SecretManager secrets.Config
Storage storage.Config
ObservabilityExporter observability.Config
Port string `env:"PORT, default=8080"`
CreateTimeout time.Duration `env:"CREATE_BATCHES_TIMEOUT, default=5m"`
WorkerTimeout time.Duration `env:"WORKER_TIMEOUT, default=5m"`
MinRecords int `env:"EXPORT_FILE_MIN_RECORDS, default=1000"`
PaddingRange int `env:"EXPORT_FILE_PADDING_RANGE, default=100"`
MaxRecords int `env:"EXPORT_FILE_MAX_RECORDS, default=500000"`
MaxInsertBatchSize int `env:"MAX_INSERT_BATCH_SIZE, default=100"`
TruncateWindow time.Duration `env:"TRUNCATE_WINDOW, default=1h"`
MinWindowAge time.Duration `env:"MIN_WINDOW_AGE, default=2h"`
TTL time.Duration `env:"CLEANUP_TTL, default=336h"`
// ReprocessCount needs to be incremented by one every time you go back and
// regenerate previously exported files.
ReprocessCount uint `env:"REPROCESS_COUNT, default=0"`
}
func (c *Config) RepressGeneration() int64 {
return int64(c.ReprocessCount)
}
func (c *Config) BlobstoreConfig() *storage.Config {
return &c.Storage
}
func (c *Config) DatabaseConfig() *database.Config {
return &c.Database
}
func (c *Config) KeyManagerConfig() *keys.Config {
return &c.KeyManager
}
func (c *Config) SecretManagerConfig() *secrets.Config {
return &c.SecretManager
}
func (c *Config) ObservabilityExporterConfig() *observability.Config {
return &c.ObservabilityExporter
}