Recreate worker as a .NET Core app
This commit is contained in:
115
worker/src/Worker/Program.cs
Normal file
115
worker/src/Worker/Program.cs
Normal file
@ -0,0 +1,115 @@
|
||||
using System;
|
||||
using System.Data.Common;
|
||||
using System.Linq;
|
||||
using System.Net;
|
||||
using System.Net.Sockets;
|
||||
using System.Threading;
|
||||
using Newtonsoft.Json;
|
||||
using Npgsql;
|
||||
using StackExchange.Redis;
|
||||
|
||||
namespace Worker
|
||||
{
|
||||
public class Program
|
||||
{
|
||||
public static int Main(string[] args)
|
||||
{
|
||||
try
|
||||
{
|
||||
var pgsql = OpenDbConnection("Server=db;Username=postgres;");
|
||||
var redis = OpenRedisConnection("redis").GetDatabase();
|
||||
|
||||
var definition = new { vote = "", voter_id = "" };
|
||||
while (true)
|
||||
{
|
||||
string json = redis.ListLeftPopAsync("votes").Result;
|
||||
if (json != null)
|
||||
{
|
||||
var vote = JsonConvert.DeserializeAnonymousType(json, definition);
|
||||
Console.WriteLine($"Processing vote for '{vote.vote}' by '{vote.voter_id}'");
|
||||
UpdateVote(pgsql, vote.voter_id, vote.vote);
|
||||
}
|
||||
}
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
Console.Error.WriteLine(ex.ToString());
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
|
||||
private static NpgsqlConnection OpenDbConnection(string connectionString)
|
||||
{
|
||||
var connection = new NpgsqlConnection(connectionString);
|
||||
while (true)
|
||||
{
|
||||
try
|
||||
{
|
||||
connection.Open();
|
||||
break;
|
||||
}
|
||||
catch (DbException)
|
||||
{
|
||||
Console.Error.WriteLine("Failed to connect to db - retrying");
|
||||
}
|
||||
}
|
||||
|
||||
var command = connection.CreateCommand();
|
||||
command.CommandText = @"CREATE TABLE IF NOT EXISTS votes (
|
||||
id VARCHAR(255) NOT NULL UNIQUE,
|
||||
vote VARCHAR(255) NOT NULL
|
||||
)";
|
||||
command.ExecuteNonQuery();
|
||||
|
||||
return connection;
|
||||
}
|
||||
|
||||
private static ConnectionMultiplexer OpenRedisConnection(string hostname)
|
||||
{
|
||||
// Use IP address to workaround hhttps://github.com/StackExchange/StackExchange.Redis/issues/410
|
||||
var ipAddress = GetIp(hostname);
|
||||
Console.WriteLine($"Found redis at {ipAddress}");
|
||||
|
||||
while (true)
|
||||
{
|
||||
try
|
||||
{
|
||||
return ConnectionMultiplexer.Connect(ipAddress);
|
||||
}
|
||||
catch (RedisConnectionException)
|
||||
{
|
||||
Console.Error.WriteLine("Failed to connect to redis - retrying");
|
||||
Thread.Sleep(1000);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private static string GetIp(string hostname)
|
||||
=> Dns.GetHostEntryAsync(hostname)
|
||||
.Result
|
||||
.AddressList
|
||||
.First(a => a.AddressFamily == AddressFamily.InterNetwork)
|
||||
.ToString();
|
||||
|
||||
private static void UpdateVote(NpgsqlConnection connection, string voterId, string vote)
|
||||
{
|
||||
var command = connection.CreateCommand();
|
||||
try
|
||||
{
|
||||
command.CommandText = "INSERT INTO votes (id, vote) VALUES (@id, @vote)";
|
||||
command.Parameters.AddWithValue("@id", voterId);
|
||||
command.Parameters.AddWithValue("@vote", vote);
|
||||
command.ExecuteNonQuery();
|
||||
}
|
||||
catch (DbException)
|
||||
{
|
||||
command.CommandText = "UPDATE votes SET vote = @vote WHERE id = @id";
|
||||
command.ExecuteNonQuery();
|
||||
}
|
||||
finally
|
||||
{
|
||||
command.Dispose();
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
24
worker/src/Worker/project.json
Normal file
24
worker/src/Worker/project.json
Normal file
@ -0,0 +1,24 @@
|
||||
{
|
||||
"name": "Worker",
|
||||
"buildOptions": {
|
||||
"emitEntryPoint": true,
|
||||
"warningsAsErrors": true
|
||||
},
|
||||
"dependencies": {
|
||||
"StackExchange.Redis": "1.1.604-alpha",
|
||||
"Npgsql": "3.1.3",
|
||||
"Newtonsoft.Json": "9.0.1-beta1",
|
||||
"Microsoft.NETCore.App": {
|
||||
"type": "platform",
|
||||
"version": "1.0.0-rc2-3002702"
|
||||
}
|
||||
},
|
||||
"frameworks": {
|
||||
"netcoreapp1.0": { }
|
||||
},
|
||||
"runtimeOptions": {
|
||||
"configProperties": {
|
||||
"System.GC.Server": true
|
||||
}
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user