-
Notifications
You must be signed in to change notification settings - Fork 4
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
21 changed files
with
1,222 additions
and
298 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,165 @@ | ||
| using CodexOpenApi; | ||
| using CodexPlugin; | ||
| using Logging; | ||
| using Newtonsoft.Json; | ||
| using Utils; | ||
|
|
||
| namespace AutoClient | ||
| { | ||
| public class AutomaticPurchaser | ||
| { | ||
| private readonly ILog log; | ||
| private readonly ICodexInstance instance; | ||
| private readonly CodexNode codex; | ||
| private Task workerTask = Task.CompletedTask; | ||
| private App app => instance.App; | ||
|
|
||
| public AutomaticPurchaser(ILog log, ICodexInstance instance, CodexNode codex) | ||
| { | ||
| this.log = log; | ||
| this.instance = instance; | ||
| this.codex = codex; | ||
| } | ||
|
|
||
| public void Start() | ||
| { | ||
| workerTask = Task.Run(Worker); | ||
| } | ||
|
|
||
| public void Stop() | ||
| { | ||
| workerTask.Wait(); | ||
| } | ||
|
|
||
| private async Task Worker() | ||
| { | ||
| log.Log("Worker started."); | ||
| while (!app.Cts.Token.IsCancellationRequested) | ||
| { | ||
| try | ||
| { | ||
| var pid = await StartNewPurchase(); | ||
| await WaitTillFinished(pid); | ||
| await DownloadForeignCid(); | ||
| } | ||
| catch (Exception ex) | ||
| { | ||
| log.Error("Worker failed with: " + ex); | ||
| await Task.Delay(TimeSpan.FromHours(6)); | ||
| } | ||
| } | ||
| } | ||
|
|
||
| private async Task DownloadForeignCid() | ||
| { | ||
| var cid = app.CidRepo.GetForeignCid(instance.NodeId); | ||
| if (cid == null) return; | ||
|
|
||
| var size = app.CidRepo.GetSizeForCid(cid); | ||
| if (size == null) return; | ||
|
|
||
| var filename = Guid.NewGuid().ToString().ToLowerInvariant(); | ||
| await codex.DownloadCid(filename, cid, size); | ||
|
|
||
| DeleteFile(filename); | ||
| } | ||
|
|
||
| private async Task<string> StartNewPurchase() | ||
| { | ||
| var file = await CreateFile(); | ||
| try | ||
| { | ||
| var cid = await codex.UploadFile(file); | ||
| var response = await codex.RequestStorage(cid); | ||
| return response.PurchaseId; | ||
| } | ||
| finally | ||
| { | ||
| DeleteFile(file); | ||
| } | ||
| } | ||
|
|
||
| private async Task<string> CreateFile() | ||
| { | ||
| return await app.Generator.Generate(); | ||
| } | ||
|
|
||
| private void DeleteFile(string file) | ||
| { | ||
| try | ||
| { | ||
| File.Delete(file); | ||
| } | ||
| catch (Exception exc) | ||
| { | ||
| app.Log.Error($"Failed to delete file '{file}': {exc}"); | ||
| } | ||
| } | ||
|
|
||
| private async Task WaitTillFinished(string pid) | ||
| { | ||
| try | ||
| { | ||
| var emptyResponseTolerance = 10; | ||
| while (!app.Cts.Token.IsCancellationRequested) | ||
| { | ||
| var purchase = await codex.GetStoragePurchase(pid); | ||
| if (purchase == null) | ||
| { | ||
| await FixedShortDelay(); | ||
| emptyResponseTolerance--; | ||
| if (emptyResponseTolerance == 0) | ||
| { | ||
| log.Log("Received 10 empty responses. Stop tracking this purchase."); | ||
| await ExpiryTimeDelay(); | ||
| return; | ||
| } | ||
| continue; | ||
| } | ||
| if (purchase.IsCancelled) | ||
| { | ||
| app.Performance.StorageContractCancelled(); | ||
| return; | ||
| } | ||
| if (purchase.IsError) | ||
| { | ||
| app.Performance.StorageContractErrored(purchase.Error); | ||
| return; | ||
| } | ||
| if (purchase.IsFinished) | ||
| { | ||
| app.Performance.StorageContractFinished(); | ||
| return; | ||
| } | ||
| if (purchase.IsStarted) | ||
| { | ||
| app.Performance.StorageContractStarted(); | ||
| await FixedDurationDelay(); | ||
| } | ||
|
|
||
| await FixedShortDelay(); | ||
| } | ||
| } | ||
| catch (Exception ex) | ||
| { | ||
| log.Log($"Wait failed with exception: {ex}. Assume contract will expire: Wait expiry time."); | ||
| await ExpiryTimeDelay(); | ||
| } | ||
| } | ||
|
|
||
| private async Task FixedDurationDelay() | ||
| { | ||
| await Task.Delay(app.Config.ContractDurationMinutes * 60 * 1000, app.Cts.Token); | ||
| } | ||
|
|
||
| private async Task ExpiryTimeDelay() | ||
| { | ||
| await Task.Delay(app.Config.ContractExpiryMinutes * 60 * 1000, app.Cts.Token); | ||
| } | ||
|
|
||
| private async Task FixedShortDelay() | ||
| { | ||
| await Task.Delay(15 * 1000, app.Cts.Token); | ||
| } | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.