From 08336be92e6b01ad5ec290c7f395b77dcb9f3174 Mon Sep 17 00:00:00 2001
From: banjiaojuhao <banjiaojuhao@qq.com>
Date: Wed, 22 Dec 2021 21:57:26 +0800
Subject: filer server: allow upload file to specific dataNode

---
 weed/operation/assign_file_id.go | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

(limited to 'weed/operation/assign_file_id.go')

diff --git a/weed/operation/assign_file_id.go b/weed/operation/assign_file_id.go
index b4d44eccf..f0f7581f3 100644
--- a/weed/operation/assign_file_id.go
+++ b/weed/operation/assign_file_id.go
@@ -133,6 +133,7 @@ type StorageOption struct {
 	Collection        string
 	DataCenter        string
 	Rack              string
+	DataNode          string
 	TtlSeconds        int32
 	Fsync             bool
 	VolumeGrowthCount uint32
@@ -151,9 +152,10 @@ func (so *StorageOption) ToAssignRequests(count int) (ar *VolumeAssignRequest, a
 		DiskType:            so.DiskType,
 		DataCenter:          so.DataCenter,
 		Rack:                so.Rack,
+		DataNode:            so.DataNode,
 		WritableVolumeCount: so.VolumeGrowthCount,
 	}
-	if so.DataCenter != "" || so.Rack != "" {
+	if so.DataCenter != "" || so.Rack != "" || so.DataNode != "" {
 		altRequest = &VolumeAssignRequest{
 			Count:               uint64(count),
 			Replication:         so.Replication,
@@ -162,6 +164,7 @@ func (so *StorageOption) ToAssignRequests(count int) (ar *VolumeAssignRequest, a
 			DiskType:            so.DiskType,
 			DataCenter:          "",
 			Rack:                "",
+			DataNode:            "",
 			WritableVolumeCount: so.VolumeGrowthCount,
 		}
 	}
-- 
cgit v1.2.3