private void StoreSnapShot(ISnapShot snapShot)
{
const string commandText = "INSERT OR REPLACE INTO SnapShots (EventProviderId, SnapShot, Version) VALUES (@eventProviderId, @snapShot, @version);";
using (var sqliteConnection = new SQLiteConnection(_sqLiteConnectionString))
{
sqliteConnection.Open();
using (var sqliteTransaction = sqliteConnection.BeginTransaction())
{
try
{
using (var sqliteCommand = new SQLiteCommand(commandText, sqliteTransaction.Connection, sqliteTransaction))
{
sqliteCommand.Parameters.Add(new SQLiteParameter("@eventProviderId", snapShot.EventProviderId));
sqliteCommand.Parameters.Add(new SQLiteParameter("@snapShot", Serialize(snapShot)));
sqliteCommand.Parameters.Add(new SQLiteParameter("@version", snapShot.Version));
sqliteCommand.ExecuteNonQuery();
}
sqliteTransaction.Commit();
}
catch (Exception)
{
sqliteTransaction.Rollback();
throw;
}
}
}
}
public ISnapShot GetSnapShot(Guid eventProviderId)
{
ISnapShot snapshot = null;
const string commandText = @"SELECT SnapShot FROM SnapShots WHERE EventProviderId = @eventProviderId AND Version != -1;";
using (var sqliteConnection = new SQLiteConnection(_sqLiteConnectionString))
{
sqliteConnection.Open();
using (var sqliteTransaction = sqliteConnection.BeginTransaction())
{
try
{
using (var sqliteCommand = new SQLiteCommand(commandText, sqliteTransaction.Connection, sqliteTransaction))
{
sqliteCommand.Parameters.Add(new SQLiteParameter("@eventProviderId", eventProviderId));
var bytes = (byte[])sqliteCommand.ExecuteScalar();
if (bytes != null)
{
snapshot = Deserialize <ISnapShot>(bytes);
}
}
sqliteTransaction.Commit();
}
catch (Exception)
{
sqliteTransaction.Rollback();
throw;
}
}
}
return(snapshot);
}
public SimpleDbSnapShot(DB db)
{
this.db = db;
this.snapshot = db.UseSnapShot();
this.batch = db.CreateWriteBatch();
Logs = new SimpleDbCache <LogKey, Log>(this.db, this.batch, TableId.DATA_Log);
CareAddrs = new SimpleDbCache <UInt160, UInt160>(this.db, this.batch, TableId.DATA_CareAddr);
CareAssets = new SimpleDbCache <UInt160, UInt160>(this.db, this.batch, TableId.DATA_CareAsset);
CareEvents = new SimpleDbCache <UInt256, CareEvent>(this.db, this.batch, TableId.DATA_CareEvent);
Transfers = new SimpleDbCache <TransferKey, TransferGroup>(this.db, this.batch, TableId.DATA_Transfer);
TransferBlockNumberList = new SimpleDbCache <UInt160, TransferBlockNumberList>(this.db, this.batch, TableId.DATA_TransferBlockNumberList);
BlockNumber = new SimpleDbCache <UInt32Wrapper, UInt32Wrapper>(this.db, this.batch, TableId.DATA_BlockNumber);
}
public ISnapShotObject(ISnapShot ISnapShotinstance)
{
ISnapShotInstance = ISnapShotinstance;
}
public NetMessage Process(UInt64 peerid, byte[] data)
{
using (MemoryStream ms = new MemoryStream(data))
{
NetMessage netMsg = NetMessage.Unpack(ms);
string cmd = netMsg.Cmd;
string id = netMsg.ID;
NetMessage netMsgBack;
try
{
switch (cmd)
{
case "_db.usesnapshot":
{
var snapshot = StorageService.maindb.UseSnapShot();
peerSnapshots[snapshot.DataHeight] = StorageService.maindb.UseSnapShot();
var p = new Param()
{
result = true, snapid = snapshot.DataHeight
};
netMsgBack = NetMessage.Create(cmd, p, id);
return(netMsgBack);
}
case "_db.disposeSnapshot":
{
ISnapShot snapshot;
peerSnapshots.TryRemove(netMsg.Param.snapid, out snapshot);
snapshot.Dispose();
var p = new Param()
{
result = true
};
netMsgBack = NetMessage.Create(cmd, p, id);
return(netMsgBack);
}
case "_db.CreateWriteBatch":
{
var wb = StorageService.maindb.CreateWriteBatch();
peerWriteBatch[wb.Wbid] = wb;
var p = new Param()
{
result = true, wbid = wb.Wbid
};
netMsgBack = NetMessage.Create(cmd, p, id);
return(netMsgBack);
}
case "_db.put":
{
IWriteBatch writeBatch = peerWriteBatch[netMsg.Param.wbid];
writeBatch.Put(netMsg.Param.tableid, netMsg.Param.key, netMsg.Param.value);
var p = new Param()
{
result = true
};
netMsgBack = NetMessage.Create(cmd, p, id);
return(netMsgBack);
}
case "_db.delete":
{
IWriteBatch writeBatch = peerWriteBatch[netMsg.Param.wbid];
writeBatch.Delete(netMsg.Param.tableid, netMsg.Param.key);
var p = new Param()
{
result = true
};
netMsgBack = NetMessage.Create(cmd, p, id);
return(netMsgBack);
}
case "_db.write":
{
IWriteBatch writeBatch = peerWriteBatch[netMsg.Param.wbid];
var p = new Param()
{
result = false
};
if (writeBatch.wbcount > 0)
{
StorageService.maindb.WriteBatch(writeBatch);
peerWriteBatch.Remove(netMsg.Param.wbid, out writeBatch);
p = new Param()
{
result = true
};
}
netMsgBack = NetMessage.Create(cmd, p, id);
return(netMsgBack);
}
case "_db.getvalue": //使用最新的snapshot 基本就是给apiserver用的
{
ISnapShot snapshot = StorageService.maindb.UseSnapShot();
var tableid = netMsg.Param.tableid;
var key = netMsg.Param.key;
var value = snapshot.GetValueData(tableid, key);
Param param = new Param();
param.snapid = snapshot.DataHeight;
param.result = true;
param.value = value;
param.key = key;
param.tableid = tableid;
netMsgBack = NetMessage.Create(cmd, param, id);
return(netMsgBack);
}
case "_db.snapshot.getvalue":
{
ISnapShot snapshot = peerSnapshots[netMsg.Param.snapid];
var tableid = netMsg.Param.tableid;
var key = netMsg.Param.key;
var value = snapshot.GetValueData(tableid, key);
Param param = new Param();
param.snapid = snapshot.DataHeight;
param.result = true;
param.value = value;
param.key = key;
param.tableid = tableid;
netMsgBack = NetMessage.Create(cmd, param, id);
return(netMsgBack);
}
case "_db.snapshot.newiterator":
{
ISnapShot snapshot = peerSnapshots[netMsg.Param.snapid];
var beginKey = netMsg.Param.key;
var endKey = netMsg.Param.value;
var tableid = netMsg.Param.tableid;
var iter = snapshot.CreateKeyIterator(tableid, beginKey, endKey);
var itid = iter.HandleID;
peerKeyIterator[itid] = iter;
var p = new Param()
{
result = true, itid = itid, snapid = netMsg.Param.snapid
};
netMsgBack = NetMessage.Create(cmd, p, id);
return(netMsgBack);
}
case "_db.iterator.current":
{
IKeyIterator keyIterator = peerKeyIterator[netMsg.Param.itid];
var cur = keyIterator.Current;
Param param = new Param();
param.itid = netMsg.Param.itid;
param.result = true;
param.value = cur;
netMsgBack = NetMessage.Create(cmd, param, id);
return(netMsgBack);
}
case "_db.iterator.next":
{
IKeyIterator keyIterator = peerKeyIterator[netMsg.Param.itid];
var result = keyIterator.MoveNext();
Param param = new Param();
param.itid = netMsg.Param.itid;
param.result = result;
netMsgBack = NetMessage.Create(cmd, param, id);
return(netMsgBack);
}
case "_db.iterator.seektofirst":
{
IKeyIterator keyIterator = peerKeyIterator[netMsg.Param.itid];
keyIterator.SeekToFirst();
var p = new Param()
{
result = true, itid = netMsg.Param.itid
};
netMsgBack = NetMessage.Create(cmd, p, id);
return(netMsgBack);
}
case "_db.iterator.reset":
{
IKeyIterator keyIterator = peerKeyIterator[netMsg.Param.itid];
keyIterator.Reset();
Param param = new Param();
param.result = true;
param.itid = netMsg.Param.itid;
netMsgBack = NetMessage.Create(cmd, param, id);
return(netMsgBack);
}
default:
throw new Exception("unknown msg cmd:" + netMsg.Cmd);
}
}
catch (Exception e)
{
netMsgBack = NetMessage.Create(cmd, new Param()
{
error = Encoding.UTF8.GetBytes(e.Message)
}, id);
return(netMsgBack);
}
}
}
