package main

import (
	"log"
	"time"

	"git.neurocipta.com/rogerferdinan/safe-web-socket/v1/server"
)

func main() {
	s, err := server.NewSafeWebsocketServerBuilder().
		BaseHost("localhost").
		BasePort(8080).
		HandleFuncWebsocket("/ws/test/", "data_1", func(c chan []byte) {
			ticker := time.NewTicker(10 * time.Millisecond)
			for range ticker.C {
				c <- []byte(time.Now().Format("2006-01-02 15:04:05") + "_data_1")
			}
		}).
		HandleFuncWebsocket("/ws/test/", "data_2", func(c chan []byte) {
			ticker := time.NewTicker(10 * time.Millisecond)
			for range ticker.C {
				c <- []byte(time.Now().Format("2006-01-02 15:04:05") + "_data_2")
			}
		}).
		Build()

	if err != nil {
		log.Fatal(err)
	}
	s.ListenAndServe()
}