-
Notifications
You must be signed in to change notification settings - Fork 874
Expand file tree
/
Copy pathCopyMessages.cs
More file actions
91 lines (74 loc) · 2.39 KB
/
CopyMessages.cs
File metadata and controls
91 lines (74 loc) · 2.39 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
using System;
using System.Collections.Generic;
using Npgsql.Internal;
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<DataFormat> ColumnFormatCodes { get; }
internal CopyResponseMessageBase()
=> ColumnFormatCodes = [];
internal void Load(NpgsqlReadBuffer buf)
{
ColumnFormatCodes.Clear();
var binaryIndicator = buf.ReadByte();
IsBinary = binaryIndicator switch
{
0 => false,
1 => true,
_ => throw new Exception("Invalid binary indicator in CopyInResponse message: " + binaryIndicator)
};
NumColumns = buf.ReadInt16();
for (var i = 0; i < NumColumns; i++)
ColumnFormatCodes.Add(DataFormatUtils.Create(buf.ReadInt16()));
}
}
sealed class CopyInResponseMessage : CopyResponseMessageBase
{
public override BackendMessageCode Code => BackendMessageCode.CopyInResponse;
internal new CopyInResponseMessage Load(NpgsqlReadBuffer buf)
{
base.Load(buf);
return this;
}
}
sealed class CopyOutResponseMessage : CopyResponseMessageBase
{
public override BackendMessageCode Code => BackendMessageCode.CopyOutResponse;
internal new CopyOutResponseMessage Load(NpgsqlReadBuffer buf)
{
base.Load(buf);
return this;
}
}
sealed 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>
sealed class CopyDataMessage : IBackendMessage
{
public BackendMessageCode Code => BackendMessageCode.CopyData;
public int Length { get; private set; }
internal CopyDataMessage Load(int len)
{
Length = len;
return this;
}
}
sealed class CopyDoneMessage : IBackendMessage
{
public BackendMessageCode Code => BackendMessageCode.CopyDone;
internal static readonly CopyDoneMessage Instance = new();
CopyDoneMessage() { }
}