Codebase list golang-github-denisenkom-go-mssqldb / ac00d471-cf13-4261-b0b5-dacf46dd0417/main tran.go
ac00d471-cf13-4261-b0b5-dacf46dd0417/main

Tree @ac00d471-cf13-4261-b0b5-dacf46dd0417/main (Download .tar.gz)

tran.go @ac00d471-cf13-4261-b0b5-dacf46dd0417/main — raw · history · blame 

  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
 99
100
101
102
103
104
105
106
107
108
109
110
111
package mssql

// Transaction Manager requests
// http://msdn.microsoft.com/en-us/library/dd339887.aspx

import (
	"encoding/binary"
)

const (
	tmGetDtcAddr    = 0
	tmPropagateXact = 1
	tmBeginXact     = 5
	tmPromoteXact   = 6
	tmCommitXact    = 7
	tmRollbackXact  = 8
	tmSaveXact      = 9
)

type isoLevel uint8

const (
	isolationUseCurrent     isoLevel = 0
	isolationReadUncommited          = 1
	isolationReadCommited            = 2
	isolationRepeatableRead          = 3
	isolationSerializable            = 4
	isolationSnapshot                = 5
)

func sendBeginXact(buf *tdsBuffer, headers []headerStruct, isolation isoLevel,
	name string) (err error) {
	buf.BeginPacket(packTransMgrReq)
	writeAllHeaders(buf, headers)
	var rqtype uint16 = tmBeginXact
	err = binary.Write(buf, binary.LittleEndian, &rqtype)
	if err != nil {
		return
	}
	err = binary.Write(buf, binary.LittleEndian, &isolation)
	if err != nil {
		return
	}
	err = writeBVarChar(buf, name)
	if err != nil {
		return
	}
	return buf.FinishPacket()
}

const (
	fBeginXact = 1
)

func sendCommitXact(buf *tdsBuffer, headers []headerStruct, name string, flags uint8, isolation uint8, newname string) error {
	buf.BeginPacket(packTransMgrReq)
	writeAllHeaders(buf, headers)
	var rqtype uint16 = tmCommitXact
	err := binary.Write(buf, binary.LittleEndian, &rqtype)
	if err != nil {
		return err
	}
	err = writeBVarChar(buf, name)
	if err != nil {
		return err
	}
	err = binary.Write(buf, binary.LittleEndian, &flags)
	if err != nil {
		return err
	}
	if flags&fBeginXact != 0 {
		err = binary.Write(buf, binary.LittleEndian, &isolation)
		if err != nil {
			return err
		}
		err = writeBVarChar(buf, name)
		if err != nil {
			return err
		}
	}
	return buf.FinishPacket()
}

func sendRollbackXact(buf *tdsBuffer, headers []headerStruct, name string, flags uint8, isolation uint8, newname string) error {
	buf.BeginPacket(packTransMgrReq)
	writeAllHeaders(buf, headers)
	var rqtype uint16 = tmRollbackXact
	err := binary.Write(buf, binary.LittleEndian, &rqtype)
	if err != nil {
		return err
	}
	err = writeBVarChar(buf, name)
	if err != nil {
		return err
	}
	err = binary.Write(buf, binary.LittleEndian, &flags)
	if err != nil {
		return err
	}
	if flags&fBeginXact != 0 {
		err = binary.Write(buf, binary.LittleEndian, &isolation)
		if err != nil {
			return err
		}
		err = writeBVarChar(buf, name)
		if err != nil {
			return err
		}
	}
	return buf.FinishPacket()
}