Fix build

Author: reshke

protos/distribution.proto

@@ -14,7 +14,7 @@ message DistributedRelation {
   repeated DistributionKeyEntry distributionKey = 2;
 }
 
-message Distribution{
+message Distribution {
   string id = 1;
   repeated string ColumnTypes = 2;
   repeated DistributedRelation relations = 3;

router/qrouter/proxy_routing.go

@@ -175,8 +175,8 @@ func (qr *ProxyQrouter) RecordDistributionKeyColumnValueOnRFQN(meta *RoutingMeta
 	} else {
 		// TODO: optimize
 		ok := false
-		for _, c := range ds.Relations[resolvedRelation.RelationName].ColumnNames {
-			if c == colname {
+		for _, c := range ds.Relations[resolvedRelation.RelationName].DistributionKey {
+			if c.Column == colname {
 				ok = true
 				break
 			}
@@ -471,10 +471,10 @@ func (qr *ProxyQrouter) deparseShardingMapping(
 				insertColsPos[c] = i
 			}
 
-			distributionKey := ds.Relations[rfqn.RelationName].ColumnNames
+			distributionKey := ds.Relations[rfqn.RelationName].DistributionKey
 			// TODO: check mapping by rules with multiple columns
 			for _, col := range distributionKey {
-				if val, ok := insertColsPos[col]; !ok {
+				if val, ok := insertColsPos[col.Column]; !ok {
 					return ComplexQuery
 				} else {
 					offsets = append(offsets, val)