Fixed Strings

Author: 1NepuNep1

internal/codegen/golang/ydb_type.go

@@ -215,7 +215,7 @@ func YDBType(req *plugin.GenerateRequest, options *opts.Options, col *plugin.Col
 			return "*uint64"
 		}
 		return "*uint64"
-	
+
 	default:
 		if strings.HasPrefix(columnType, "decimal") {
 			if notNull {
@@ -227,7 +227,7 @@ func YDBType(req *plugin.GenerateRequest, options *opts.Options, col *plugin.Col
 			}
 			return "*types.Decimal"
 		}
-		
+
 		if debug.Active {
 			log.Printf("unknown YDB type: %s\n", columnType)
 		}

internal/engine/ydb/convert.go

@@ -3523,11 +3523,14 @@ func (c *cc) VisitLiteral_value(n *parser.Literal_valueContext) interface{} {
 		return &ast.A_Const{Val: &ast.Float{Str: text}, Location: c.pos(n.GetStart())}
 
 	case n.STRING_VALUE() != nil: // !!! debug !!! (problem with quoted strings)
-		val := n.STRING_VALUE().GetText()
-		if len(val) >= 2 {
-			val = val[1 : len(val)-1]
+		originalText := n.STRING_VALUE().GetText()
+		content, hasSuffix := parseStringValue(originalText)
+
+		if hasSuffix {
+			return &ast.A_Const{Val: &ast.String{Str: originalText}, Location: c.pos(n.GetStart())}
+		} else {
+			return &ast.A_Const{Val: &ast.String{Str: content}, Location: c.pos(n.GetStart())}
 		}
-		return &ast.A_Const{Val: &ast.String{Str: val}, Location: c.pos(n.GetStart())}
 
 	case n.Bool_value() != nil:
 		var i bool

internal/engine/ydb/utils.go

@@ -295,3 +295,48 @@ func (c *cc) collectEqualityOps(ctx parser.ICond_exprContext) []antlr.TerminalNo
 	}
 	return ops
 }
+
+
+func parseStringValue(s string) (value string, hasSuffix bool) {
+	if len(s) < 2 {
+		return s, false
+	}
+
+	quote := s[0]
+	if quote != '\'' && quote != '"' {
+		return s, false
+	}
+
+	quotePos := -1
+	for i := len(s) - 1; i >= 0; i-- {
+		if s[i] == quote {
+			quotePos = i
+			break
+		}
+	}
+
+	if quotePos == -1 || quotePos == 0 {
+		return s, false
+	}
+
+	content := s[1:quotePos]
+
+	if quotePos < len(s)-1 {
+		suffix := s[quotePos+1:]
+		if isValidYDBStringSuffix(suffix) {
+			return content, true
+		}
+	}
+
+	return content, false
+}
+
+func isValidYDBStringSuffix(suffix string) bool {
+	switch suffix {
+	case "s", "S", "u", "U", "y", "Y", "j", "J",
+		"pt", "PT", "pb", "PB", "pv", "PV":
+		return true
+	default:
+		return false
+	}
+}