Use a workerpool for configuration file updates
Co-Authored-By: Jakob <schrej@users.noreply.github.com>
This commit is contained in:
		
							parent
							
								
									5e8425ad6a
								
							
						
					
					
						commit
						5366d0f652
					
				| 
						 | 
				
			
			@ -1,34 +1,32 @@
 | 
			
		|||
package server
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"github.com/pterodactyl/wings/parser"
 | 
			
		||||
	"sync"
 | 
			
		||||
	"github.com/gammazero/workerpool"
 | 
			
		||||
	"runtime"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// Parent function that will update all of the defined configuration files for a server
 | 
			
		||||
// automatically to ensure that they always use the specified values.
 | 
			
		||||
func (s *Server) UpdateConfigurationFiles() {
 | 
			
		||||
	wg := new(sync.WaitGroup)
 | 
			
		||||
	pool := workerpool.New(runtime.GOMAXPROCS(0))
 | 
			
		||||
 | 
			
		||||
	files := s.ProcessConfiguration().ConfigurationFiles
 | 
			
		||||
	for _, v := range files {
 | 
			
		||||
		wg.Add(1)
 | 
			
		||||
	for _, cf := range files {
 | 
			
		||||
		f := cf
 | 
			
		||||
 | 
			
		||||
		go func(f parser.ConfigurationFile, server *Server) {
 | 
			
		||||
			defer wg.Done()
 | 
			
		||||
 | 
			
		||||
			p, err := server.Filesystem.SafePath(f.FileName)
 | 
			
		||||
		pool.Submit(func() {
 | 
			
		||||
			p, err := s.Filesystem.SafePath(f.FileName)
 | 
			
		||||
			if err != nil {
 | 
			
		||||
				server.Log().WithField("error", err).Error("failed to generate safe path for configuration file")
 | 
			
		||||
				s.Log().WithField("error", err).Error("failed to generate safe path for configuration file")
 | 
			
		||||
 | 
			
		||||
				return
 | 
			
		||||
			}
 | 
			
		||||
 | 
			
		||||
			if err := f.Parse(p, false); err != nil {
 | 
			
		||||
				server.Log().WithField("error", err).Error("failed to parse and update server configuration file")
 | 
			
		||||
				s.Log().WithField("error", err).Error("failed to parse and update server configuration file")
 | 
			
		||||
			}
 | 
			
		||||
		}(v, s)
 | 
			
		||||
		})
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	wg.Wait()
 | 
			
		||||
	pool.StopWait()
 | 
			
		||||
}
 | 
			
		||||
		Loading…
	
		Reference in New Issue
	
	Block a user