Possibilidade de criar os índices extras via CLI

cmd/cmd.go

@@ -67,6 +67,19 @@ var dropCmd = &cobra.Command{
 	},
 }
 
+var extraIndexesCmd = &cobra.Command{
+	Use:   "extra-indexes",
+	Short: "Creates indexes in database", //Acho que essa explicacao pode melhorar
+	RunE: func(_ *cobra.Command, args []string) error {
+		db, err := loadDatabase()
+		if err != nil {
+			return fmt.Errorf("could not find database: %w", err)
+		}
+		defer db.Close()
+		return db.ExtraIndexes(args)
+	},
+}
+
 func addDataDir(c *cobra.Command) *cobra.Command {
 	c.Flags().StringVarP(&dir, "directory", "d", defaultDataDir, "directory of the downloaded files")
 	return c
@@ -80,7 +93,7 @@ func addDatabase(c *cobra.Command) *cobra.Command {
 
 // CLI returns the root command from Cobra CLI tool.
 func CLI() *cobra.Command {
-	for _, c := range []*cobra.Command{createCmd, dropCmd} {
+	for _, c := range []*cobra.Command{createCmd, dropCmd, extraIndexesCmd} {
 		addDatabase(c)
 	}
 	for _, c := range []*cobra.Command{
@@ -92,6 +105,7 @@ func CLI() *cobra.Command {
 		checkCLI(),
 		createCmd,
 		dropCmd,
+		extraIndexesCmd,
 		transformCLI(),
 		sampleCLI(),
 		mirrorCLI(),

cmd/db.go

@@ -22,6 +22,7 @@ type database interface {
 	CreateCompanies([][]string) error
 	PostLoad() error
 	MetaSave(string, string) error
+	ExtraIndexes(idxs []string) error
 	// api
 	GetCompany(string) (string, error)
 	MetaRead(string) (string, error)

db/mongodb.go

@@ -223,3 +223,21 @@ func (m *MongoDB) GetCompany(cnpj string) (string, error) {
 	}
 	return string(b), nil
 }
+
+func (m *MongoDB) ExtraIndexes(idxs []string) error {
+	log.Output(1, "Creating the indexes...")
+	c := m.db.Collection(companyTableName)
+	var i []mongo.IndexModel
+	for _, field := range idxs {
+		i = append(i, mongo.IndexModel{
+			Keys: bson.D{{Key: field, Value: 1}},
+		})
+	}
+	_, err := c.Indexes().CreateMany(m.ctx, i)
+	if err != nil {
+		return fmt.Errorf("error creating indexes: %w", err)
+	}
+	log.Output(1, fmt.Sprintf("Indexes successfully created in the collection %s", companyTableName))
+	return nil
+
+}

db/postgres.go

@@ -212,3 +212,36 @@ func NewPostgreSQL(uri, schema string) (PostgreSQL, error) {
 	}
 	return p, nil
 }
+
+func (p *PostgreSQL) ExtraIndexes(idxs []string) error {
+	for _, v := range idxs {
+		t := "((json->'" + v + "'))"
+		iname := "json_"
+		qsa := []string{"pais", "nome_socio", "codigo_pais", "faixa_etaria", "cnpj_cpf_do_socio", "qualificacao_socio", "codigo_faixa_etaria", "data_entrada_sociedade", "identificador_de_socio", "cpf_representante_legal", "nome_representante_legal", "codigo_qualificacao_socio", "qualificacao_representante_legal", "codigo_qualificacao_representante_legal"}
+		for _, v_qsa := range qsa {
+			if v == v_qsa {
+				t = "(jsonb_extract_path(json, 'qsa', '" + v_qsa + "') jsonb_ops)"
+				iname += "qsa_"
+				break
+			}
+		}
+		cnae := []string{"codigo", "descricao"}
+		for _, v_cnae := range cnae {
+			if v == v_cnae {
+				t = "(jsonb_extract_path(json, 'cnae', '" + v_cnae + "') jsonb_ops)"
+				iname += "cnae_"
+				break
+			}
+		}
+		q := fmt.Sprintf(
+			"CREATE INDEX IF NOT EXISTS idx_%s%s ON %s USING GIN %s;",
+			iname, v, p.CompanyTableName, t,
+		)
+		if _, err := p.pool.Exec(context.Background(), q); err != nil {
+			return fmt.Errorf("error to create indexe %s: %w", v, err)
+		}
+		log.Output(1, "Indexe "+v+" created")
+	}
+
+	return nil
+}