diff options
Diffstat (limited to 'zenserver')
| -rw-r--r-- | zenserver/cache/structuredcache.cpp | 837 | ||||
| -rw-r--r-- | zenserver/cache/structuredcache.h | 9 | ||||
| -rw-r--r-- | zenserver/cache/structuredcachestore.cpp | 7 | ||||
| -rw-r--r-- | zenserver/cache/structuredcachestore.h | 1 | ||||
| -rw-r--r-- | zenserver/compute/apply.cpp | 9 | ||||
| -rw-r--r-- | zenserver/config.cpp | 21 | ||||
| -rw-r--r-- | zenserver/resource.h | 18 | ||||
| -rw-r--r-- | zenserver/upstream/jupiter.cpp | 30 | ||||
| -rw-r--r-- | zenserver/upstream/jupiter.h | 10 | ||||
| -rw-r--r-- | zenserver/upstream/upstreamcache.cpp | 206 | ||||
| -rw-r--r-- | zenserver/upstream/upstreamcache.h | 44 | ||||
| -rw-r--r-- | zenserver/upstream/zen.cpp | 25 | ||||
| -rw-r--r-- | zenserver/upstream/zen.h | 10 | ||||
| -rw-r--r-- | zenserver/zenserver.cpp | 4 | ||||
| -rw-r--r-- | zenserver/zenserver.rc | 81 | ||||
| -rw-r--r-- | zenserver/zenserver.vcxproj | 7 | ||||
| -rw-r--r-- | zenserver/zenserver.vcxproj.filters | 7 |
17 files changed, 797 insertions, 529 deletions
diff --git a/zenserver/cache/structuredcache.cpp b/zenserver/cache/structuredcache.cpp index 7f1fe7b44..3ac1ec37f 100644 --- a/zenserver/cache/structuredcache.cpp +++ b/zenserver/cache/structuredcache.cpp @@ -173,10 +173,19 @@ HttpStructuredCacheService::Flush() { } -void +void HttpStructuredCacheService::Scrub(ScrubContext& Ctx) { - ZEN_UNUSED(Ctx); + if (m_LastScrubTime == Ctx.ScrubTimestamp()) + { + return; + } + + m_LastScrubTime = Ctx.ScrubTimestamp(); + + m_CasStore.Scrub(Ctx); + m_CidStore.Scrub(Ctx); + m_CacheStore.Scrub(Ctx); } void @@ -244,7 +253,7 @@ HttpStructuredCacheService::HandleCacheBucketRequest(HttpServerRequest& Request, } void -HttpStructuredCacheService::HandleCacheRecordRequest(HttpServerRequest& Request, CacheRef& Ref, CachePolicy Policy) +HttpStructuredCacheService::HandleCacheRecordRequest(HttpServerRequest& Request, const CacheRef& Ref, CachePolicy Policy) { switch (auto Verb = Request.RequestVerb()) { @@ -253,439 +262,380 @@ HttpStructuredCacheService::HandleCacheRecordRequest(HttpServerRequest& Request, case kHead: case kGet: { - const ZenContentType AcceptType = Request.AcceptContentType(); + HandleGetCacheRecord(Request, Ref, Policy); + if (Verb == kHead) + { + Request.SetSuppressResponseBody(); + } + } + break; + case kPut: + HandlePutCacheRecord(Request, Ref, Policy); + break; + default: + break; + } +} - ZenCacheValue Value; - bool Success = m_CacheStore.Get(Ref.BucketSegment, Ref.HashKey, /* out */ Value); - bool InUpstreamCache = false; +void +HttpStructuredCacheService::HandleGetCacheRecord(zen::HttpServerRequest& Request, const CacheRef& Ref, CachePolicy Policy) +{ + const ZenContentType AcceptType = Request.AcceptContentType(); - const bool QueryUpstream = !Success && m_UpstreamCache && (CachePolicy::QueryRemote == (Policy & CachePolicy::QueryRemote)); + ZenCacheValue Value; + bool Success = m_CacheStore.Get(Ref.BucketSegment, Ref.HashKey, /* out */ Value); + bool InUpstreamCache = false; - if (QueryUpstream) - { - const ZenContentType CacheRecordType = Ref.BucketSegment == "legacy"sv ? ZenContentType::kBinary - : AcceptType == ZenContentType::kCbPackage ? ZenContentType::kCbPackage - : ZenContentType::kCbObject; + const bool QueryUpstream = !Success && m_UpstreamCache && (CachePolicy::QueryRemote == (Policy & CachePolicy::QueryRemote)); - if (auto UpstreamResult = m_UpstreamCache->GetCacheRecord({Ref.BucketSegment, Ref.HashKey}, CacheRecordType); - UpstreamResult.Success) - { - Value.Value = UpstreamResult.Value; - Success = true; - InUpstreamCache = true; + if (QueryUpstream) + { + const ZenContentType CacheRecordType = Ref.BucketSegment == "legacy"sv ? ZenContentType::kBinary + : AcceptType == ZenContentType::kCbPackage ? ZenContentType::kCbPackage + : ZenContentType::kCbObject; - if (CacheRecordType == ZenContentType::kBinary || CacheRecordType == ZenContentType::kCbObject) - { - if (CacheRecordType == ZenContentType::kCbObject) - { - const CbValidateError ValidationResult = ValidateCompactBinary(UpstreamResult.Value, CbValidateMode::All); - - if (ValidationResult == CbValidateError::None) - { - CbObjectView CacheRecord(UpstreamResult.Value.Data()); - - CbObjectWriter IndexData; - IndexData.BeginArray("references"); - CacheRecord.IterateAttachments([&](CbFieldView Attachment) { IndexData.AddHash(Attachment.AsHash()); }); - IndexData.EndArray(); - - Value.IndexData = IndexData.Save(); - } - else - { - Success = false; - ZEN_WARN("Get - cache record '{}/{}' FAILED, invalid compact binary object from upstream", - Ref.BucketSegment, - Ref.HashKey); - } - } + if (auto UpstreamResult = m_UpstreamCache->GetCacheRecord({Ref.BucketSegment, Ref.HashKey}, CacheRecordType); + UpstreamResult.Success) + { + Value.Value = UpstreamResult.Value; + Success = true; + InUpstreamCache = true; - if (Success) - { - m_CacheStore.Put(Ref.BucketSegment, Ref.HashKey, Value); - } - } - else - { - ZEN_ASSERT(CacheRecordType == ZenContentType::kCbPackage); + if (CacheRecordType == ZenContentType::kBinary || CacheRecordType == ZenContentType::kCbObject) + { + if (CacheRecordType == ZenContentType::kCbObject) + { + const CbValidateError ValidationResult = ValidateCompactBinary(UpstreamResult.Value, CbValidateMode::All); - CbPackage Package; - if (Package.TryLoad(UpstreamResult.Value)) - { - uint32_t AttachmentCount = 0; - uint32_t FoundCount = 0; - CbObject CacheRecord = Package.GetObject(); - - CacheRecord.IterateAttachments( - [this, &Package, &Ref, &AttachmentCount, &FoundCount](CbFieldView AttachmentHash) { - if (const CbAttachment* Attachment = Package.FindAttachment(AttachmentHash.AsHash())) - { - if (CompressedBuffer Chunk = Attachment->AsCompressedBinary()) - { - m_CidStore.AddChunk(Chunk); - FoundCount++; - } - else - { - ZEN_WARN("GET - cache record '{}/{}' FAILED, upstream attachment not compressed", - Ref.BucketSegment, - Ref.HashKey); - } - } - AttachmentCount++; - }); - - if (FoundCount == AttachmentCount) - { - m_CacheStore.Put(Ref.BucketSegment, Ref.HashKey, {.Value = CacheRecord.GetBuffer().AsIoBuffer()}); - - if (zen::CachePolicy::SkipAttachments == (Policy & zen::CachePolicy::SkipAttachments)) - { - CbPackage PackageWithoutAttachments; - PackageWithoutAttachments.SetObject(CacheRecord); - - MemoryOutStream MemStream; - BinaryWriter Writer(MemStream); - PackageWithoutAttachments.Save(Writer); - - Value.Value = IoBuffer(IoBuffer::Clone, MemStream.Data(), MemStream.Size()); - } - } - else - { - Success = false; - ZEN_WARN("GET - cache record '{}/{}' FAILED, attachments missing in upstream package", - Ref.BucketSegment, - Ref.HashKey); - } - } - else - { - Success = false; - ZEN_WARN("GET - cache record '{}/{}' FAILED, invalid upstream package", Ref.BucketSegment, Ref.HashKey); - } - } - } - } + if (ValidationResult == CbValidateError::None) + { + CbObjectView CacheRecord(UpstreamResult.Value.Data()); - if (!Success) - { - ZEN_DEBUG("MISS - '{}/{}'", Ref.BucketSegment, Ref.HashKey); + CbObjectWriter IndexData; + IndexData.BeginArray("references"); + CacheRecord.IterateAttachments([&](CbFieldView Attachment) { IndexData.AddHash(Attachment.AsHash()); }); + IndexData.EndArray(); - return Request.WriteResponse(HttpResponseCode::NotFound); + Value.IndexData = IndexData.Save(); + } + else + { + Success = false; + ZEN_WARN("Get - cache record '{}/{}' FAILED, invalid compact binary object from upstream", + Ref.BucketSegment, + Ref.HashKey); + } } - if (Verb == kHead) + if (Success) { - Request.SetSuppressResponseBody(); + m_CacheStore.Put(Ref.BucketSegment, Ref.HashKey, Value); } + } + else + { + ZEN_ASSERT(CacheRecordType == ZenContentType::kCbPackage); - if (AcceptType == ZenContentType::kCbPackage && !InUpstreamCache) + CbPackage Package; + if (Package.TryLoad(UpstreamResult.Value)) { - CbObjectView CacheRecord(Value.Value.Data()); + uint32_t AttachmentCount = 0; + uint32_t ValidCount = 0; + CbObject CacheRecord = Package.GetObject(); - const CbValidateError ValidationResult = ValidateCompactBinary(Value.Value, CbValidateMode::All); + CacheRecord.IterateAttachments([this, &Package, &Ref, &AttachmentCount, &ValidCount](CbFieldView AttachmentHash) { + if (const CbAttachment* Attachment = Package.FindAttachment(AttachmentHash.AsHash())) + { + if (CompressedBuffer Chunk = Attachment->AsCompressedBinary()) + { + m_CidStore.AddChunk(Chunk); + ValidCount++; + } + else + { + ZEN_WARN("GET - cache record '{}/{}' FAILED, upstream attachment not compressed", + Ref.BucketSegment, + Ref.HashKey); + } + } + AttachmentCount++; + }); - if (ValidationResult != CbValidateError::None) + if (ValidCount == AttachmentCount) { - ZEN_WARN("GET - cache record '{}/{}' FAILED, invalid compact binary object", Ref.BucketSegment, Ref.HashKey); - - return Request.WriteResponse(HttpResponseCode::NotFound, HttpContentType::kText, "Invalid cache record"sv); - } - - const bool SkipAttachments = zen::CachePolicy::SkipAttachments == (Policy & zen::CachePolicy::SkipAttachments); - uint32_t AttachmentCount = 0; - uint32_t FoundCount = 0; - uint64_t AttachmentBytes = 0ull; + m_CacheStore.Put(Ref.BucketSegment, Ref.HashKey, {.Value = CacheRecord.GetBuffer().AsIoBuffer()}); - CbPackage Package; - - if (!SkipAttachments) - { - CacheRecord.IterateAttachments( - [this, &Ref, &Package, &AttachmentCount, &FoundCount, &AttachmentBytes](CbFieldView AttachmentHash) { - if (IoBuffer Chunk = m_CidStore.FindChunkByCid(AttachmentHash.AsHash())) - { - Package.AddAttachment(CbAttachment(CompressedBuffer::FromCompressed(SharedBuffer(Chunk)))); - AttachmentBytes += Chunk.Size(); - FoundCount++; - } - AttachmentCount++; - }); - - if (FoundCount != AttachmentCount) + if (zen::CachePolicy::SkipAttachments == (Policy & zen::CachePolicy::SkipAttachments)) { - ZEN_WARN("GET - cache record '{}/{}' FAILED, found '{}' of '{}' attachments", - Ref.BucketSegment, - Ref.HashKey, - FoundCount, - AttachmentCount); + CbPackage PackageWithoutAttachments; + PackageWithoutAttachments.SetObject(CacheRecord); + + MemoryOutStream MemStream; + BinaryWriter Writer(MemStream); + PackageWithoutAttachments.Save(Writer); - return Request.WriteResponse(HttpResponseCode::NotFound, HttpContentType::kText, "Missing attachments"sv); + Value.Value = IoBuffer(IoBuffer::Clone, MemStream.Data(), MemStream.Size()); } } - - Package.SetObject(LoadCompactBinaryObject(Value.Value)); - - ZEN_DEBUG("HIT - '{}/{}' {}, {} attachments ({})", - Ref.BucketSegment, - Ref.HashKey, - NiceBytes(AttachmentBytes + Value.Value.Size()), - AttachmentCount, - InUpstreamCache ? "UPSTREAM" : "LOCAL"); - - MemoryOutStream MemStream; - BinaryWriter Writer(MemStream); - Package.Save(Writer); - - IoBuffer Response(IoBuffer::Clone, MemStream.Data(), MemStream.Size()); - - return Request.WriteResponse(HttpResponseCode::OK, HttpContentType::kCbPackage, Response); + else + { + Success = false; + ZEN_WARN("GET - cache record '{}/{}' FAILED, attachments missing in upstream package", + Ref.BucketSegment, + Ref.HashKey); + } } else { - ZEN_DEBUG("HIT - '{}/{}' {} ({})", - Ref.BucketSegment, - Ref.HashKey, - NiceBytes(Value.Value.Size()), - InUpstreamCache ? "UPSTREAM" : "LOCAL"); - - return Request.WriteResponse(HttpResponseCode::OK, Value.Value.GetContentType(), Value.Value); + Success = false; + ZEN_WARN("GET - cache record '{}/{}' FAILED, invalid upstream package", Ref.BucketSegment, Ref.HashKey); } } - break; + } + } - case kPut: - { - IoBuffer Body = Request.ReadPayload(); + if (!Success) + { + ZEN_DEBUG("MISS - '{}/{}'", Ref.BucketSegment, Ref.HashKey); - if (!Body || Body.Size() == 0) - { - return Request.WriteResponse(HttpResponseCode::BadRequest); - } + return Request.WriteResponse(HttpResponseCode::NotFound); + } - const HttpContentType ContentType = Request.RequestContentType(); + if (AcceptType == ZenContentType::kCbPackage && !InUpstreamCache) + { + CbObjectView CacheRecord(Value.Value.Data()); - const bool StoreUpstream = m_UpstreamCache && (CachePolicy::StoreRemote == (Policy & CachePolicy::StoreRemote)); + const CbValidateError ValidationResult = ValidateCompactBinary(Value.Value, CbValidateMode::All); - if (ContentType == HttpContentType::kBinary || ContentType == HttpContentType::kUnknownContentType) - { - // TODO: create a cache record and put value in CAS? - m_CacheStore.Put(Ref.BucketSegment, Ref.HashKey, {.Value = Body}); - ZEN_DEBUG("PUT - binary '{}/{}' {}", Ref.BucketSegment, Ref.HashKey, NiceBytes(Body.Size())); + if (ValidationResult != CbValidateError::None) + { + ZEN_WARN("GET - cache record '{}/{}' FAILED, invalid compact binary object", Ref.BucketSegment, Ref.HashKey); + return Request.WriteResponse(HttpResponseCode::NotFound, HttpContentType::kText, "Invalid cache record"sv); + } + + const bool SkipAttachments = zen::CachePolicy::SkipAttachments == (Policy & zen::CachePolicy::SkipAttachments); + uint32_t AttachmentCount = 0; + uint32_t ValidCount = 0; + uint64_t AttachmentBytes = 0ull; - if (StoreUpstream) + CbPackage Package; + + if (!SkipAttachments) + { + CacheRecord.IterateAttachments( + [this, &Ref, &Package, &AttachmentCount, &ValidCount, &AttachmentBytes](CbFieldView AttachmentHash) { + if (IoBuffer Chunk = m_CidStore.FindChunkByCid(AttachmentHash.AsHash())) { - auto Result = m_UpstreamCache->EnqueueUpstream( - {.Type = ZenContentType::kBinary, .CacheKey = {Ref.BucketSegment, Ref.HashKey}}); + Package.AddAttachment(CbAttachment(CompressedBuffer::FromCompressed(SharedBuffer(Chunk)))); + AttachmentBytes += Chunk.Size(); + ValidCount++; } + AttachmentCount++; + }); - return Request.WriteResponse(HttpResponseCode::Created); - } - else if (ContentType == HttpContentType::kCbObject) - { - // Validate payload before accessing it - const CbValidateError ValidationResult = - ValidateCompactBinary(MemoryView(Body.Data(), Body.Size()), CbValidateMode::All); + if (ValidCount != AttachmentCount) + { + ZEN_WARN("GET - cache record '{}/{}' FAILED, found '{}' of '{}' attachments", + Ref.BucketSegment, + Ref.HashKey, + ValidCount, + AttachmentCount); - if (ValidationResult != CbValidateError::None) - { - ZEN_WARN("PUT - cache record '{}/{}' ({} bytes) FAILED, invalid compact binary", - Ref.BucketSegment, - Ref.HashKey, - Body.Size()); + return Request.WriteResponse(HttpResponseCode::NotFound, HttpContentType::kText, "Missing attachments"sv); + } + } - // TODO: add details in response, kText || kCbObject? - return Request.WriteResponse(HttpResponseCode::BadRequest, - HttpContentType::kText, - "Compact binary validation failed"sv); - } + Package.SetObject(LoadCompactBinaryObject(Value.Value)); - // Extract referenced payload hashes - CbObjectView Cbo(Body.Data()); + ZEN_DEBUG("HIT - '{}/{}' {}, {} attachments (LOCAL)", + Ref.BucketSegment, + Ref.HashKey, + NiceBytes(AttachmentBytes + Value.Value.Size()), + AttachmentCount); - std::vector<IoHash> References; - std::vector<IoHash> MissingRefs; - Cbo.IterateAttachments([&](CbFieldView AttachmentView) { References.push_back(AttachmentView.AsHash()); }); + MemoryOutStream MemStream; + BinaryWriter Writer(MemStream); + Package.Save(Writer); - ZenCacheValue CacheValue; - CacheValue.Value = Body; + IoBuffer Response(IoBuffer::Clone, MemStream.Data(), MemStream.Size()); - if (!References.empty()) - { - CbObjectWriter Idx; - Idx.BeginArray("references"); + Request.WriteResponse(HttpResponseCode::OK, HttpContentType::kCbPackage, Response); + } + else + { + ZEN_DEBUG("HIT - '{}/{}' {} ({})", + Ref.BucketSegment, + Ref.HashKey, + NiceBytes(Value.Value.Size()), + InUpstreamCache ? "UPSTREAM" : "LOCAL"); - for (const IoHash& Hash : References) - { - Idx.AddHash(Hash); - if (!m_CidStore.ContainsChunk(Hash)) - { - MissingRefs.push_back(Hash); - } - } + Request.WriteResponse(HttpResponseCode::OK, Value.Value.GetContentType(), Value.Value); + } +} - Idx.EndArray(); +void +HttpStructuredCacheService::HandlePutCacheRecord(zen::HttpServerRequest& Request, const CacheRef& Ref, CachePolicy Policy) +{ + IoBuffer Body = Request.ReadPayload(); - CacheValue.IndexData = Idx.Save(); - } + if (!Body || Body.Size() == 0) + { + return Request.WriteResponse(HttpResponseCode::BadRequest); + } - m_CacheStore.Put(Ref.BucketSegment, Ref.HashKey, CacheValue); + const HttpContentType ContentType = Request.RequestContentType(); + const bool StoreUpstream = m_UpstreamCache && (CachePolicy::StoreRemote == (Policy & CachePolicy::StoreRemote)); - ZEN_DEBUG("PUT - cache record '{}/{}' {}, {}/{} attachments missing", - Ref.BucketSegment, - Ref.HashKey, - NiceBytes(CacheValue.Value.Size()), - MissingRefs.size(), - References.size()); + if (ContentType == HttpContentType::kBinary || ContentType == HttpContentType::kUnknownContentType) + { + m_CacheStore.Put(Ref.BucketSegment, Ref.HashKey, {.Value = Body}); + ZEN_DEBUG("PUT - binary '{}/{}' {}", Ref.BucketSegment, Ref.HashKey, NiceBytes(Body.Size())); - if (MissingRefs.empty() && StoreUpstream) - { - ZEN_ASSERT(m_UpstreamCache); - auto Result = m_UpstreamCache->EnqueueUpstream({.Type = ZenContentType::kCbObject, - .CacheKey = {Ref.BucketSegment, Ref.HashKey}, - .PayloadIds = std::move(References)}); + if (StoreUpstream) + { + ZEN_ASSERT(m_UpstreamCache); + auto Result = m_UpstreamCache->EnqueueUpstream({.Type = ZenContentType::kBinary, .CacheKey = {Ref.BucketSegment, Ref.HashKey}}); + } - return Request.WriteResponse(HttpResponseCode::Created); - } - else - { - // TODO: Binary attachments? - CbObjectWriter Response; - Response.BeginArray("needs"); - for (const IoHash& MissingRef : MissingRefs) - { - Response.AddHash(MissingRef); - ZEN_DEBUG("cache record '{}/{}' is missing reference '{}'", Ref.BucketSegment, Ref.HashKey, MissingRef); - } - Response.EndArray(); + Request.WriteResponse(HttpResponseCode::Created); + } + else if (ContentType == HttpContentType::kCbObject) + { + const CbValidateError ValidationResult = ValidateCompactBinary(MemoryView(Body.GetData(), Body.GetSize()), CbValidateMode::All); - // Return Created | BadRequest? - return Request.WriteResponse(HttpResponseCode::Created, Response.Save()); - } - } - else if (ContentType == HttpContentType::kCbPackage) - { - CbPackage Package; + if (ValidationResult != CbValidateError::None) + { + ZEN_WARN("PUT - cache record '{}/{}' ({} bytes) FAILED, invalid compact binary", Ref.BucketSegment, Ref.HashKey, Body.Size()); + return Request.WriteResponse(HttpResponseCode::BadRequest, HttpContentType::kText, "Compact binary validation failed"sv); + } - if (!Package.TryLoad(Body)) - { - ZEN_WARN("PUT - cache record '{}/{}' FAILED, invalid package", Ref.BucketSegment, Ref.HashKey); - return Request.WriteResponse(HttpResponseCode::BadRequest, HttpContentType::kText, "Invalid package"); - } + CbObjectView CacheRecord(Body.Data()); + std::vector<IoHash> ValidAttachments; + uint32_t AttachmentCount = 0; - CbObject CacheRecord = Package.GetObject(); + CacheRecord.IterateAttachments([this, &AttachmentCount, &ValidAttachments](CbFieldView AttachmentHash) { + const IoHash Hash = AttachmentHash.AsHash(); + if (m_CidStore.ContainsChunk(Hash)) + { + ValidAttachments.emplace_back(Hash); + } + AttachmentCount++; + }); - struct AttachmentInsertResult - { - int32_t Count = 0; - int32_t NewCount = 0; - uint64_t Bytes = 0; - uint64_t NewBytes = 0; - bool Ok = false; - }; + const uint32_t ValidCount = static_cast<uint32_t>(ValidAttachments.size()); + const bool ValidCacheRecord = ValidCount == AttachmentCount; - AttachmentInsertResult AttachmentResult{.Ok = true}; - std::span<const CbAttachment> Attachments = Package.GetAttachments(); - std::vector<IoHash> PayloadIds; + if (ValidCacheRecord) + { + ZEN_DEBUG("PUT - cache record '{}/{}' {}, {} attachments", Ref.BucketSegment, Ref.HashKey, NiceBytes(Body.Size()), ValidCount); + m_CacheStore.Put(Ref.BucketSegment, Ref.HashKey, {.Value = Body}); - PayloadIds.reserve(Attachments.size()); + if (StoreUpstream) + { + ZEN_ASSERT(m_UpstreamCache); + auto Result = m_UpstreamCache->EnqueueUpstream({.Type = ZenContentType::kCbObject, + .CacheKey = {Ref.BucketSegment, Ref.HashKey}, + .PayloadIds = std::move(ValidAttachments)}); + } - CacheRecord.IterateAttachments([this, &Ref, &Package, &AttachmentResult, &PayloadIds](CbFieldView AttachmentHash) { - if (const CbAttachment* Attachment = Package.FindAttachment(AttachmentHash.AsHash())) - { - if (Attachment->IsCompressedBinary()) - { - CompressedBuffer Chunk = Attachment->AsCompressedBinary(); - const uint64_t ChunkSize = Chunk.GetCompressed().GetSize(); - CidStore::InsertResult InsertResult = m_CidStore.AddChunk(Chunk); + Request.WriteResponse(HttpResponseCode::Created); + } + else + { + ZEN_WARN("PUT - cache record '{}/{}' FAILED, found {}/{} attachments", + Ref.BucketSegment, + Ref.HashKey, + ValidCount, + AttachmentCount); - PayloadIds.emplace_back(InsertResult.DecompressedId); + Request.WriteResponse(HttpResponseCode::BadRequest, HttpContentType::kText, "Missing attachments"sv); + } + } + else if (ContentType == HttpContentType::kCbPackage) + { + CbPackage Package; - if (InsertResult.New) - { - AttachmentResult.NewBytes += ChunkSize; - AttachmentResult.NewCount++; - } + if (!Package.TryLoad(Body)) + { + ZEN_WARN("PUT - cache record '{}/{}' FAILED, invalid package", Ref.BucketSegment, Ref.HashKey); + return Request.WriteResponse(HttpResponseCode::BadRequest, HttpContentType::kText, "Invalid package"sv); + } - AttachmentResult.Bytes += ChunkSize; - AttachmentResult.Count++; - } - else - { - ZEN_WARN("PUT - cache record '{}/{}' FAILED, attachment '{}' is not compressed", - Ref.BucketSegment, - Ref.HashKey, - AttachmentHash.AsHash()); - AttachmentResult.Ok = false; - } - } - else - { - ZEN_WARN("PUT - cache record '{}/{}' FAILED, missing attachment '{}'", - Ref.BucketSegment, - Ref.HashKey, - AttachmentHash.AsHash()); - AttachmentResult.Ok = false; - } - }); + CbObject CacheRecord = Package.GetObject(); - if (!AttachmentResult.Ok) - { - return Request.WriteResponse(HttpResponseCode::BadRequest, HttpContentType::kText, "Invalid attachments"); - } + std::span<const CbAttachment> Attachments = Package.GetAttachments(); + std::vector<IoHash> ValidAttachments; + int32_t NewAttachmentCount = 0; - IoBuffer CacheRecordChunk = CacheRecord.GetBuffer().AsIoBuffer(); - const uint64_t TotalPackageBytes = AttachmentResult.Bytes + CacheRecordChunk.Size(); + ValidAttachments.reserve(Attachments.size()); - ZenCacheValue CacheValue{.Value = CacheRecordChunk}; - m_CacheStore.Put(Ref.BucketSegment, Ref.HashKey, CacheValue); + CacheRecord.IterateAttachments([this, &Ref, &Package, &ValidAttachments, &NewAttachmentCount](CbFieldView AttachmentHash) { + if (const CbAttachment* Attachment = Package.FindAttachment(AttachmentHash.AsHash())) + { + if (Attachment->IsCompressedBinary()) + { + CompressedBuffer Chunk = Attachment->AsCompressedBinary(); + CidStore::InsertResult InsertResult = m_CidStore.AddChunk(Chunk); + + ValidAttachments.emplace_back(InsertResult.DecompressedId); - if (StoreUpstream) + if (InsertResult.New) { - ZEN_ASSERT(m_UpstreamCache); - auto Result = m_UpstreamCache->EnqueueUpstream({.Type = ZenContentType::kCbPackage, - .CacheKey = {Ref.BucketSegment, Ref.HashKey}, - .PayloadIds = std::move(PayloadIds)}); + NewAttachmentCount++; } - - ZEN_DEBUG("PUT - cache record '{}/{}' {}, {}/{} ({}/{}) new attachments", - Ref.BucketSegment, - Ref.HashKey, - NiceBytes(TotalPackageBytes), - AttachmentResult.NewCount, - AttachmentResult.Count, - NiceBytes(AttachmentResult.NewBytes), - NiceBytes(AttachmentResult.Bytes)); - - return Request.WriteResponse(HttpResponseCode::Created); } else { - return Request.WriteResponse(HttpResponseCode::BadRequest); + ZEN_WARN("PUT - cache record '{}/{}' FAILED, attachment '{}' is not compressed", + Ref.BucketSegment, + Ref.HashKey, + AttachmentHash.AsHash()); } } - break; + else + { + ZEN_WARN("PUT - cache record '{}/{}' FAILED, missing attachment '{}'", + Ref.BucketSegment, + Ref.HashKey, + AttachmentHash.AsHash()); + } + }); - case kPost: - break; + const bool AttachmentsValid = ValidAttachments.size() == Attachments.size(); - default: - break; + if (!AttachmentsValid) + { + return Request.WriteResponse(HttpResponseCode::BadRequest, HttpContentType::kText, "Invalid attachments"sv); + } + + ZEN_DEBUG("PUT - cache record '{}/{}' {}, {}/{} new attachments", + Ref.BucketSegment, + Ref.HashKey, + NiceBytes(Body.GetSize()), + NewAttachmentCount, + Attachments.size()); + + m_CacheStore.Put(Ref.BucketSegment, Ref.HashKey, {.Value = CacheRecord.GetBuffer().AsIoBuffer()}); + + if (StoreUpstream) + { + ZEN_ASSERT(m_UpstreamCache); + auto Result = m_UpstreamCache->EnqueueUpstream({.Type = ZenContentType::kCbPackage, + .CacheKey = {Ref.BucketSegment, Ref.HashKey}, + .PayloadIds = std::move(ValidAttachments)}); + } + + Request.WriteResponse(HttpResponseCode::Created); + } + else + { + Request.WriteResponse(HttpResponseCode::BadRequest); } } void -HttpStructuredCacheService::HandleCachePayloadRequest(HttpServerRequest& Request, CacheRef& Ref, CachePolicy Policy) +HttpStructuredCacheService::HandleCachePayloadRequest(HttpServerRequest& Request, const CacheRef& Ref, CachePolicy Policy) { - // Note: the URL references the uncompressed payload hash - so this maintains the mapping - // from uncompressed CAS identity (aka CID/Content ID) to the stored payload hash - // - // this is a PITA but a consequence of the fact that the client side code is not able to - // address data by compressed hash - - ZEN_UNUSED(Policy); - switch (auto Verb = Request.RequestVerb()) { using enum HttpVerb; @@ -693,112 +643,107 @@ HttpStructuredCacheService::HandleCachePayloadRequest(HttpServerRequest& Request case kHead: case kGet: { - IoBuffer Payload = m_CidStore.FindChunkByCid(Ref.PayloadId); - bool InUpstreamCache = false; - - if (!Payload && m_UpstreamCache) - { - if (auto UpstreamResult = m_UpstreamCache->GetCachePayload({{Ref.BucketSegment, Ref.HashKey}, Ref.PayloadId}); - UpstreamResult.Success) - { - if (CompressedBuffer Compressed = CompressedBuffer::FromCompressed(SharedBuffer(UpstreamResult.Value))) - { - Payload = UpstreamResult.Value; - IoHash ChunkHash = IoHash::HashBuffer(Payload); - CasStore::InsertResult Result = m_CasStore.InsertChunk(Payload, ChunkHash); - InUpstreamCache = true; - - m_CidStore.AddCompressedCid(Ref.PayloadId, ChunkHash); - } - else - { - ZEN_WARN("got uncompressed upstream cache payload"); - } - } - } - - if (!Payload) - { - ZEN_DEBUG("MISS - '{}/{}/{}'", Ref.BucketSegment, Ref.HashKey, Ref.PayloadId); - return Request.WriteResponse(HttpResponseCode::NotFound); - } - - ZEN_DEBUG("HIT - '{}/{}/{}' {} (type: {}) ({})", - Ref.BucketSegment, - Ref.HashKey, - Ref.PayloadId, - NiceBytes(Payload.Size()), - Payload.GetContentType(), - InUpstreamCache ? "UPSTREAM" : "LOCAL"); - + HandleGetCachePayload(Request, Ref, Policy); if (Verb == kHead) { Request.SetSuppressResponseBody(); } - - return Request.WriteResponse(HttpResponseCode::OK, HttpContentType::kBinary, Payload); } break; - case kPut: + HandlePutCachePayload(Request, Ref, Policy); + break; + default: + break; + } +} + +void +HttpStructuredCacheService::HandleGetCachePayload(zen::HttpServerRequest& Request, const CacheRef& Ref, CachePolicy Policy) +{ + IoBuffer Payload = m_CidStore.FindChunkByCid(Ref.PayloadId); + bool InUpstreamCache = false; + const bool QueryUpstream = !Payload && m_UpstreamCache && (CachePolicy::QueryRemote == (Policy & CachePolicy::QueryRemote)); + + if (QueryUpstream) + { + if (auto UpstreamResult = m_UpstreamCache->GetCachePayload({{Ref.BucketSegment, Ref.HashKey}, Ref.PayloadId}); + UpstreamResult.Success) + { + if (CompressedBuffer Compressed = CompressedBuffer::FromCompressed(SharedBuffer(UpstreamResult.Value))) { - if (IoBuffer Body = Request.ReadPayload()) - { - if (Body.Size() == 0) - { - return Request.WriteResponse(HttpResponseCode::BadRequest, HttpContentType::kText, "Empty payload not permitted"); - } + Payload = UpstreamResult.Value; + IoHash ChunkHash = IoHash::HashBuffer(Payload); + CasStore::InsertResult Result = m_CasStore.InsertChunk(Payload, ChunkHash); + InUpstreamCache = true; - IoHash ChunkHash = IoHash::HashBuffer(Body); + m_CidStore.AddCompressedCid(Ref.PayloadId, ChunkHash); + } + else + { + ZEN_WARN("got uncompressed upstream cache payload"); + } + } + } + + if (!Payload) + { + ZEN_DEBUG("MISS - '{}/{}/{}'", Ref.BucketSegment, Ref.HashKey, Ref.PayloadId); + return Request.WriteResponse(HttpResponseCode::NotFound); + } - CompressedBuffer Compressed = CompressedBuffer::FromCompressed(SharedBuffer(Body)); + ZEN_DEBUG("HIT - '{}/{}/{}' {} (type: {}) ({})", + Ref.BucketSegment, + Ref.HashKey, + Ref.PayloadId, + NiceBytes(Payload.Size()), + Payload.GetContentType(), + InUpstreamCache ? "UPSTREAM" : "LOCAL"); - if (!Compressed) - { - // All attachment payloads need to be in compressed buffer format - return Request.WriteResponse(HttpResponseCode::BadRequest, - HttpContentType::kText, - "Attachments must be compressed"); - } - else - { - if (IoHash::FromBLAKE3(Compressed.GetRawHash()) != Ref.PayloadId) - { - // the URL specified content id and content hashes don't match! - return Request.WriteResponse(HttpResponseCode::BadRequest); - } + Request.WriteResponse(HttpResponseCode::OK, HttpContentType::kBinary, Payload); +} - CasStore::InsertResult Result = m_CasStore.InsertChunk(Body, ChunkHash); +void +HttpStructuredCacheService::HandlePutCachePayload(zen::HttpServerRequest& Request, const CacheRef& Ref, CachePolicy Policy) +{ + // Note: Individual cache payloads are not propagated upstream until a valid cache record has been stored + ZEN_UNUSED(Policy); - m_CidStore.AddCompressedCid(Ref.PayloadId, ChunkHash); + IoBuffer Body = Request.ReadPayload(); - ZEN_DEBUG("PUT - payload '{}/{}/{}' {} (type: {}) {}", - Ref.BucketSegment, - Ref.HashKey, - Ref.PayloadId, - NiceBytes(Body.Size()), - Body.GetContentType(), - Result.New ? "NEW" : "OLD"); + if (!Body || Body.Size() == 0) + { + return Request.WriteResponse(HttpResponseCode::BadRequest); + } - if (Result.New) - { - return Request.WriteResponse(HttpResponseCode::Created); - } - else - { - return Request.WriteResponse(HttpResponseCode::OK); - } - } - } - } - break; + IoHash ChunkHash = IoHash::HashBuffer(Body); + CompressedBuffer Compressed = CompressedBuffer::FromCompressed(SharedBuffer(Body)); - case kPost: - break; + if (!Compressed) + { + return Request.WriteResponse(HttpResponseCode::BadRequest, HttpContentType::kText, "Attachments must be compressed"sv); + } - default: - break; + if (IoHash::FromBLAKE3(Compressed.GetRawHash()) != Ref.PayloadId) + { + return Request.WriteResponse(HttpResponseCode::BadRequest, HttpContentType::kText, "Payload ID does not match attachment hash"sv); } + + CasStore::InsertResult Result = m_CasStore.InsertChunk(Body, ChunkHash); + + m_CidStore.AddCompressedCid(Ref.PayloadId, ChunkHash); + + ZEN_DEBUG("PUT - payload '{}/{}/{}' {} (type: {}) {}", + Ref.BucketSegment, + Ref.HashKey, + Ref.PayloadId, + NiceBytes(Body.Size()), + Body.GetContentType(), + Result.New ? "NEW" : "OLD"); + + const HttpResponseCode ResponseCode = Result.New ? HttpResponseCode::Created : HttpResponseCode::OK; + + Request.WriteResponse(ResponseCode); } bool diff --git a/zenserver/cache/structuredcache.h b/zenserver/cache/structuredcache.h index bd163dd1d..3fdaa1236 100644 --- a/zenserver/cache/structuredcache.h +++ b/zenserver/cache/structuredcache.h @@ -71,8 +71,12 @@ private: }; [[nodiscard]] bool ValidateKeyUri(zen::HttpServerRequest& Request, CacheRef& OutRef); - void HandleCacheRecordRequest(zen::HttpServerRequest& Request, CacheRef& Ref, CachePolicy Policy); - void HandleCachePayloadRequest(zen::HttpServerRequest& Request, CacheRef& Ref, CachePolicy Policy); + void HandleCacheRecordRequest(zen::HttpServerRequest& Request, const CacheRef& Ref, CachePolicy Policy); + void HandleGetCacheRecord(zen::HttpServerRequest& Request, const CacheRef& Ref, CachePolicy Policy); + void HandlePutCacheRecord(zen::HttpServerRequest& Request, const CacheRef& Ref, CachePolicy Policy); + void HandleCachePayloadRequest(zen::HttpServerRequest& Request, const CacheRef& Ref, CachePolicy Policy); + void HandleGetCachePayload(zen::HttpServerRequest& Request, const CacheRef& Ref, CachePolicy Policy); + void HandlePutCachePayload(zen::HttpServerRequest& Request, const CacheRef& Ref, CachePolicy Policy); void HandleCacheBucketRequest(zen::HttpServerRequest& Request, std::string_view Bucket); spdlog::logger& Log() { return m_Log; } @@ -81,6 +85,7 @@ private: zen::CasStore& m_CasStore; zen::CidStore& m_CidStore; std::unique_ptr<UpstreamCache> m_UpstreamCache; + uint64_t m_LastScrubTime = 0; }; } // namespace zen diff --git a/zenserver/cache/structuredcachestore.cpp b/zenserver/cache/structuredcachestore.cpp index 502ca6605..3d80bb14c 100644 --- a/zenserver/cache/structuredcachestore.cpp +++ b/zenserver/cache/structuredcachestore.cpp @@ -108,6 +108,13 @@ ZenCacheStore::Flush() void ZenCacheStore::Scrub(ScrubContext& Ctx) { + if (m_LastScrubTime == Ctx.ScrubTimestamp()) + { + return; + } + + m_LastScrubTime = Ctx.ScrubTimestamp(); + m_DiskLayer.Scrub(Ctx); m_MemLayer.Scrub(Ctx); } diff --git a/zenserver/cache/structuredcachestore.h b/zenserver/cache/structuredcachestore.h index fdf4a8cfe..2cc3abb53 100644 --- a/zenserver/cache/structuredcachestore.h +++ b/zenserver/cache/structuredcachestore.h @@ -112,6 +112,7 @@ private: ZenCacheMemoryLayer m_MemLayer; ZenCacheDiskLayer m_DiskLayer; uint64_t m_DiskLayerSizeThreshold = 4 * 1024; + uint64_t m_LastScrubTime = 0; }; /** Tracks cache entry access, stats and orchestrates cleanup activities diff --git a/zenserver/compute/apply.cpp b/zenserver/compute/apply.cpp index 3197eaee4..15d9e0141 100644 --- a/zenserver/compute/apply.cpp +++ b/zenserver/compute/apply.cpp @@ -375,7 +375,7 @@ HttpFunctionService::HttpFunctionService(CasStore& Store, CidStore& InCidStore, FunctionSpec.IterateAttachments([&](CbFieldView Field) { const IoHash Hash = Field.AsHash(); - ChunkSet.AddChunk(Hash); + ChunkSet.AddChunkToSet(Hash); }); // Note that we store executables uncompressed to make it @@ -399,16 +399,15 @@ HttpFunctionService::HttpFunctionService(CasStore& Store, CidStore& InCidStore, CbObjectWriter ResponseWriter; ResponseWriter.BeginArray("need"); - for (const IoHash& Hash : ChunkSet.GetChunkSet()) - { + ChunkSet.IterateChunks([&](const IoHash& Hash) { ZEN_DEBUG("worker {}: need chunk {}", WorkerId, Hash); ResponseWriter.AddHash(Hash); - } + }); ResponseWriter.EndArray(); - ZEN_DEBUG("worker {}: need {} attachments", WorkerId, ChunkSet.GetChunkSet().size()); + ZEN_DEBUG("worker {}: need {} attachments", WorkerId, ChunkSet.GetSize()); return HttpReq.WriteResponse(HttpResponseCode::NotFound, ResponseWriter.Save()); } diff --git a/zenserver/config.cpp b/zenserver/config.cpp index 164d2a792..c21638258 100644 --- a/zenserver/config.cpp +++ b/zenserver/config.cpp @@ -17,6 +17,8 @@ #include <zencore/logging.h> #include <sol/sol.hpp> +#include <conio.h> + #if ZEN_PLATFORM_WINDOWS // Used for getting My Documents for default data directory @@ -219,7 +221,8 @@ ParseGlobalCliOptions(int argc, char* argv[], ZenServerOptions& GlobalOptions, Z if (result.count("help")) { zen::logging::ConsoleLog().info("{}", options.help()); - + zen::logging::ConsoleLog().info("Press any key to exit!"); + _getch(); exit(0); } @@ -274,14 +277,24 @@ ParseServiceConfig(const std::filesystem::path& DataRoot, ZenServiceConfig& Serv try { - sol::load_result config = lua.load(std::string_view((const char*)LuaScript.Data(), LuaScript.Size()), "zencfg"); + sol::load_result config = lua.load(std::string_view((const char*)LuaScript.Data(), LuaScript.Size()), "zen_cfg"); + + if (!config.valid()) + { + sol::error err = config; + + std::string ErrorString = sol::to_string(config.status()); + + throw std::runtime_error("{} error: {}"_format(ErrorString, err.what())); + } + config(); } catch (std::exception& e) { - ZEN_ERROR("config script failure: {}", e.what()); + ZEN_ERROR("config failure: {}", e.what()); - throw std::runtime_error("fatal zen global config script ({}) failure: {}"_format(ConfigScript, e.what()).c_str()); + throw std::runtime_error("failed to run global config script ('{}'): {}"_format(ConfigScript, e.what()).c_str()); } ServiceConfig.MeshEnabled = lua["mesh"]["enable"].get_or(ServiceConfig.MeshEnabled); diff --git a/zenserver/resource.h b/zenserver/resource.h new file mode 100644 index 000000000..f2e3b471b --- /dev/null +++ b/zenserver/resource.h @@ -0,0 +1,18 @@ +// Copyright Epic Games, Inc. All Rights Reserved. + +//{{NO_DEPENDENCIES}} +// Microsoft Visual C++ generated include file. +// Used by zenserver.rc +// +#define IDI_ICON1 101 + +// Next default values for new objects +// +#ifdef APSTUDIO_INVOKED +# ifndef APSTUDIO_READONLY_SYMBOLS +# define _APS_NEXT_RESOURCE_VALUE 102 +# define _APS_NEXT_COMMAND_VALUE 40001 +# define _APS_NEXT_CONTROL_VALUE 1001 +# define _APS_NEXT_SYMED_VALUE 101 +# endif +#endif diff --git a/zenserver/upstream/jupiter.cpp b/zenserver/upstream/jupiter.cpp index 2e74602db..14da8cbcc 100644 --- a/zenserver/upstream/jupiter.cpp +++ b/zenserver/upstream/jupiter.cpp @@ -90,6 +90,11 @@ CloudCacheSession::GetDerivedData(std::string_view BucketId, std::string_view Ke cpr::Response Response = Session.Get(); ZEN_DEBUG("GET {}", Response); + if (Response.error) + { + return {.ErrorCode = static_cast<int32_t>(Response.error.code), .Reason = Response.error.message}; + } + const bool Success = Response.status_code == 200; const IoBuffer Buffer = Success ? IoBufferBuilder::MakeCloneFromMemory(Response.text.data(), Response.text.size()) : IoBuffer(); @@ -121,6 +126,11 @@ CloudCacheSession::GetRef(std::string_view BucketId, const IoHash& Key, ZenConte cpr::Response Response = Session.Get(); ZEN_DEBUG("GET {}", Response); + if (Response.error) + { + return {.ErrorCode = static_cast<int32_t>(Response.error.code), .Reason = Response.error.message}; + } + const bool Success = Response.status_code == 200; const IoBuffer Buffer = Success ? IoBufferBuilder::MakeCloneFromMemory(Response.text.data(), Response.text.size()) : IoBuffer(); @@ -144,6 +154,11 @@ CloudCacheSession::GetCompressedBlob(const IoHash& Key) cpr::Response Response = Session.Get(); ZEN_DEBUG("GET {}", Response); + if (Response.error) + { + return {.ErrorCode = static_cast<int32_t>(Response.error.code), .Reason = Response.error.message}; + } + const bool Success = Response.status_code == 200; const IoBuffer Buffer = Success ? IoBufferBuilder::MakeCloneFromMemory(Response.text.data(), Response.text.size()) : IoBuffer(); @@ -171,6 +186,11 @@ CloudCacheSession::PutDerivedData(std::string_view BucketId, std::string_view Ke cpr::Response Response = Session.Put(); ZEN_DEBUG("PUT {}", Response); + if (Response.error) + { + return {.ErrorCode = static_cast<int32_t>(Response.error.code), .Reason = Response.error.message}; + } + return {.Bytes = Response.uploaded_bytes, .ElapsedSeconds = Response.elapsed, .Success = (Response.status_code == 200 || Response.status_code == 201)}; @@ -204,6 +224,11 @@ CloudCacheSession::PutRef(std::string_view BucketId, const IoHash& Key, IoBuffer cpr::Response Response = Session.Put(); ZEN_DEBUG("PUT {}", Response); + if (Response.error) + { + return {.ErrorCode = static_cast<int32_t>(Response.error.code), .Reason = Response.error.message}; + } + return {.Bytes = Response.uploaded_bytes, .ElapsedSeconds = Response.elapsed, .Success = (Response.status_code == 200 || Response.status_code == 201)}; @@ -227,6 +252,11 @@ CloudCacheSession::PutCompressedBlob(const IoHash& Key, IoBuffer Blob) cpr::Response Response = Session.Put(); ZEN_DEBUG("PUT {}", Response); + if (Response.error) + { + return {.ErrorCode = static_cast<int32_t>(Response.error.code), .Reason = Response.error.message}; + } + return {.Bytes = Response.uploaded_bytes, .ElapsedSeconds = Response.elapsed, .Success = (Response.status_code == 200 || Response.status_code == 201)}; diff --git a/zenserver/upstream/jupiter.h b/zenserver/upstream/jupiter.h index 21217387c..94e7e7680 100644 --- a/zenserver/upstream/jupiter.h +++ b/zenserver/upstream/jupiter.h @@ -42,10 +42,12 @@ private: struct CloudCacheResult { - IoBuffer Response; - int64_t Bytes = {}; - double ElapsedSeconds = {}; - bool Success = false; + IoBuffer Response; + int64_t Bytes = {}; + double ElapsedSeconds = {}; + int32_t ErrorCode = {}; + std::string Reason; + bool Success = false; }; /** diff --git a/zenserver/upstream/upstreamcache.cpp b/zenserver/upstream/upstreamcache.cpp index d6b6d44be..a889fb984 100644 --- a/zenserver/upstream/upstreamcache.cpp +++ b/zenserver/upstream/upstreamcache.cpp @@ -23,7 +23,6 @@ #include <algorithm> #include <atomic> #include <deque> -#include <limits> #include <thread> #include <unordered_map> @@ -106,11 +105,23 @@ namespace detail { virtual ~JupiterUpstreamEndpoint() = default; - virtual bool Initialize() override + virtual bool IsHealthy() const override { return m_HealthOk.load(); } + + virtual UpstreamEndpointHealth CheckHealth() override { - CloudCacheSession Session(m_Client); - const CloudCacheResult Result = Session.Authenticate(); - return Result.Success; + try + { + CloudCacheSession Session(m_Client); + const CloudCacheResult Result = Session.Authenticate(); + + m_HealthOk = Result.ErrorCode == 0; + + return {.Reason = std::move(Result.Reason), .Ok = Result.Success}; + } + catch (std::exception& Err) + { + return {.Reason = Err.what(), .Ok = false}; + } } virtual std::string_view DisplayName() const override { return m_DisplayName; } @@ -143,6 +154,7 @@ namespace detail { CacheRecord.IterateAttachments([&Session, &Result, &Package](CbFieldView AttachmentHash) { CloudCacheResult AttachmentResult = Session.GetCompressedBlob(AttachmentHash.AsHash()); Result.ElapsedSeconds += AttachmentResult.ElapsedSeconds; + Result.ErrorCode = AttachmentResult.ErrorCode; if (CompressedBuffer Chunk = CompressedBuffer::FromCompressed(SharedBuffer(AttachmentResult.Response))) { @@ -169,14 +181,16 @@ namespace detail { } } + m_HealthOk = Result.ErrorCode == 0; + return {.Value = Result.Response, .Bytes = Result.Bytes, .ElapsedSeconds = Result.ElapsedSeconds, .Success = Result.Success}; } - catch (std::exception& e) + catch (std::exception& Err) { - return {.Reason = std::string(e.what()), .Success = false}; + return {.Error{.StatusCode = UpstreamStatusCode::Error, .Reason = Err.what()}}; } } @@ -187,14 +201,16 @@ namespace detail { CloudCacheSession Session(m_Client); const CloudCacheResult Result = Session.GetCompressedBlob(PayloadKey.PayloadId); + m_HealthOk = Result.ErrorCode == 0; + return {.Value = Result.Response, .Bytes = Result.Bytes, .ElapsedSeconds = Result.ElapsedSeconds, .Success = Result.Success}; } - catch (std::exception& e) + catch (std::exception& Err) { - return {.Reason = std::string(e.what()), .Success = false}; + return {.Error{.StatusCode = UpstreamStatusCode::Error, .Reason = Err.what()}}; } } @@ -277,6 +293,7 @@ namespace detail { bool m_UseLegacyDdc; std::string m_DisplayName; RefPtr<CloudCacheClient> m_Client; + std::atomic_bool m_HealthOk{false}; }; class ZenUpstreamEndpoint final : public UpstreamEndpoint @@ -285,27 +302,33 @@ namespace detail { ZenUpstreamEndpoint(std::string_view ServiceUrl) { using namespace fmt::literals; - m_DisplayName = "Zen - '{}'"_format(ServiceUrl); + m_DisplayName = "Zen - {}"_format(ServiceUrl); m_Client = new ZenStructuredCacheClient(ServiceUrl); } ~ZenUpstreamEndpoint() = default; - virtual bool Initialize() override + virtual bool IsHealthy() const override { return m_HealthOk; } + + virtual UpstreamEndpointHealth CheckHealth() override { try { ZenStructuredCacheSession Session(*m_Client); ZenCacheResult Result; + for (int32_t Attempt = 0, MaxAttempts = 3; Attempt < MaxAttempts && !Result.Success; ++Attempt) { Result = Session.SayHello(); } - return Result.Success; + + m_HealthOk = Result.ErrorCode == 0; + + return {.Reason = std::move(Result.Reason), .Ok = m_HealthOk}; } - catch (std::exception&) + catch (std::exception& Err) { - return false; + return {.Reason = Err.what(), .Ok = false}; } } @@ -318,14 +341,16 @@ namespace detail { ZenStructuredCacheSession Session(*m_Client); const ZenCacheResult Result = Session.GetCacheRecord(CacheKey.Bucket, CacheKey.Hash, Type); + m_HealthOk = Result.ErrorCode == 0; + return {.Value = Result.Response, .Bytes = Result.Bytes, .ElapsedSeconds = Result.ElapsedSeconds, .Success = Result.Success}; } - catch (std::exception& e) + catch (std::exception& Err) { - return {.Reason = std::string(e.what()), .Success = false}; + return {.Error{.StatusCode = UpstreamStatusCode::Error, .Reason = Err.what()}}; } } @@ -337,14 +362,16 @@ namespace detail { const ZenCacheResult Result = Session.GetCachePayload(PayloadKey.CacheKey.Bucket, PayloadKey.CacheKey.Hash, PayloadKey.PayloadId); + m_HealthOk = Result.ErrorCode == 0; + return {.Value = Result.Response, .Bytes = Result.Bytes, .ElapsedSeconds = Result.ElapsedSeconds, .Success = Result.Success}; } - catch (std::exception& e) + catch (std::exception& Err) { - return {.Reason = std::string(e.what()), .Success = false}; + return {.Error{.StatusCode = UpstreamStatusCode::Error, .Reason = Err.what()}}; } } @@ -390,6 +417,8 @@ namespace detail { CacheRecord.CacheKey.Hash, PackagePayload, CacheRecord.Type); + + m_HealthOk = Result.ErrorCode == 0; } TotalBytes = Result.Bytes; @@ -406,6 +435,8 @@ namespace detail { CacheRecord.CacheKey.Hash, CacheRecord.PayloadIds[Idx], Payloads[Idx]); + + m_HealthOk = Result.ErrorCode == 0; } TotalBytes += Result.Bytes; @@ -425,6 +456,8 @@ namespace detail { { Result = Session.PutCacheRecord(CacheRecord.CacheKey.Bucket, CacheRecord.CacheKey.Hash, RecordValue, CacheRecord.Type); + + m_HealthOk = Result.ErrorCode == 0; } TotalBytes += Result.Bytes; @@ -435,6 +468,7 @@ namespace detail { } catch (std::exception& e) { + m_HealthOk = false; return {.Reason = std::string(e.what()), .Success = false}; } } @@ -442,6 +476,7 @@ namespace detail { private: std::string m_DisplayName; RefPtr<ZenStructuredCacheClient> m_Client; + std::atomic_bool m_HealthOk{false}; }; } // namespace detail @@ -548,27 +583,35 @@ public: virtual bool Initialize() override { - auto NewEnd = std::remove_if(std::begin(m_Endpoints), std::end(m_Endpoints), [this](auto& Endpoint) { - const bool Ok = Endpoint->Initialize(); - ZEN_INFO("{} [{}]", Endpoint->DisplayName(), Ok ? "OK" : "FAILED"); - return !Ok; - }); + for (auto& Endpoint : m_Endpoints) + { + const UpstreamEndpointHealth Health = Endpoint->CheckHealth(); + if (Health.Ok) + { + ZEN_INFO("initialize endpoint '{}' OK", Endpoint->DisplayName()); + } + else + { + ZEN_WARN("initialize endpoint '{}' FAILED, reason '{}'", Endpoint->DisplayName(), Health.Reason); + } + } - m_Endpoints.erase(NewEnd, std::end(m_Endpoints)); - m_IsRunning = !m_Endpoints.empty(); + m_RunState.IsRunning = !m_Endpoints.empty(); - if (m_IsRunning) + if (m_RunState.IsRunning) { for (uint32_t Idx = 0; Idx < m_Options.ThreadCount; Idx++) { m_UpstreamThreads.emplace_back(&DefaultUpstreamCache::ProcessUpstreamQueue, this); } + + m_EndpointMonitorThread = std::thread(&DefaultUpstreamCache::MonitorEndpoints, this); } - return m_IsRunning; + return m_RunState.IsRunning; } - virtual void AddEndpoint(std::unique_ptr<UpstreamEndpoint> Endpoint) override { m_Endpoints.emplace_back(std::move(Endpoint)); } + virtual void RegisterEndpoint(std::unique_ptr<UpstreamEndpoint> Endpoint) override { m_Endpoints.emplace_back(std::move(Endpoint)); } virtual GetUpstreamCacheResult GetCacheRecord(UpstreamCacheKey CacheKey, ZenContentType Type) override { @@ -576,10 +619,13 @@ public: { for (auto& Endpoint : m_Endpoints) { - if (GetUpstreamCacheResult Result = Endpoint->GetCacheRecord(CacheKey, Type); Result.Success) + if (Endpoint->IsHealthy()) { - m_Stats.Add(*Endpoint, Result); - return Result; + if (GetUpstreamCacheResult Result = Endpoint->GetCacheRecord(CacheKey, Type); Result.Success) + { + m_Stats.Add(*Endpoint, Result); + return Result; + } } } } @@ -593,10 +639,13 @@ public: { for (auto& Endpoint : m_Endpoints) { - if (GetUpstreamCacheResult Result = Endpoint->GetCachePayload(PayloadKey); Result.Success) + if (Endpoint->IsHealthy()) { - m_Stats.Add(*Endpoint, Result); - return Result; + if (GetUpstreamCacheResult Result = Endpoint->GetCachePayload(PayloadKey); Result.Success) + { + m_Stats.Add(*Endpoint, Result); + return Result; + } } } } @@ -606,7 +655,7 @@ public: virtual EnqueueResult EnqueueUpstream(UpstreamCacheRecord CacheRecord) override { - if (m_IsRunning.load() && m_Options.WriteUpstream) + if (m_RunState.IsRunning && m_Options.WriteUpstream) { if (!m_UpstreamThreads.empty()) { @@ -655,18 +704,21 @@ private: for (auto& Endpoint : m_Endpoints) { - const PutUpstreamCacheResult Result = Endpoint->PutCacheRecord(CacheRecord, CacheValue.Value, std::span(Payloads)); - if (Result.Success) + if (Endpoint->IsHealthy()) { - m_Stats.Add(*Endpoint, Result); - } - else - { - ZEN_WARN("process upstream FAILED, '{}/{}' FAILED, endpoint '{}', reason: '{}'", - CacheRecord.CacheKey.Bucket, - CacheRecord.CacheKey.Hash, - Endpoint->DisplayName(), - Result.Reason); + const PutUpstreamCacheResult Result = Endpoint->PutCacheRecord(CacheRecord, CacheValue.Value, std::span(Payloads)); + if (Result.Success) + { + m_Stats.Add(*Endpoint, Result); + } + else + { + ZEN_WARN("process upstream FAILED, '{}/{}' FAILED, endpoint '{}', reason: '{}'", + CacheRecord.CacheKey.Bucket, + CacheRecord.CacheKey.Hash, + Endpoint->DisplayName(), + Result.Reason); + } } } } @@ -688,33 +740,82 @@ private: } } - if (!m_IsRunning.load()) + if (!m_RunState.IsRunning) { break; } } } + void MonitorEndpoints() + { + for (;;) + { + { + std::unique_lock lk(m_RunState.Mutex); + if (m_RunState.ExitSignal.wait_for(lk, m_Options.HealthCheckInterval, [this]() { return !m_RunState.IsRunning.load(); })) + { + break; + } + } + + for (auto& Endpoint : m_Endpoints) + { + if (!Endpoint->IsHealthy()) + { + if (const UpstreamEndpointHealth Health = Endpoint->CheckHealth(); Health.Ok) + { + ZEN_INFO("health check endpoint '{}' OK", Endpoint->DisplayName(), Health.Reason); + } + else + { + ZEN_WARN("health check endpoint '{}' FAILED, reason '{}'", Endpoint->DisplayName(), Health.Reason); + } + } + } + } + } + void Shutdown() { - if (m_IsRunning.load()) + if (m_RunState.Stop()) { - m_IsRunning.store(false); m_UpstreamQueue.CompleteAdding(); - for (std::thread& Thread : m_UpstreamThreads) { Thread.join(); } + m_EndpointMonitorThread.join(); m_UpstreamThreads.clear(); m_Endpoints.clear(); } } + spdlog::logger& Log() { return m_Log; } + using UpstreamQueue = detail::BlockingQueue<UpstreamCacheRecord>; - spdlog::logger& Log() { return m_Log; } + struct RunState + { + std::mutex Mutex; + std::condition_variable ExitSignal; + std::atomic_bool IsRunning{false}; + + bool Stop() + { + bool Stopped = false; + { + std::lock_guard _(Mutex); + Stopped = IsRunning.exchange(false); + } + if (Stopped) + { + ExitSignal.notify_all(); + } + return Stopped; + } + }; spdlog::logger& m_Log; UpstreamCacheOptions m_Options; @@ -724,7 +825,8 @@ private: UpstreamStats m_Stats; std::vector<std::unique_ptr<UpstreamEndpoint>> m_Endpoints; std::vector<std::thread> m_UpstreamThreads; - std::atomic_bool m_IsRunning{false}; + std::thread m_EndpointMonitorThread; + RunState m_RunState; }; ////////////////////////////////////////////////////////////////////////// diff --git a/zenserver/upstream/upstreamcache.h b/zenserver/upstream/upstreamcache.h index 142fe260f..96ee8bddc 100644 --- a/zenserver/upstream/upstreamcache.h +++ b/zenserver/upstream/upstreamcache.h @@ -6,6 +6,7 @@ #include <zencore/iohash.h> #include <zencore/zencore.h> +#include <chrono> #include <memory> namespace zen { @@ -35,18 +36,33 @@ struct UpstreamCacheRecord struct UpstreamCacheOptions { - uint32_t ThreadCount = 4; - bool ReadUpstream = true; - bool WriteUpstream = true; + std::chrono::seconds HealthCheckInterval{5}; + uint32_t ThreadCount = 4; + bool ReadUpstream = true; + bool WriteUpstream = true; +}; + +enum class UpstreamStatusCode : uint8_t +{ + Ok, + Error +}; + +struct UpstreamError +{ + UpstreamStatusCode StatusCode = UpstreamStatusCode::Ok; + std::string Reason; + + explicit operator bool() const { return StatusCode != UpstreamStatusCode::Ok; } }; struct GetUpstreamCacheResult { - IoBuffer Value; - std::string Reason; - int64_t Bytes = {}; - double ElapsedSeconds = {}; - bool Success = false; + IoBuffer Value; + UpstreamError Error; + int64_t Bytes = {}; + double ElapsedSeconds = {}; + bool Success = false; }; struct PutUpstreamCacheResult @@ -57,6 +73,12 @@ struct PutUpstreamCacheResult bool Success = false; }; +struct UpstreamEndpointHealth +{ + std::string Reason; + bool Ok = false; +}; + /** * The upstream endpont is responsible for handling upload/downloading of cache records. */ @@ -65,7 +87,9 @@ class UpstreamEndpoint public: virtual ~UpstreamEndpoint() = default; - virtual bool Initialize() = 0; + virtual bool IsHealthy() const = 0; + + virtual UpstreamEndpointHealth CheckHealth() = 0; virtual std::string_view DisplayName() const = 0; @@ -88,7 +112,7 @@ public: virtual bool Initialize() = 0; - virtual void AddEndpoint(std::unique_ptr<UpstreamEndpoint> Endpoint) = 0; + virtual void RegisterEndpoint(std::unique_ptr<UpstreamEndpoint> Endpoint) = 0; virtual GetUpstreamCacheResult GetCacheRecord(UpstreamCacheKey CacheKey, ZenContentType Type) = 0; diff --git a/zenserver/upstream/zen.cpp b/zenserver/upstream/zen.cpp index 7f689d7f3..710d381c6 100644 --- a/zenserver/upstream/zen.cpp +++ b/zenserver/upstream/zen.cpp @@ -391,6 +391,11 @@ ZenStructuredCacheSession::SayHello() Session.SetOption(cpr::Url{Uri.c_str()}); cpr::Response Response = Session.Get(); + if (Response.error) + { + return {.ErrorCode = static_cast<int32_t>(Response.error.code), .Reason = std::move(Response.error.message)}; + } + return {.Bytes = Response.downloaded_bytes, .ElapsedSeconds = Response.elapsed, .Success = Response.status_code == 200}; } @@ -411,6 +416,11 @@ ZenStructuredCacheSession::GetCacheRecord(std::string_view BucketId, const IoHas cpr::Response Response = Session.Get(); ZEN_DEBUG("GET {}", Response); + if (Response.error) + { + return {.ErrorCode = static_cast<int32_t>(Response.error.code), .Reason = std::move(Response.error.message)}; + } + const bool Success = Response.status_code == 200; const IoBuffer Buffer = Success ? IoBufferBuilder::MakeCloneFromMemory(Response.text.data(), Response.text.size()) : IoBuffer(); @@ -431,6 +441,11 @@ ZenStructuredCacheSession::GetCachePayload(std::string_view BucketId, const IoHa cpr::Response Response = Session.Get(); ZEN_DEBUG("GET {}", Response); + if (Response.error) + { + return {.ErrorCode = static_cast<int32_t>(Response.error.code), .Reason = std::move(Response.error.message)}; + } + const bool Success = Response.status_code == 200; const IoBuffer Buffer = Success ? IoBufferBuilder::MakeCloneFromMemory(Response.text.data(), Response.text.size()) : IoBuffer(); @@ -455,6 +470,11 @@ ZenStructuredCacheSession::PutCacheRecord(std::string_view BucketId, const IoHas cpr::Response Response = Session.Put(); ZEN_DEBUG("PUT {}", Response); + if (Response.error) + { + return {.ErrorCode = static_cast<int32_t>(Response.error.code), .Reason = std::move(Response.error.message)}; + } + return {.Bytes = Response.uploaded_bytes, .ElapsedSeconds = Response.elapsed, .Success = (Response.status_code == 200 || Response.status_code == 201)}; @@ -475,6 +495,11 @@ ZenStructuredCacheSession::PutCachePayload(std::string_view BucketId, const IoHa cpr::Response Response = Session.Put(); ZEN_DEBUG("PUT {}", Response); + if (Response.error) + { + return {.ErrorCode = static_cast<int32_t>(Response.error.code), .Reason = std::move(Response.error.message)}; + } + return {.Bytes = Response.uploaded_bytes, .ElapsedSeconds = Response.elapsed, .Success = (Response.status_code == 200 || Response.status_code == 201)}; diff --git a/zenserver/upstream/zen.h b/zenserver/upstream/zen.h index 36cfd1217..48886096d 100644 --- a/zenserver/upstream/zen.h +++ b/zenserver/upstream/zen.h @@ -91,10 +91,12 @@ namespace detail { struct ZenCacheResult { - IoBuffer Response; - int64_t Bytes = {}; - double ElapsedSeconds = {}; - bool Success = false; + IoBuffer Response; + int64_t Bytes = {}; + double ElapsedSeconds = {}; + int32_t ErrorCode = {}; + std::string Reason; + bool Success = false; }; /** Zen Structured Cache session diff --git a/zenserver/zenserver.cpp b/zenserver/zenserver.cpp index cf24dc224..e3b61568f 100644 --- a/zenserver/zenserver.cpp +++ b/zenserver/zenserver.cpp @@ -193,7 +193,7 @@ public: if (!UpstreamConfig.ZenConfig.Url.empty()) { std::unique_ptr<zen::UpstreamEndpoint> ZenEndpoint = zen::MakeZenUpstreamEndpoint(UpstreamConfig.ZenConfig.Url); - UpstreamCache->AddEndpoint(std::move(ZenEndpoint)); + UpstreamCache->RegisterEndpoint(std::move(ZenEndpoint)); } { @@ -221,7 +221,7 @@ public: if (!Options.ServiceUrl.empty()) { std::unique_ptr<zen::UpstreamEndpoint> JupiterEndpoint = zen::MakeJupiterUpstreamEndpoint(Options); - UpstreamCache->AddEndpoint(std::move(JupiterEndpoint)); + UpstreamCache->RegisterEndpoint(std::move(JupiterEndpoint)); } } diff --git a/zenserver/zenserver.rc b/zenserver/zenserver.rc new file mode 100644 index 000000000..c063436ef --- /dev/null +++ b/zenserver/zenserver.rc @@ -0,0 +1,81 @@ +// Microsoft Visual C++ generated resource script. +// +#include "resource.h" + +#define APSTUDIO_READONLY_SYMBOLS +///////////////////////////////////////////////////////////////////////////// +// +// Generated from the TEXTINCLUDE 2 resource. +// +#include "winres.h" + +///////////////////////////////////////////////////////////////////////////// +#undef APSTUDIO_READONLY_SYMBOLS + +///////////////////////////////////////////////////////////////////////////// +// English (United States) resources + +#if !defined(AFX_RESOURCE_DLL) || defined(AFX_TARG_ENU) +LANGUAGE LANG_ENGLISH, SUBLANG_ENGLISH_US +#pragma code_page(1252) + +///////////////////////////////////////////////////////////////////////////// +// +// Icon +// + +// Icon with lowest ID value placed first to ensure application icon +// remains consistent on all systems. +IDI_ICON1 ICON "..\\UnrealEngine.ico" + +#endif // English (United States) resources +///////////////////////////////////////////////////////////////////////////// + + +///////////////////////////////////////////////////////////////////////////// +// English (United Kingdom) resources + +#if !defined(AFX_RESOURCE_DLL) || defined(AFX_TARG_ENG) +LANGUAGE LANG_ENGLISH, SUBLANG_ENGLISH_UK +#pragma code_page(1252) + +#ifdef APSTUDIO_INVOKED +///////////////////////////////////////////////////////////////////////////// +// +// TEXTINCLUDE +// + +1 TEXTINCLUDE +BEGIN + "resource.h\0" +END + +2 TEXTINCLUDE +BEGIN + "#include ""winres.h""\r\n" + "\0" +END + +3 TEXTINCLUDE +BEGIN + "\r\n" + "\0" +END + +#endif // APSTUDIO_INVOKED + +#endif // English (United Kingdom) resources +///////////////////////////////////////////////////////////////////////////// + + + +#ifndef APSTUDIO_INVOKED +///////////////////////////////////////////////////////////////////////////// +// +// Generated from the TEXTINCLUDE 3 resource. +// + + +///////////////////////////////////////////////////////////////////////////// +#endif // not APSTUDIO_INVOKED + diff --git a/zenserver/zenserver.vcxproj b/zenserver/zenserver.vcxproj index db657d192..1671d98a6 100644 --- a/zenserver/zenserver.vcxproj +++ b/zenserver/zenserver.vcxproj @@ -109,6 +109,7 @@ <ClInclude Include="compute\apply.h" /> <ClInclude Include="config.h" /> <ClInclude Include="diag\logging.h" /> + <ClInclude Include="resource.h" /> <ClInclude Include="sos\sos.h" /> <ClInclude Include="testing\httptest.h" /> <ClInclude Include="upstream\jupiter.h" /> @@ -163,6 +164,12 @@ <ItemGroup> <None Include="xmake.lua" /> </ItemGroup> + <ItemGroup> + <ResourceCompile Include="zenserver.rc" /> + </ItemGroup> + <ItemGroup> + <Image Include="..\UnrealEngine.ico" /> + </ItemGroup> <Import Project="$(VCTargetsPath)\Microsoft.Cpp.targets" /> <ImportGroup Label="ExtensionTargets"> </ImportGroup> diff --git a/zenserver/zenserver.vcxproj.filters b/zenserver/zenserver.vcxproj.filters index 250c55812..c51a8eb76 100644 --- a/zenserver/zenserver.vcxproj.filters +++ b/zenserver/zenserver.vcxproj.filters @@ -40,6 +40,7 @@ </ClInclude> <ClInclude Include="testing\httptest.h" /> <ClInclude Include="windows\service.h" /> + <ClInclude Include="resource.h" /> </ItemGroup> <ItemGroup> <ClCompile Include="zenserver.cpp" /> @@ -96,4 +97,10 @@ <ItemGroup> <None Include="xmake.lua" /> </ItemGroup> + <ItemGroup> + <ResourceCompile Include="zenserver.rc" /> + </ItemGroup> + <ItemGroup> + <Image Include="..\UnrealEngine.ico" /> + </ItemGroup> </Project>
\ No newline at end of file |