(ctx expression.BuildContext, col *model.ColumnInfo, defaultVal any, args *getColOriginDefaultValue)
| 621 | } |
| 622 | |
| 623 | func 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 | |
| 668 | func getColDefaultValueFromNil(ctx expression.BuildContext, col *model.ColumnInfo, args *getColOriginDefaultValue) (types.Datum, error) { |
| 669 | if !mysql.HasNotNullFlag(col.GetFlag()) { |
no test coverage detected