You can not select more than 25 topics
			Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
		
		
		
		
		
			
		
			
				
					
					
						
							54 lines
						
					
					
						
							1.2 KiB
						
					
					
				
			
		
		
		
			
			
			
		
		
	
	
							54 lines
						
					
					
						
							1.2 KiB
						
					
					
				
								package mount
							 | 
						|
								
							 | 
						|
								import (
							 | 
						|
									"context"
							 | 
						|
									"fmt"
							 | 
						|
									"github.com/seaweedfs/seaweedfs/weed/glog"
							 | 
						|
									"github.com/seaweedfs/seaweedfs/weed/pb/filer_pb"
							 | 
						|
									"time"
							 | 
						|
								)
							 | 
						|
								
							 | 
						|
								func (wfs *WFS) loopCheckQuota() {
							 | 
						|
								
							 | 
						|
									for {
							 | 
						|
								
							 | 
						|
										time.Sleep(61 * time.Second)
							 | 
						|
								
							 | 
						|
										if wfs.option.Quota <= 0 {
							 | 
						|
											continue
							 | 
						|
										}
							 | 
						|
								
							 | 
						|
										err := wfs.WithFilerClient(false, func(client filer_pb.SeaweedFilerClient) error {
							 | 
						|
								
							 | 
						|
											request := &filer_pb.StatisticsRequest{
							 | 
						|
												Collection:  wfs.option.Collection,
							 | 
						|
												Replication: wfs.option.Replication,
							 | 
						|
												Ttl:         fmt.Sprintf("%ds", wfs.option.TtlSec),
							 | 
						|
												DiskType:    string(wfs.option.DiskType),
							 | 
						|
											}
							 | 
						|
								
							 | 
						|
											resp, err := client.Statistics(context.Background(), request)
							 | 
						|
											if err != nil {
							 | 
						|
												glog.V(0).Infof("reading quota usage %v: %v", request, err)
							 | 
						|
												return err
							 | 
						|
											}
							 | 
						|
											glog.V(4).Infof("read quota usage: %+v", resp)
							 | 
						|
								
							 | 
						|
											isOverQuota := int64(resp.UsedSize) > wfs.option.Quota
							 | 
						|
											if isOverQuota && !wfs.IsOverQuota {
							 | 
						|
												glog.Warningf("Quota Exceeded! quota:%d used:%d", wfs.option.Quota, resp.UsedSize)
							 | 
						|
											} else if !isOverQuota && wfs.IsOverQuota {
							 | 
						|
												glog.Warningf("Within quota limit! quota:%d used:%d", wfs.option.Quota, resp.UsedSize)
							 | 
						|
											}
							 | 
						|
											wfs.IsOverQuota = isOverQuota
							 | 
						|
								
							 | 
						|
											return nil
							 | 
						|
										})
							 | 
						|
								
							 | 
						|
										if err != nil {
							 | 
						|
											glog.Warningf("read quota usage: %v", err)
							 | 
						|
										}
							 | 
						|
								
							 | 
						|
									}
							 | 
						|
								
							 | 
						|
								}
							 |