You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
314 lines
10 KiB
314 lines
10 KiB
#if MYSQL_6_9
|
|
|
|
// Copyright © 2004, 2013, Oracle and/or its affiliates. All rights reserved.
|
|
//
|
|
// MySQL Connector/NET is licensed under the terms of the GPLv2
|
|
// <http://www.gnu.org/licenses/old-licenses/gpl-2.0.html>, like most
|
|
// MySQL Connectors. There are special exceptions to the terms and
|
|
// conditions of the GPLv2 as it is applied to this software, see the
|
|
// FLOSS License Exception
|
|
// <http://www.mysql.com/about/legal/licensing/foss-exception.html>.
|
|
//
|
|
// This program is free software; you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published
|
|
// by the Free Software Foundation; version 2 of the License.
|
|
//
|
|
// This program is distributed in the hope that it will be useful, but
|
|
// WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
|
|
// or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
|
|
// for more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License along
|
|
// with this program; if not, write to the Free Software Foundation, Inc.,
|
|
// 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
|
|
|
|
namespace Externals.MySql.Data.MySqlClient.Memcached
|
|
{
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Text;
|
|
using System.IO;
|
|
|
|
/// <summary>
|
|
/// Implementation of the Memcached text client protocol.
|
|
/// </summary>
|
|
internal class TextClient : Client
|
|
{
|
|
private Encoding encoding;
|
|
|
|
private static readonly string PROTOCOL_ADD = "add";
|
|
private static readonly string PROTOCOL_APPEND = "append";
|
|
private static readonly string PROTOCOL_CAS = "cas";
|
|
private static readonly string PROTOCOL_DECREMENT = "decr";
|
|
private static readonly string PROTOCOL_DELETE = "delete";
|
|
private static readonly string PROTOCOL_FLUSHALL = "flush_all";
|
|
// private static readonly string PROTOCOL_GET = "get";
|
|
private static readonly string PROTOCOL_GETS = "gets";
|
|
private static readonly string PROTOCOL_INCREMENT = "incr";
|
|
private static readonly string PROTOCOL_PREPEND = "prepend";
|
|
private static readonly string PROTOCOL_REPLACE = "replace";
|
|
private static readonly string PROTOCOL_SET = "set";
|
|
|
|
private static readonly string VALUE = "VALUE";
|
|
private static readonly string END = "END";
|
|
// Errors
|
|
private static readonly string ERR_ERROR = "ERROR";
|
|
private static readonly string ERR_CLIENT_ERROR = "CLIENT_ERROR";
|
|
private static readonly string ERR_SERVER_ERROR = "SERVER_ERROR";
|
|
|
|
internal protected TextClient( string server, uint port ) : base( server, port )
|
|
{
|
|
encoding = Encoding.UTF8;
|
|
}
|
|
|
|
#region Memcached protocol interface
|
|
|
|
public override void Add(string key, object data, TimeSpan expiration)
|
|
{
|
|
SendCommand(PROTOCOL_ADD, key, data, expiration);
|
|
}
|
|
|
|
public override void Append(string key, object data )
|
|
{
|
|
SendCommand(PROTOCOL_APPEND, key, data);
|
|
}
|
|
|
|
public override void Cas(string key, object data, TimeSpan expiration, ulong casUnique)
|
|
{
|
|
SendCommand(PROTOCOL_CAS, key, data, expiration, casUnique);
|
|
}
|
|
|
|
public override void Decrement(string key, int amount)
|
|
{
|
|
SendCommand(PROTOCOL_DECREMENT, key, amount);
|
|
}
|
|
|
|
public override void Delete(string key)
|
|
{
|
|
SendCommand(PROTOCOL_DELETE, key);
|
|
}
|
|
|
|
public override void FlushAll(TimeSpan delay)
|
|
{
|
|
SendCommand(PROTOCOL_FLUSHALL, delay);
|
|
}
|
|
|
|
public override KeyValuePair<string, object> Get(string key)
|
|
{
|
|
KeyValuePair<string, object>[] kvp = Gets(key);
|
|
if (kvp.Length == 0)
|
|
throw new MemcachedException("Item does not exists.");
|
|
else
|
|
return kvp[0];
|
|
}
|
|
|
|
private KeyValuePair<string, object>[] Gets(params string[] keys)
|
|
{
|
|
StringBuilder sb = new StringBuilder();
|
|
sb.Append(string.Format("{0}", PROTOCOL_GETS));
|
|
for (int i = 0; i < keys.Length; i++)
|
|
{
|
|
sb.Append(string.Format(" {0}", keys[i]));
|
|
}
|
|
sb.Append("\r\n");
|
|
SendData(sb.ToString());
|
|
byte[] res = GetResponse();
|
|
return ParseGetResponse(res);
|
|
}
|
|
|
|
public override void Increment(string key, int amount)
|
|
{
|
|
SendCommand(PROTOCOL_INCREMENT, key, amount);
|
|
}
|
|
|
|
public override void Prepend(string key, object data)
|
|
{
|
|
SendCommand(PROTOCOL_PREPEND, key, data);
|
|
}
|
|
|
|
public override void Replace(string key, object data, TimeSpan expiration)
|
|
{
|
|
SendCommand(PROTOCOL_REPLACE, key, data, expiration);
|
|
}
|
|
|
|
public override void Set(string key, object data, TimeSpan expiration)
|
|
{
|
|
SendCommand(PROTOCOL_SET, key, data, expiration);
|
|
}
|
|
|
|
#endregion
|
|
|
|
#region Support methods
|
|
|
|
|
|
/// <summary>
|
|
/// Sends a command to the memcached server.
|
|
/// </summary>
|
|
/// <remarks>This version is for commands that take a key, data, expiration and casUnique.</remarks>
|
|
private void SendCommand(string cmd, string key, object data, TimeSpan expiration, ulong casUnique)
|
|
{
|
|
StringBuilder sb = new StringBuilder();
|
|
// set key flags exptime
|
|
sb.Append(string.Format("{0} {1} 0 {2} ", cmd, key, (int)(expiration.TotalSeconds)));
|
|
byte[] buf = encoding.GetBytes(data.ToString());
|
|
string s = encoding.GetString(buf, 0, buf.Length);
|
|
sb.Append(s.Length.ToString());
|
|
sb.AppendFormat(" {0}", casUnique);
|
|
sb.Append("\r\n");
|
|
sb.Append(s);
|
|
sb.Append("\r\n");
|
|
SendData(sb.ToString());
|
|
GetResponse();
|
|
}
|
|
|
|
/// <summary>
|
|
/// Sends a command to the memcached server.
|
|
/// </summary>
|
|
/// <param name="cmd"></param>
|
|
/// <param name="key"></param>
|
|
/// <param name="data"></param>
|
|
/// <param name="expiration"></param>
|
|
/// <remarks>This version is for commands that take a key, data and expiration</remarks>
|
|
private void SendCommand(string cmd, string key, object data, TimeSpan expiration)
|
|
{
|
|
StringBuilder sb = new StringBuilder();
|
|
// set key flags exptime
|
|
sb.Append(string.Format("{0} {1} 0 {2} ", cmd, key, (int)(expiration.TotalSeconds)));
|
|
byte[] buf = encoding.GetBytes(data.ToString());
|
|
string s = encoding.GetString(buf, 0, buf.Length);
|
|
sb.Append(s.Length.ToString());
|
|
sb.Append("\r\n");
|
|
sb.Append(s);
|
|
sb.Append("\r\n");
|
|
SendData(sb.ToString());
|
|
GetResponse();
|
|
}
|
|
|
|
/// <summary>
|
|
/// Send a command to memcached server.
|
|
/// </summary>
|
|
/// <remarks>This version is for commands that don't need flags neither expiration fields.</remarks>
|
|
private void SendCommand(string cmd, string key, object data )
|
|
{
|
|
StringBuilder sb = new StringBuilder();
|
|
// set key
|
|
sb.Append(string.Format("{0} {1} ", cmd, key ));
|
|
byte[] buf = encoding.GetBytes(data.ToString());
|
|
string s = encoding.GetString(buf, 0, buf.Length);
|
|
if ((cmd == PROTOCOL_APPEND) || (cmd == PROTOCOL_PREPEND))
|
|
{
|
|
sb.Append("0 0 ");
|
|
}
|
|
sb.Append(s.Length.ToString());
|
|
sb.Append("\r\n");
|
|
sb.Append(s);
|
|
sb.Append("\r\n");
|
|
SendData(sb.ToString());
|
|
GetResponse();
|
|
}
|
|
|
|
/// <summary>
|
|
/// Sends a command to the server.
|
|
/// </summary>
|
|
/// <param name="cmd"></param>
|
|
/// <param name="key"></param>
|
|
/// <remarks>This version is for commands that only require a key</remarks>
|
|
private void SendCommand(string cmd, string key )
|
|
{
|
|
StringBuilder sb = new StringBuilder();
|
|
// set key
|
|
sb.Append(string.Format("{0} {1} ", cmd, key ));
|
|
sb.Append("\r\n");
|
|
SendData(sb.ToString());
|
|
GetResponse();
|
|
}
|
|
|
|
/// <summary>
|
|
/// Sends a command to the server.
|
|
/// </summary>
|
|
/// <remarks>This version is for commands that only require a key and an integer value.</remarks>
|
|
private void SendCommand(string cmd, string key, int amount )
|
|
{
|
|
StringBuilder sb = new StringBuilder();
|
|
// set key
|
|
sb.Append(string.Format("{0} {1} {2}", cmd, key, amount));
|
|
sb.Append("\r\n");
|
|
SendData(sb.ToString());
|
|
GetResponse();
|
|
}
|
|
|
|
/// <summary>
|
|
/// Sends a command to the server.
|
|
/// </summary>
|
|
/// <remarks>This version is for commands that only require a key and expiration.</remarks>
|
|
private void SendCommand(string cmd, TimeSpan expiration)
|
|
{
|
|
StringBuilder sb = new StringBuilder();
|
|
sb.Append(string.Format("{0} {1}\r\n", PROTOCOL_FLUSHALL, expiration.TotalSeconds));
|
|
SendData(sb.ToString());
|
|
GetResponse();
|
|
}
|
|
|
|
private void ValidateErrorResponse(byte[] res)
|
|
{
|
|
string s = encoding.GetString(res, 0, res.Length);
|
|
if ((s.StartsWith(ERR_ERROR, StringComparison.OrdinalIgnoreCase)) ||
|
|
(s.StartsWith(ERR_CLIENT_ERROR, StringComparison.OrdinalIgnoreCase)) ||
|
|
(s.StartsWith(ERR_SERVER_ERROR, StringComparison.OrdinalIgnoreCase)))
|
|
{
|
|
throw new MemcachedException(s);
|
|
}
|
|
}
|
|
|
|
private void SendData(string sData)
|
|
{
|
|
byte[] data = encoding.GetBytes(sData);
|
|
stream.Write(data, 0, data.Length);
|
|
}
|
|
|
|
private KeyValuePair<string, object>[] ParseGetResponse(byte[] input)
|
|
{
|
|
// VALUE key2 10 9 2\r\n111222333\r\nEND\r\n
|
|
string[] sInput = encoding.GetString(input, 0, input.Length).Split(new string[] { "\r\n" }, StringSplitOptions.None);
|
|
List<KeyValuePair<string, object>> l = new List<KeyValuePair<string, object>>();
|
|
int i = 0;
|
|
string key = "";
|
|
KeyValuePair<string, object> kvp;
|
|
while ((sInput[i] != END) && (i < sInput.Length))
|
|
{
|
|
if (sInput[i].StartsWith(VALUE, StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
key = sInput[i].Split(' ')[1];
|
|
}
|
|
else
|
|
{
|
|
kvp = new KeyValuePair<string, object>(key, sInput[i]);
|
|
l.Add(kvp);
|
|
}
|
|
i++;
|
|
}
|
|
return l.ToArray();
|
|
}
|
|
|
|
private byte[] GetResponse()
|
|
{
|
|
byte[] res = new byte[1024];
|
|
MemoryStream ms = new MemoryStream();
|
|
int cnt = stream.Read(res, 0, 1024);
|
|
while (cnt > 0)
|
|
{
|
|
ms.Write(res, 0, cnt);
|
|
if (cnt < 1024) break;
|
|
cnt = stream.Read(res, 0, 1024);
|
|
}
|
|
byte[] res2 = ms.ToArray();
|
|
ValidateErrorResponse(res2);
|
|
return res2;
|
|
}
|
|
|
|
#endregion
|
|
}
|
|
}
|
|
|
|
|
|
#endif
|
|
|