func recentHandlerWs(
w http.ResponseWriter, r *http.Request, stream *wsutil.WebSocketStream) {
args, err := ParseArguments(r)
if err != nil {
stream.Fatalf("Invalid arguments; %v", err)
return
}
if err := sendRecent(stream, args); err != nil {
stream.Fatalf("%v", err)
return
}
}
func sendRecent(stream *wsutil.WebSocketStream, args *Arguments) error {
if args.Num <= 0 {
// First authorize with the CC by fetching something
_, err := recentLogs(args.Token, args.GUID, 1)
if err != nil {
return err
}
} else {
// Recent history requested?
recentLogs, err := recentLogs(args.Token, args.GUID, args.Num)
if err != nil {
return err
}
for _, line := range recentLogs {
err = stream.Send(line)
if err != nil {
return err
}
}
}
return nil
}
func tailHandlerWs(
w http.ResponseWriter, r *http.Request, stream *wsutil.WebSocketStream) {
args, err := ParseArguments(r)
if err != nil {
stream.Fatalf("Invalid arguments; %v", err)
return
}
if err := sendRecent(stream, args); err != nil {
stream.Fatalf("%v", err)
return
}
d, err := drain.NewAppLogDrain(args.GUID)
if err != nil {
stream.Fatalf("Unable to create drain: %v", err)
return
}
ch, err := d.Start()
if err != nil {
stream.Fatalf("Unable to start drain: %v", err)
}
err = stream.Forward(ch)
if err != nil {
log.Infof("%v", err)
d.Stop(err)
}
// We expect drain.Wait to not block at this point.
if err := d.Wait(); err != nil {
if _, ok := err.(wsutil.WebSocketStreamError); !ok {
log.Warnf("Error from app log drain server: %v", err)
}
}
}