39 lines
979 B
JavaScript
39 lines
979 B
JavaScript
/**
|
|
* Copyright (c) 2010-2017 Brian Carlson (brian.m.carlson@gmail.com)
|
|
* All rights reserved.
|
|
*
|
|
* This source code is licensed under the MIT license found in the
|
|
* README.md file in the root directory of this source tree.
|
|
*/
|
|
|
|
var types = require('pg-types');
|
|
|
|
function TypeOverrides(userTypes) {
|
|
this._types = userTypes || types;
|
|
this.text = {};
|
|
this.binary = {};
|
|
}
|
|
|
|
TypeOverrides.prototype.getOverrides = function(format) {
|
|
switch(format) {
|
|
case 'text': return this.text;
|
|
case 'binary': return this.binary;
|
|
default: return {};
|
|
}
|
|
};
|
|
|
|
TypeOverrides.prototype.setTypeParser = function(oid, format, parseFn) {
|
|
if(typeof format == 'function') {
|
|
parseFn = format;
|
|
format = 'text';
|
|
}
|
|
this.getOverrides(format)[oid] = parseFn;
|
|
};
|
|
|
|
TypeOverrides.prototype.getTypeParser = function(oid, format) {
|
|
format = format || 'text';
|
|
return this.getOverrides(format)[oid] || this._types.getTypeParser(oid, format);
|
|
};
|
|
|
|
module.exports = TypeOverrides;
|