251 lines
7.5 KiB
C#
251 lines
7.5 KiB
C#
using System;
|
||
using System.Collections.Generic;
|
||
using System.Linq;
|
||
using AntJob.Data;
|
||
using AntJob.Data.Entity;
|
||
using NewLife;
|
||
using NewLife.Caching;
|
||
using NewLife.Log;
|
||
using NewLife.Net;
|
||
using NewLife.Security;
|
||
using AntJob.Models;
|
||
using NewLife.Data;
|
||
using NewLife.Remoting;
|
||
using NewLife.Web;
|
||
using System.Reflection;
|
||
using System.Xml.Linq;
|
||
|
||
namespace AntJob.Server.Services;
|
||
|
||
public class AppService
|
||
{
|
||
private readonly ICacheProvider _cacheProvider;
|
||
private readonly ITracer _tracer;
|
||
private readonly ILog _log;
|
||
|
||
public AppService(ICacheProvider cacheProvider, ITracer tracer, ILog log)
|
||
{
|
||
_cacheProvider = cacheProvider;
|
||
_tracer = tracer;
|
||
_log = log;
|
||
}
|
||
|
||
#region 登录
|
||
/// <summary>应用登录</summary>
|
||
/// <param name="model">模型</param>
|
||
/// <returns></returns>
|
||
public (App, LoginResponse) Login(LoginModel model, String ip)
|
||
{
|
||
if (model.User.IsNullOrEmpty()) throw new ArgumentNullException(nameof(model.User));
|
||
|
||
_log.Info("[{0}]从[{1}]登录[{2}@{3}]", model.User, ip, model.Machine, model.ProcessId);
|
||
|
||
// 找应用
|
||
var autoReg = false;
|
||
var app = App.FindByName(model.User);
|
||
if (app == null || app.Secret.MD5() != model.Pass)
|
||
{
|
||
app = CheckApp(app, model.User, model.Pass, ip);
|
||
if (app == null) throw new ArgumentOutOfRangeException(nameof(model.User));
|
||
|
||
autoReg = true;
|
||
}
|
||
|
||
if (app == null) throw new Exception($"应用[{model.User}]不存在！");
|
||
if (!app.Enable) throw new Exception("已禁用！");
|
||
|
||
// 核对密码
|
||
if (!autoReg && !app.Secret.IsNullOrEmpty())
|
||
{
|
||
var pass2 = app.Secret.MD5();
|
||
if (model.Pass != pass2) throw new Exception("密码错误！");
|
||
}
|
||
|
||
// 版本和编译时间
|
||
if (app.Version.IsNullOrEmpty() || app.Version.CompareTo(model.Version) < 0) app.Version = model.Version;
|
||
if (app.CompileTime < model.Compile) app.CompileTime = model.Compile;
|
||
if (app.DisplayName.IsNullOrEmpty()) app.DisplayName = model.DisplayName;
|
||
|
||
app.Save();
|
||
|
||
// 应用上线
|
||
var online = CreateOnline(app, ip, model.Machine, model.ProcessId);
|
||
online.Version = model.Version;
|
||
online.CompileTime = model.Compile;
|
||
online.Save();
|
||
|
||
WriteHistory(app, autoReg ? "注册" : "登录", true, $"[{model.User}/{model.Pass}]在[{model.Machine}@{model.ProcessId}]登录[{app}]成功");
|
||
|
||
var rs = new LoginResponse { Name = app.Name, DisplayName = app.DisplayName };
|
||
if (autoReg) rs.Secret = app.Secret;
|
||
|
||
return (app, rs);
|
||
}
|
||
|
||
protected virtual App CheckApp(App app, String user, String pass, String ip)
|
||
{
|
||
// 本地账号不存在时
|
||
var name = user;
|
||
if (app == null)
|
||
{
|
||
// 是否支持自动注册
|
||
var set = AntJobSetting.Current;
|
||
if (!set.AutoRegistry) throw new Exception($"找不到应用[{name}]");
|
||
|
||
app = new App
|
||
{
|
||
Secret = Rand.NextString(16)
|
||
};
|
||
}
|
||
else if (app.Secret.MD5() != pass)
|
||
{
|
||
// 是否支持自动注册
|
||
var set = AntJobSetting.Current;
|
||
if (!set.AutoRegistry) throw new Exception($"应用[{name}]申请重新激活，但服务器设置禁止自动注册");
|
||
|
||
if (app.Secret.IsNullOrEmpty()) app.Secret = Rand.NextString(16);
|
||
}
|
||
|
||
if (app.ID == 0)
|
||
{
|
||
app.Name = name;
|
||
app.CreateIP = ip;
|
||
app.CreateTime = DateTime.Now;
|
||
|
||
// 首次打开
|
||
app.Enable = true;
|
||
}
|
||
|
||
app.UpdateIP = ip;
|
||
app.UpdateTime = DateTime.Now;
|
||
|
||
//app.Save();
|
||
|
||
return app;
|
||
}
|
||
#endregion
|
||
|
||
#region 在线状态
|
||
/// <summary>获取当前应用的所有在线实例</summary>
|
||
/// <returns></returns>
|
||
public PeerModel[] GetPeers(App app)
|
||
{
|
||
var olts = AppOnline.FindAllByAppID(app.ID);
|
||
|
||
return olts.Select(e => e.ToModel()).ToArray();
|
||
}
|
||
|
||
AppOnline CreateOnline(App app, String ip, String machine, Int32 pid)
|
||
{
|
||
var online = GetOnline(app, ip);
|
||
online.Client = $"{(ip.IsNullOrEmpty() ? machine : ip)}@{pid}";
|
||
online.Name = machine;
|
||
online.ProcessId = pid;
|
||
online.UpdateIP = ip;
|
||
//online.Version = version;
|
||
|
||
online.Server = Environment.MachineName;
|
||
//online.Save();
|
||
|
||
return online;
|
||
}
|
||
|
||
public AppOnline GetOnline(App app, String ip)
|
||
{
|
||
var ins = $"{app.Name}@{ip}";
|
||
var online = AppOnline.FindByInstance(ins) ?? new AppOnline { CreateIP = ip };
|
||
online.AppID = app.ID;
|
||
online.Instance = ins;
|
||
|
||
return online;
|
||
}
|
||
|
||
public void UpdateOnline(App app, JobTask ji, String ip)
|
||
{
|
||
var online = GetOnline(app, ip);
|
||
online.Total += ji.Total;
|
||
online.Success += ji.Success;
|
||
online.Error += ji.Error;
|
||
online.Cost += ji.Cost;
|
||
online.Speed = ji.Speed;
|
||
online.LastKey = ji.Key;
|
||
online.SaveAsync();
|
||
}
|
||
#endregion
|
||
|
||
#region 写历史
|
||
public void WriteHistory(App app, String action, Boolean success, String remark, String ip = null) =>
|
||
AppHistory.Create(app, action, success, remark, Environment.MachineName, ip);
|
||
#endregion
|
||
|
||
#region 辅助
|
||
public TokenModel IssueToken(String name, AntJobSetting set)
|
||
{
|
||
// 颁发令牌
|
||
var ss = set.TokenSecret.Split(':');
|
||
var jwt = new JwtBuilder
|
||
{
|
||
Issuer = Assembly.GetEntryAssembly().GetName().Name,
|
||
Subject = name,
|
||
Id = Rand.NextString(8),
|
||
Expire = DateTime.Now.AddSeconds(set.TokenExpire),
|
||
|
||
Algorithm = ss[0],
|
||
Secret = ss[1],
|
||
};
|
||
|
||
return new TokenModel
|
||
{
|
||
AccessToken = jwt.Encode(null),
|
||
TokenType = jwt.Type ?? "JWT",
|
||
ExpireIn = set.TokenExpire,
|
||
RefreshToken = jwt.Encode(null),
|
||
};
|
||
}
|
||
|
||
public (App, Exception) DecodeToken(String token, String tokenSecret)
|
||
{
|
||
if (token.IsNullOrEmpty()) throw new ArgumentNullException(nameof(token));
|
||
//if (token.IsNullOrEmpty()) throw new ApiException(401, $"节点未登录[ip={UserHost}]");
|
||
|
||
// 解码令牌
|
||
var ss = tokenSecret.Split(':');
|
||
var jwt = new JwtBuilder
|
||
{
|
||
Algorithm = ss[0],
|
||
Secret = ss[1],
|
||
};
|
||
|
||
var rs = jwt.TryDecode(token, out var message);
|
||
var app = App.FindByName(jwt.Subject);
|
||
|
||
Exception ex = null;
|
||
if (!rs || app == null)
|
||
{
|
||
if (app != null)
|
||
ex = new ApiException(403, $"[{app.Name}/{app.DisplayName}]非法访问 {message}");
|
||
else
|
||
ex = new ApiException(403, $"[{jwt.Subject}]非法访问 {message}");
|
||
}
|
||
|
||
return (app, ex);
|
||
}
|
||
|
||
public TokenModel ValidAndIssueToken(String deviceCode, String token, AntJobSetting set)
|
||
{
|
||
if (token.IsNullOrEmpty()) return null;
|
||
//var set = Setting.Current;
|
||
|
||
// 令牌有效期检查，10分钟内过期者，重新颁发令牌
|
||
var ss = set.TokenSecret.Split(':');
|
||
var jwt = new JwtBuilder
|
||
{
|
||
Algorithm = ss[0],
|
||
Secret = ss[1],
|
||
};
|
||
var rs = jwt.TryDecode(token, out var message);
|
||
return !rs || jwt == null ? null : DateTime.Now.AddMinutes(10) > jwt.Expire ? IssueToken(deviceCode, set) : null;
|
||
}
|
||
#endregion
|
||
}
|