/
vector-protocol.lisp
197 lines (166 loc) · 7.51 KB
/
vector-protocol.lisp
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
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
;;; -*- Mode: lisp; Syntax: ansi-common-lisp; Base: 10; Package: org.apache.thrift.implementation; -*-
(in-package :org.apache.thrift.implementation)
;;; define a binary stream to wrap a vector for use in tests.
;;; adapted from the cl-xml version to restrict i/o to unsigned byte operations.
;;; this version uses a signed byte stream, as that's the basis of the thrift binary transport
;;;
;;;
;;; abstract
(defclass vector-stream ()
((position
:initform 0
:reader get-stream-position :writer setf-stream-position)
(vector
:reader get-vector-stream-vector :writer setf-vector-stream-vector
:type vector)
(force-output-hook
:initform nil :initarg :force-output-hook
:accessor stream-force-output-hook
:documentation "A function of one argument, the stream, called as the
base implementation of stream-force-output.")
#+(or CMU sbcl lispworks) (direction :initarg :direction)
)
(:default-initargs
#+CormanLisp :element-type #+CormanLisp 'character))
(defClass vector-input-stream (vector-stream
#+ALLEGRO excl::fundamental-binary-input-stream
#+LispWorks stream:fundamental-stream
#+(and MCL digitool) ccl::input-binary-stream
#+(and MCL openmcl) fundamental-binary-input-stream
#+CMU extensions:fundamental-binary-input-stream
#+sbcl sb-gray:fundamental-binary-input-stream
#+CormanLisp stream
)
()
(:default-initargs :direction :input))
(defClass vector-output-stream (vector-stream
#+ALLEGRO excl::fundamental-binary-output-stream
#+LispWorks stream:fundamental-stream
#+(and MCL digitool) ccl::output-binary-stream
#+(and MCL openmcl) fundamental-binary-output-stream
#+CMU extensions:fundamental-binary-output-stream
#+sbcl sb-gray:fundamental-binary-output-stream
#+CormanLisp stream
)
()
(:default-initargs :direction :output))
(defclass vector-stream-transport (vector-input-stream vector-output-stream binary-transport)
((stream :initform nil)))
(defun make-vector-stream-buffer (length &optional (type *binary-transport-element-type*))
(make-array length :element-type type :initial-element 0))
(defmethod shared-initialize
((instance vector-stream) (slots t) &key (vector nil vector-s) (length 128))
(with-slots (position) instance
(setf position 0)
(when vector-s
(setf-vector-stream-vector
(etypecase vector
(string (map-into (make-vector-stream-buffer (length vector)) #'char-code vector))
(cl:cons (map-into (make-vector-stream-buffer (length vector))
#'(lambda (datum)
(etypecase datum
(fixnum datum)
(character (char-code datum))))
vector))
(vector vector)
(null (make-vector-stream-buffer length)))
instance))
(call-next-method)
(unless (slot-boundp instance 'vector)
(setf-vector-stream-vector (make-vector-stream-buffer length) instance))))
#+cmu
(let ((old-definition (fdefinition 'stream-element-type)))
(unless (typep old-definition 'generic-function)
(fmakunbound 'stream-element-type)
(defgeneric stream-element-type (stream))
(setf (documentation 'stream-element-type 'function)
(documentation old-definition 'function))
(defmethod stream-element-type (stream)
(funcall old-definition stream))))
(defmethod stream-element-type ((stream vector-stream))
*binary-transport-element-type*)
(defmethod stream-position ((stream vector-stream) &optional new)
(with-slots (vector) stream
(if new
(setf-stream-position (min (length vector) new) stream)
(get-stream-position stream))))
(defmethod stream-eofp ((stream vector-stream))
(with-slots (position vector) stream
(>= position (length vector))))
(defmethod stream-finish-output ((stream vector-stream))
nil)
(defmethod print-object
((vs vector-stream) (stream t)
&aux (*print-array* t) (*print-length* 32) (*print-base* 16))
(print-unreadable-object (vs stream :type t)
(princ (get-vector-stream-vector vs) stream)))
(defmethod stream-force-output ((stream vector-stream))
(let ((hook (stream-force-output-hook stream)))
(when hook (funcall hook stream))))
#-mcl
(defmethod open-stream-p ((stream vector-stream))
t)
(defgeneric vector-stream-vector (vector-stream)
(:documentation "Return the written subsequence and reset the position")
(:method ((stream vector-stream))
(with-slots (vector position) stream
(prog1 (subseq vector 0 position)
(setf position 0)))))
(defgeneric (setf vector-stream-vector) (vector vector-stream)
(:method ((new-vector vector) (stream vector-stream))
(assert (equal (array-element-type new-vector) *binary-transport-element-type*) ()
"Invalid vector stream element type: ~s." (array-element-type new-vector))
(with-slots (vector position) stream
(setf position 0
vector new-vector))))
;;;
;;; input
(defmethod stream-read-byte ((stream vector-stream-transport))
(with-slots (position vector) stream
(when (< position (length vector))
(let ((byte (aref vector position)))
(incf position)
(if (> byte 127)
(- (logxor 255 (1- byte)))
byte)))))
(defmethod stream-read-sequence ((stream vector-input-stream) (sequence vector)
#+mcl &key #-mcl &optional (start 0) (end nil))
(unless end (setf end (length sequence)))
(assert (typep start '(integer 0)))
(assert (>= end start))
(with-slots (vector position) stream
(let* ((new-position (min (+ position (- end start)) (length vector))))
(when (> new-position position)
(replace sequence vector
:start1 start :end1 end
:start2 position :end2 new-position)
(setf position new-position))
new-position)))
;;;
;;; output
(defmethod stream-write-byte ((stream vector-output-stream) (datum integer) &aux next)
(with-slots (position vector) stream
(unless (< (setf next (1+ position)) (length vector))
(setf vector
(adjust-array vector (+ next (floor (/ next 4)))
:element-type *binary-transport-element-type*)))
(setf (aref vector position)
(logand #xff datum))
(setf position next)))
(defmethod stream-write-sequence ((stream vector-output-stream) (sequence vector)
#+mcl &key #-mcl &optional (start 0) (end nil))
(unless end (setf end (length sequence)))
(assert (typep start '(integer 0)))
(assert (>= end start))
(with-slots (vector position) stream
(let* ((new-position (+ position (- end start))))
(when (> new-position position)
(unless (< new-position (length vector))
(setf vector
(adjust-array vector (floor (+ new-position (floor (/ new-position 4))))
:element-type *binary-transport-element-type*)))
(replace vector sequence
:start1 position :end1 new-position
:start2 start :end2 end)
(setf position new-position))
new-position)))