grain/main.go

37 lines
745 B
Go

package main
import (
"fmt"
"log"
"net/http"
"grain/relay"
"grain/relay/db"
"grain/relay/utils"
"golang.org/x/net/websocket"
)
func main() {
// Load configuration
config, err := utils.LoadConfig("config.yml")
if err != nil {
log.Fatal("Error loading config: ", err)
}
// Initialize MongoDB client
_, err = db.InitDB(config.MongoDB.URI, config.MongoDB.Database)
if err != nil {
log.Fatal("Error initializing database: ", err)
}
defer db.DisconnectDB()
// Start WebSocket relay
http.Handle("/", websocket.Handler(relay.Listener))
fmt.Println("WebSocket server started on", config.Server.Address)
err = http.ListenAndServe(config.Server.Address, nil)
if err != nil {
fmt.Println("Error starting server:", err)
}
}