/// <summary>
/// Build new beat information.
/// </summary>
/// <param name="serviceName">service name with group name, format: ${groupName}@@${serviceName}</param>
/// <param name="instance">instance</param>
/// <returns>new beat information</returns>
public BeatInfo BuildBeatInfo(string serviceName, RegisterInstanceRequest instance)
{
BeatInfo beatInfo = new BeatInfo
{
serviceName = serviceName,
ip = instance.Ip,
port = instance.Port,
cluster = instance.ClusterName,
weight = instance.Weight,
metadata = instance.Metadata,
scheduled = false,
// using the default value at first, the unit is ms.
period = 5000
};
return(beatInfo);
}
/// <summary>
/// Add beat information.
/// </summary>
/// <param name="serviceName">service name </param>
/// <param name="beatInfo">beat information </param>
public Task AddBeatInfo(string serviceName, BeatInfo beatInfo)
{
_logger.LogInformation("[BEAT] adding beat: {0} to beat map.", beatInfo.ToJsonString());
string key = BuildKey(serviceName, beatInfo.ip, beatInfo.port);
if (_dom2Beat.TryRemove(key, out var existBeat))
{
existBeat.stopped = true;
}
_dom2Beat.AddOrUpdate(key, beatInfo, (x, y) => beatInfo);
var timer = new Timer(
async x =>
{
var info = x as BeatInfo;
await BeatTask(info);
}, beatInfo, beatInfo.period, beatInfo.period);
_beatTimer.AddOrUpdate(key, timer, (x, y) => timer);
return(Task.CompletedTask);
}
public async Task <bool> RegisterInstanceAsync(RegisterInstanceRequest request)
{
if (request == null)
{
throw new NacosException(NacosException.CLIENT_INVALID_PARAM, "request param invalid");
}
request.CheckParam();
if (request.Ephemeral.HasValue && request.Ephemeral.Value)
{
BeatInfo beatInfo = _beatReactor.BuildBeatInfo(request.ServiceName, request);
await _beatReactor.AddBeatInfo(GetGroupedName(request.ServiceName, request.GroupName), beatInfo);
}
var responseMessage = await _proxy.ReqApiAsync(HttpMethod.Post, RequestPathValue.INSTANCE, null, request.ToDict(), _options.DefaultTimeOut);
switch (responseMessage.StatusCode)
{
case System.Net.HttpStatusCode.OK:
var result = await responseMessage.Content.ReadAsStringAsync();
if (result.Equals("ok", StringComparison.OrdinalIgnoreCase))
{
return(true);
}
else
{
_logger.LogWarning($"[client.RegisterInstance] server return {result} ");
return(false);
}
default:
_logger.LogWarning($"[client.RegisterInstance] Register an instance to service failed {responseMessage.StatusCode.ToString()}");
throw new NacosException((int)responseMessage.StatusCode, $"Register an instance to service failed {responseMessage.StatusCode.ToString()}");
}
}
private async Task BeatTask(BeatInfo beatInfo)
{
if (beatInfo.stopped)
{
return;
}
try
{
// send heart beat will register instance
var request = new SendHeartbeatRequest
{
Ephemeral = true,
ServiceName = beatInfo.serviceName,
BeatInfo = beatInfo,
NameSpaceId = _options.Namespace,
};
if (request == null)
{
throw new NacosException(ConstValue.CLIENT_INVALID_PARAM, "request param invalid");
}
request.CheckParam();
var responseMessage = await _proxy.ReqApiAsync(HttpMethod.Put, RequestPathValue.INSTANCE_BEAT, null, request.ToDict(), _options.DefaultTimeOut);
switch (responseMessage.StatusCode)
{
case System.Net.HttpStatusCode.OK:
var result = await responseMessage.Content.ReadAsStringAsync();
var jObj = Newtonsoft.Json.Linq.JObject.Parse(result);
if (jObj.ContainsKey("code"))
{
int code = int.Parse(jObj["code"].ToString());
var flag = code == 10200;
if (!flag)
{
_logger.LogWarning($"[CLIENT-BEAT] server return {result} ");
}
}
else
{
_logger.LogWarning($"[CLIENT-BEAT] server return {result} ");
}
break;
default:
_logger.LogWarning("[CLIENT-BEAT] failed to send beat {0}, {1}", beatInfo.ToJsonString(), responseMessage.StatusCode.ToString());
throw new NacosException((int)responseMessage.StatusCode, $"Send instance beat failed {responseMessage.StatusCode.ToString()}");
}
}
catch (Exception ex)
{
_logger.LogWarning(ex, "Send heart beat to Nacos error");
}
}
