feat: implement sponge commands

pkg/sql2code/parser/option.go

@@ -0,0 +1,123 @@
+package parser
+
+// NullStyle null type
+type NullStyle int
+
+// nolint
+const (
+	NullDisable NullStyle = iota
+	NullInSql
+	NullInPointer
+)
+
+// Option function
+type Option func(*options)
+
+type options struct {
+	Charset        string
+	Collation      string
+	JSONTag        bool
+	JSONNamedType  int // json命名类型，0:默认，其他值表示驼峰
+	TablePrefix    string
+	ColumnPrefix   string
+	NoNullType     bool
+	NullStyle      NullStyle
+	Package        string
+	GormType       bool
+	ForceTableName bool
+	IsEmbed        bool // 是否嵌入gorm.Model
+}
+
+var defaultOptions = options{
+	NullStyle: NullInSql,
+	Package:   "model",
+}
+
+// WithCharset  set charset
+func WithCharset(charset string) Option {
+	return func(o *options) {
+		o.Charset = charset
+	}
+}
+
+// WithCollation set collation
+func WithCollation(collation string) Option {
+	return func(o *options) {
+		o.Collation = collation
+	}
+}
+
+// WithTablePrefix set table prefix
+func WithTablePrefix(p string) Option {
+	return func(o *options) {
+		o.TablePrefix = p
+	}
+}
+
+// WithColumnPrefix set column prefix
+func WithColumnPrefix(p string) Option {
+	return func(o *options) {
+		o.ColumnPrefix = p
+	}
+}
+
+// WithJSONTag json名称命名类型，0:表示默认，其他值表示驼峰
+func WithJSONTag(namedType int) Option {
+	return func(o *options) {
+		o.JSONTag = true
+		o.JSONNamedType = namedType
+	}
+}
+
+// WithNoNullType set NoNullType
+func WithNoNullType() Option {
+	return func(o *options) {
+		o.NoNullType = true
+	}
+}
+
+// WithNullStyle set NullType
+func WithNullStyle(s NullStyle) Option {
+	return func(o *options) {
+		o.NullStyle = s
+	}
+}
+
+// WithPackage set package name
+func WithPackage(pkg string) Option {
+	return func(o *options) {
+		o.Package = pkg
+	}
+}
+
+// WithGormType will write type in gorm tag
+func WithGormType() Option {
+	return func(o *options) {
+		o.GormType = true
+	}
+}
+
+// WithForceTableName set forceFloats
+func WithForceTableName() Option {
+	return func(o *options) {
+		o.ForceTableName = true
+	}
+}
+
+// WithEmbed is embed gorm.Model
+func WithEmbed() Option {
+	return func(o *options) {
+		o.IsEmbed = true
+	}
+}
+
+func parseOption(options []Option) options {
+	o := defaultOptions
+	for _, f := range options {
+		f(&o)
+	}
+	if o.NoNullType {
+		o.NullStyle = NullDisable
+	}
+	return o
+}