Method handleConnection

agent/boundarylogproxy/proxy.go:143–184

(ctx context.Context, conn net.Conn)

}

func (s *Server) handleConnection(ctx context.Context, conn net.Conn) {
	defer s.wg.Done()

	ctx, cancel := context.WithCancel(ctx)
	defer cancel()

	s.wg.Add(1)
	go func() {
		defer s.wg.Done()
		<-ctx.Done()
		_ = conn.Close()
	}()

	// This is intended to be a sane starting point for the read buffer size.
	// It may be grown by codec.ReadMessage if necessary.
	const initBufSize = 1 << 10
	buf := make([]byte, initBufSize)

	for {
		select {
		case <-ctx.Done():
			return
		default:
		}

		var err error
		var msg proto.Message
		msg, buf, err = codec.ReadMessage(conn, buf)
		switch {
		case errors.Is(err, io.EOF) || errors.Is(err, net.ErrClosed):
			return
		case errors.Is(err, codec.ErrUnsupportedTag) || errors.Is(err, codec.ErrMessageTooLarge):
			s.logger.Warn(ctx, "read frame error", slog.Error(err))
			return
		case err != nil:
			s.logger.Warn(ctx, "read message error", slog.Error(err))
			continue
		}

		s.handleMessage(ctx, msg)
	}
}

func (s *Server) handleMessage(ctx context.Context, msg proto.Message) {
	switch m := msg.(type) {