public static async Task <UserDto> Register(UserDto userDto)
{
LoadManagerContext dbContext = new LoadManagerContext();
User usr;
using (dbContext)
{
using (var transaction = dbContext.Database.BeginTransaction())
{
User user = new User
{
UserName = userDto.UserName,
Password = userDto.Password,
};
await dbContext.AddAsync <User>(user);
await dbContext.SaveChangesAsync();
usr = user;
transaction.Commit();
}
}
return(new UserDto
{
UserId = usr.UserId,
UserName = usr.UserName,
Password = usr.Password,
});
}
public static async Task UpdateTaskPercent(int taskId, byte percent)
{
using (var dbContext = new LoadManagerContext())
{
var entity = await dbContext.Tasks.FirstOrDefaultAsync(e => e.TaskID == taskId);
entity.DonePercent = percent;
dbContext.Tasks.Update(entity);
await dbContext.SaveChangesAsync();
}
}
public static async Task <int> CreateNewTask(int size, int userId)
{
var emptyTask = new Models.Task()
{
Status = "Waiting", DonePercent = (byte)0, Size = size, UserId = userId
};
using (var dbContext = new LoadManagerContext())
{
dbContext.Tasks.Add(emptyTask);
await dbContext.SaveChangesAsync();
return(emptyTask.TaskID);
}
}
public static async Task UnAssignTask(int taskId)
{
using (var dbContext = new LoadManagerContext())
{
using (var transaction = dbContext.Database.BeginTransaction())
{
var entity = await dbContext.Tasks.FirstOrDefaultAsync(e => e.TaskID == taskId);
entity.ServerID = null;
dbContext.Tasks.Update(entity);
await dbContext.SaveChangesAsync();
transaction.Commit();
}
}
}
public static async Task UpdateTaskStatus(int taskId, string status)
{
using (var dbContext = new LoadManagerContext())
{
using (var transaction = dbContext.Database.BeginTransaction())
{
var entity = await dbContext.Tasks.FirstOrDefaultAsync(e => e.TaskID == taskId);
entity.Status = status;
dbContext.Tasks.Update(entity);
await dbContext.SaveChangesAsync();
transaction.Commit();
}
}
}
public static async Task <bool> AddNewWorkerServer(WorkerServerDto server)
{
using (var dbContext = new LoadManagerContext())
{
if (dbContext.WorkerServers.Count(b => b.IpAddress == server.IpAddress && b.Port == server.Port) != 0)
{
return(true);
}
dbContext.WorkerServers.Add(new WorkerServer()
{
IpAddress = server.IpAddress,
IsConnected = server.IsConnected,
Port = server.Port
});
return(await dbContext.SaveChangesAsync() == 1);
}
}
