package main

import (
	"encoding/json"
	"log"
	"time"

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

type ExampleData struct {
	Time time.Time `json:"time"`
	Data string    `json:"data"`
}

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 {
				jsonBytes, err := json.Marshal(ExampleData{
					Time: time.Now(),
					Data: "data_1",
				})
				if err != nil {
					continue
				}
				c <- jsonBytes
			}
		}).
		HandleFuncWebsocket("/ws/test/", "data_2", func(c chan []byte) {
			ticker := time.NewTicker(10 * time.Millisecond)
			for range ticker.C {
				jsonBytes, err := json.Marshal(ExampleData{
					Time: time.Now(),
					Data: "data_2",
				})
				if err != nil {
					continue
				}
				c <- jsonBytes
			}
		}).
		Build()

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