using System; using System.Threading; namespace Preparation.Utility { //理论上结构体最好不可变,这里采用了可变结构。 //其对应属性不应当有set访问器,避免不安全的=赋值 public struct AtomicInt { private int v; public AtomicInt(int x) { v = x; } public override string ToString() => Interlocked.CompareExchange(ref v, -1, -1).ToString(); public int Get() => Interlocked.CompareExchange(ref v, -1, -1); public static implicit operator int(AtomicInt aint) => Interlocked.CompareExchange(ref aint.v, -1, -1); public int Set(int value) => Interlocked.Exchange(ref v, value); public int Add(int x) => Interlocked.Add(ref v, x); public int Sub(int x) => Interlocked.Add(ref v, -x); public int Inc() => Interlocked.Increment(ref v); public int Dec() => Interlocked.Decrement(ref v); public void CompareExchange(int newV, int compareTo) => Interlocked.CompareExchange(ref v, newV, compareTo); /// 返回操作前的值 public int CompareExReturnOri(int newV, int compareTo) => Interlocked.CompareExchange(ref v, newV, compareTo); } public struct AtomicBool { private int v;//v==0为false,v!=0(v==1或v==-1)为true public AtomicBool(bool x) { v = x ? 1 : 0; } public override string ToString() => (Interlocked.CompareExchange(ref v, -2, -2) == 0) ? "false" : "true"; public bool Get() => (Interlocked.CompareExchange(ref v, -1, -1) != 0); public static implicit operator bool(AtomicBool abool) => (Interlocked.CompareExchange(ref abool.v, -1, -1) != 0); public bool Set(bool value) => (Interlocked.Exchange(ref v, value ? 1 : 0) != 0); /// 赋值前的值是否与将赋予的值不相同 public bool TrySet(bool value) { return (Interlocked.CompareExchange(ref v, value ? 1 : 0, value ? 0 : 1) ^ (value ? 1 : 0)) != 0; } public bool Invert() => Interlocked.Add(ref v, -1) != 0; public bool And(bool x) => Interlocked.And(ref v, x ? 1 : 0) != 0; public bool Or(bool x) => Interlocked.Or(ref v, x ? 1 : 0) != 0; } /// /// 一个能记录Start后完成多少进度的进度条(int), /// 只允许Start时修改needTime(请确保大于0); /// 支持TrySet0使未完成的进度条终止清零;支持Set0使进度条强制终止清零; /// 不支持暂停 /// public struct IntProgressContinuously { private long endT = long.MaxValue; private long needT; public IntProgressContinuously(long needTime) { if (needTime <= 0) Debugger.Output("Bug:IntProgressContinuously.needTime (" + needTime.ToString() + ") is less than 0."); this.needT = needTime; } public long GetEndTime() => Interlocked.CompareExchange(ref endT, -2, -2); public long GetNeedTime() => Interlocked.CompareExchange(ref needT, -2, -2); public override string ToString() => "EndTime:" + Interlocked.CompareExchange(ref endT, -2, -2).ToString() + " ms, NeedTime:" + Interlocked.CompareExchange(ref needT, -2, -2).ToString() + " ms"; public bool IsFinished() { return Interlocked.CompareExchange(ref endT, -2, -2) <= Environment.TickCount64; } /// /// GetProgress<0则表明未开始 /// public long GetProgress() { long cutime = Interlocked.CompareExchange(ref endT, -2, -2) - Environment.TickCount64; if (cutime <= 0) return Interlocked.CompareExchange(ref needT, -2, -2); return Interlocked.CompareExchange(ref needT, -2, -2) - cutime; } /// /// GetProgressDouble<0则表明未开始 /// public double GetProgressDouble() { long cutime = Interlocked.CompareExchange(ref endT, -2, -2) - Environment.TickCount64; if (cutime <= 0) return 1; return 1.0 - ((double)cutime / Interlocked.CompareExchange(ref needT, -2, -2)); } public bool Start(long needTime) { if (needTime <= 0) { Debugger.Output("Warning:Start IntProgressContinuously with the needTime (" + needTime.ToString() + ") which is less than 0."); return false; } //规定只有Start可以修改needT,且需要先访问endTime,从而避免锁(某种程度上endTime可以认为是needTime的锁) if (Interlocked.CompareExchange(ref endT, Environment.TickCount64 + needTime, long.MaxValue) != long.MaxValue) return false; if (needTime <= 2) Debugger.Output("Warning:the field of IntProgressContinuously is " + needTime.ToString() + ",which is too small."); Interlocked.Exchange(ref this.needT, needTime); return true; } public bool Start() { long needTime = Interlocked.CompareExchange(ref needT, -2, -2); if (Interlocked.CompareExchange(ref endT, Environment.TickCount64 + needTime, long.MaxValue) != long.MaxValue) return false; return true; } public void Set0() => Interlocked.Exchange(ref endT, long.MaxValue); public bool TrySet0() { if (Environment.TickCount64 < Interlocked.CompareExchange(ref endT, -2, -2)) { Interlocked.Exchange(ref endT, long.MaxValue); return true; } return false; } //增加其他新的写操作可能导致不安全 } /// /// 一个保证在[0,maxValue]的可变int,支持可变的maxValue(请确保大于0) /// public struct IntWithVariableRange { private int v; private int maxV; private readonly object vLock = new(); public IntWithVariableRange(int value, int maxValue) { if (maxValue < 0) { Debugger.Output("Warning:Try to set IntWithVariableRange.maxValue to " + maxValue.ToString() + "."); maxValue = 0; } v = value < maxValue ? value : maxValue; this.maxV = maxValue; } public override string ToString() { lock (vLock) { return "value:" + v.ToString() + " ,maxValue:" + maxV.ToString(); } } public int GetValue() { lock (vLock) return v; } public int GetMaxV() { lock (vLock) return maxV; } /// /// 若maxValue<=0则maxValue设为0并返回False /// public bool SetMaxV(int maxValue) { if (maxValue < 0) maxValue = 0; lock (vLock) { maxV = maxValue; if (v > maxValue) v = maxValue; } return maxValue > 0; } /// /// 应当保证该maxValue>=0 /// public void SetPositiveMaxV(int maxValue) { lock (vLock) { maxV = maxValue; if (v > maxValue) v = maxValue; } } /// /// 应当保证该value>=0 /// public int SetPositiveV(int value) { lock (vLock) { return v = (value > maxV) ? maxV : value; } } public int SetV(int value) { if (value < 0) value = 0; lock (vLock) { return v = (value > maxV) ? maxV : value; } } public int AddV(int addV) { lock (vLock) { v += addV; if (v < 0) return v = 0; if (v > maxV) return v = maxV; return v; } } /// /// 应当保证该增加值大于0 /// public int AddPositiveV(int addPositiveV) { lock (vLock) { v += addPositiveV; if (v > maxV) return v = maxV; return v; } } /// /// 应当保证该减少值大于0 /// public int SubPositiveV(int subPositiveV) { lock (vLock) { v += subPositiveV; if (v < 0) return v = 0; return v; } } } /// /// 一个保证在[0,maxNum],每CDms自动更新的可变int,支持可变的CD、maxNum(请确保大于0) /// public struct IntNumUpdateByCD { private int num; private int maxNum; private int cd; private long updateTime = 0; private object numLock = new(); public IntNumUpdateByCD(int num, int maxNum, int cd) { if (num < 0) Debugger.Output("Bug:IntNumUpdateByCD.num (" + num.ToString() + ") is less than 0."); if (maxNum < 0) Debugger.Output("Bug:IntNumUpdateByCD.maxNum (" + maxNum.ToString() + ") is less than 0."); if (cd <= 0) Debugger.Output("Bug:IntNumUpdateByCD.cd (" + cd.ToString() + ") is less than 0."); this.num = num; this.maxNum = maxNum; this.cd = cd; this.updateTime = Environment.TickCount64; } public IntNumUpdateByCD(int maxNum, int cd) { if (maxNum < 0) Debugger.Output("Bug:IntNumUpdateByCD.maxNum (" + maxNum.ToString() + ") is less than 0."); if (cd <= 0) Debugger.Output("Bug:IntNumUpdateByCD.cd (" + cd.ToString() + ") is less than 0."); this.num = this.maxNum = maxNum; this.cd = cd; } public int GetMaxNum() { lock (numLock) return maxNum; } public int GetCD() { lock (numLock) return cd; } public int GetNum(long time) { lock (numLock) { if (num < maxNum && time - updateTime >= cd) { int add = (int)Math.Min(maxNum - num, (time - updateTime) / cd); updateTime += add * cd; return (num += add); } return num; } } /// /// 应当保证该subV>=0 /// public int TrySub(int subV) { if (subV < 0) Debugger.Output("Bug:IntNumUpdateByCD Try to sub " + subV.ToString() + ", which is less than 0."); long time = Environment.TickCount64; lock (numLock) { if (num < maxNum && time - updateTime >= cd) { int add = (int)Math.Min(maxNum - num, (time - updateTime) / cd); updateTime += add * cd; num += add; } if (num == maxNum) updateTime = time; num -= subV = Math.Min(subV, num); } return subV; } /// /// 应当保证该addV>=0 /// public void TryAdd(int addV) { if (addV < 0) Debugger.Output("Bug:IntNumUpdateByCD Try to add " + addV.ToString() + ", which is less than 0."); lock (numLock) { num += Math.Min(addV, maxNum - num); } } /// /// 若maxNum<=0则maxNum及Num设为0并返回False /// public bool SetMaxNumAndNum(int maxNum) { if (maxNum < 0) maxNum = 0; lock (numLock) { this.num = this.maxNum = maxNum; } return maxNum > 0; } /// /// 应当保证该maxnum>=0 /// public void SetPositiveMaxNumAndNum(int maxNum) { lock (numLock) { this.num = this.maxNum = maxNum; } } /// /// 应当保证该maxnum>=0 /// public void SetPositiveMaxNum(int maxNum) { lock (numLock) { if ((this.maxNum = maxNum) < num) num = maxNum; } } /// /// 若maxNum<=0则maxNum及Num设为0并返回False /// public bool SetMaxNum(int maxNum) { if (maxNum < 0) maxNum = 0; lock (numLock) { if ((this.maxNum = maxNum) < num) num = maxNum; } return maxNum > 0; } /// /// 若num<0则num设为0并返回False /// public bool SetNum(int num) { lock (numLock) { if (num < 0) { this.num = 0; return false; } this.num = num; return true; } } /// /// 应当保证该num>=0 /// public void SetPositiveNum(int num) { lock (numLock) { this.num = num; } } public void SetCD(int cd) { lock (numLock) { if (cd <= 0) Debugger.Output("Bug:Set IntNumUpdateByCD.cd to " + cd.ToString() + "."); this.cd = cd; } } } }