.Net Core Configuration Etcd数据源

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
public static class EtcdConfigurationExtensions
{
/// <summary>
/// AddEtcd扩展方法
/// </summary>
/// <param name="serverAddress">Etcd地址</param>
/// <param name="path">读取路径</param>
/// <returns></returns>
public static IConfigurationBuilder AddEtcd(this IConfigurationBuilder builder, string serverAddress,string path)
{
return AddEtcd(builder, serverAddress:serverAddress, path: path,reloadOnChange: false);
}
/// <summary>
/// AddEtcd扩展方法
/// </summary>
/// <param name="serverAddress">Etcd地址</param>
/// <param name="path">读取路径</param>
/// <param name="reloadOnChange">如果数据发送改变是否刷新</param>
/// <returns></returns>
public static IConfigurationBuilder AddEtcd(this IConfigurationBuilder builder, string serverAddress, string path, bool reloadOnChange)
{
return AddEtcd(builder,options => {
options.Address = serverAddress;
options.Path = path;
options.ReloadOnChange = reloadOnChange;
});
}
public static IConfigurationBuilder AddEtcd(this IConfigurationBuilder builder, Action<EtcdOptions> options)
{
EtcdOptions etcdOptions = new EtcdOptions();
options.Invoke(etcdOptions);
return builder.Add(new EtcdConfigurationSource { EtcdOptions = etcdOptions });
}
}

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
public class EtcdOptions
{
/// <summary>
/// Etcd地址
/// </summary>
public string Address { get; set; }
/// <summary>
/// Etcd访问用户名
/// </summary>
public string UserName { get; set; }
/// <summary>
/// Etcd访问密码
/// </summary>
public string PassWord { get; set; }
/// <summary>
/// Etcd读取路径
/// </summary>
public string Path { get; set; }
/// <summary>
/// 数据变更是否刷新读取
/// </summary>
public bool ReloadOnChange { get; set; }
}

1
2
3
4
5
6
7
8
9
public class EtcdConfigurationSource : IConfigurationSource
{
public EtcdOptions EtcdOptions { get; set; }
public IConfigurationProvider Build(IConfigurationBuilder builder)
{
return new EtcdConfigurationProvider(EtcdOptions);
}
}

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
public class EtcdConfigurationProvider : ConfigurationProvider
{
private readonly string _path;
private readonly bool _reloadOnChange;
private readonly EtcdClient _etcdClient;
public EtcdConfigurationProvider(EtcdOptions options)
{
//实例化EtcdClient
_etcdClient = new EtcdClient(options.Address,username: options.UserName,password: options.PassWord);
_path = options.Path;
_reloadOnChange = options.ReloadOnChange;
}
/// <summary>
/// 重写加载方法
/// </summary>
public override void Load()
{
//读取数据
LoadData();
//数据发生变化是否重新加载
if (_reloadOnChange)
{
ReloadData();
}
}
private void LoadData()
{
//读取Etcd里的数据
string result = _etcdClient.GetValAsync(_path).GetAwaiter().GetResult();
if (string.IsNullOrEmpty(result))
{
return;
}
//转换一下数据结构，这里我使用的是json格式
//读取的数据只要赋值到Data属性上即可,IConfiguration真正读取的数据就是存储到Data的字典数据
Data = ConvertData(result);
}
private IDictionary<string,string> ConvertData(string result)
{
byte[] array = Encoding.UTF8.GetBytes(result);
MemoryStream stream = new MemoryStream(array);
//JsonConfigurationFileParser是将json数据转换为Configuration可读取的结构(复制JsonConfiguration类库里的????????????)
return JsonConfigurationFileParser.Parse(stream);
}
private void ReloadData()
{
WatchRequest request = new WatchRequest()
{
CreateRequest = new WatchCreateRequest()
{
//需要转换一个格式,因为etcd v3版本的接口都包含在grpc的定义中
Key = ByteString.CopyFromUtf8(_path)
}
};
//监听Etcd节点变化,获取变更数据，更新配置
_etcdClient.Watch(request, rsp =>
{
if (rsp.Events.Any())
{
var @event = rsp.Events[0];
//需要转换一个格式,因为etcd v3版本的接口都包含在grpc的定义中
Data = ConvertData(@event.Kv.Value.ToStringUtf8());
//需要调用ConfigurationProvider的OnReload方法触发ConfigurationReloadToken通知
//这样才能对使用Configuration的类发送数据变更通知
//比如IOptionsMonitor就是通过ConfigurationReloadToken通知变更数据的
OnReload();
}
});
}
}

1
2
builder.AddEtcd("http://127.0.0.1:2379", "service/mydemo", true);