Làm thế nào để tôi kết nối lại một cách hiệu quả boost :: socket sau khi ngắt kết nối?

Question

Ứng dụng khách của tôi sử dụng boost::asio::ip::tcp::socket để kết nối với máy chủ từ xa. Nếu ứng dụng mất kết nối với máy chủ này (ví dụ: do máy chủ gặp sự cố hoặc bị tắt), tôi muốn thử kết nối lại theo khoảng thời gian đều đặn cho đến khi ứng dụng thành công.

Tôi cần làm gì ở phía máy khách để xử lý ngắt kết nối, dọn dẹp và sau đó liên tục thử kết nối lại?

Hiện tại, các bit thú vị của mã của tôi trông giống như thế này.

tôi connect như thế này:

bool MyClient::myconnect() 
{ 
    bool isConnected = false; 

    // Attempt connection 
    socket.connect(server_endpoint, errorcode); 

    if (errorcode) 
    { 
     cerr << "Connection failed: " << errorcode.message() << endl; 
     mydisconnect(); 
    } 
    else 
    { 
     isConnected = true; 

     // Connected so setup async read for an incoming message. 
     startReadMessage(); 

     // And start the io_service_thread 
     io_service_thread = new boost::thread(
      boost::bind(&MyClient::runIOService, this, boost::ref(io_service))); 
    } 
    return (isConnected) 
} 

Trường hợp phương pháp runIOServer() chỉ là:

void MyClient::runIOService(boost::asio::io_service& io_service) 
{ 
    size_t executedCount = io_service.run(); 
    cout << "io_service: " << executedCount << " handlers executed." << endl; 
    io_service.reset(); 
} 

Và nếu bất kỳ của các bộ xử lý async đọc trả về một lỗi sau đó họ chỉ cần gọi phương thức disconnect này:

void MyClient::mydisconnect(void) 
{ 
    boost::system::error_code errorcode; 

    if (socket.is_open()) 
    { 
     // Boost documentation recommends calling shutdown first 
     // for "graceful" closing of socket. 
     socket.shutdown(boost::asio::ip::tcp::socket::shutdown_both, errorcode); 
     if (errorcode) 
     { 
      cerr << "socket.shutdown error: " << errorcode.message() << endl; 
     } 

     socket.close(errorcode); 
     if (errorcode) 
     { 
      cerr << "socket.close error: " << errorcode.message() << endl; 
     }  

     // Notify the observer we have disconnected 
     myObserver->disconnected();    
    } 

..những lần thử ngắt kết nối hoàn toàn và sau đó thông báo cho một người quan sát, sẽ bắt đầu gọi connect() sau năm giây cho đến khi nó được kết nối lại.

Tôi còn cần phải làm gì khác không?

Hiện tại, điều này có nghĩa là dường như để hoạt động. Nếu tôi giết máy chủ mà nó được kết nối với tôi nhận được lỗi "End of file" được mong đợi tại các trình xử lý đọc của tôi và mydisconnect() được gọi mà không có bất kỳ vấn đề nào.

Nhưng khi nó cố gắng kết nối lại và không thành công, tôi thấy báo cáo "socket.shutdown error: Invalid argument". Đây có phải chỉ vì tôi đang cố gắng tắt một ổ cắm không có đọc/ghi đang chờ xử lý không? Hay là một cái gì đó nhiều hơn?

Answer 1

Bạn cần tạo boost::asio::ip::tcp::socket mới mỗi khi bạn kết nối lại. Cách dễ nhất để làm điều này có lẽ là chỉ phân bổ ổ cắm trên heap bằng cách sử dụng boost::shared_ptr (bạn có thể cũng có thể lấy đi với scoped_ptr nếu ổ cắm của bạn được đóng gói hoàn toàn trong một lớp). Ví dụ:

bool MyClient::myconnect() 
{ 
    bool isConnected = false; 

    // Attempt connection 
    // socket is of type boost::shared_ptr<boost::asio::ip::tcp::socket> 
    socket.reset(new boost::asio::ip::tcp::socket(...)); 
    socket->connect(server_endpoint, errorcode); 
    // ... 
} 

Sau đó, khi mydisconnect được gọi, bạn có thể deallocate ổ cắm:

void MyClient::mydisconnect(void) 
{ 
    // ... 
    // deallocate socket. will close any open descriptors 
    socket.reset(); 
} 

Các lỗi mà bạn thấy có lẽ là một kết quả của hệ điều hành dọn dẹp các mô tả tập tin sau khi bạn' đã gọi là close. Khi bạn gọi close và sau đó thử đến connect trên cùng một ổ cắm, có thể bạn đang cố gắng kết nối một bộ mô tả tệp không hợp lệ. Tại thời điểm này, bạn sẽ thấy thông báo lỗi bắt đầu bằng "Kết nối không thành công: ..." dựa trên logic của bạn, nhưng sau đó bạn gọi mydisconnect có thể sau đó đang cố gắng gọi shutdown trên bộ mô tả tệp không hợp lệ. Vòng luẩn quẩn!

Answer 2

Tôi đã thực hiện điều gì đó tương tự bằng cách sử dụng Boost.Asio trong quá khứ. Tôi sử dụng các phương thức không đồng bộ, do đó, kết nối lại thường cho phép đối tượng ip :: tcp :: socket hiện tại của tôi ra khỏi phạm vi, sau đó tạo một đối tượng mới cho các cuộc gọi đến async_connect. Nếu async_connect không thành công, tôi sử dụng bộ hẹn giờ để ngủ một chút rồi thử lại.

Answer 3

Vì lợi ích của sự rõ ràng đây là cách tiếp cận chính thức tôi đã sử dụng (nhưng điều này được dựa trên câu trả lời bjlaub, vì vậy xin vui lòng cho bất kỳ upvotes để anh ta):

tôi tuyên bố là thành viên socket như một scoped_ptr:

boost::scoped_ptr<boost::asio::ip::tcp::socket> socket; 

Sau đó, tôi đổi phương pháp connect của tôi là:

bool MyClient::myconnect() 
{ 
    bool isConnected = false; 

    // Create new socket (old one is destroyed automatically) 
    socket.reset(new boost::asio::ip::tcp::socket(io_service)); 

    // Attempt connection 
    socket->connect(server_endpoint, errorcode); 

    if (errorcode) 
    { 
     cerr << "Connection failed: " << errorcode.message() << endl; 
     socket->close(); 
    } 
    else 
    { 
     isConnected = true; 

     // Connected so setup async read for an incoming message. 
     startReadMessage(); 

     // And start the io_service_thread 
     io_service_thread = new boost::thread(
      boost::bind(&MyClient::runIOService, this, boost::ref(io_service))); 
    } 
    return (isConnected) 
} 

Answer 4

tôi đã thử cả hai phương pháp chặt chẽ() và phương pháp tắt máy và họ chỉ là khó khăn đối với tôi. Close() có thể ném một lỗi mà bạn cần để nắm bắt và là cách thô lỗ để làm những gì bạn muốn :) và tắt máy() có vẻ là tốt nhất nhưng trên phần mềm đa luồng, tôi thấy nó có thể được cầu kỳ. Vì vậy, cách tốt nhất là, như Sam nói, để cho nó đi ra khỏi phạm vi. Nếu socket là thành viên của lớp, bạn có thể 1) thiết kế lại để lớp sử dụng đối tượng 'kết nối' để bọc ổ cắm và để nó ra khỏi phạm vi hoặc 2) quấn nó vào con trỏ thông minh và đặt lại con trỏ thông minh. Nếu bạn sử dụng boost, bao gồm shared_ptr là rẻ và hoạt động như một sự quyến rũ. Không bao giờ có một vấn đề ổ cắm làm sạch nó với một shared_ptr. Chỉ là kinh nghiệm của tôi.