package com.yb.controller;

import java.io.*;

import com.yb.message.WebSocketServer;
import org.springframework.web.bind.annotation.*;

@RestController
@RequestMapping({ "/api/ws" })
public class WebSocketController
{
    @RequestMapping(value = { "/sendAll" }, method = { RequestMethod.GET })
    public String sendAllMessage(@RequestParam(required = true) final String message) {
        try {
            WebSocketServer.BroadCastInfo(message);
        }
        catch (IOException e) {
            e.printStackTrace();
        }
        return "ok";
    }
    
    @RequestMapping(value = { "/sendOne" }, method = { RequestMethod.GET })
    public String sendOneMessage(@RequestParam(required = true) final String message, @RequestParam(required = true) final String id) {
        try {
            WebSocketServer.SendMessage(message, id);
        }
        catch (IOException e) {
            e.printStackTrace();
        }
        return "ok";
    }
}