/
connection_info.go
61 lines (50 loc) · 1.22 KB
/
connection_info.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
package nfs
import (
"strconv"
"google.golang.org/grpc/codes"
"google.golang.org/grpc/status"
)
// NFSConnectionInfo class
type NFSConnectionInfo struct {
Hostname string
Port int
Path string
}
func getConnectionInfoFromMap(params map[string]string, connInfo *NFSConnectionInfo) error {
for k, v := range params {
switch k {
case "host":
connInfo.Hostname = v
case "port":
p, err := strconv.Atoi(v)
if err != nil {
return status.Errorf(codes.InvalidArgument, "Argument %q must be a valid port number - %v", k, err)
}
connInfo.Port = p
case "path":
connInfo.Path = v
default:
// ignore
}
}
return nil
}
// GetConnectionInfo returns NFSConnectionInfo value from param map
func GetConnectionInfo(configs map[string]string) (*NFSConnectionInfo, error) {
connInfo := NFSConnectionInfo{}
err := getConnectionInfoFromMap(configs, &connInfo)
if err != nil {
return nil, err
}
if len(connInfo.Hostname) == 0 {
return nil, status.Error(codes.InvalidArgument, "Argument host is empty")
}
if len(connInfo.Path) == 0 {
return nil, status.Error(codes.InvalidArgument, "Argument path is empty")
}
if connInfo.Port <= 0 {
// default
connInfo.Port = 2049
}
return &connInfo, nil
}