Method UpdateStream

agent/app/service/website.go:2418–2477

(req request.StreamUpdate)

}

func (w WebsiteService) UpdateStream(req request.StreamUpdate) error {
	if req.StreamConfig.StreamPorts == "" {
		return buserr.New("ErrTypePortRange")
	}
	website, err := websiteRepo.GetFirst(repo.WithByID(req.WebsiteID))
	if err != nil {
		return err
	}
	nginxFull, err := getNginxFull(&website)
	if err != nil {
		return nil
	}
	website.StreamPorts = req.StreamConfig.StreamPorts
	ports := strings.Split(req.StreamConfig.StreamPorts, ",")
	for _, port := range ports {
		portNum, _ := strconv.Atoi(port)
		if err = checkWebsitePort(nginxFull.Install.HttpsPort, portNum, website.Type); err != nil {
			return err
		}
	}

	config := nginxFull.SiteConfig.Config
	servers := config.FindServers()
	if len(servers) == 0 {
		return errors.New("nginx config is not valid")
	}
	server := servers[0]
	server.Listens = []*components.ServerListen{}
	var params []string
	if req.UDP {
		params = []string{"udp"}
	}
	for _, port := range ports {
		server.UpdateListen(port, false, params...)
		if website.IPV6 {
			server.UpdateListen("[::]:"+port, false, params...)
		}
	}
	upstream := components.Upstream{
		UpstreamName: website.Alias,
	}
	if req.Algorithm != "default" {
		upstream.UpdateDirective(req.Algorithm, []string{})
	}
	upstream.UpstreamServers = parseUpstreamServers(req.Servers)
	for i, dir := range config.Block.Directives {
		if dir.GetName() == "upstream" && dir.GetParameters()[0] == website.Alias {
			config.Block.Directives[i] = &upstream
		}
	}

	if err := nginx.WriteConfig(config, nginx.IndentedStyle); err != nil {
		return err
	}
	if err = nginxCheckAndReload(nginxFull.SiteConfig.OldContent, config.FilePath, nginxFull.Install.ContainerName); err != nil {
		return err
	}
	website.StreamPorts = req.StreamConfig.StreamPorts