MCPcopy
hub / github.com/pingcap/tidb / getColDefaultValue

Function getColDefaultValue

pkg/table/column.go:623–666  ·  view source on GitHub ↗
(ctx expression.BuildContext, col *model.ColumnInfo, defaultVal any, args *getColOriginDefaultValue)

Source from the content-addressed store, hash-verified

621}
622
623func getColDefaultValue(ctx expression.BuildContext, col *model.ColumnInfo, defaultVal any, args *getColOriginDefaultValue) (types.Datum, error) {
624 if defaultVal == nil {
625 return getColDefaultValueFromNil(ctx, col, args)
626 }
627
628 switch col.GetType() {
629 case mysql.TypeTimestamp, mysql.TypeDate, mysql.TypeDatetime:
630 default:
631 value, err := CastColumnValue(ctx, types.NewDatum(defaultVal), col, false, false)
632 if err != nil {
633 return types.Datum{}, err
634 }
635 return value, nil
636 }
637
638 // Check and get timestamp/datetime default value.
639 var needChangeTimeZone bool
640 var explicitTz *time.Location
641 // If the column's default value is not ZeroDatetimeStr nor CurrentTimestamp, should use the time zone of the default value itself.
642 if col.GetType() == mysql.TypeTimestamp {
643 if vv, ok := defaultVal.(string); ok && vv != types.ZeroDatetimeStr && !strings.EqualFold(vv, ast.CurrentTimestamp) {
644 needChangeTimeZone = true
645 // For col.Version = 0, the timezone information of default value is already lost, so use the system timezone as the default value timezone.
646 explicitTz = timeutil.SystemLocation()
647 if col.Version >= model.ColumnInfoVersion1 {
648 explicitTz = time.UTC
649 }
650 }
651 }
652 value, err := expression.GetTimeValue(ctx, defaultVal, col.GetType(), col.GetDecimal(), explicitTz)
653 if err != nil {
654 return types.Datum{}, errGetDefaultFailed.GenWithStackByArgs(col.Name)
655 }
656 // If the column's default value is not ZeroDatetimeStr or CurrentTimestamp, convert the default value to the current session time zone.
657 if needChangeTimeZone {
658 t := value.GetMysqlTime()
659 err = t.ConvertTimeZone(explicitTz, ctx.GetEvalCtx().Location())
660 if err != nil {
661 return value, err
662 }
663 value.SetMysqlTime(t)
664 }
665 return value, nil
666}
667
668func getColDefaultValueFromNil(ctx expression.BuildContext, col *model.ColumnInfo, args *getColOriginDefaultValue) (types.Datum, error) {
669 if !mysql.HasNotNullFlag(col.GetFlag()) {

Calls 12

NewDatumFunction · 0.92
SystemLocationFunction · 0.92
GetTimeValueFunction · 0.92
CastColumnValueFunction · 0.85
GetMysqlTimeMethod · 0.80
ConvertTimeZoneMethod · 0.80
SetMysqlTimeMethod · 0.80
GetTypeMethod · 0.65
LocationMethod · 0.65
GetEvalCtxMethod · 0.65
GetDecimalMethod · 0.45

Tested by

no test coverage detected