tinymq/examples/client-ws2-proxy.go

//go:build ignore
// +build ignore

package main

import (
	"log"
	"time"

	"git.me9.top/git/tinymq"
	"git.me9.top/git/tinymq/config"
	"git.me9.top/git/tinymq/conn"
)

func main() {
	cf := config.NewConfig()
	localChannel := "/tinymq/client/ws2/proxy"
	remoteChannel := "/tinymq/server"
	remoteFilter := tinymq.StrChannelFilter(remoteChannel)

	hostProxy := &conn.HostInfo{
		Proto:   "ws",
		Version: 2,
		Host:    "127.0.0.1",
		Port:    44211,
		// Path:    "/tinymq",
		Path:  "/tinymq-xor",
		Hash:  "xor:1qaz2wsx3edc",
		Proxy: true,
	}

	hostServer := &conn.HostInfo{
		Proto:   "ws",
		Version: 2,
		Host:    "127.0.0.1",
		Port:    34211,
		// Path:    "/tinymq",
		Path: "/tinymq-xor",
		Hash: "xor:1qaz2wsx3edc",
	}

	hub := tinymq.NewHub(
		cf,
		localChannel,
		func(channel string, hostType tinymq.HostType) (hostInfo *conn.HostInfo, err error) {
			log.Println("[connectHostFunc]", channel, hostType)
			if hostType == tinymq.HostTypeDirect {
				return hostServer, nil
			}
			return hostProxy, nil
		},
		func(host *conn.HostInfo, proto string, version uint8, channel string, remoteAuth []byte) (auth []byte) {
			log.Println("[AuthFunc]", host, proto, version, channel, string(remoteAuth))
			return []byte("tinymq-client")
		},
		func(host *conn.HostInfo, proto string, version uint8, channel string, auth []byte) bool {
			log.Println("[CheckAuthFunc]", host, proto, version, channel, string(auth))
			return string(auth) == "tinymq-server"
		},
		func(conn *tinymq.Line) {
			log.Println("[Connect state change]", conn.Channel(), conn.State(), time.Since(conn.Started()))
		},
	)

	// 订阅频道
	hub.Subscribe(remoteFilter, "hello", func(request *tinymq.RequestData) (state uint8, result any) {
		log.Println("[client RECV]<-", string(request.Data))
		return 1, "tiny client"
	},
	)
	hub.Subscribe(remoteFilter, "nodata", func(request *tinymq.RequestData) (state uint8, result any) {
		log.Println("[client RECV]<-", string(request.Data))
		return 1, nil
	},
	)

	err := hub.ConnectToServer(remoteChannel, true, nil, false)
	if err != nil {
		log.Fatalln("[client ConnectToServer ERROR]", err)
	}

	log.Println("start get data >>>")
	count, err := hub.Get(remoteFilter, "hello", "hello in get model", func(response *tinymq.ResponseData) (ok bool) {
		log.Println("get state and data: ", response.State, string(response.Data))
		return true
	})
	log.Println("end get data with count and err:", count, err)

	// 获取信息
	rsp := hub.GetOne(remoteFilter, "hello", "hello from client，hello from client，hello from client")
	if rsp.State != tinymq.STATE_OK {
		log.Println("error state:", rsp.State)
		return
	}
	log.Println("[RESULT]<-", string(rsp.Data))
	rsp = hub.GetOne(remoteFilter, "hello", func() ([]byte, error) {
		return []byte("hello from data function"), nil
	})
	if rsp.State != tinymq.STATE_OK {
		log.Println("error state:", rsp.State)
		return
	}
	log.Println("[RESULT]<-", string(rsp.Data))
	// 获取长数据
	rsp = hub.GetOne(remoteFilter, "bigdata", nil)
	if rsp.State != tinymq.STATE_OK {
		log.Println("error state:", rsp.State)
		return
	} else {
		log.Println("get bigdata ok")
	}

	time.Sleep(time.Second * 5)
	hub.Push(remoteFilter, "push", []byte(time.Now().GoString()))

	log.Println("test finished")

	time.Sleep(time.Second * 10)
	log.Println("client exit")
}