custom_types.go

package go_ora

import (
	"database/sql/driver"
	"encoding/json"
)

type ValueEncoder interface {
	EncodeValue(param *ParameterInfo, connection *Connection) error
}
type NVarChar string

func (val *NVarChar) Value() (driver.Value, error) {
	return val, nil
}
func (val *NVarChar) Scan(value interface{}) error {
	*val = NVarChar(getString(value))
	return nil
}
func (val NVarChar) MarshalJSON() ([]byte, error) {
	return json.Marshal(string(val))
}
func (val *NVarChar) UnmarshalJSON(data []byte) error {
	var temp string
	err := json.Unmarshal(data, &temp)
	if err != nil {
		return err
	}
	*val = NVarChar(temp)
	return nil
}

//func (val *NVarChar) ValueDecoder(buffer []byte) error {
//
//}
//func (val *NVarChar) EncodeValue(param *ParameterInfo, connection *Connection) error {
//	strVal := string(*val)
//	param.DataType = NCHAR
//	param.CharsetForm = 2
//	param.ContFlag = 0x10
//	param.CharsetID = connection.tcpNego.ServernCharset
//	param.MaxCharLen = len([]rune(strVal))
//	if len(string(*val)) == 0 {
//		param.BValue = nil
//	}
//	if param.CharsetID != connection.strConv.GetLangID() {
//		tempCharset := connection.strConv.SetLangID(param.CharsetID)
//		defer connection.strConv.SetLangID(tempCharset)
//		param.BValue = connection.strConv.Encode(strVal)
//	} else {
//		param.BValue = connection.strConv.Encode(strVal)
//	}
//	if param.MaxLen < len(param.BValue) {
//		param.MaxLen = len(param.BValue)
//	}
//	if param.BValue == nil && param.MaxLen == 0 {
//		param.MaxLen = 1
//	}
//	if (param.Direction == Output && param.MaxLen == 0) || param.MaxLen > converters.MAX_LEN_NVARCHAR2 {
//		param.MaxLen = converters.MAX_LEN_NVARCHAR2
//	}
//	return nil
//}
//func (val *TimeStamp) EncodeValue(param *ParameterInfo, _ *Connection) error {
//	param.setForTime()
//	param.DataType = TIMESTAMP
//	param.BValue = converters.EncodeTimeStamp(time.Time(*val))
//	return nil
//}

type NullNVarChar struct {
	NVarChar NVarChar
	Valid    bool
}

func (val NullNVarChar) Value() (driver.Value, error) {
	if val.Valid {
		return val.NVarChar.Value()
	} else {
		return nil, nil
	}
}

func (val *NullNVarChar) Scan(value interface{}) error {
	if value == nil {
		val.Valid = false
		return nil
	}
	val.Valid = true
	return val.NVarChar.Scan(value)
}

func (val NullNVarChar) MarshalJSON() ([]byte, error) {
	if val.Valid {
		return json.Marshal(string(val.NVarChar))
	}
	return json.Marshal(nil)
}

func (val *NullNVarChar) UnmarshalJSON(data []byte) error {
	var temp = new(string)
	err := json.Unmarshal(data, temp)
	if err != nil {
		return err
	}
	if temp == nil {
		val.Valid = false
	} else {
		val.Valid = true
		val.NVarChar = NVarChar(*temp)
	}
	return nil
}

func (val NClob) MarshalJSON() ([]byte, error) {
	if val.Valid {
		return json.Marshal(val.String)
	}
	return json.Marshal(nil)
}

func (val *NClob) UnmarshalJSON(data []byte) error {
	var temp = new(string)
	err := json.Unmarshal(data, temp)
	if err != nil {
		return err
	}
	if temp == nil {
		val.Valid = false
	} else {
		val.Valid = true
		val.String = *temp
	}
	return nil
}

func (val Clob) MarshalJSON() ([]byte, error) {
	if val.Valid {
		return json.Marshal(val.String)
	}
	return json.Marshal(nil)
}

func (val *Clob) UnmarshalJSON(data []byte) error {
	var temp = new(string)
	err := json.Unmarshal(data, temp)
	if err != nil {
		return err
	}
	if temp == nil {
		val.Valid = false
	} else {
		val.Valid = true
		val.String = *temp
	}
	return nil
}

func (val Blob) MarshalJSON() ([]byte, error) {
	if val.Valid {
		return json.Marshal(val.Data)
	}
	return json.Marshal(nil)
}

func (val *Blob) UnmarshalJSON(data []byte) error {
	var temp = new([]byte)
	err := json.Unmarshal(data, temp)
	if err != nil {
		return err
	}
	if temp == nil {
		val.Valid = false
	} else {
		val.Valid = true
		val.Data = *temp
	}
	return nil
}