public void Setup()
{
kafkaConsumer = new KafkaConsumer("kafka:9092", "test_consumer");
probe = this.CreateTestProbe();
kafkaConsumer.CreateConsumer(probe).Start();
kafkaProduce = new KafkaProduce("kafka:9092", "test_consumer");
// Wait for SystemLoad
Task.Delay(3000).Wait();
}
// This method gets called by the runtime. Use this method to add services to the container.
public void ConfigureServices(IServiceCollection services)
{
services.AddMvc();
// 设置数据源，读取appsettings.json配置文件
GlobalSystemConfig.DBConnectionString = this.Configuration.GetConnectionString("DataSource");
// 注册缓存对象
services.AddMemoryCache();
IConfigurationSection connectionStringsSection = this.Configuration.GetSection("ConnectionStrings");
IConfigurationSection cacheProviderSection = connectionStringsSection.GetSection("CacheProvider");
if (cacheProviderSection != null)
{
try
{
string type = cacheProviderSection.GetValue <string>("type");
if ("redis".Equals(type, StringComparison.OrdinalIgnoreCase))
{
// redis 分布式缓存
RedisCacheProvider provider = cacheProviderSection.GetSection("provider").Get <RedisCacheProvider>();
services.AddSingleton(typeof(ICacheService), new RedisCacheService(new RedisCacheOptions
{
Configuration = provider.HostName + ":" + provider.Port,
InstanceName = provider.InstanceName
}, provider.DataBase));
}
else if ("redis.r.w".Equals(type, StringComparison.OrdinalIgnoreCase))
{
// redis读写分离
RedisRWConfigInfo provider = cacheProviderSection.GetSection("provider").Get <RedisRWConfigInfo>();
RedisManager redisManager = new RedisManager(provider);
services.AddSingleton(typeof(ICacheService), new RedisRWCacheService(redisManager.GetClient()));
}
}
catch (Exception ex)
{
LOG.Error(ex, "redis连接失败，准备启用MemoryCache服务");
SetCacheService(services);
}
}
else
{
SetCacheService(services);
}
// mongo
IConfigurationSection mongoSection = this.Configuration.GetSection("ConnectionStrings").GetSection("Mongo");
if (mongoSection != null)
{
try
{
// 注册mongo连接信息
MongodbHost mongohost = mongoSection.Get <MongodbHost>();
services.AddSingleton(typeof(MongodbHost), mongohost);
// IMongoClient mongoClient = new MongoClient(mongohost.Connection);
IMongoClient mongoClient = MongodbManager <object> .GetMongodbClient(mongohost.ConnectionString);
services.AddSingleton(typeof(IMongoClient), mongoClient);
IMongoDatabase mongoDatabase = mongoClient.GetDatabase(mongohost.DataBase);
services.AddSingleton(typeof(IMongoDatabase), mongoDatabase);
}
catch (Exception ex)
{
LOG.Error(ex, "mongo连接失败");
}
}
// 注册知识库
services.AddSingleton <IRepository, Repository>();
// 注册版本服务
services.AddSingleton <IVersionService, VersionService>();
// 注册用户服务
services.AddSingleton <IUserService, UserService>();
// 注册模型服务
services.AddSingleton <IModelService, ModelService> ();
// 注册数据源服务
services.AddSingleton <IDataSourceService, DataSourceService>();
// 注册算法服务
services.AddSingleton <IAlgorithmService, AlgorithmService>();
// 注册业务日志服务
services.AddSingleton <ILogService, LogService>();
// 注册任务服务
services.AddSingleton <ITaskService, TaskService>();
// 设置全局
IServiceProvider serviceProvider = services.BuildServiceProvider();
ServiceFactory.CacheService = serviceProvider.GetService <ICacheService>();
ServiceFactory.MongoHost = serviceProvider.GetService <MongodbHost>();
ServiceFactory.MongoClient = serviceProvider.GetService <IMongoClient>();
ServiceFactory.MongoDatabase = serviceProvider.GetService <IMongoDatabase>();
// 消息相关
IConfigurationSection messageSection = connectionStringsSection.GetSection("Message");
if (messageSection != null)
{
// 消息
IConfigurationSection mqSection = messageSection.GetSection("MQ");
if (mqSection != null)
{
KafkaSetting kafkaSetting = mqSection.Get <KafkaSetting>();
if (kafkaSetting.Switch)
{
KafkaConsumer.CreateConsumer(kafkaSetting.Opinion.GroupId, kafkaSetting.Opinion.Servers, kafkaSetting.Opinion.Topics);
KafkaConsumer.Start();
KafkaProducer.CreateProducer(kafkaSetting.Opinion.Servers);
}
}
// websocket
IConfigurationSection websocketSection = messageSection.GetSection("Websocket");
if (websocketSection != null)
{
WebsocketSetting websocketSetting = websocketSection.Get <WebsocketSetting>();
WebsocketFleckServer.CreateWebsocketServer(websocketSetting.NodeId, websocketSetting.Port, websocketSetting.Host);
}
}
// scheduler，注入参数设置
IConfigurationSection schedulerSection = this.Configuration.GetSection("Scheduler");
if (schedulerSection != null)
{
if (schedulerSection.GetValue <Boolean>("switch"))
{
IConfigurationSection propertiesSection = schedulerSection.GetSection("properties");
var values = propertiesSection.GetChildren()
.Select(item => new KeyValuePair <string, string>(item.Key,
item.Value.Contains("$") ? Configuration.GetValue <string>(item.Value.Replace("${", "").Replace("}", "")) : item.Value))
.ToDictionary(x => x.Key, x => x.Value);
DmeQuartzScheduler <TaskRunnerJob> .SetSchedulerProperties(DataUtil.ToNameValueCollection(values));
// 调取开启。如果不开启，则不会执行。
DmeQuartzScheduler <TaskRunnerJob> .Start().GetAwaiter();
}
}
// DemoScheduler.RunProOracle().Wait();
services.AddSwaggerGen(c =>
{
c.SwaggerDoc("v1", new Info
{
Version = "v1",
Title = "Dist Model Engine接口文档",
Description = "RESTful API for DME",
TermsOfService = "None",
Contact = new Contact {
Name = "weifj", Email = "*****@*****.**", Url = "https://github.com/leoterry-ulrica/dme"
}
});
//Set the comments path for the swagger json and ui.
var basePath = PlatformServices.Default.Application.ApplicationBasePath;
var xmlPath = Path.Combine(basePath, "Dist.Dme.WebApi.xml");
c.IncludeXmlComments(xmlPath);
// c.OperationFilter<HttpHeaderOperation>(); // 添加httpHeader参数
});
// 配置日志服务
services.AddLogging();
}