grain/requests/requests.go

package requests

import (
	"context"
	"encoding/json"
	"fmt"

	"grain/events"

	"go.mongodb.org/mongo-driver/mongo"
	"golang.org/x/net/websocket"
)

var eventKind0Collection, eventKind1Collection *mongo.Collection

func SetCollections(collections map[string]*mongo.Collection) {
	eventKind0Collection = collections["eventKind0"]
	eventKind1Collection = collections["eventKind1"]
}

func Handler(ws *websocket.Conn) {
	var msg string
	for {
		err := websocket.Message.Receive(ws, &msg)
		if err != nil {
			fmt.Println("Error receiving message:", err)
			return
		}
		fmt.Println("Received message:", msg)

		// Parse the received message
		var event []interface{}
		err = json.Unmarshal([]byte(msg), &event)
		if err != nil {
			fmt.Println("Error parsing message:", err)
			return
		}

		if len(event) < 2 || event[0] != "EVENT" {
			fmt.Println("Invalid event format")
			continue
		}

		// Convert the event map to an Event struct
		eventData, ok := event[1].(map[string]interface{})
		if !ok {
			fmt.Println("Invalid event data format")
			continue
		}
		eventBytes, err := json.Marshal(eventData)
		if err != nil {
			fmt.Println("Error marshaling event data:", err)
			continue
		}

		var evt events.Event
		err = json.Unmarshal(eventBytes, &evt)
		if err != nil {
			fmt.Println("Error unmarshaling event data:", err)
			continue
		}

		// Store the event in the appropriate MongoDB collection
		//var collection *mongo.Collection
		//switch evt.Kind {
		//case 0:
		//	collection = eventKind0Collection
		//case 1:
		//	collection = eventKind1Collection
		//default:
		//	fmt.Println("Unknown event kind:", evt.Kind)
		//	continue
		//}

		err = events.HandleEvent(context.TODO(), evt)
		if err != nil {
			fmt.Println("Error handling event:", err)
			continue
		}

		err = websocket.Message.Send(ws, "Echo: "+msg)
		if err != nil {
			fmt.Println("Error sending message:", err)
			return
		}
	}
}