library/cpp/containers/comptrie/loader/loader.h


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

#pragma once 
 
#include <library/cpp/archive/yarchive.h>
#include <util/generic/string.h>
#include <util/generic/ptr.h> 
#include <util/generic/yexception.h> 
#include <util/memory/blob.h> 
 
template <class TrieType, size_t N> 
TrieType LoadTrieFromArchive(const TString& key,
                             const unsigned char (&data)[N], 
                             bool ignoreErrors = false) {
    TArchiveReader archive(TBlob::NoCopy(data, sizeof(data))); 
    if (archive.Has(key)) { 
        TAutoPtr<IInputStream> trie = archive.ObjectByKey(key);
        return TrieType(TBlob::FromStream(*trie)); 
    } 
    if (!ignoreErrors) { 
        ythrow yexception() << "Resource " << key << " not found"; 
    } 
    return TrieType(); 
}