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
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
|
using Google.Protobuf;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using Tango.DataStore;
using Tango.DataStore.Remote;
using Tango.FSE.Common.Connection;
namespace Tango.FSE.UI.DataStore
{
public class RemoteDataStoreCollection : IDataStoreCollection
{
private IMachineProvider _machineProvider;
public string Name { get; }
public RemoteDataStoreCollection(String name, IMachineProvider machineProvider)
{
Name = name;
_machineProvider = machineProvider;
}
public void Put<T>(string key, T value)
{
Put(key, (object)value);
}
public void Put(string key, object value)
{
Put(key, DataStoreHelper.GetDataType(value), value);
}
public void Put(string key, DataType type, object value)
{
if (type == DataType.Proto)
{
if (value is IMessage protoMessage)
{
value = DataStoreProtoObject.FromMessage(protoMessage);
}
else
{
throw new InvalidOperationException("DataType if Proto but actual message is not part of the PMR.");
}
}
var result = _machineProvider.MachineOperator.SendGenericRequest<RemoteDataStorePutRequest, RemoteDataStorePutResponse>(new RemoteDataStorePutRequest()
{
Collection = Name,
DataType = type,
Key = key,
Value = value
}).Result;
}
public T Get<T>(string key)
{
return (T)Convert.ChangeType(Get(key, null), typeof(T));
}
public T Get<T>(string key, T defaultValue)
{
return (T)Convert.ChangeType(Get(key, (object)defaultValue), typeof(T));
}
public object Get(string key)
{
return Get(key, null);
}
public object Get(string key, object defaultValue)
{
if (defaultValue is IMessage defaultValueMessage)
{
defaultValue = DataStoreProtoObject.FromMessage(defaultValueMessage);
}
var result = _machineProvider.MachineOperator.SendGenericRequest<RemoteDataStoreGetRequest, RemoteDataStoreGetResponse>(new RemoteDataStoreGetRequest()
{
Collection = Name,
Key = key,
DefaultValue = defaultValue,
}).Result;
if (result.DataType == DataType.Proto)
{
return result.ProtoObject.Message;
}
return result.Value;
}
public List<IDataStoreItem> GetAll()
{
var result = _machineProvider.MachineOperator.SendGenericRequest<RemoteDataStoreGetAllRequest, RemoteDataStoreGetAllResponse>(new RemoteDataStoreGetAllRequest()
{
Collection = Name
}).Result;
return result.Items.Cast<IDataStoreItem>().ToList();
}
public void Delete(string key)
{
var result = _machineProvider.MachineOperator.SendGenericRequest<RemoteDataStoreDeleteRequest, RemoteDataStoreDeleteResponse>(new RemoteDataStoreDeleteRequest()
{
Collection = Name,
Key = key
}).Result;
}
public void DeleteAll()
{
var result = _machineProvider.MachineOperator.SendGenericRequest<RemoteDataStoreDeleteAllRequest, RemoteDataStoreDeleteAllResponse>(new RemoteDataStoreDeleteAllRequest()
{
Collection = Name,
}).Result;
}
public int Count()
{
var result = _machineProvider.MachineOperator.SendGenericRequest<RemoteDataStoreCountRequest, RemoteDataStoreCountResponse>(new RemoteDataStoreCountRequest()
{
Collection = Name
}).Result;
return result.Count;
}
public IDataStoreItem GetItem(string key)
{
return GetItem(key, null);
}
public IDataStoreItem GetItem(string key, object defaultValue)
{
if (defaultValue is IMessage defaultValueMessage)
{
defaultValue = DataStoreProtoObject.FromMessage(defaultValueMessage);
}
var result = _machineProvider.MachineOperator.SendGenericRequest<RemoteDataStoreGetItemRequest, RemoteDataStoreGetItemResponse>(new RemoteDataStoreGetItemRequest()
{
Collection = Name,
Key = key,
DefaultValue = defaultValue
}).Result;
return result.Item;
}
public List<IDataStoreItem> GetUnsynchronized()
{
return GetAll().Where(x => !x.IsSynchronized).ToList();
}
}
}
|
