WIP: initial prototype

- Added querystring support
- Added support for already serialized bodies
This commit is contained in:
delvedor
2018-10-26 19:04:45 +02:00
parent 8429ad28d9
commit 51e6c76511
3 changed files with 34 additions and 8 deletions

View File

@ -32,6 +32,9 @@ class Connection {
request (params, callback) {
params.url = resolve(this.host.href, params.path)
if (params.querystring != null && params.querystring.length > 0) {
params.url += '?' + params.querystring
}
params.agent = this._agent
debug('Starting a new request', params)
return makeRequest(params, callback)

View File

@ -1,5 +1,6 @@
'use strict'
const { stringify } = require('querystring')
const debug = require('debug')('elasticsearch')
const { SerializationError, DeserializationError } = require('./errors')
@ -35,6 +36,19 @@ class Serializer {
}
return ndjson
}
qserialize (object) {
debug('qserialize', object)
// arrays should be serialized as comma separated list
const keys = Object.keys(object)
for (var i = 0, len = keys.length; i < len; i++) {
var key = keys[i]
if (Array.isArray(object[key]) === true) {
object[key] = object[key].join(',')
}
}
return stringify(object)
}
}
module.exports = Serializer

View File

@ -42,28 +42,37 @@ class Transport {
// handle json body
if (params.body != null) {
try {
params.body = this.serializer.serialize(params.body)
} catch (err) {
return callback(err)
if (typeof params.body !== 'string') {
try {
params.body = this.serializer.serialize(params.body)
} catch (err) {
return callback(err)
}
}
params.headers = params.headers || {}
params.headers['Content-Type'] = 'application/json'
params.headers['Content-Length'] = '' + Buffer.byteLength(params.body)
// handle ndjson body
} else if (params.bulkBody != null) {
try {
params.body = this.serializer.ndserialize(params.bulkBody)
} catch (err) {
return callback(err)
if (typeof params.bulkBody !== 'string') {
try {
params.body = this.serializer.ndserialize(params.bulkBody)
} catch (err) {
return callback(err)
}
} else {
params.body = params.bulkBody
}
params.headers = params.headers || {}
params.headers['Content-Type'] = 'application/x-ndjson'
params.headers['Content-Length'] = '' + Buffer.byteLength(params.body)
}
// serializes the querystring
params.querystring = this.serializer.qserialize(params.querystring)
params.timeout = params.timeout || this.requestTimeout
this.emit('request', params)
const request = connection.request(params, (err, response) => {
if (err != null) {
this.connectionPool.markDead(connection)