reload.go

package main

import (
	"log"

	"github.com/Qv2ray/mmp-go/config"
)

func ReloadConfig(oldConf *config.Config) {
	log.Println("Reloading configuration")
	mPortDispatcher.Lock()
	defer mPortDispatcher.Unlock()

	// rebuild config
	confPath := oldConf.ConfPath
	httpClient := oldConf.HttpClient
	newConf, err := config.BuildConfig(confPath, httpClient)
	if err != nil {
		log.Printf("failed to reload configuration: %v", err)
		return
	}
	// check if there is any net error when pulling the upstream configurations
	for i := range newConf.Groups {
		newGroup := &newConf.Groups[i]
		for j := range newGroup.Upstreams {
			newUpstream := newGroup.Upstreams[j]
			pErr := newUpstream.PullingError
			if pErr != nil {
				log.Printf("skip to update some servers in group %v , error on upstream %v: %v", newGroup.Name, newUpstream.Name, pErr)
				// error occurred, remain those servers

				// find the group in the oldConf
				var oldGroup *config.Group
				for k := range oldConf.Groups {
					// they should have the same port
					if oldConf.Groups[k].Port != newGroup.Port {
						continue
					}
					oldGroup = &oldConf.Groups[k]
					break
				}
				if oldGroup == nil {
					// cannot find the corresponding old group
					continue
				}
				// check if upstreamConf can match
				for k := range oldGroup.Servers {
					oldServer := oldGroup.Servers[k]
					if oldServer.UpstreamConf != nil && newUpstream.Equal(*oldServer.UpstreamConf) {
						// remain the server
						newGroup.Servers = append(newGroup.Servers, oldServer)
					}
				}
			}
		}
	}
	config.SetConfig(newConf)
	c := newConf

	// update dispatchers
	newConfPortSet := make(map[int]struct{})
	for i := range c.Groups {
		newConfPortSet[c.Groups[i].Port] = struct{}{}

		if t, ok := mPortDispatcher.Map[c.Groups[i].Port]; ok {
			// update the existing dispatcher
			for j := range protocols {
				t[j].UpdateGroup(&c.Groups[i])
			}
		} else {
			// add a new port dispatcher
			groupWG.Add(1)
			go func(group *config.Group) {
				listenGroup(group)
				groupWG.Done()
			}(&c.Groups[i])
		}
	}
	// close all removed port dispatcher
	for port := range mPortDispatcher.Map {
		if _, ok := newConfPortSet[port]; !ok {
			t := mPortDispatcher.Map[port]
			delete(mPortDispatcher.Map, port)
			for j := range protocols {
				_ = (*t)[j].Close()
			}
		}
	}
	log.Println("Reloaded configuration")
}