添加推送服务

Author: hero

go_push/gateway/http_handle.go

@@ -0,0 +1,3 @@
+package gateway
+
+

go_push/gateway/http_server.go

@@ -0,0 +1,22 @@
+package gateway
+
+import (
+	"fmt"
+	"net/http"
+	"time"
+)
+
+func InitHttpServer() {
+	mux := http.NewServeMux()
+	mux.HandleFunc("/push/all", getConn)
+	mux.HandleFunc("/push/room", getConn)
+	// HTTP服务
+	server := http.Server{
+		Addr:         "0.0.0.0:9999",
+		ReadTimeout:  time.Duration(10) * time.Millisecond,
+		WriteTimeout: time.Duration(10) * time.Millisecond,
+		Handler:      mux,
+	}
+	fmt.Println("启动HTTP服务器成功 ：",9999)
+	_ = server.ListenAndServe()
+}

go_push/gateway/push_job.go

@@ -7,5 +7,20 @@ type PushJob struct {
 }
 
 type PushTask struct {
-	
+	JobChan          []chan *PushJob
+	DistributionTask chan *PushJob
+}
+
+func NewPushTask() *PushTask {
+	var p *PushTask
+	p.DistributionTask = make(chan *PushJob, 10)
+	p.JobChan = make([]chan *PushJob, len(roomTitle)) //一个房间对应一个job
+	GetRoomManage().AllRoom.Range(func(key, value interface{}) bool {
+		return true
+	})
+	return p
+}
+
+func (p *PushTask) distributionTask() {
+
 }

go_push/gateway/room.go

@@ -5,12 +5,14 @@ import "sync"
 //一个房间代表一个订阅推送类型
 type Room struct {
 	id    string
+	title string
 	RConn sync.Map
 }
 
-func NewRoom(id string) *Room {
+func NewRoom(id,title string) *Room {
 	return &Room{
 		id: id,
+		title:title,
 	}
 }
 

go_push/gateway/room_manage.go

@@ -31,13 +31,14 @@ func (r *RoomManage) DelConn(ws *WsConnection) {
 }
 
 func (r *RoomManage) AddRoom(id string, ws *WsConnection) error {
+	var room *Room
 	val, ok := r.AllRoom.Load(id)
 	if !ok {
 		//room = NewRoom(id)
 		//r.AllRoom.Store(id, room)
 		return errors.New("not find room")
 	}
-	room := val.(*Room)
+	room = val.(*Room)
 	room.JoinRoom(ws)
 	return nil
 }

go_push/gateway/ws_server.go

@@ -1,9 +1,11 @@
 package gateway
 
 import (
+	"fmt"
 	"github.com/gorilla/websocket"
 	"net"
 	"net/http"
+	"strconv"
 	"strings"
 	"time"
 )
@@ -14,18 +16,25 @@ var wsUpgrader = websocket.Upgrader{
 		return true
 	},
 }
+var roomTitle = []string{"健身", "体育", "电影", "音乐"}
 
 func InitWsServer() {
 	mux := http.NewServeMux()
 	mux.HandleFunc("/connect", getConn)
 	NewRoomManage()
+	//初始化房间
+	for i := 0; i < 4; i++ {
+		NewRoom(strconv.Itoa(i), roomTitle[i])
+		fmt.Println("新建房间推送类型", roomTitle[i])
+	}
 	// HTTP服务
 	server := http.Server{
 		Addr:         "0.0.0.0:8888",
 		ReadTimeout:  time.Duration(10) * time.Millisecond,
 		WriteTimeout: time.Duration(10) * time.Millisecond,
 		Handler:      mux,
 	}
+	fmt.Println("启动HTTP服务器成功 ：", 8888)
 	_ = server.ListenAndServe()
 
 }

go_push/main.go

@@ -4,4 +4,5 @@ import "learning_tools/go_push/gateway"
 
 func main() {
 	gateway.InitWsServer()
+	gateway.InitHttpServer()
 }