diff options
| author | chrislu <chris.lu@gmail.com> | 2023-09-10 22:03:44 -0700 |
|---|---|---|
| committer | chrislu <chris.lu@gmail.com> | 2023-09-10 22:03:44 -0700 |
| commit | 07c2d75ac7cc2b8652203d9d4d59461903d20d4d (patch) | |
| tree | bfb20243eea6ba57c617c94987466add0dc47d8f /weed/operation/assign_file_id.go | |
| parent | f24c7e803f0a14d35e2bb40a2ed838c80b8cf2e8 (diff) | |
| download | seaweedfs-07c2d75ac7cc2b8652203d9d4d59461903d20d4d.tar.xz seaweedfs-07c2d75ac7cc2b8652203d9d4d59461903d20d4d.zip | |
ensure single threaded access
Diffstat (limited to 'weed/operation/assign_file_id.go')
| -rw-r--r-- | weed/operation/assign_file_id.go | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/weed/operation/assign_file_id.go b/weed/operation/assign_file_id.go index a4753d234..f6d0e0110 100644 --- a/weed/operation/assign_file_id.go +++ b/weed/operation/assign_file_id.go @@ -8,6 +8,7 @@ import ( "github.com/seaweedfs/seaweedfs/weed/security" "github.com/seaweedfs/seaweedfs/weed/storage/needle" "google.golang.org/grpc" + "sync" ) type VolumeAssignRequest struct { @@ -66,9 +67,13 @@ func (ap *AssignProxy) Assign(primaryRequest *VolumeAssignRequest, alternativeRe type singleThreadAssignProxy struct { assignClient master_pb.Seaweed_StreamAssignClient + sync.Mutex } func (ap *singleThreadAssignProxy) doAssign(grpcConnection *grpc.ClientConn, primaryRequest *VolumeAssignRequest, alternativeRequests ...*VolumeAssignRequest) (ret *AssignResult, err error) { + ap.Lock() + defer ap.Unlock() + if ap.assignClient == nil { client := master_pb.NewSeaweedClient(grpcConnection) ap.assignClient, err = client.StreamAssign(context.Background()) |