agency/request.h at main · compoflow/agency · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
/*
 * @Author: Lavender
 * @Date: 2022-02-23 20:22:38
 * @LastEditors: Lavender
 * @LastEditTime: 2022-02-25 16:07:33
 * @Description: 发送网络请求的头文件
 * @FilePath: /Microservice-Workflows-Agency/request.h
 */

#pragma once

#include <boost/beast/core.hpp>
#include <boost/beast/http.hpp>
#include <boost/beast/version.hpp>
#include <boost/asio.hpp>

#include <iostream>
#include <fstream>
#include <sstream>
#include <chrono>

#include <cstring>

namespace beast = boost::beast;
namespace http = beast::http;
namespace asio = boost::asio;
using tcp = boost::asio::ip::tcp;

//
// 报告错误的函数
//
void fail(beast::error_code ec, const char* what);

//
// 客户端与服务端建立连接的类
//
class session : public std::enable_shared_from_this<session> {
public:
    explicit session(asio::io_context& io)
        : _resolver(asio::make_strand(io)), _stream(asio::make_strand(io)), _state(nfile) {
        // std::cout << "nfile模式" << std::endl;
    }
    explicit session(asio::io_context& io, std::ifstream&& input_stream)
        : _resolver(asio::make_strand(io)), _stream(asio::make_strand(io)), _state(ifile), _input_stream(std::move(input_stream)) {
        // std::cout << "ifile模式" << std::endl;
    }
    explicit session(asio::io_context& io, std::ofstream&& output_stream)
        : _resolver(asio::make_strand(io)), _stream(asio::make_strand(io)), _state(ofile), _output_stream(std::move(output_stream)) {
        // std::cout << "ofile模式" << std::endl;
    }
    explicit session(asio::io_context& io, std::ifstream&& input_stream, std::ofstream&& output_stream)
        : _resolver(asio::make_strand(io)), _stream(asio::make_strand(io)), _state(iofile), _input_stream(std::move(input_stream)), _output_stream(std::move(output_stream)) {
        // std::cout << "iofile模式" << std::endl;
    }
    ~session() {
        if (_state == ifile) {
            _input_stream.close();
        } else if (_state == ofile) {
            _output_stream.close();
        } else {
            _input_stream.close();
            _output_stream.close();
        }
    }
    enum io_state {
        nfile,
        ifile,
        ofile,
        iofile
    };
    //
    // 启动连接
    //
    void run(const char* host, const char* port, const char* target, const char* verb);
private:
    //
    // 解析域名转换为ip
    //
    void on_resolve(beast::error_code ec, tcp::resolver::results_type results);
    //
    // 连接服务器
    //
    void on_connect(beast::error_code ec, tcp::resolver::results_type::endpoint_type);
    //
    // 保存服务器返回的数据到buffer
    //
    void on_write(beast::error_code ec, std::size_t bytes_transferred);
    //
    // 从buffer读取服务器返回的数据
    //
    void on_read(beast::error_code ec, std::size_t bytes_transferred);
    tcp::resolver _resolver;
    beast::tcp_stream _stream;
    beast::flat_buffer _buffer;
    http::request<http::string_body> _req;
    http::response<http::string_body> _res;
    std::ifstream _input_stream;
    std::ofstream _output_stream;
    io_state _state;
};


// // Get
// std::string GetRequest(char* host, char* path);
// std::string GetRequest(string url);

// // Post
// std::string PostRequest(char* host, char* path, std::string form);