library/cpp/http/client/fetch/coctx.h


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

#pragma once

#include <library/cpp/coroutine/engine/impl.h>

#include <util/thread/singleton.h>

namespace NAsyncDns {
    class TContResolver;
    class TContDnsCache;
}

namespace NHttpFetcher {
    struct TCoCtx {
        TContExecutor* Executor;
        NAsyncDns::TContResolver* Resolver;
        NAsyncDns::TContDnsCache* DnsCache;

        TCoCtx(TContExecutor* executor, NAsyncDns::TContResolver* resolver, NAsyncDns::TContDnsCache* dnsCache = nullptr)
            : Executor(executor)
            , Resolver(resolver)
            , DnsCache(dnsCache)
        {
        }

        TCont* Cont() {
            return Executor->Running();
        }
    };

    inline TCoCtx*& CoCtx() {
        return *FastTlsSingletonWithPriority<TCoCtx*, 0>();
    }

    class TCoCtxSetter {
    public:
        TCoCtxSetter(TContExecutor* executor, NAsyncDns::TContResolver* resolver, NAsyncDns::TContDnsCache* dnsCache = nullptr)
            : Instance(executor, resolver, dnsCache)
        {
            Y_VERIFY(!CoCtx(), "coCtx already exists");
            CoCtx() = &Instance;
        }

        ~TCoCtxSetter() {
            CoCtx() = nullptr;
        }

    private:
        TCoCtx Instance;
    };
}