Repository navigation
Expand file tree
/
Copy pathinit.go
More file actions
125 lines (106 loc) · 3.13 KB
/
Copy pathinit.go
File metadata and controls
125 lines (106 loc) · 3.13 KB
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
115
116
117
118
119
120
121
122
123
124
125
/*
* Copyright 2016-2018 Dgraph Labs, Inc. and Contributors
*
* 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 x
import (
"crypto/sha256"
"fmt"
"io"
"os"
"runtime"
"strings"
"github.com/golang/glog"
)
var (
initFunc []func()
isTest bool
// These variables are set using -ldflags
dgraphVersion string
gitBranch string
lastCommitSHA string
lastCommitTime string
)
// SetTestRun sets a variable to indicate that the current execution is a test.
func SetTestRun() {
isTest = true
}
// IsTestRun indicates whether a test is being executed. Useful to handle special
// conditions during tests that differ from normal execution.
func IsTestRun() bool {
return isTest
}
// AddInit adds a function to be run in x.Init, which should be called at the
// beginning of all mains.
func AddInit(f func()) {
initFunc = append(initFunc, f)
}
// Init initializes flags and run all functions in initFunc.
func Init() {
// Default value, would be overwritten by flag.
Config.QueryEdgeLimit = 1e6
// Next, run all the init functions that have been added.
for _, f := range initFunc {
f()
}
}
// BuildDetails returns a string containing details about the Dgraph binary.
func BuildDetails() string {
licenseInfo := `Licensed under the Apache Public License 2.0`
if !strings.HasSuffix(dgraphVersion, "-oss") {
licenseInfo = "Licensed variously under the Apache Public License 2.0 and Dgraph " +
"Community License"
}
return fmt.Sprintf(`
Dgraph version : %v
Dgraph SHA-256 : %x
Commit SHA-1 : %v
Commit timestamp : %v
Branch : %v
Go version : %v
For Dgraph official documentation, visit https://docs.dgraph.io.
For discussions about Dgraph , visit https://discuss.dgraph.io.
To say hi to the community , visit https://dgraph.slack.com.
%s.
Copyright 2015-2018 Dgraph Labs, Inc.
`,
dgraphVersion, ExecutableChecksum(), lastCommitSHA, lastCommitTime, gitBranch,
runtime.Version(), licenseInfo)
}
// PrintVersion prints version and other helpful information if --version.
func PrintVersion() {
glog.Infof("\n%s\n", BuildDetails())
}
// Version returns a string containing the dgraphVersion.
func Version() string {
return dgraphVersion
}
// ExecutableChecksum returns a byte slice containing the SHA256 checksum of the executable.
// It returns a nil slice if there's an error trying to calculate the checksum.
func ExecutableChecksum() []byte {
execPath, err := os.Executable()
if err != nil {
return nil
}
execFile, err := os.Open(execPath)
if err != nil {
return nil
}
defer execFile.Close()
h := sha256.New()
if _, err := io.Copy(h, execFile); err != nil {
return nil
}
return h.Sum(nil)
}