-
Notifications
You must be signed in to change notification settings - Fork 4
/
environment.go
47 lines (37 loc) · 1.12 KB
/
environment.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
// Copyright 2022 Azugo. All rights reserved.
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.
package core
import (
"os"
)
// Environment type.
type Environment string
const (
EnvironmentDevelopment Environment = "Development"
EnvironmentStaging Environment = "Staging"
EnvironmentProduction Environment = "Production"
)
// NewEnvironment creates new Environment instance.
func NewEnvironment(defaultMode Environment) Environment {
env := Environment(os.Getenv("ENVIRONMENT"))
if len(env) == 0 {
env = defaultMode
}
if env == EnvironmentProduction || env == EnvironmentStaging || env == EnvironmentDevelopment {
return env
}
return defaultMode
}
// IsProduction checks if current environment is production.
func (e Environment) IsProduction() bool {
return e == EnvironmentProduction
}
// IsStaging checks if current environment is staging.
func (e Environment) IsStaging() bool {
return e == EnvironmentStaging
}
// IsDevelopment checks if current environment is development.
func (e Environment) IsDevelopment() bool {
return e == EnvironmentDevelopment
}