diff options
Diffstat (limited to 'pkg/driver')
| -rw-r--r-- | pkg/driver/driver.go | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/pkg/driver/driver.go b/pkg/driver/driver.go index 1cc3832..5a375e1 100644 --- a/pkg/driver/driver.go +++ b/pkg/driver/driver.go @@ -5,12 +5,12 @@ import ( "os" "github.com/container-storage-interface/spec/lib/go/csi" + "github.com/seaweedfs/seaweedfs-csi-driver/pkg/datalocality" "github.com/seaweedfs/seaweedfs/weed/glog" "github.com/seaweedfs/seaweedfs/weed/pb" "github.com/seaweedfs/seaweedfs/weed/pb/filer_pb" "github.com/seaweedfs/seaweedfs/weed/security" "github.com/seaweedfs/seaweedfs/weed/util" - "github.com/seaweedfs/seaweedfs-csi-driver/pkg/datalocality" "google.golang.org/grpc" "google.golang.org/grpc/codes" "google.golang.org/grpc/status" @@ -44,8 +44,9 @@ type SeaweedFsDriver struct { CacheDir string UidMap string GidMap string + signature int32 DataCenter string - DataLocality datalocality.DataLocality + DataLocality datalocality.DataLocality } func NewSeaweedFsDriver(filer, nodeID, endpoint string) *SeaweedFsDriver { @@ -61,6 +62,7 @@ func NewSeaweedFsDriver(filer, nodeID, endpoint string) *SeaweedFsDriver { version: version, filers: pb.ServerAddresses(filer).ToAddresses(), grpcDialOption: security.LoadClientTLS(util.GetViper(), "grpc.client"), + signature: util.RandomInt32(), } n.AddVolumeCapabilityAccessModes([]csi.VolumeCapability_AccessMode_Mode{ @@ -143,7 +145,7 @@ func (d *SeaweedFsDriver) WithFilerClient(streamingMode bool, fn func(filer_pb.S var err error for x := 0; x < n; x++ { - err = pb.WithGrpcClient(streamingMode, func(grpcConnection *grpc.ClientConn) error { + err = pb.WithGrpcClient(streamingMode, d.signature, func(grpcConnection *grpc.ClientConn) error { client := filer_pb.NewSeaweedFilerClient(grpcConnection) return fn(client) }, d.filers[i].ToGrpcAddress(), false, d.grpcDialOption) |