51 lines
1.6 KiB
Plaintext
51 lines
1.6 KiB
Plaintext
import { entityKind } from "../../entity.js";
|
|
import { getColumnNameAndConfig } from "../../utils.js";
|
|
import { SingleStoreColumn, SingleStoreColumnBuilder } from "./common.js";
|
|
class SingleStoreCustomColumnBuilder extends SingleStoreColumnBuilder {
|
|
static [entityKind] = "SingleStoreCustomColumnBuilder";
|
|
constructor(name, fieldConfig, customTypeParams) {
|
|
super(name, "custom", "SingleStoreCustomColumn");
|
|
this.config.fieldConfig = fieldConfig;
|
|
this.config.customTypeParams = customTypeParams;
|
|
}
|
|
/** @internal */
|
|
build(table) {
|
|
return new SingleStoreCustomColumn(
|
|
table,
|
|
this.config
|
|
);
|
|
}
|
|
}
|
|
class SingleStoreCustomColumn extends SingleStoreColumn {
|
|
static [entityKind] = "SingleStoreCustomColumn";
|
|
sqlName;
|
|
mapTo;
|
|
mapFrom;
|
|
constructor(table, config) {
|
|
super(table, config);
|
|
this.sqlName = config.customTypeParams.dataType(config.fieldConfig);
|
|
this.mapTo = config.customTypeParams.toDriver;
|
|
this.mapFrom = config.customTypeParams.fromDriver;
|
|
}
|
|
getSQLType() {
|
|
return this.sqlName;
|
|
}
|
|
mapFromDriverValue(value) {
|
|
return typeof this.mapFrom === "function" ? this.mapFrom(value) : value;
|
|
}
|
|
mapToDriverValue(value) {
|
|
return typeof this.mapTo === "function" ? this.mapTo(value) : value;
|
|
}
|
|
}
|
|
function customType(customTypeParams) {
|
|
return (a, b) => {
|
|
const { name, config } = getColumnNameAndConfig(a, b);
|
|
return new SingleStoreCustomColumnBuilder(name, config, customTypeParams);
|
|
};
|
|
}
|
|
export {
|
|
SingleStoreCustomColumn,
|
|
SingleStoreCustomColumnBuilder,
|
|
customType
|
|
};
|
|
//# sourceMappingURL=custom.js.map |