func client() { // cellid: 1.1 acceptor // cellid: 1.2 connector // cellid: 1.3 cid := cellnet.Spawn(func(src cellnet.CellID, data interface{}) { switch d := data.(type) { case *cellnet.Packet: log.Println("recv node msg", src.String(), cellnet.ReflectContent(d)) } }) log.Println(cid.String()) nexus.Register(nexus.Dispatcher) // cellid: 1.4 dispatcher.RegisterMessage(nexus.Dispatcher, coredef.TestEchoACK{}, func(src cellnet.CellID, content interface{}) { msg := content.(*coredef.TestEchoACK) log.Println("recv msg callback :", msg.GetContent()) }) }
func joinNexus(addr string) { ltvsocket.SpawnConnector(addr, dispatcher.PeerHandler(Dispatcher)) dispatcher.RegisterMessage(Dispatcher, coredef.ConnectedACK{}, func(src cellnet.CellID, _ interface{}) { cellnet.Send(src, &coredef.RegionLinkREQ{ Profile: &coredef.Region{ ID: proto.Int32(cellnet.RegionID), Address: proto.String(config.Listen), }, }) }) dispatcher.RegisterMessage(Dispatcher, coredef.RegionLinkACK{}, func(src cellnet.CellID, content interface{}) { msg := content.(*coredef.RegionLinkACK) status := msg.GetStatus() if status.GetID() == cellnet.RegionID { log.Printf("[nexus] duplicate regionid: %d@%s", status.GetID(), status.GetAddress()) return } addRegion(src, status) for _, rg := range msg.GetAddressList() { //log.Printf("address: %d@%s", rg.GetID(), rg.GetAddress()) // 不能是自己 if rg.GetID() == cellnet.RegionID { continue } // 已经连上了, 不再连接 if GetRegion(rg.GetID()) != nil { continue } // 连接地址中的服务器 joinNexus(rg.GetAddress()) } }) }
func listenNexus() { listenAddr := config.Listen ltvsocket.SpawnAcceptor(listenAddr, dispatcher.PeerHandler(Dispatcher)) dispatcher.RegisterMessage(Dispatcher, coredef.RegionLinkREQ{}, func(ses cellnet.CellID, content interface{}) { msg := content.(*coredef.RegionLinkREQ) profile := msg.GetProfile() if profile.GetID() == cellnet.RegionID { log.Printf("duplicate regionid: %d@%s", profile.GetID(), profile.GetAddress()) return } addRegion(ses, profile) ack := coredef.RegionLinkACK{ AddressList: make([]*coredef.Region, 0), Status: &coredef.Region{ ID: proto.Int32(cellnet.RegionID), Address: proto.String(config.Listen), }, } IterateRegion(func(profile *RegionData) { ack.AddressList = append(ack.AddressList, profile.Region) }) cellnet.Send(ses, &ack) }) dispatcher.RegisterMessage(Dispatcher, coredef.ClosedACK{}, func(ses cellnet.CellID, _ interface{}) { removeRegion(ses) }) }
func client() { disp := dispatcher.NewPacketDispatcher() dispatcher.RegisterMessage(disp, coredef.TestEchoACK{}, func(ses cellnet.CellID, content interface{}) { msg := content.(*coredef.TestEchoACK) log.Println("client recv:", msg.String()) done <- true }) dispatcher.RegisterMessage(disp, coredef.ConnectedACK{}, func(ses cellnet.CellID, content interface{}) { cellnet.Send(ses, &coredef.TestEchoACK{ Content: proto.String("hello"), }) }) ltvsocket.SpawnConnector("127.0.0.1:8001", dispatcher.PeerHandler(disp)) }
func server() { disp := dispatcher.NewPacketDispatcher() dispatcher.RegisterMessage(disp, coredef.TestEchoACK{}, func(ses cellnet.CellID, content interface{}) { msg := content.(*coredef.TestEchoACK) log.Println("server recv:", msg.String()) cellnet.Send(ses, &coredef.TestEchoACK{ Content: proto.String("world"), }) }) ltvsocket.SpawnAcceptor("127.0.0.1:8001", dispatcher.PeerHandler(disp)) }