응답 반환 후 HTTP 서버 종료

명령줄에 최적화되지 않은 OAuth를 사용하여 Instagram API와 상호 작용하는 명령줄 봇을 개발 중입니다. 응용 프로그램. 이 문제를 극복하기 위해 인증을 위한 리디렉션 URI로 로컬 HTTP 서버를 설정했습니다. 사용자가 애플리케이션을 승인하고 서버로 리디렉션된 후 액세스 토큰이 표시되면 서버를 종료하려고 합니다.

종료 메커니즘을 구현할 때 문제가 발생했습니다. 특히 요청 처리기(showTokenToUser)에서 srv.Shutdown(nil)을 호출하면 오류가 발생합니다.

2017/11/23 16:02:03 Httpserver: ListenAndServe() error: http: Server closed
2017/11/23 16:02:03 http: panic serving [::1]:61793: runtime error: invalid memory address or nil pointer dereference

솔루션

이 문제는 다음 때문에 발생합니다. srv.Shutdown(nil)을 여러 번 호출합니다:

1. showTokenToUser 핸들러에서 클로저 내에 위치합니다.
2. 핸들러가 기본 함수에서 실행을 마친 후.

서버가 여전히 연결을 수신하는 동안 srv.Shutdown을 호출하면 경쟁 조건이 발생합니다. ListenAndServe() 고루틴은 열려 있는 리스너와 유휴 연결을 닫으려고 시도하지만 핸들러 클로저에서 Shutdown에 대한 후속 호출로 인해 중단됩니다. 이 상태 불일치로 인해 패닉이 발생합니다.

이 문제를 해결하려면 다음 두 가지 방법 중 하나를 사용할 수 있습니다.

1. context.WithCancel:

를 사용합니다. 이 접근 방식에서는 취소 기능을 사용하여 context.Context를 만듭니다. 컨텍스트는 ListenAndServe 고루틴과 showTokenToUser 핸들러 함수에 전달됩니다. 핸들러 내에서 액세스 토큰이 사용자에게 표시되면 취소 함수를 호출하여 컨텍스트를 종료합니다. ListenAndServe 고루틴은 컨텍스트가 취소되면 서버를 정상적으로 종료합니다.

2. 동일한 Context를 사용하세요:

context.WithCancel을 호출하는 대신 동일한 context.Context를 ListenAndServe 고루틴과 핸들러 함수에 전달할 수 있습니다. 액세스 토큰이 표시되면 컨텍스트에서 cancel()을 호출하여 핸들러와 ListenAndServe 고루틴 모두의 종료를 트리거합니다.

두 가지 접근 방식 중 하나를 구현한 후 srv.Shutdown 함수를 기다려야 합니다. 프로그램을 종료하기 전에 완료하십시오.