Repository navigation
Expand file tree
/
Copy pathAgentStorageInFile.cs
More file actions
118 lines (102 loc) · 3.69 KB
/
Copy pathAgentStorageInFile.cs
File metadata and controls
118 lines (102 loc) · 3.69 KB
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
109
110
111
112
113
114
115
116
117
118
using BotSharp.Platform.Abstractions;
using BotSharp.Platform.Models;
using CSRedis;
using Microsoft.Extensions.Configuration;
using Newtonsoft.Json;
using Newtonsoft.Json.Serialization;
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
namespace BotSharp.Core.AgentStorage
{
public class AgentStorageInFile<TAgent> : IAgentStorage<TAgent>
where TAgent : AgentBase
{
private static string storageDir;
public AgentStorageInFile()
{
IConfiguration config = (IConfiguration)AppDomain.CurrentDomain.GetData("Configuration");
var db = config.GetSection("Database:Default").Value;
storageDir = config.GetSection($"Database:ConnectionStrings:{db}").Value;
string contentDir = AppDomain.CurrentDomain.GetData("DataPath").ToString();
storageDir = storageDir.Replace("|DataDirectory|", contentDir + Path.DirectorySeparatorChar + "AgentStorage" + Path.DirectorySeparatorChar);
if (!Directory.Exists(storageDir))
{
Directory.CreateDirectory(storageDir);
}
}
public async Task<TAgent> FetchById(string agentId)
{
string dataPath = Path.Combine(storageDir, agentId + ".json");
if (File.Exists(dataPath))
{
string json = File.ReadAllText(dataPath);
return JsonConvert.DeserializeObject<TAgent>(json);
}
else
{
return default(TAgent);
}
}
public async Task<TAgent> FetchByName(string agentName)
{
var files = Directory.GetFiles(storageDir);
for (int i = 0; i < files.Length; i++)
{
var file = files[i];
string json = File.ReadAllText(file);
var agent = JsonConvert.DeserializeObject<TAgent>(json);
if (agent.Name.ToLower() == agentName.ToLower())
{
return agent;
}
}
return default(TAgent);
}
public async Task<bool> Persist(TAgent agent)
{
if (String.IsNullOrEmpty(agent.Id))
{
agent.Id = Guid.NewGuid().ToString();
}
var json = JsonConvert.SerializeObject(agent, new JsonSerializerSettings
{
NullValueHandling = NullValueHandling.Ignore,
Formatting = Formatting.Indented,
ContractResolver = new CamelCasePropertyNamesContractResolver(),
Converters = new List<JsonConverter>
{
new Newtonsoft.Json.Converters.StringEnumConverter()
}
});
string dataPath = Path.Combine(storageDir, agent.Id + ".json");
File.WriteAllText(dataPath, json);
return true;
}
public async Task<int> PurgeAllAgents()
{
var files = Directory.GetFiles(storageDir);
for (int i = 0; i < files.Length; i++)
{
File.Delete(files[i]);
}
return files.Length;
}
public async Task<List<TAgent>> Query()
{
var agents = new List<TAgent>();
var files = Directory.GetFiles(storageDir);
for (int i = 0; i < files.Length; i++)
{
var file = files[i];
string json = File.ReadAllText(file);
var agent = JsonConvert.DeserializeObject<TAgent>(json);
agents.Add(agent);
}
return agents;
}
}
}