lc-smartX/lc/server.go

package lc

import (
	"fmt"
	"github.com/sirupsen/logrus"
	"lc-smartX/util"
	"lc-smartX/util/gopool"
	"net"
	"strings"
	"time"
)

type Controller interface {
	StartServer()
}

var Ctl = IntersectionCtl{
	Main:     time.NewTicker(5 * time.Second),
	Sub:      time.NewTicker(5 * time.Second),
	ReTicker: time.NewTicker(18 * time.Second),
}

type IntersectionCtl struct {
	MainState   byte
	SubState    byte
	MainDevices []IntersectionDevice
	SubDevices  []IntersectionDevice
	ReTicker    *time.Ticker
	Main        *time.Ticker
	Sub         *time.Ticker
}

type MainNotifier struct{}

// Notify 主路来车,通知支路设备
func (m MainNotifier) Notify() {
	Ctl.Main.Reset(5 * time.Second)
	if Ctl.MainState != 1 {
		Ctl.MainState = 1
		for _, v := range Ctl.SubDevices {
			if v.S.IsLive {
				gopool.Go(v.Call)
			}
		}
	}
}

type SubNotifier struct{}

// Notify 支路来车,通知主路设备
func (s SubNotifier) Notify() {
	Ctl.Sub.Reset(5 * time.Second)
	if Ctl.SubState != 1 {
		Ctl.SubState = 1
		for _, v := range Ctl.MainDevices {
			if v.S.IsLive {
				gopool.Go(v.Call)
			}
		}
	}
}

func (ctl *IntersectionCtl) StartServer() {
	RegisterCallback(1, &SubNotifier{})
	RegisterCallback(0, &MainNotifier{})
	//先创建响应设备
	for _, v := range util.Config.OutputDevices {
		iDevice := IntersectionDevice{
			Info: v,
		}
		iDevice.ReConnect() //初次连接
		if iDevice.Info.Branch == 1 {
			ctl.MainDevices = append(ctl.MainDevices, iDevice)
		} else {
			ctl.SubDevices = append(ctl.SubDevices, iDevice)
		}
	}
	logrus.Info("主路输出设备列表:", ctl.MainDevices)
	logrus.Info("支路输出设备列表:", ctl.SubDevices)
	fmt.Println("主路输出设备列表:", ctl.MainDevices)
	fmt.Println("支路输出设备列表:", ctl.SubDevices)
	for {
		select {
		case <-ctl.Main.C: //检查主路状态->支路输出设备作出响应
			for _, v := range Ctl.SubDevices {
				if v.S.IsLive && ctl.MainState == 1 {
					back := func() {
						v.S.Lock(1, "P000")
					}
					gopool.Go(back)
					//go v.S.Lock(1, "P000")
				}
			}
			ctl.MainState = 0
		case <-ctl.Sub.C: //检查支路状态->主路输出设备作出响应
			for _, v := range Ctl.MainDevices {
				if v.S.IsLive && ctl.SubState == 1 {
					back := func() {
						v.S.Lock(1, "P000")
					}
					gopool.Go(back)
					//go v.S.Lock(1, "P000")
				}
			}
			ctl.SubState = 0
		case <-ctl.ReTicker.C: //每18s尝试重连
			gopool.Go(func() {
				for _, v := range ctl.MainDevices {
					if v.S.IsLive {
						continue
					}
					logrus.Info("reconnect")
					fmt.Println("reconnect")
					v.ReConnect()
				}
			})
			gopool.Go(func() {
				for _, v := range ctl.SubDevices {
					if v.S.IsLive {
						continue
					}
					logrus.Info("reconnect")
					fmt.Println("reconnect")
					v.ReConnect()
				}
			})
		}
	}
}

// LServer ###
// ===
// ===
// ===
// === todo 服务器模式没测通,屏没有连接服务器
var LServer LedServer

type LedServer struct {
}

func (ls LedServer) Start() {
	listener, err := net.Listen("tcp", "192.168.110.69"+util.Config.LedServerAddr)
	if err != nil {
		fmt.Println("服务启动失败:", err)
		return
	}
	fmt.Println("led屏服务启动成功!addr:", listener.Addr())
	for {
		conn, err := listener.Accept()
		if err != nil {
			continue
		}
		go func(c net.Conn) {
			ip := strings.Split(c.RemoteAddr().String(), ":")[0]
			for i, v := range Ctl.MainDevices {
				if v.Info.ScreenIp == ip {
					fmt.Println("主路屏幕注册,ip:", ip)
					screen := NewScreen(v.Info.Name, c)
					Ctl.MainDevices[i].S = screen
				}
			}
			for i, v := range Ctl.SubDevices {
				if v.Info.ScreenIp == ip {
					fmt.Println("支路屏幕注册,ip:", ip)
					screen := NewScreen(v.Info.Name, c)
					Ctl.SubDevices[i].S = screen
				}
			}
		}(conn)
	}
}