forked from npgsql/npgsql
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathCopyMessages.cs
More file actions
98 lines (82 loc) · 2.76 KB
/
CopyMessages.cs
File metadata and controls
98 lines (82 loc) · 2.76 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
using System;
using System.Collections.Generic;
using Npgsql.Util;
namespace Npgsql.BackendMessages
{
abstract class CopyResponseMessageBase : IBackendMessage
{
public abstract BackendMessageCode Code { get; }
internal bool IsBinary { get; private set; }
internal short NumColumns { get; private set; }
internal List<FormatCode> ColumnFormatCodes { get; }
internal CopyResponseMessageBase()
{
ColumnFormatCodes = new List<FormatCode>();
}
internal void Load(NpgsqlReadBuffer buf)
{
ColumnFormatCodes.Clear();
var binaryIndicator = buf.ReadByte();
switch (binaryIndicator) {
case 0:
IsBinary = false;
break;
case 1:
IsBinary = true;
break;
default:
throw new Exception("Invalid binary indicator in CopyInResponse message: " + binaryIndicator);
}
NumColumns = buf.ReadInt16();
for (var i = 0; i < NumColumns; i++)
ColumnFormatCodes.Add((FormatCode)buf.ReadInt16());
}
}
class CopyInResponseMessage : CopyResponseMessageBase
{
public override BackendMessageCode Code => BackendMessageCode.CopyInResponse;
internal new CopyInResponseMessage Load(NpgsqlReadBuffer buf)
{
base.Load(buf);
return this;
}
}
class CopyOutResponseMessage : CopyResponseMessageBase
{
public override BackendMessageCode Code => BackendMessageCode.CopyOutResponse;
internal new CopyOutResponseMessage Load(NpgsqlReadBuffer buf)
{
base.Load(buf);
return this;
}
}
class CopyBothResponseMessage : CopyResponseMessageBase
{
public override BackendMessageCode Code => BackendMessageCode.CopyBothResponse;
internal new CopyBothResponseMessage Load(NpgsqlReadBuffer buf)
{
base.Load(buf);
return this;
}
}
/// <summary>
/// Note that this message doesn't actually contain the data, but only the length. Data is processed
/// directly from the connector's buffer.
/// </summary>
class CopyDataMessage : IBackendMessage
{
public BackendMessageCode Code => BackendMessageCode.CopyData;
public int Length { get; private set; }
internal CopyDataMessage Load(int len)
{
Length = len;
return this;
}
}
class CopyDoneMessage : IBackendMessage
{
public BackendMessageCode Code => BackendMessageCode.CopyDone;
internal static readonly CopyDoneMessage Instance = new CopyDoneMessage();
CopyDoneMessage() { }
}
}