From 75a7327a2f1d779c1d05ca57491065f25dbf9884 Mon Sep 17 00:00:00 2001
From: tangenta <tangenta@126.com>
Date: Fri, 3 Feb 2023 14:35:56 +0800
Subject: [PATCH] This is an automated cherry-pick of #41029

Signed-off-by: ti-chi-bot <ti-community-prow-bot@tidb.io>
---
 ddl/ingest/disk_root.go | 24 ++++++++++++++++++++----
 1 file changed, 20 insertions(+), 4 deletions(-)

diff --git a/ddl/ingest/disk_root.go b/ddl/ingest/disk_root.go
index c1c98f3fe681a..c26442824dfcc 100644
--- a/ddl/ingest/disk_root.go
+++ b/ddl/ingest/disk_root.go
@@ -15,7 +15,12 @@
 package ingest
 
 import (
+<<<<<<< HEAD
 	"github.com/pingcap/errors"
+=======
+	"sync"
+
+>>>>>>> f7d5db24b3 (ddl/ingest: add mutex to disk root (#41029))
 	lcom "github.com/pingcap/tidb/br/pkg/lightning/common"
 	"github.com/pingcap/tidb/sessionctx/variable"
 	"github.com/pingcap/tidb/util/logutil"
@@ -38,6 +43,7 @@ type diskRootImpl struct {
 	currentUsage uint64
 	maxQuota     uint64
 	bcCtx        *backendCtxManager
+	mu           sync.RWMutex
 }
 
 // NewDiskRootImpl creates a new DiskRoot.
@@ -50,22 +56,32 @@ func NewDiskRootImpl(path string, bcCtx *backendCtxManager) DiskRoot {
 
 // CurrentUsage implements DiskRoot interface.
 func (d *diskRootImpl) CurrentUsage() uint64 {
-	return d.currentUsage
+	d.mu.RLock()
+	usage := d.currentUsage
+	d.mu.RUnlock()
+	return usage
 }
 
 // MaxQuota implements DiskRoot interface.
 func (d *diskRootImpl) MaxQuota() uint64 {
-	return d.maxQuota
+	d.mu.RLock()
+	quota := d.maxQuota
+	d.mu.RUnlock()
+	return quota
 }
 
 // UpdateUsageAndQuota implements DiskRoot interface.
 func (d *diskRootImpl) UpdateUsageAndQuota() error {
-	d.currentUsage = d.bcCtx.TotalDiskUsage()
+	totalDiskUsage := d.bcCtx.TotalDiskUsage()
 	sz, err := lcom.GetStorageSize(d.path)
 	if err != nil {
 		logutil.BgLogger().Error(LitErrGetStorageQuota, zap.Error(err))
 		return errors.New(LitErrGetStorageQuota)
 	}
-	d.maxQuota = mathutil.Min(variable.DDLDiskQuota.Load(), uint64(capacityThreshold*float64(sz.Capacity)))
+	maxQuota := mathutil.Min(variable.DDLDiskQuota.Load(), uint64(capacityThreshold*float64(sz.Capacity)))
+	d.mu.Lock()
+	d.currentUsage = totalDiskUsage
+	d.maxQuota = maxQuota
+	d.mu.Unlock()
 	return nil
 }