-
Notifications
You must be signed in to change notification settings - Fork 8.8k
/
plugin_activation.go
89 lines (74 loc) · 2.69 KB
/
plugin_activation.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
85
86
87
88
89
/*
Copyright IBM Corp. All Rights Reserved.
SPDX-License-Identifier: Apache-2.0
*/
package pluggable
import (
"fmt"
"io/ioutil"
"os"
"path/filepath"
"github.com/spf13/viper"
)
const (
EndorsementPluginEnvVar = "ENDORSEMENT_PLUGIN_ENV_VAR"
ValidationPluginEnvVar = "VALIDATION_PLUGIN_ENV_VAR"
)
// EndorsementPluginActivationFolder returns the name of the folder that if
// the file of the peer's id in it exists - it indicates that the endorsement plugin was activated
// for that peer
func EndorsementPluginActivationFolder() string {
return os.Getenv(EndorsementPluginEnvVar)
}
// SetEndorsementPluginActivationFolder sets the name of the folder
// that if the file of the peer's id in it exists - it indicates that the endorsement plugin was activated
// for that peer
func SetEndorsementPluginActivationFolder(path string) {
os.Setenv(EndorsementPluginEnvVar, path)
}
// ValidationPluginActivationFilePath returns the name of the folder that if
// the file of the peer's id in it exists - it indicates that the validation plugin was activated
// for that peer
func ValidationPluginActivationFolder() string {
return os.Getenv(ValidationPluginEnvVar)
}
// SetValidationPluginActivationFolder sets the name of the folder
// that if the file of the peer's id in it exists - it indicates that the validation plugin was activated
// for that peer
func SetValidationPluginActivationFolder(path string) {
os.Setenv(ValidationPluginEnvVar, path)
}
func markPluginActivation(dir string) {
fileName := filepath.Join(dir, viper.GetString("peer.id"))
_, err := os.Create(fileName)
if err != nil {
panic(fmt.Sprintf("failed to create file %s: %v", fileName, err))
}
}
// PublishEndorsementPluginActivation makes it known that the endorsement plugin
// was activated for the peer that is invoking this function
func PublishEndorsementPluginActivation() {
markPluginActivation(EndorsementPluginActivationFolder())
}
// PublishValidationPluginActivation makes it known that the validation plugin
// was activated for the peer that is invoking this function
func PublishValidationPluginActivation() {
markPluginActivation(ValidationPluginActivationFolder())
}
// CountEndorsementPluginActivations returns the number of peers that activated
// the endorsement plugin
func CountEndorsementPluginActivations() int {
return listDir(EndorsementPluginActivationFolder())
}
// CountValidationPluginActivations returns the number of peers that activated
// the validation plugin
func CountValidationPluginActivations() int {
return listDir(ValidationPluginActivationFolder())
}
func listDir(d string) int {
dir, err := ioutil.ReadDir(d)
if err != nil {
panic(fmt.Sprintf("failed listing directory %s: %v", d, err))
}
return len(dir)
}