odbtest/test/update_stats.go

/*
	Before you execute the program, Launch `cqlsh` and execute:

create keyspace example with replication = { 'class' : 'SimpleStrategy', 'replication_factor' : 1 };
create table example.tweet(timeline text, id UUID, text text, PRIMARY KEY(id));
create index on example.tweet(timeline);
*/
package main

import (
	// "time"
	//"fmt"
	"flag"
	// "strconv"
	// "strings"
	"git.wecise.com/wecise/odbserver/odb/test"
	"gitee.com/wecisecode/util/logger"
)

func main() {
	// connect to the cluster
	//cluster := gocql.NewCluster("192.168.40.14")
	keyspace := flag.String("keyspace", "matrix", "keyspace")
	flag.Parse()

	option := &Option{Cache: CacheAll, Keyspace: *keyspace, DisableInitialHostLookup: true}
	g, err := test.NewG(option)

	if err != nil {
		logger.Error(err.Error())
	} else {
		defer g.Close()
	}

	//logger.SetRollingDaily("C:/test/zkcron/src/test", "test.log")
	logger.SetConsole(true)

	if rows, err := g.RawQuery(` SELECT * FROM relation_stats`); err == nil {
		if _, err := g.RawQuery("truncate relation_stats"); err != nil {
			logger.Warnf("Update RelationsStats Error [%s]", err)
			logger.Info(rows)
		} else if _, err := g.RawQuery("drop table relation_stats"); err != nil {
			logger.Warnf("Update RelationsStats Error [%s]", err)
			logger.Info(rows)
		} else if _, err := g.RawQuery(`CREATE TABLE IF NOT EXISTS relation_stats (
				inclass				varchar,
				relation			varchar,
				outclass			varchar,		
				constrain			list<int>,	
				
				morph_info			set<varchar>,
				PRIMARY KEY(inclass, relation, outclass)
			)`); err != nil {

			logger.Warnf("Update RelationsStats Error [%s]", err)
			logger.Info(rows)

		} else {
			logger.Info("total =>", len(rows))

			g.RawQuery("CREATE INDEX IF NOT EXISTS relation_stats_outclass ON relation_stats( outclass )")
			g.RawQuery("CREATE INDEX IF NOT EXISTS relation_stats_relation ON relation_stats( relation )")

			for _, row := range rows {
				g.RawQuery(`insert into relation_stats(inclass, relation, outclass, constrain) values (?,?,?,?)`, row["inclass"], row["relation"], row["outclass"], row["constrain"])
			}

			//g.RawQuery("INSERT INTO rels (name, colname, remedy, mtime) VALUES ('_morph', '_morph', '多态', 1458113899033)")
		}
	}

}